mirror of
https://github.com/bitburner-official/bitburner-src.git
synced 2024-10-23 18:23:15 +02:00
ef23033b2c
For some time now, filenames have included the extension, so there's no need to append ".js" either on a single-file download or zip download. While I was in the area, I also fixed .txt files in a folder - they had the same bug as scripts used to (appear in the zip as an underscore). I consolidated the code since it was the same for script files and text files. I also added the ability to download *.js and *.ns (previously only supported *, *.script, and *.txt) I also made the wildcard matching just a bit more flexible, so if someone really wanted to, they could go "download *blah.js" and get all scripts ending with that pattern ¯\_(ツ)_/¯
70 lines
2.8 KiB
TypeScript
70 lines
2.8 KiB
TypeScript
import { ITerminal } from "../ITerminal";
|
|
import { IRouter } from "../../ui/Router";
|
|
import { IPlayer } from "../../PersonObjects/IPlayer";
|
|
import { BaseServer } from "../../Server/BaseServer";
|
|
import { isScriptFilename } from "../../Script/isScriptFilename";
|
|
import FileSaver from "file-saver";
|
|
import JSZip from "jszip";
|
|
|
|
export function download(
|
|
terminal: ITerminal,
|
|
router: IRouter,
|
|
player: IPlayer,
|
|
server: BaseServer,
|
|
args: (string | number)[],
|
|
): void {
|
|
try {
|
|
if (args.length !== 1) {
|
|
terminal.error("Incorrect usage of download command. Usage: download [script/text file]");
|
|
return;
|
|
}
|
|
const fn = args[0] + "";
|
|
// If the parameter starts with *, download all files that match the wildcard pattern
|
|
if (fn.startsWith("*")) {
|
|
const matchEnding = fn.length == 1 || fn === "*.*" ? null : fn.slice(1); // Treat *.* the same as *
|
|
const zip = new JSZip();
|
|
// Helper function to zip any file contents whose name matches the pattern
|
|
let zipFiles = (fileNames: string[], fileContents: string[]) => {
|
|
for (let i = 0; i < fileContents.length; ++i) {
|
|
let name = fileNames[i];
|
|
if (name.startsWith("/")) name = name.slice(1);
|
|
if (!matchEnding || name.endsWith(matchEnding))
|
|
zip.file(name, new Blob([fileContents[i]], { type: "text/plain" }));
|
|
}
|
|
};
|
|
// In the case of script files, we pull from the server.scripts array
|
|
if (!matchEnding || isScriptFilename(matchEnding))
|
|
zipFiles(server.scripts.map(s => s.filename), server.scripts.map(s => s.code));
|
|
// In the case of text files, we pull from the server.scripts array
|
|
if (!matchEnding || matchEnding.endsWith(".txt"))
|
|
zipFiles(server.textFiles.map(s => s.fn), server.textFiles.map(s => s.text));
|
|
// Return an error if no files matched, rather than an empty zip folder
|
|
if (Object.keys(zip.files).length == 0)
|
|
return terminal.error(`No files match the pattern ${fn}`);
|
|
const zipFn = `bitburner${isScriptFilename(fn) ? "Scripts" : fn === "*.txt" ? "Texts" : "Files"}.zip`;
|
|
zip.generateAsync({ type: "blob" }).then((content: any) => FileSaver.saveAs(content, zipFn));
|
|
return;
|
|
} else if (isScriptFilename(fn)) {
|
|
// Download a single script
|
|
const script = terminal.getScript(player, fn);
|
|
if (script != null) {
|
|
return script.download();
|
|
}
|
|
} else if (fn.endsWith(".txt")) {
|
|
// Download a single text file
|
|
const txt = terminal.getTextFile(player, fn);
|
|
if (txt != null) {
|
|
return txt.download();
|
|
}
|
|
} else {
|
|
terminal.error(`Cannot download this filetype`);
|
|
return;
|
|
}
|
|
terminal.error(`${fn} does not exist`);
|
|
return;
|
|
} catch (e) {
|
|
terminal.error(e + "");
|
|
return;
|
|
}
|
|
}
|