myserver.js

export function myServerPriceList(ns) {
  let purchasedServerCache = ns.getPurchasedServers();

  let ramNumber = 1, upgradeRamCost = 0, purchaseRamCost = 0;
  for (let i = 0; i < 20; i++) {
    ramNumber *= 2;

    for(let srv=0; srv < purchasedServerCache.length; srv++){
      let srvname = purchasedServerCache[srv];
      upgradeRamCost = ns.getPurchasedServerUpgradeCost(srvname, ramNumber);
      if(upgradeRamCost > 0) {
        break;
      }
    }
    purchaseRamCost = ns.getPurchasedServerCost(ramNumber);

    ns.tprintf("-----------------------------------------------------"); 
    if (upgradeRamCost > 0) {
      ns.tprintf("%s -> %i GB RAM would cost you\n\tNEW $%s / UPGRADE $%s", ns.formatRam(ramNumber), ramNumber, ns.formatNumber(purchaseRamCost), ns.formatNumber(upgradeRamCost));
    } else {
      ns.tprintf("%s -> %i GB RAM would cost you\n\tNEW $%s", ns.formatRam(ramNumber), ramNumber, ns.formatNumber(purchaseRamCost));
    }
  }
}

export function myServerPurchase(ns, ramNumber){
  const hostName = "srv";
  let ramCost = ns.getPurchasedServerCost(ramNumber);
  let buySuccess = ns.purchaseServer(hostName, ramNumber);

  ns.tprintf("Buying a server with (%s RAM) for $%s -> %s.", ns.formatRam(ramNumber), ns.formatNumber(ramCost), buySuccess ? "Success" : "Failed");
  return buySuccess;
}

export function myServerPurchaseAll(ns, ramNumber){
  let successCount = 0;

  while(true) {
    const upgradeSuccess = myServerPurchase(ns, ramNumber);

    if(upgradeSuccess) {
      successCount ++;
    } else {
      break; // cannot buy any more servers for any reason, breaks the loop.
    }
  }

  return successCount;
}

export function myServerUpgrade(ns, hostName, ramNumber) {
  let ramCost = ns.getPurchasedServerUpgradeCost(hostName, ramNumber);
  let upgradeSuccess = ns.upgradePurchasedServer(hostName, ramNumber);
  ns.tprintf("Upgrade %s to %s RAM for $%s -> %s.", hostName, ns.formatRam(ramNumber), ns.formatNumber(ramCost), upgradeSuccess ? "Success" : "Fail");
  return upgradeSuccess;
}

export function myServerUpgradeAll(ns, ramNumber){
  const purchasedServerCache = ns.getPurchasedServers();
  let successCount = 0;

  for(let srv=0; srv < purchasedServerCache.length; srv++){
    const serverName = purchasedServerCache[srv];
    const upgradeSuccess = myServerUpgrade(ns, serverName, ramNumber);

    if(upgradeSuccess) {
      successCount ++;
    }
  }
  return successCount;
}

export function myServerRename(ns, curHostName, newHostName) {
  let renameSuccess = ns.renamePurchasedServer(curHostName, newHostName);
  ns.tprintf("Rename server from %s to %s -> %s.", curHostName, newHostName, renameSuccess ? "Success" : "Failed");
  return renameSuccess;
}

export function myServerRenameAll(ns, templateHostName) {
  let renameSuccessCount = 0;
  let purchasedServerCache = ns.getPurchasedServers();
  for(let srv=0; srv < purchasedServerCache.length; srv++){
    let curHostName = purchasedServerCache[srv];
    let newHostName = ns.sprintf("%s-%i", templateHostName, srv+1)
    let renameSuccess = ns.renamePurchasedServer(curHostName, newHostName);
    if(renameSuccess) {
      renameSuccessCount++;
    }
  }
  return renameSuccessCount;
}

export function printMyServers(ns) {
  let purchasedServerCache = ns.getPurchasedServers();
  ns.tprintf("%i server(s)...", purchasedServerCache.length);
  for(let srv=0; srv < purchasedServerCache.length; srv++){
    const serverName = purchasedServerCache[srv];
    const serverRam = ns.getServerMaxRam(serverName);
    ns.tprintf("%s (%s RAM)", serverName, ns.formatRam(serverRam));
  }
}

/** @param {NS} ns */
export async function main(ns) {
  if(ns.args.length >= 1 && ns.args[0] == "prices") {
    myServerPriceList(ns);
  } else if(ns.args.length >= 1 && ns.args[0] == "purchase") {
    if(ns.args.length >= 2) {
      myServerPurchase(ns, ns.args[1]);
    } else {
      ns.tprintf("Parameters: <RAM GB>");
    }
  } else if(ns.args.length >= 1 && ns.args[0] == "purchaseall") {
    if(ns.args.length >= 2) {
      const successCount = myServerPurchaseAll(ns, ns.args[1]);
      if(successCount > 0) {
        printMyServers(ns);
      }
    } else {
      ns.tprintf("Parameters: <RAM GB>");
    }
  } else if(ns.args.length >= 1 && ns.args[0] == "upgrade") {
    if (ns.args.length >= 3) {
      myServerUpgrade(ns, ns.args[1], ns.args[2]); 
    } else {
      ns.tprintf("Parameters: <hostname> <RAM GB>");
    }
  } else if(ns.args.length >= 1 && ns.args[0] == "upgradeall") {
    if (ns.args.length >= 2) {
      const successCount = myServerUpgradeAll(ns, ns.args[1]); 
      if(successCount > 0) {
        printMyServers(ns);
      }
    } else {
      ns.tprintf("Parameters: <RAM GB>");
    }
  } else if(ns.args.length >= 1 && ns.args[0] == "rename") {
    if (ns.args.length >= 3) {
      myServerRename(ns, ns.args[1], ns.args[2]);
    } else {
      ns.tprintf("Parameters: <hostname> <new hostname>");
    }
  } else if(ns.args.length >= 1 && ns.args[0] == "renameall") {
    if (ns.args.length >= 2) {
      const templateName = ns.args[1];
      const successCount = myServerRenameAll(ns, templateName);
      if(successCount > 0) {
        printMyServers(ns);
      }
      ns.tprintf("%i servers were renamed with the template %s.", successCount, templateName);
    } else {
      ns.tprintf("Parameters: <new template hostname>");
    }
  } else if(ns.args.length >= 1 && ns.args[0] == "get") {
    printMyServers(ns);
  }
}