forked from sent/waves
97 lines
5.4 KiB
JavaScript
97 lines
5.4 KiB
JavaScript
class EJS_COMPRESSION {
|
|
constructor(EJS) {
|
|
this.EJS = EJS;
|
|
}
|
|
isCompressed(data) { //https://www.garykessler.net/library/file_sigs.html
|
|
//todo. Use hex instead of numbers
|
|
if ((data[0] === 80 && data[1] === 75) && ((data[2] === 3 && data[3] === 4) || (data[2] === 5 && data[3] === 6) || (data[2] === 7 && data[3] === 8))) {
|
|
return 'zip';
|
|
} else if (data[0] === 55 && data[1] === 122 && data[2] === 188 && data[3] === 175 && data[4] === 39 && data[5] === 28) {
|
|
return '7z';
|
|
} else if ((data[0] === 82 && data[1] === 97 && data[2] === 114 && data[3] === 33 && data[4] === 26 && data[5] === 7) && ((data[6] === 0) || (data[6] === 1 && data[7] == 0))) {
|
|
return 'rar';
|
|
}
|
|
return null;
|
|
}
|
|
decompress(data, updateMsg, fileCbFunc) {
|
|
const compressed = this.isCompressed(data.slice(0, 10));
|
|
if (compressed === null) {
|
|
if (typeof fileCbFunc === "function") {
|
|
fileCbFunc("!!notCompressedData", data);
|
|
}
|
|
return new Promise(resolve => resolve({"!!notCompressedData": data}));
|
|
}
|
|
return this.decompressFile(compressed, data, updateMsg, fileCbFunc);
|
|
}
|
|
getWorkerFile(method) {
|
|
return new Promise(async (resolve, reject) => {
|
|
let path, obj;
|
|
if (method === "7z") {
|
|
path = "compression/extract7z.js";
|
|
obj = "sevenZip";
|
|
} else if (method === "zip") {
|
|
path = "compression/extractzip.js";
|
|
obj = "zip";
|
|
} else if (method === "rar") {
|
|
path = "compression/libunrar.js";
|
|
obj = "rar";
|
|
}
|
|
const res = await this.EJS.downloadFile(path, null, false, {responseType: "text", method: "GET"});
|
|
if (res === -1) {
|
|
this.EJS.startGameError(this.EJS.localization('Network Error'));
|
|
return;
|
|
}
|
|
if (method === "rar") {
|
|
const res2 = await this.EJS.downloadFile("compression/libunrar.wasm", null, false, {responseType: "arraybuffer", method: "GET"});
|
|
if (res2 === -1) {
|
|
this.EJS.startGameError(this.EJS.localization('Network Error'));
|
|
return;
|
|
}
|
|
const path = URL.createObjectURL(new Blob([res2.data], {type: "application/wasm"}));
|
|
let data = '\nlet dataToPass = [];\nModule = {\n monitorRunDependencies: function(left) {\n if (left == 0) {\n setTimeout(function() {\n unrar(dataToPass, null);\n }, 100);\n }\n },\n onRuntimeInitialized: function() {\n },\n locateFile: function(file) {\n return \''+path+'\';\n }\n};\n'+res.data+'\nlet unrar = function(data, password) {\n let cb = function(fileName, fileSize, progress) {\n postMessage({"t":4,"current":progress,"total":fileSize, "name": fileName});\n };\n\n let rarContent = readRARContent(data.map(function(d) {\n return {\n name: d.name,\n content: new Uint8Array(d.content)\n }\n }), password, cb)\n let rec = function(entry) {\n if (!entry) return;\n if (entry.type === \'file\') {\n postMessage({"t":2,"file":entry.fullFileName,"size":entry.fileSize,"data":entry.fileContent});\n } else if (entry.type === \'dir\') {\n Object.keys(entry.ls).forEach(function(k) {\n rec(entry.ls[k]);\n })\n } else {\n throw "Unknown type";\n }\n }\n rec(rarContent);\n postMessage({"t":1});\n return rarContent;\n};\nonmessage = function(data) {\n dataToPass.push({name: \'test.rar\', content: data.data});\n};\n ';
|
|
const blob = new Blob([data], {
|
|
type: 'application/javascript'
|
|
})
|
|
resolve(blob);
|
|
} else {
|
|
const blob = new Blob([res.data], {
|
|
type: 'application/javascript'
|
|
})
|
|
resolve(blob);
|
|
}
|
|
})
|
|
}
|
|
decompressFile(method, data, updateMsg, fileCbFunc) {
|
|
return new Promise(async callback => {
|
|
const file = await this.getWorkerFile(method);
|
|
const worker = new Worker(URL.createObjectURL(file));
|
|
const files = {};
|
|
worker.onmessage = (data) => {
|
|
if (!data.data) return;
|
|
//data.data.t/ 4=progress, 2 is file, 1 is zip done
|
|
if (data.data.t === 4) {
|
|
const pg = data.data;
|
|
const num = Math.floor(pg.current / pg.total * 100);
|
|
if (isNaN(num)) return;
|
|
const progress = ' '+num.toString()+'%';
|
|
updateMsg(progress, true);
|
|
}
|
|
if (data.data.t === 2) {
|
|
if (typeof fileCbFunc === "function") {
|
|
fileCbFunc(data.data.file, data.data.data);
|
|
files[data.data.file] = true;
|
|
} else {
|
|
files[data.data.file] = data.data.data;
|
|
}
|
|
}
|
|
if (data.data.t === 1) {
|
|
callback(files);
|
|
}
|
|
}
|
|
worker.postMessage(data);
|
|
});
|
|
}
|
|
}
|
|
|
|
window.EJS_COMPRESSION = EJS_COMPRESSION;
|