h3ml-update.js

/*
    h3ml update script, this module must not have any library dependency
    upload this script at home computer and run:
        wget https://raw.githubusercontent.com/AlexBurnes/h3ml/devel/h3ml-update.js h3ml-update.js
        run h3ml-update.js --version
        run h3ml-update.js
*/
"use strict";
const Module  = '/h3ml-update.js';
const Version = '0.3.5.12'; // update this every time when edit the code!!!

const baseUrl    = "https://raw.githubusercontent.com/AlexBurnes/h3ml/devel";

// core files required for updater
const files_list = [
    "/h3ml/var/files.js",
    "/h3ml/sbin/update-fetch.js",
    "/h3ml/etc/settings.js",
    "/h3ml/etc/scripts.js",
    "/h3ml/etc/servers.js",
    "/h3ml/etc/security.js",
    "/h3ml/lib/constants.js",
    "/h3ml/lib/log.js"
];

const update_fetch = "/h3ml/sbin/update-fetch.js";

async function version(ns, port) {
    if (port !== undefined && port) {
        const data = ns.sprintf("%d|%s|%s", Date.now(), Module, Version);
        return ns.tryWritePort(port, data);
    }
    ns.tprintf("module %s version %s", Module, Version);
    return;
}

function help(ns) {
    ns.tprintf("usage: %s --version [--update-port] | --help", Module);
    ns.tprintf("update script from github");
    return;
}

/** @param {NS} ns **/
export async function main(ns) {
    const args = ns.flags([
        [ 'version'     , false ],
        [ 'update-port' , 0     ],
        [ 'help'        , false ]
    ]);

    if (args['version']) {
        return version(ns, args['update-port']);
    }
    if (args['help']) {
        return help(ns);
    }
    ns.tprint(Module, " ", Version);
    const result = await update(ns);
    if (!result) {
        ns.tprintf("failed update");
        return;
    }
    ns.tprintf("done updating");
}

/** @param {import("Ns").NS } ns */
async function update(ns) {
    const host = ns.getHostname();
    const update_success = true;
    ns.tprintf("uploading core files from %s", baseUrl);
    for(let i = 0; i < files_list.length; i++) {
        const file = files_list[i];
        if (! await ns.wget(`${baseUrl}${file}`, file)) {
            ns.tprintf("[%d/%d] failed get %s", i+1, files_list.length, file);
            return false;
        }
        ns.tprintf("[%d/%d] %s uploaded", i+1, files_list.length, file);
    }

    // settings files, if not exists copy it, is user configurated file
    if (!ns.fileExists("h3ml-settings", host)) {
        await ns.mv(host, "/h3ml/etc/settings.js", "h3ml-settings.js");
    }

    ns.tprintf("run h3ml update-fetch to complite updating");
    const pid = ns.run(update_fetch, 1, baseUrl, host);
    if (pid == 0) return false;
    return true;
}