Personal_mirrors/bitburner-src
1
0
Fork 0
mirror of https://github.com/bitburner-official/bitburner-src.git synced 2024-11-10 09:43:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor scp function

Browse Source
This commit is contained in:
Snarling 2022-08-17 20:31:26 -04:00
parent f6a8d5d337
commit 07e9cb9277
1 changed files with 86 additions and 112 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

198
src/NetscriptFunctions.ts
View File

@ -829,132 +829,106 @@ const base: InternalAPI<NS> = {
}, },
scp: scp:
(ctx: NetscriptContext) => (ctx: NetscriptContext) =>
(_scriptname: unknown, _destination: unknown, _source: unknown = ctx.workerScript.hostname): boolean => { (_files: unknown, _destination: unknown, _source: unknown = ctx.workerScript.hostname): boolean => {
const destination = helpers.string(ctx, "destination", _destination); const destination = helpers.string(ctx, "destination", _destination);
const source = helpers.string(ctx, "source", _source); const source = helpers.string(ctx, "source", _source);
if (Array.isArray(_scriptname)) {
// Recursively call scp on all elements of array
const scripts: string[] = _scriptname;
if (scripts.length === 0) {
throw helpers.makeRuntimeErrorMsg(ctx, "No scripts to copy");
}
let res = true;
scripts.forEach(function (script) {
//TODO: This array method still needs to be reworked because it is terribly inefficient.
if (!NetscriptFunctions(ctx.workerScript).scp(script, destination, source)) {
res = false;
}
});
return res;
}
const scriptName = helpers.string(ctx, "scriptName", _scriptname);
// Invalid file type
if (!isValidFilePath(scriptName)) {
throw helpers.makeRuntimeErrorMsg(ctx, `Invalid filename: '${scriptName}'`);
}
// Invalid file name
if (!scriptName.endsWith(".lit") && !isScriptFilename(scriptName) && !scriptName.endsWith("txt")) {
throw helpers.makeRuntimeErrorMsg(ctx, "Only works for scripts, .lit and .txt files");
}
const destServer = helpers.getServer(ctx, destination); const destServer = helpers.getServer(ctx, destination);
const currServ = helpers.getServer(ctx, source); const sourceServ = helpers.getServer(ctx, source);
const files = Array.isArray(_files) ? _files : [_files];
// Scp for lit files //First loop through filenames to find all errors before moving anything.
if (scriptName.endsWith(".lit")) { for (const file of files) {
let found = false; // Not a string
for (let i = 0; i < currServ.messages.length; ++i) { if (typeof file !== "string")
if (currServ.messages[i] == scriptName) { throw helpers.makeRuntimeErrorMsg(ctx, "files should be a string or an array of strings.");
found = true;
break; // Invalid file name
if (!isValidFilePath(file)) throw helpers.makeRuntimeErrorMsg(ctx, `Invalid filename: '${file}'`);
// Invalid file type
if (!file.endsWith(".lit") && !isScriptFilename(file) && !file.endsWith("txt")) {
throw helpers.makeRuntimeErrorMsg(ctx, "Only works for scripts, .lit and .txt files.");
}
}
let noFailures = true;
//ts detects files as any[] here even though we would have thrown in the above loop if it wasn't string[]
for (const file of files as string[]) {
// Scp for lit files
if (file.endsWith(".lit")) {
const sourceMessage = sourceServ.messages.find((message) => message === file);
if (!sourceMessage) {
helpers.log(ctx, () => `File '${file}' does not exist.`);
noFailures = false;
continue;
} }
}
if (!found) { const destMessage = destServer.messages.find((message) => message === file);
helpers.log(ctx, () => `File '${scriptName}' does not exist.`); if (destMessage) {
return false; helpers.log(ctx, () => `File '${file}' was already on '${destServer?.hostname}'.`);
} continue;
for (let i = 0; i < destServer.messages.length; ++i) {
if (destServer.messages[i] === scriptName) {
helpers.log(ctx, () => `File '${scriptName}' copied over to '${destServer?.hostname}'.`);
return true; // Already exists
} }
}
destServer.messages.push(scriptName);
helpers.log(ctx, () => `File '${scriptName}' copied over to '${destServer?.hostname}'.`);
return true;
}
// Scp for text files destServer.messages.push(file);
if (scriptName.endsWith(".txt")) { helpers.log(ctx, () => `File '${file}' copied over to '${destServer?.hostname}'.`);
let txtFile; continue;
for (let i = 0; i < currServ.textFiles.length; ++i) { }
if (currServ.textFiles[i].fn === scriptName) {
txtFile = currServ.textFiles[i]; // Scp for text files
break; if (file.endsWith(".txt")) {
const sourceTextFile = sourceServ.textFiles.find((textFile) => textFile.fn === file);
if (!sourceTextFile) {
helpers.log(ctx, () => `File '${file}' does not exist.`);
noFailures = false;
continue;
} }
}
if (txtFile === undefined) {
helpers.log(ctx, () => `File '${scriptName}' does not exist.`);
return false;
}
for (let i = 0; i < destServer.textFiles.length; ++i) { const destTextFile = destServer.textFiles.find((textFile) => textFile.fn === file);
if (destServer.textFiles[i].fn === scriptName) { if (destTextFile) {
// Overwrite destTextFile.text = sourceTextFile.text;
destServer.textFiles[i].text = txtFile.text; helpers.log(ctx, () => `File '${file}' overwritten on '${destServer?.hostname}'.`);
helpers.log(ctx, () => `File '${scriptName}' copied over to '${destServer?.hostname}'.`); continue;
return true;
} }
const newFile = new TextFile(sourceTextFile.fn, sourceTextFile.text);
destServer.textFiles.push(newFile);
helpers.log(ctx, () => `File '${file}' copied over to '${destServer?.hostname}'.`);
continue;
} }
const newFile = new TextFile(txtFile.fn, txtFile.text);
destServer.textFiles.push(newFile); // Scp for script files
helpers.log(ctx, () => `File '${scriptName}' copied over to '${destServer?.hostname}'.`); const sourceScript = sourceServ.scripts.find((script) => script.filename === file);
return true; if (!sourceScript) {
helpers.log(ctx, () => `File '${file}' does not exist.`);
noFailures = false;
continue;
}
// Overwrite script if it already exists
const destScript = destServer.scripts.find((script) => script.filename === file);
if (destScript) {
if (destScript.code === sourceScript.code) {
helpers.log(ctx, () => `Identical file '${file}' was already on '${destServer?.hostname}'`);
continue;
}
destScript.code = sourceScript.code;
destScript.ramUsage = destScript.ramUsage;
destScript.markUpdated();
helpers.log(ctx, () => `WARNING: File '${file}' overwritten on '${destServer?.hostname}'`);
continue;
}
// Create new script if it does not already exist
const newScript = new Script(Player, file);
newScript.code = sourceScript.code;
newScript.ramUsage = sourceScript.ramUsage;
newScript.server = destServer.hostname;
destServer.scripts.push(newScript);
helpers.log(ctx, () => `File '${file}' copied over to '${destServer?.hostname}'.`);
newScript.updateRamUsage(Player, destServer.scripts);
} }
// Scp for script files return noFailures;
let sourceScript = null;
for (let i = 0; i < currServ.scripts.length; ++i) {
if (scriptName == currServ.scripts[i].filename) {
sourceScript = currServ.scripts[i];
break;
}
}
if (sourceScript == null) {
helpers.log(ctx, () => `File '${scriptName}' does not exist.`);
return false;
}
// Overwrite script if it already exists
for (let i = 0; i < destServer.scripts.length; ++i) {
if (scriptName == destServer.scripts[i].filename) {
helpers.log(ctx, () => `WARNING: File '${scriptName}' overwritten on '${destServer?.hostname}'`);
const oldScript = destServer.scripts[i];
// If it's the exact same file don't actually perform the
// copy to avoid recompiling uselessly. Players tend to scp
// liberally.
if (oldScript.code === sourceScript.code) return true;
oldScript.code = sourceScript.code;
oldScript.ramUsage = sourceScript.ramUsage;
oldScript.markUpdated();
return true;
}
}
// Create new script if it does not already exist
const newScript = new Script(Player, scriptName);
newScript.code = sourceScript.code;
newScript.ramUsage = sourceScript.ramUsage;
newScript.server = destServer.hostname;
destServer.scripts.push(newScript);
helpers.log(ctx, () => `File '${scriptName}' copied over to '${destServer?.hostname}'.`);
newScript.updateRamUsage(Player, destServer.scripts);
return true;
}, },
ls: ls:
(ctx: NetscriptContext) => (ctx: NetscriptContext) =>