90 lines
2.5 KiB
TypeScript
90 lines
2.5 KiB
TypeScript
|
import {NS} from "NetscriptDefinitions";
|
||
|
|
|
||
|
|
enum Type {
|
||
|
|
newNode = "node",
|
||
|
|
level = "level",
|
||
|
|
ram = "ram",
|
||
|
|
core = "code"
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function main(ns: NS) {
|
||
|
|
let timeout: number = <number>ns.args[0]
|
||
|
|
|
||
|
|
let nodes = ns.hacknet.numNodes()
|
||
|
|
let costs: {type: Type, cost: number}[] = []
|
||
|
|
while (true) {
|
||
|
|
costs = []
|
||
|
|
// Go through each node and get the cheapest upgrade
|
||
|
|
for (let i = 0; i < nodes; i++) {
|
||
|
|
costs.push(getCheapestCost(ns, i))
|
||
|
|
}
|
||
|
|
// Buy the cheapest upgrade from all nodes
|
||
|
|
let cheapest = Math.min(...costs.map(c => c.cost))
|
||
|
|
// Find the index of the cheapest cost in the list
|
||
|
|
let index = costs.findIndex(c => c.cost === cheapest)
|
||
|
|
// Wait to buy the cheapest upgrade
|
||
|
|
try {
|
||
|
|
await buy(ns, costs[index].type, costs[index].cost, index, timeout)
|
||
|
|
} catch (e) {
|
||
|
|
ns.tprint(e)
|
||
|
|
ns.exit()
|
||
|
|
}
|
||
|
|
// Make sure that the number of nodes is up-to-date
|
||
|
|
nodes = ns.hacknet.numNodes()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Wait until the player has enough money to buy something
|
||
|
|
* @param ns Global NS object
|
||
|
|
* @param money Amount of money to wait for
|
||
|
|
* @param timeout=-1 Number of seconds to wait before timing out.
|
||
|
|
* @throws Error if the timeout is reached
|
||
|
|
* @note This function will wait forever by default
|
||
|
|
*/
|
||
|
|
async function waitUntilMoney(ns: NS, money: number, timeout: number = -1) {
|
||
|
|
while (ns.getServerMoneyAvailable("home") < money) {
|
||
|
|
await ns.sleep(1000)
|
||
|
|
if (timeout == 0) {
|
||
|
|
throw new Error("Timed out waiting for money")
|
||
|
|
} else if (timeout > -1) {
|
||
|
|
timeout--;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
async function buy(ns: NS, type: Type, cost: number, node: number, timeout: number) {
|
||
|
|
await waitUntilMoney(ns, cost, timeout)
|
||
|
|
switch (type) {
|
||
|
|
case Type.newNode:
|
||
|
|
ns.hacknet.purchaseNode()
|
||
|
|
break
|
||
|
|
case Type.level:
|
||
|
|
ns.hacknet.upgradeLevel(node)
|
||
|
|
break
|
||
|
|
case Type.ram:
|
||
|
|
ns.hacknet.upgradeRam(node)
|
||
|
|
break
|
||
|
|
case Type.core:
|
||
|
|
ns.hacknet.upgradeCore(node)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function getCheapestCost(ns: NS, node: number) {
|
||
|
|
let nodeCost = ns.hacknet.getPurchaseNodeCost()
|
||
|
|
let levelCost = ns.hacknet.getLevelUpgradeCost(node)
|
||
|
|
let ramCost = ns.hacknet.getRamUpgradeCost(node)
|
||
|
|
let coreCost = ns.hacknet.getCoreUpgradeCost(node)
|
||
|
|
let cheapest = Math.min(nodeCost, levelCost, ramCost, coreCost)
|
||
|
|
switch (cheapest) {
|
||
|
|
case nodeCost:
|
||
|
|
return {type: Type.newNode, cost: nodeCost}
|
||
|
|
case levelCost:
|
||
|
|
return {type: Type.level, cost: levelCost}
|
||
|
|
case ramCost:
|
||
|
|
return {type: Type.ram, cost: ramCost}
|
||
|
|
case coreCost:
|
||
|
|
return {type: Type.core, cost: coreCost}
|
||
|
|
}
|
||
|
|
}
|