From 72ea910c8da0bbd38fd68c351d736ac1787d8c82 Mon Sep 17 00:00:00 2001
From: Zak Timson <zaktimson@gmail.com>
Date: Wed, 9 Mar 2022 19:24:05 +0000
Subject: [PATCH] Updated update script

---
 scripts/lib/utils.js |  2 +-
 scripts/update.js    | 17 ++++++++++++-----
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/scripts/lib/utils.js b/scripts/lib/utils.js
index d09ff2b..dc7ef5b 100644
--- a/scripts/lib/utils.js
+++ b/scripts/lib/utils.js
@@ -5,7 +5,7 @@
  */
 export async function downloadPrint(ns, file) {
 	const speed = ~~(Math.random() * 100) / 10;
-	const spacing = Array(5 - Math.floor((file.length) / 8)).fill('\t').join('');
+	const spacing = Array(40 - file.length || 1).fill(' ').join('');
 	await slowPrint(ns, `${file}${spacing}[==================>] 100% \t (${speed} MB/s)`);
 }
 
diff --git a/scripts/update.js b/scripts/update.js
index ba1dc28..c1d1534 100644
--- a/scripts/update.js
+++ b/scripts/update.js
@@ -121,10 +121,11 @@ export async function main(ns) {
 	ns.disableLog('ALL');
     const updateFile = 'update.js';
 	const argParser = new ArgParser(updateFile, 'Download the latest script updates from the repository using wget.', null, [
-		{name: 'target', desc: 'Target device to update, defaults to current machine', optional: true, default: ns.getHostname(), type: 'string'}
+		{name: 'skip', desc: 'Skip updating self (for debugging)', optional: true, type: 'bool'},
+		{name: 'device', desc: 'Device to update, defaults to current machine', flags: ['-d', '--device'], default: ns.getHostname(), type: 'string'}
 	]);
 	const src = 'https://gitlab.zakscode.com/ztimson/BitBurner/-/raw/develop/scripts/';
-    const dest = '/scripts2/';
+    const dest = '/scripts/';
     const fileList = [
 		'lib/arg-parser.js',
 		'lib/utils.js',
@@ -143,17 +144,23 @@ export async function main(ns) {
 		throw err;
 	}
     
-    if(!ns.args.length) { // Update self & restart
+	// Banner
+	ns.tprint('===================================================');
+	ns.tprint(`Updating: ${args['device']}`);
+	ns.tprint('===================================================');
+
+	// Run
+    if(!args['skip']) { // Update self & restart
         await slowPrint(ns, 'Updating self:');
         await ns.wget(`${src}${updateFile}`, `${dest}${updateFile}`, args['target']);
         await downloadPrint(ns, `${dest}${updateFile}`);
         ns.tprint('');
         await slowPrint(ns, 'Restarting...');
+		ns.tprint('');
         return ns.exec(`${dest}${updateFile}`, args['target'], 1, 1);
     } else { // Update everything else
-        ns.tprint('');
         await slowPrint(ns, 'Downloading scripts:');
-        for(let file in fileList) {
+        for(let file of fileList) {
             await ns.wget(`${src}${file}`, `${dest}${file}`, args['target']);
             await downloadPrint(ns, `${dest}${file}`);
         }