WIP download bugfix

This commit is contained in:
FinnHornhoover 2023-09-21 18:01:55 +03:00
parent 01e329e621
commit ee309319c9
3 changed files with 153 additions and 37 deletions

View File

@ -5,13 +5,15 @@ var path = remote.require("path");
var url = remote.require("url"); var url = remote.require("url");
var http = remote.require("http"); var http = remote.require("http");
var createHash = remote.require("crypto").createHash; var createHash = remote.require("crypto").createHash;
var async = remote.require("async");
var userData = remote.require("app").getPath("userData"); var userData = remote.require("app").getPath("userData");
var configPath = path.join(userData, "config.json"); var configPath = path.join(userData, "config.json");
var serversPath = path.join(userData, "servers.json"); var serversPath = path.join(userData, "servers.json");
var versionsPath = path.join(userData, "versions.json"); var versionsPath = path.join(userData, "versions.json");
var hashPath = path.join(userData, "hash.txt");
var chunk_size = 1 << 16; var chunkSize = 1 << 16;
var gb = 1 << 30; var gb = 1 << 30;
var cdn = "cdn.dexlabs.systems"; var cdn = "cdn.dexlabs.systems";
@ -180,7 +182,7 @@ function getCacheLabelText(sizes) {
var labelText = (sizes.intact / gb).toFixed(2) + " / " + (sizes.total / gb).toFixed(2) + " GB"; var labelText = (sizes.intact / gb).toFixed(2) + " / " + (sizes.total / gb).toFixed(2) + " GB";
if (sizes.altered > 0) { if (sizes.altered > 0) {
labelText += " (" + (sizes.altered / gb).toFixed(2) + " GB Altered)"; labelText += "\n(" + (sizes.altered / gb).toFixed(2) + " GB Altered)";
} }
return labelText; return labelText;
@ -253,13 +255,12 @@ function getCacheInfoCell(versionString, cacheMode) {
} }
function getFileHash(filePath) { function getFileHash(filePath) {
var totalReadCount = 0, readCount = 0; var readCount = 0;
var buff = new Buffer(chunk_size); var buff = new Buffer(chunkSize);
var hash = createHash("sha256"); var hash = createHash("sha256");
var file = remotefs.openSync(filePath, "r"); var file = remotefs.openSync(filePath, "r");
while ((readCount = remotefs.readSync(file, buff, 0, chunk_size, totalReadCount)) > 0) { while ((readCount = remotefs.readSync(file, buff, 0, chunkSize, null)) > 0) {
totalReadCount += readCount;
hash.update(buff.slice(0, readCount)); hash.update(buff.slice(0, readCount));
} }
@ -267,6 +268,7 @@ function getFileHash(filePath) {
return hash.digest(encoding="hex"); return hash.digest(encoding="hex");
} }
/*
function downloadFiles(root, client, sizes, hashes, updateCallback, endCallback) { function downloadFiles(root, client, sizes, hashes, updateCallback, endCallback) {
if (hashes.length === 0) { if (hashes.length === 0) {
endCallback(); endCallback();
@ -278,7 +280,7 @@ function downloadFiles(root, client, sizes, hashes, updateCallback, endCallback)
delete hashes[filePath]; delete hashes[filePath];
var fullFilePath = path.join(root, filePath); var fullFilePath = path.join(root, filePath);
var fullCDNPath = ["http:/", cdn, "ff", "big", filePath].join("/"); var fullCDNPath = "http://" + [cdn, "ff", "big", filePath].join("/");
if (remotefs.existsSync(fullFilePath) && fileHash === getFileHash(fullFilePath)) { if (remotefs.existsSync(fullFilePath) && fileHash === getFileHash(fullFilePath)) {
console.log(fullFilePath + " is intact, skipping..."); console.log(fullFilePath + " is intact, skipping...");
@ -308,8 +310,10 @@ function downloadFile(client, fullCDNPath, fullFilePath, callback) {
var urlParts = url.parse(fullCDNPath); var urlParts = url.parse(fullCDNPath);
var req = client.request("GET", urlParts.path, { var req = client.request("GET", urlParts.path, {
"host": urlParts.hostname, "Host": urlParts.hostname,
"Content-Type": "application/octet-stream" "Content-Type": "application/octet-stream",
"Referer": fullCDNPath.split("/").slice(0, 4).join("/") + "/",
"Connection": "keep-alive",
}); });
var writeStream = remotefs.createWriteStream(fullFilePath); var writeStream = remotefs.createWriteStream(fullFilePath);
@ -330,7 +334,7 @@ function downloadFile(client, fullCDNPath, fullFilePath, callback) {
writeStream.on("error", retry); writeStream.on("error", retry);
req.on("response", function (res) { req.on("response", function (res) {
if (res.statusCode !== 200) { if (res.statusCode >= 300) {
console.log("Error in fetching file " + fullFilePath + " from " + fullCDNPath); console.log("Error in fetching file " + fullFilePath + " from " + fullCDNPath);
retry("Status Code: " + res.statusCode); retry("Status Code: " + res.statusCode);
return; return;
@ -346,15 +350,91 @@ function downloadFile(client, fullCDNPath, fullFilePath, callback) {
req.end(); req.end();
} }
*/
function downloadFile(nginxDir, localDir, relativePath, callback) {
var nginxUrl = path.dirname(nginxDir) + "/" + relativePath;
var localPath = path.join(localDir, relativePath);
// Create directories if they don't exist
var dirName = path.dirname(localPath);
remotefs.ensureDirSync(dirName);
// HTTP request to download the file
var fileStream = remotefs.createWriteStream(localPath);
var urlParse = url.parse(nginxUrl);
var client = http.createClient(80, urlParse.hostname);
var options = {
url: urlParse.hostname,
port: 80,
path: urlParse.path,
headers: {
"Host": urlParse.hostname,
"Content-Type": "application/octet-stream",
"Referer": nginxDir,
"Connection": "keep-alive",
}
};
var request = client.request("GET", urlParse.path, options, function(response) {
// Handle errors
response.on("error", function(err) {
console.error("Error downloading " + relativePath + ": " + err.message);
retryDownload(nginxDir, localDir, relativePath, callback); // Retry download
});
response.pipe(fileStream);
// When the download is complete, invoke the callback
response.on("end", function() {
fileStream.end();
callback(null, relativePath);
});
});
// Handle HTTP errors
request.on("error", function(err) {
console.error("Error downloading " + relativePath + ": " + err.message);
retryDownload(nginxDir, localDir, relativePath, callback); // Retry download
});
request.end();
}
// Function to retry downloading a file after a delay
function retryDownload(nginxDir, localDir, relativePath, callback) {
setTimeout(function() {
downloadFile(nginxDir, localDir, relativePath, callback);
}, 1000); // Retry after 1 second
}
// Function to download multiple files in parallel
function downloadFiles(nginxDir, localDir, fileRelativePaths, allDoneCallback) {
async.eachLimit(
fileRelativePaths,
5, // Number of parallel downloads
function(relativePath, callback) {
downloadFile(nginxDir, localDir, relativePath, callback);
},
function(err) {
if (err) {
console.error("Download failed: " + err);
} else {
console.log("All files downloaded successfully.");
allDoneCallback();
}
}
);
}
function loadCacheList() { function loadCacheList() {
var versionjson = JSON.parse( var versionjson = remotefs.readJsonSync(versionsPath);
remotefs.readFileSync(path.join(userDir, "versions.json"))
);
versionArray = versionjson["versions"]; versionArray = versionjson["versions"];
versionHashes = { playable: {}, offline: {} }; versionHashes = { playable: {}, offline: {} };
var hashlines = remotefs.readFileSync(path.join(userDir, "hash.txt"), "utf-8"); var hashlines = remotefs.readFileSync(hashPath, ( encoding = "utf-8" ));
$.each(hashlines.split(/\r\n|\r|\n/), function (key, line) { $.each(hashlines.split(/\r\n|\r|\n/), function (key, line) {
if (line.length === 0) { if (line.length === 0) {
return; return;
@ -404,7 +484,10 @@ function loadCacheList() {
} }
function deletePlayableCache(versionString) { function deletePlayableCache(versionString) {
var cacheroot = path.join(userDir, "..", "..", "LocalLow", "Unity", "Web Player", "Cache"); var cacheRoot = path.join(
userData,
"/../../LocalLow/Unity/Web Player/Cache"
);
resetCacheNames(); resetCacheNames();
@ -412,26 +495,30 @@ function deletePlayableCache(versionString) {
return; return;
} }
remotefs.removeSync(path.join(cacheroot, versionString)); remotefs.removeSync(path.join(cacheRoot, versionString));
console.log("Playable cache " + versionString + " has been removed!"); console.log("Playable cache " + versionString + " has been removed!");
checkPlayableCache(versionString); checkPlayableCache(versionString);
} }
function downloadOfflineCache(versionString) { function downloadOfflineCache(versionString) {
var offlineroot = path.join(userDir, "..", "..", "LocalLow", "Unity", "Web Player", "Cache", "Offline"); var offlineRoot = path.join(
userData,
"/../../LocalLow/Unity/Web Player/Cache/Offline"
);
var buttonDownload = document.getElementById(getCacheButtonID(versionString, "offline", "download")); var buttonDownload = document.getElementById(getCacheButtonID(versionString, "offline", "download"));
var buttonFix = document.getElementById(getCacheButtonID(versionString, "offline", "fix")); var buttonFix = document.getElementById(getCacheButtonID(versionString, "offline", "fix"));
var buttonDelete = document.getElementById(getCacheButtonID(versionString, "offline", "delete")); var buttonDelete = document.getElementById(getCacheButtonID(versionString, "offline", "delete"));
var label = document.getElementById(getCacheElemID(versionString, "offline", "label")); var label = document.getElementById(getCacheElemID(versionString, "offline", "label"));
var version = versionArray.filter(function (value) {
return value.name === versionString;
})[0];
var sizes = { var sizes = {
intact: 0, intact: 0,
altered: 0, altered: 0,
total: versionArray.filter(function (value) { total: version.offline_size
return value.name === versionString;
})[0].offline_size
}; };
buttonDownload.setAttribute("disabled", ""); buttonDownload.setAttribute("disabled", "");
@ -441,9 +528,12 @@ function downloadOfflineCache(versionString) {
buttonDownload.children[0].setAttribute("class", "fas fa-spinner fa-spin fa-fw"); buttonDownload.children[0].setAttribute("class", "fas fa-spinner fa-spin fa-fw");
buttonFix.children[0].setAttribute("class", "fas fa-spinner fa-spin fa-fw"); buttonFix.children[0].setAttribute("class", "fas fa-spinner fa-spin fa-fw");
/*
var client = http.createClient(80, cdn);
downloadFiles( downloadFiles(
offlineroot, offlineRoot,
http.createClient(80, cdn), client,
sizes, sizes,
JSON.parse(JSON.stringify(versionHashes.offline[versionString])), JSON.parse(JSON.stringify(versionHashes.offline[versionString])),
function (sizesNow) { function (sizesNow) {
@ -455,19 +545,37 @@ function downloadOfflineCache(versionString) {
checkOfflineCache(versionString); checkOfflineCache(versionString);
} }
); );
*/
downloadFiles(
version.url,
offlineRoot,
Object.keys(JSON.parse(JSON.stringify(versionHashes.offline[versionString]))),
function () {
buttonDownload.children[0].setAttribute("class", "fas fa-download");
buttonFix.children[0].setAttribute("class", "fas fa-hammer");
checkOfflineCache(versionString);
}
);
} }
function deleteOfflineCache(versionString) { function deleteOfflineCache(versionString) {
var offlineroot = path.join(userDir, "..", "..", "LocalLow", "Unity", "Web Player", "Cache", "Offline"); var offlineRoot = path.join(
userData,
"/../../LocalLow/Unity/Web Player/Cache/Offline"
);
remotefs.removeSync(path.join(offlineroot, versionString)); remotefs.removeSync(path.join(offlineRoot, versionString));
console.log("Offline cache " + versionString + " has been removed!"); console.log("Offline cache " + versionString + " has been removed!");
checkOfflineCache(versionString); checkOfflineCache(versionString);
} }
function checkPlayableCache(versionString) { function checkPlayableCache(versionString) {
var cacheroot = path.join(userDir, "..", "..", "LocalLow", "Unity", "Web Player", "Cache"); var cacheRoot = path.join(
userData,
"/../../LocalLow/Unity/Web Player/Cache"
);
var button = document.getElementById(getCacheButtonID(versionString, "playable", "delete")); var button = document.getElementById(getCacheButtonID(versionString, "playable", "delete"));
var label = document.getElementById(getCacheElemID(versionString, "playable", "label")); var label = document.getElementById(getCacheElemID(versionString, "playable", "label"));
@ -483,7 +591,7 @@ function checkPlayableCache(versionString) {
resetCacheNames(); resetCacheNames();
$.each(versionHashes.playable[versionString], function (filePath, fileHash) { $.each(versionHashes.playable[versionString], function (filePath, fileHash) {
var fullFilePath = path.join(cacheroot, filePath); var fullFilePath = path.join(cacheRoot, filePath);
if (remotefs.existsSync(fullFilePath)) { if (remotefs.existsSync(fullFilePath)) {
var fileSize = remotefs.statSync(fullFilePath).size; var fileSize = remotefs.statSync(fullFilePath).size;
@ -506,7 +614,10 @@ function checkPlayableCache(versionString) {
} }
function checkOfflineCache(versionString) { function checkOfflineCache(versionString) {
var offlineroot = path.join(userDir, "..", "..", "LocalLow", "Unity", "Web Player", "Cache", "Offline"); var offlineRoot = path.join(
userData,
"/../../LocalLow/Unity/Web Player/Cache/Offline"
);
var buttonDownload = document.getElementById(getCacheButtonID(versionString, "offline", "download")); var buttonDownload = document.getElementById(getCacheButtonID(versionString, "offline", "download"));
var buttonFix = document.getElementById(getCacheButtonID(versionString, "offline", "fix")); var buttonFix = document.getElementById(getCacheButtonID(versionString, "offline", "fix"));
@ -522,7 +633,7 @@ function checkOfflineCache(versionString) {
}; };
$.each(versionHashes.offline[versionString], function (filePath, fileHash) { $.each(versionHashes.offline[versionString], function (filePath, fileHash) {
var fullFilePath = path.join(offlineroot, filePath); var fullFilePath = path.join(offlineRoot, filePath);
if (remotefs.existsSync(fullFilePath)) { if (remotefs.existsSync(fullFilePath)) {
var fileSize = remotefs.statSync(fullFilePath).size; var fileSize = remotefs.statSync(fullFilePath).size;
@ -553,20 +664,23 @@ function checkOfflineCache(versionString) {
} }
function resetCacheNames() { function resetCacheNames() {
var cacheroot = path.join(userDir, "..", "..", "LocalLow", "Unity", "Web Player", "Cache"); var cacheRoot = path.join(
var currentcache = path.join(cacheroot, "Fusionfall"); userData,
var record = path.join(userDir, ".lastver"); "/../../LocalLow/Unity/Web Player/Cache"
);
var currentCache = path.join(cacheRoot, "Fusionfall");
var record = path.join(userData, ".lastver");
if (!remotefs.existsSync(currentcache)) { if (!remotefs.existsSync(currentCache)) {
return; return;
} }
lastversion = remotefs.readFileSync(record); var lastVersion = remotefs.readFileSync(record, (encoding = "utf8"));
remotefs.renameSync( remotefs.renameSync(
currentcache, currentCache,
path.join(cacheroot, lastversion) path.join(cacheRoot, lastVersion)
); );
console.log("Current cache " + lastversion + " has been renamed to its original name."); console.log("Current cache " + lastVersion + " has been renamed to its original name.");
} }
function performCacheSwap(newVersion) { function performCacheSwap(newVersion) {

View File

@ -16,6 +16,7 @@ if (process.env.npm_node_execpath) {
process.env["UNITY_HOME_DIR"] = unityHomeDir; process.env["UNITY_HOME_DIR"] = unityHomeDir;
process.env["UNITY_DISABLE_PLUGIN_UPDATES"] = "yes"; process.env["UNITY_DISABLE_PLUGIN_UPDATES"] = "yes";
process.env["WINE_LARGE_ADDRESS_AWARE"] = "1";
app.commandLine.appendSwitch("enable-npapi"); app.commandLine.appendSwitch("enable-npapi");
app.commandLine.appendSwitch( app.commandLine.appendSwitch(

View File

@ -74,6 +74,7 @@
"afterPack": "./build/after-pack.js" "afterPack": "./build/after-pack.js"
}, },
"dependencies": { "dependencies": {
"fs-extra": "2.0.0" "fs-extra": "2.0.0",
"async": "1.5.2"
} }
} }