bitburner-src/src/HacknetNode.js

import {BitNodeMultipliers}                     from "./BitNode";
import {CONSTANTS}                              from "./Constants";
import {Engine}                                 from "./engine";
import {iTutorialSteps, iTutorialNextStep,
        iTutorialIsRunning, currITutorialStep}  from "./InteractiveTutorial";
import {Player}                                 from "./Player";
import {dialogBoxCreate}                        from "../utils/DialogBox";
import {clearEventListeners,
        getElementById}                         from "../utils/HelperFunctions";
import {Reviver, Generic_toJSON,
        Generic_fromJSON}                       from "../utils/JSONReviver";
import {createElement}                          from "../utils/uiHelpers/createElement";
import {formatNumber}                           from "../utils/StringHelperFunctions";

/**
 * Overwrites the inner text of the specified HTML element if it is different from what currently exists.
 * @param {string} elementId The HTML ID to find the first instance of.
 * @param {string} text The inner text that should be set.
 */
function updateText(elementId, text) {
    var el = getElementById(elementId);
    if (el.innerText != text) {
        el.innerText = text;
    }
};

/* HacknetNode.js */
function hacknetNodesInit() {
    var performMapping = function(x) {
        getElementById("hacknet-nodes-" + x.id + "-multiplier")
            .addEventListener("click", function() {
                hacknetNodePurchaseMultiplier = x.multiplier;
                updateHacknetNodesMultiplierButtons();
                updateHacknetNodesContent();
                return false;
            });
    };

    var mappings = [
        { id: "1x", multiplier: 1 },
        { id: "5x", multiplier: 5 },
        { id: "10x", multiplier: 10 },
        { id: "max", multiplier: 0 }
    ];
    for (var elem of mappings) {
        // Encapsulate in a function so that the appropriate scope is kept in the click handler.
        performMapping(elem);
    }
}

document.addEventListener("DOMContentLoaded", hacknetNodesInit, false);

function HacknetNode(name) {
    this.level      = 1;
    this.ram        = 1; //GB
    this.cores      = 1;

    this.name       = name;

    this.totalMoneyGenerated    = 0;
    this.onlineTimeSeconds      = 0;

    this.moneyGainRatePerSecond = 0;
}


HacknetNode.prototype.updateMoneyGainRate = function() {
    //How much extra $/s is gained per level
    var gainPerLevel = CONSTANTS.HacknetNodeMoneyGainPerLevel;

    this.moneyGainRatePerSecond = (this.level * gainPerLevel) *
                                  Math.pow(1.035, this.ram-1) *
                                  ((this.cores + 5) / 6) *
                                  Player.hacknet_node_money_mult *
                                  BitNodeMultipliers.HacknetNodeMoney;
    if (isNaN(this.moneyGainRatePerSecond)) {
        this.moneyGainRatePerSecond = 0;
        dialogBoxCreate("Error in calculating Hacknet Node production. Please report to game developer");
    }

    updateTotalHacknetProduction();
}

HacknetNode.prototype.calculateLevelUpgradeCost = function(levels=1) {
    levels = Math.round(levels);
    if (isNaN(levels) || levels < 1) {
        return 0;
    }

    var mult = CONSTANTS.HacknetNodeUpgradeLevelMult;
    var totalMultiplier = 0; //Summed
    var currLevel = this.level;
    for (var i = 0; i < levels; ++i) {
        totalMultiplier += Math.pow(mult, currLevel);
        ++currLevel;
    }

    return CONSTANTS.BaseCostForHacknetNode / 2 * totalMultiplier * Player.hacknet_node_level_cost_mult;
}

//Wrapper function for Netscript
HacknetNode.prototype.getLevelUpgradeCost = function(levels=1) {
    return this.calculateLevelUpgradeCost(levels);
}

HacknetNode.prototype.purchaseLevelUpgrade = function(levels=1) {
    levels = Math.round(levels);
    var cost = this.calculateLevelUpgradeCost(levels);
    if (isNaN(cost) || levels < 0) {
        return false;
    }

    if (this.level + levels > CONSTANTS.HacknetNodeMaxLevel) {
        var diff = Math.max(0, CONSTANTS.HacknetNodeMaxLevel - this.level);
        return this.purchaseLevelUpgrade(diff);
    }

    if (Player.money.lt(cost)) {
        return false;
    }

    Player.loseMoney(cost);
    this.level += levels;
    this.updateMoneyGainRate();
    return true;
}

//Wrapper function for Netscript
HacknetNode.prototype.upgradeLevel = function(levels=1) {
    let res = this.purchaseLevelUpgrade(levels);
    createPlayerHacknetNodeWrappers();
    return res;
}

HacknetNode.prototype.calculateRamUpgradeCost = function() {
    var numUpgrades = Math.log2(this.ram);

    //Calculate cost
    //Base cost of RAM is 50k per 1GB, increased by some multiplier for each time RAM is upgraded
    var baseCost = this.ram * CONSTANTS.BaseCostFor1GBOfRamHacknetNode;
    var mult = Math.pow(CONSTANTS.HacknetNodeUpgradeRamMult, numUpgrades);
    return baseCost * mult * Player.hacknet_node_ram_cost_mult;
}

//Wrapper function for Netscript
HacknetNode.prototype.getRamUpgradeCost = function() {
    return this.calculateRamUpgradeCost();
}

HacknetNode.prototype.purchaseRamUpgrade = function() {
    var cost = this.calculateRamUpgradeCost();
    if (isNaN(cost)) {
        return false;
    }

    if (Player.money.lt(cost)) {
        return false;
    }

    if (this.ram >= CONSTANTS.HacknetNodeMaxRam) {
        return false;
    }

    Player.loseMoney(cost);
    this.ram *= 2; //Ram is always doubled
    this.updateMoneyGainRate();
    return true;
}

//Wrapper function for Netscript
HacknetNode.prototype.upgradeRam = function() {
    let res = this.purchaseRamUpgrade();
    createPlayerHacknetNodeWrappers();
    return res;
}

HacknetNode.prototype.calculateCoreUpgradeCost = function() {
    var coreBaseCost = CONSTANTS.BaseCostForHacknetNodeCore;
    var mult = CONSTANTS.HacknetNodeUpgradeCoreMult;
    return coreBaseCost * Math.pow(mult, this.cores - 1) * Player.hacknet_node_core_cost_mult;
}

//Wrapper function for Netscript
HacknetNode.prototype.getCoreUpgradeCost = function() {
    let res = this.calculateCoreUpgradeCost();
    createPlayerHacknetNodeWrappers();
    return res;
}

HacknetNode.prototype.purchaseCoreUpgrade = function() {
    var cost = this.calculateCoreUpgradeCost();
    if (isNaN(cost)) {
        return false;
    }

    if (Player.money.lt(cost)) {
        return false;
    }

    if (this.cores >= CONSTANTS.HacknetNodeMaxCores) {
        return false;
    }

    Player.loseMoney(cost);
    ++this.cores;
    this.updateMoneyGainRate();
    return true;
}

//Wrapper function for Netscript
HacknetNode.prototype.upgradeCore = function() {
    return this.purchaseCoreUpgrade();
}

/* Saving and loading HackNets */
HacknetNode.prototype.toJSON = function() {
    return Generic_toJSON("HacknetNode", this);
}

HacknetNode.fromJSON = function(value) {
    return Generic_fromJSON(HacknetNode, value.data);
}

Reviver.constructors.HacknetNode = HacknetNode;

var HacknetNodeWrapper = function(hacknetNodeObj) {
    var _node                   = hacknetNodeObj;
    return {
        name                   : _node.name,
        level                  : _node.level,
        ram                    : _node.ram,
        cores                  : _node.cores,
        totalMoneyGenerated    : _node.totalMoneyGenerated,
        onlineTimeSeconds      : _node.onlineTimeSeconds,
        moneyGainRatePerSecond : _node.moneyGainRatePerSecond,
        upgradeLevel : function(n) {
            return _node.upgradeLevel(n);
        },
        upgradeRam : function() {
            return _node.upgradeRam();
        },
        upgradeCore : function() {
            return _node.upgradeCore();
        },
        getLevelUpgradeCost : function(n) {
            return _node.getLevelUpgradeCost(n);
        },
        getRamUpgradeCost : function() {
            return _node.getRamUpgradeCost();
        },
        getCoreUpgradeCost : function() {
            return _node.getCoreUpgradeCost();
        }
    }
}

function purchaseHacknet() {
    /* INTERACTIVE TUTORIAL */
    if (iTutorialIsRunning) {
        if (currITutorialStep == iTutorialSteps.HacknetNodesIntroduction) {
            iTutorialNextStep();
        } else {
            return;
        }
    }

    /* END INTERACTIVE TUTORIAL */

    var cost = getCostOfNextHacknetNode();
    if (isNaN(cost)) {
        throw new Error("Cost is NaN");
    }

    if (Player.money.lt(cost)) {
        //dialogBoxCreate("You cannot afford to purchase a Hacknet Node!");
        return false;
    }

    //Auto generate a name for the node for now...TODO
    var numOwned = Player.hacknetNodes.length;
    var name = "hacknet-node-" + numOwned;
    var node = new HacknetNode(name);
    node.updateMoneyGainRate();

    Player.loseMoney(cost);
    Player.hacknetNodes.push(node);

    if (Engine.currentPage === Engine.Page.HacknetNodes) {
        displayHacknetNodesContent();
    }
    createPlayerHacknetNodeWrappers();
    updateTotalHacknetProduction();
    return numOwned;
}

//Calculates the total production from all HacknetNodes
function updateTotalHacknetProduction() {
    var total = 0;
    for (var i = 0; i < Player.hacknetNodes.length; ++i) {
        total += Player.hacknetNodes[i].moneyGainRatePerSecond;
    }
    Player.totalHacknetNodeProduction = total;
}

function getCostOfNextHacknetNode() {
    //Cost increases exponentially based on how many you own
    var numOwned = Player.hacknetNodes.length;
    var mult = CONSTANTS.HacknetNodePurchaseNextMult;
    return CONSTANTS.BaseCostForHacknetNode * Math.pow(mult, numOwned) * Player.hacknet_node_purchase_cost_mult;
}

var hacknetNodePurchaseMultiplier = 1;
function updateHacknetNodesMultiplierButtons() {
    var mult1x = document.getElementById("hacknet-nodes-1x-multiplier");
    var mult5x = document.getElementById("hacknet-nodes-5x-multiplier");
    var mult10x = document.getElementById("hacknet-nodes-10x-multiplier");
    var multMax = document.getElementById("hacknet-nodes-max-multiplier");
    mult1x.setAttribute("class", "a-link-button");
    mult5x.setAttribute("class", "a-link-button");
    mult10x.setAttribute("class", "a-link-button");
    multMax.setAttribute("class", "a-link-button");
    if (Player.hacknetNodes.length == 0) {
        mult1x.setAttribute("class", "a-link-button-inactive");
        mult5x.setAttribute("class", "a-link-button-inactive");
        mult10x.setAttribute("class", "a-link-button-inactive");
        multMax.setAttribute("class", "a-link-button-inactive");
    } else if (hacknetNodePurchaseMultiplier == 1) {
        mult1x.setAttribute("class", "a-link-button-inactive");
    } else if (hacknetNodePurchaseMultiplier == 5) {
        mult5x.setAttribute("class", "a-link-button-inactive");
    } else if (hacknetNodePurchaseMultiplier == 10) {
        mult10x.setAttribute("class", "a-link-button-inactive");
    } else {
        multMax.setAttribute("class", "a-link-button-inactive");
    }
}

//Calculate the maximum number of times the Player can afford to upgrade
//a Hacknet Node's level"
function getMaxNumberLevelUpgrades(nodeObj) {
    if (Player.money.lt(nodeObj.calculateLevelUpgradeCost(1))) {
        return 0;
    }

    var min = 1;
    var max = CONSTANTS.HacknetNodeMaxLevel - 1;
    var levelsToMax = CONSTANTS.HacknetNodeMaxLevel - nodeObj.level;
    if (Player.money.gt(nodeObj.calculateLevelUpgradeCost(levelsToMax))) {
        return levelsToMax;
    }

    while (min <= max) {
        var curr = (min + max) / 2 | 0;
        if (curr != CONSTANTS.HacknetNodeMaxLevel &&
            Player.money.gt(nodeObj.calculateLevelUpgradeCost(curr)) &&
            Player.money.lt(nodeObj.calculateLevelUpgradeCost(curr + 1))) {
            return Math.min(levelsToMax, curr);
        } else if (Player.money.lt(nodeObj.calculateLevelUpgradeCost(curr))) {
            max = curr - 1;
        } else if (Player.money.gt(nodeObj.calculateLevelUpgradeCost(curr))) {
            min = curr + 1;
        } else {
            return Math.min(levelsToMax, curr);
        }
    }
}

//Creates Hacknet Node DOM elements when the page is opened
function displayHacknetNodesContent() {
    //Update Hacknet Nodes button
    var newPurchaseButton = clearEventListeners("hacknet-nodes-purchase-button");

    newPurchaseButton.addEventListener("click", function() {
        purchaseHacknet();
        return false;
    });

    //Handle Purchase multiplier buttons
    updateHacknetNodesMultiplierButtons();

    //Remove all old hacknet Node DOM elements
    var hacknetNodesList = document.getElementById("hacknet-nodes-list");
    while (hacknetNodesList.firstChild) {
        hacknetNodesList.removeChild(hacknetNodesList.firstChild);
    }

    //Then re-create them
    for (var i = 0; i < Player.hacknetNodes.length; ++i) {
        createHacknetNodeDomElement(Player.hacknetNodes[i]);
    }

    updateHacknetNodesContent();
}

//Update information on all Hacknet Node DOM elements
function updateHacknetNodesContent() {
    //Set purchase button to inactive if not enough money, and update its price display
    var cost = getCostOfNextHacknetNode();
    var purchaseButton = getElementById("hacknet-nodes-purchase-button");
    var formattedCost = formatNumber(cost, 2);

    updateText("hacknet-nodes-purchase-button", "Purchase Hacknet Node - $" + formattedCost);

    if (Player.money.lt(cost)) {
        purchaseButton.setAttribute("class", "a-link-button-inactive");
    } else {
        purchaseButton.setAttribute("class", "a-link-button");
    }

    //Update player's money
    updateText("hacknet-nodes-player-money", "$" + formatNumber(Player.money.toNumber(), 2));
    updateText("hacknet-nodes-total-production", "$" + formatNumber(Player.totalHacknetNodeProduction, 2) + " / second");

    //Update information in each owned hacknet node
    for (var i = 0; i < Player.hacknetNodes.length; ++i) {
        updateHacknetNodeDomElement(Player.hacknetNodes[i]);
    }
}

//Creates a single Hacknet Node DOM element
function createHacknetNodeDomElement(nodeObj) {
    var nodeName = nodeObj.name;

    var nodeLevelContainer = createElement("div", {
        class: "hacknet-node-level-container row",
        innerHTML: "<p>Level:</p><span class=\"text upgradable-info\" id=\"hacknet-node-level-" + nodeName + "\"></span>"
    });

    var nodeRamContainer = createElement("div", {
        class: "hacknet-node-ram-container row",
        innerHTML: "<p>RAM:</p><span class=\"text upgradable-info\" id=\"hacknet-node-ram-" + nodeName + "\"></span>"
    });

    var nodeCoresContainer = createElement("div", {
        class: "hacknet-node-cores-container row",
        innerHTML: "<p>Cores:</p><span class=\"text upgradable-info\" id=\"hacknet-node-cores-" + nodeName + "\"><span>"
    })
    var containingDiv = createElement("div", {
        class: "hacknet-node-container",
        innerHTML: "<div class=\"hacknet-node-name-container row\">" +
            "<p>Node name:</p>" +
            "<span class=\"text\" id=\"hacknet-node-name-" + nodeName + "\"></span>" +
            "</div>" + 
            "<div class=\"hacknet-node-production-container row\">" +
            "<p>Production:</p>" +
            "<span class=\"text\" id=\"hacknet-node-total-production-" + nodeName + "\"></span>" +
            "<span class=\"text\" id=\"hacknet-node-production-rate-" + nodeName + "\"></span>" +
            "</div>"
    });
    containingDiv.appendChild(nodeLevelContainer);
    containingDiv.appendChild(nodeRamContainer);
    containingDiv.appendChild(nodeCoresContainer);

    var listItem = createElement("li", {
        class: "hacknet-node"
    });
    listItem.appendChild(containingDiv);

    //Upgrade buttons
    nodeLevelContainer.appendChild(createElement("a", {
        id: "hacknet-node-upgrade-level-" + nodeName,
        class: "a-link-button-inactive",
        clickListener: function() {
            var numUpgrades = hacknetNodePurchaseMultiplier;
            if (hacknetNodePurchaseMultiplier == 0) {
                numUpgrades = getMaxNumberLevelUpgrades(nodeObj);
            }
            nodeObj.purchaseLevelUpgrade(numUpgrades);
            updateHacknetNodesContent();
            return false;
        }
    }));

    nodeRamContainer.appendChild(createElement("a", {
        id: "hacknet-node-upgrade-ram-" + nodeName,
        class: "a-link-button-inactive",
        clickListener: function() {
            nodeObj.purchaseRamUpgrade();
            updateHacknetNodesContent();
            return false;
        }
    }));

    nodeCoresContainer.appendChild(createElement("a", {
        id: "hacknet-node-upgrade-core-" + nodeName,
        class: "a-link-button-inactive",
        clickListener: function() {
            nodeObj.purchaseCoreUpgrade();
            updateHacknetNodesContent();
            return false;
        }
    }));

    document.getElementById("hacknet-nodes-list").appendChild(listItem);

    //Set the text and stuff inside the DOM element
    updateHacknetNodeDomElement(nodeObj);
}

//Updates information on a single hacknet node DOM element
function updateHacknetNodeDomElement(nodeObj) {
    var nodeName = nodeObj.name;

    updateText("hacknet-node-name-" + nodeName, nodeName);
    updateText("hacknet-node-total-production-" + nodeName, "$" + formatNumber(nodeObj.totalMoneyGenerated, 2));
    updateText("hacknet-node-production-rate-" + nodeName, "($" + formatNumber(nodeObj.moneyGainRatePerSecond, 2) + " / second)");
    updateText("hacknet-node-level-" + nodeName, nodeObj.level);
    updateText("hacknet-node-ram-" + nodeName, nodeObj.ram + "GB");
    updateText("hacknet-node-cores-" + nodeName, nodeObj.cores);

    //Upgrade level
    var upgradeLevelButton = getElementById("hacknet-node-upgrade-level-" + nodeName);

    if (nodeObj.level >= CONSTANTS.HacknetNodeMaxLevel) {
        updateText("hacknet-node-upgrade-level-" + nodeName, "MAX LEVEL");
        upgradeLevelButton.setAttribute("class", "a-link-button-inactive");
    } else {
        var multiplier = 0;
        if (hacknetNodePurchaseMultiplier == 0) {
            //Max
            multiplier = getMaxNumberLevelUpgrades(nodeObj);
        } else {
            var levelsToMax = CONSTANTS.HacknetNodeMaxLevel - nodeObj.level;
            multiplier = Math.min(levelsToMax, hacknetNodePurchaseMultiplier);
        }

        var upgradeLevelCost = nodeObj.calculateLevelUpgradeCost(multiplier);
        updateText("hacknet-node-upgrade-level-" + nodeName, "Upgrade x" + multiplier + " - $" + formatNumber(upgradeLevelCost, 2))
        if (Player.money.lt(upgradeLevelCost)) {
            upgradeLevelButton.setAttribute("class", "a-link-button-inactive");
        } else {
            upgradeLevelButton.setAttribute("class", "a-link-button");
        }
    }

    //Upgrade RAM
    var upgradeRamButton = getElementById("hacknet-node-upgrade-ram-" + nodeName);

    if (nodeObj.ram >= CONSTANTS.HacknetNodeMaxRam) {
        updateText("hacknet-node-upgrade-ram-" + nodeName, "MAX RAM");
        upgradeRamButton.setAttribute("class", "a-link-button-inactive");
    } else {
        var upgradeRamCost = nodeObj.calculateRamUpgradeCost();
        updateText("hacknet-node-upgrade-ram-" + nodeName, "Upgrade - $" + formatNumber(upgradeRamCost, 2));
        if (Player.money.lt(upgradeRamCost)) {
            upgradeRamButton.setAttribute("class", "a-link-button-inactive");
        } else {
            upgradeRamButton.setAttribute("class", "a-link-button");
        }
    }

    //Upgrade Cores
    var upgradeCoreButton = getElementById("hacknet-node-upgrade-core-" + nodeName);

    if (nodeObj.cores >= CONSTANTS.HacknetNodeMaxCores) {
        updateText("hacknet-node-upgrade-core-" + nodeName, "MAX CORES");
        upgradeCoreButton.setAttribute("class", "a-link-button-inactive");
    } else {
        var upgradeCoreCost = nodeObj.calculateCoreUpgradeCost();
        updateText("hacknet-node-upgrade-core-" + nodeName, "Upgrade - $" + formatNumber(upgradeCoreCost, 2));
        if (Player.money.lt(upgradeCoreCost)) {
            upgradeCoreButton.setAttribute("class", "a-link-button-inactive");
        } else {
            upgradeCoreButton.setAttribute("class", "a-link-button");
        }
    }
}

function createPlayerHacknetNodeWrappers() {
    Player.hacknetNodeWrappers.length = Player.hacknetNodes.length;
    for (var i = 0; i < Player.hacknetNodes.length; ++i) {
        Player.hacknetNodeWrappers[i] = new HacknetNodeWrapper(Player.hacknetNodes[i]);
    }
}

function updatePlayerHacknetNodeWrappers() {
    if (Player.hacknetNodeWrappers.length !== Player.hacknetNodes.length) {
        return createPlayerHacknetNodeWrappers();
    }

    for (var i = 0; i < Player.hacknetNodeWrappers.length; ++i) {
        if (!(Player.hacknetNodeWrappers[i] instanceof HacknetNodeWrapper)) {
            return createPlayerHacknetNodeWrappers();
        }

        Player.hacknetNodeWrappers[i].level                  = Player.hacknetNodes[i].level;
        Player.hacknetNodeWrappers[i].ram                    = Player.hacknetNodes[i].ram;
        Player.hacknetNodeWrappers[i].cores                  = Player.hacknetNodes[i].cores;
        Player.hacknetNodeWrappers[i].totalMoneyGenerated    = Player.hacknetNodes[i].totalMoneyGenerated;
        Player.hacknetNodeWrappers[i].onlineTimeSeconds      = Player.hacknetNodes[i].onlineTimeSeconds;
        Player.hacknetNodeWrappers[i].moneyGainRatePerSecond = Player.hacknetNodes[i].moneyGainRatePerSecond;
    }
}

function processAllHacknetNodeEarnings(numCycles) {
    var total = 0;
    updatePlayerHacknetNodeWrappers();
    for (var i = 0; i < Player.hacknetNodes.length; ++i) {
        total += processSingleHacknetNodeEarnings(numCycles, Player.hacknetNodes[i]);
    }

    return total;
}

function processSingleHacknetNodeEarnings(numCycles, nodeObj) {
    var cyclesPerSecond = 1000 / Engine._idleSpeed;
    var earningPerCycle = nodeObj.moneyGainRatePerSecond / cyclesPerSecond;
    if (isNaN(earningPerCycle)) {
        console.error("Hacknet Node '" + nodeObj.name + "' Calculated earnings is NaN");
        earningPerCycle = 0;
    }

    var totalEarnings = numCycles * earningPerCycle;
    nodeObj.totalMoneyGenerated += totalEarnings;
    nodeObj.onlineTimeSeconds += (numCycles * (Engine._idleSpeed / 1000));
    Player.gainMoney(totalEarnings);
    return totalEarnings;
}

function getHacknetNode(name) {
    for (var i = 0; i < Player.hacknetNodes.length; ++i) {
        if (Player.hacknetNodes[i].name == name) {
            return Player.hacknetNodes[i];
        }
    }

    return null;
}

export {
    HacknetNode,
    createPlayerHacknetNodeWrappers,
    displayHacknetNodesContent,
    getCostOfNextHacknetNode,
    getHacknetNode,
    getMaxNumberLevelUpgrades,
    hacknetNodesInit,
    processAllHacknetNodeEarnings,
    purchaseHacknet,
    updateHacknetNodesContent,
    updateHacknetNodesMultiplierButtons,
    updateTotalHacknetProduction
};