bitburner/scripts/copy.js
2022-03-17 12:19:54 -04:00

62 lines
2.1 KiB
JavaScript

import {ArgError, ArgParser} from '/scripts/lib/arg-parser';
import {copyWithDependencies, progressBar} from '/scripts/lib/utils';
/**
* BitBurner autocomplete
* @param data {server: string[], txts: string[], scripts: string[], flags: string[]} - Contextual information
* @returns {string[]} - Pool of autocomplete options
*/
export function autocomplete(data) {
return [...data.servers, ...data.scripts];
}
/** @param {NS} ns **/
export async function main(ns) {
// Setup
ns.disableLog('ALL');
const argParser = new ArgParser('copy.js', 'Copy a file/script to a device along with any dependencies.', null, [
{name: 'file', desc: 'File to copy', type: 'string'},
{name: 'device', desc: 'Device to copy file(s) to', type: 'string'},
{name: 'noDeps', desc: 'Skip copying dependencies', flags: ['-d', '--no-deps'], type: 'bool'},
{name: 'silent', desc: 'Suppress program output', flags: ['-s', '--silent'], type: 'bool'}
], true);
try {
// Run
const args = argParser.parse(ns.args);
// Banner
if(!args['silent']) {
ns.tprint('===================================================');
ns.tprint(`Copying: ${args['device']}`);
ns.tprint('===================================================');
ns.tprint('');
ns.tprint('Copying Files:');
await ns.sleep(500);
}
// Copy files & create download bar
if(args['noDeps']) {
await ns.scp(args['file'], args['device']);
if(!args['silent']) await progressBar(ns, args['file']);
} else {
const files = await copyWithDependencies(ns, args['file'], args['device']);
if(!args['silent']) {
for(let file of files) {
await progressBar(ns, file);
}
}
}
// Done message
if(!args['silent']) {
ns.tprint('');
ns.tprint('Done!');
ns.tprint('');
}
} catch(err) {
if(err instanceof ArgError) return ns.tprint(argParser.help(err.message));
throw err;
}
}