@consensys-software/permissioning-smart-contracts/migrations/3_deploy_node_ingress_rules_contract.js

Smart contracts and dapp implementing EEA spec onchain permissioning

const Web3Utils = require("web3-utils");
const AllowlistUtils = require('../scripts/allowlist_utils');

const Rules = artifacts.require("./NodeRules.sol");
const NodeIngress = artifacts.require("./NodeIngress.sol");
const NodeStorage = artifacts.require("./NodeStorage.sol");
const Admin = artifacts.require("./Admin.sol");

const adminContractName = Web3Utils.utf8ToHex("administration");
const rulesContractName = Web3Utils.utf8ToHex("rules");

/* The address of the node ingress contract if pre deployed */
let nodeIngress = process.env.NODE_INGRESS_CONTRACT_ADDRESS;
/* The address of the node storage contract if pre deployed */
let nodeStorage = process.env.NODE_STORAGE_CONTRACT_ADDRESS;
let retainCurrentRulesContract = AllowlistUtils.getRetainNodeRulesContract();

async function logCurrentAllowlist(instance) {
    let s = await instance.getSize();
    console.log("\n<<< current NODE allowlist >>>");
    for (i = 0; i < s; i ++) {
        let x = await instance.getByIndex(i);
        console.log("enode://" + x[0] + "@" + x[1] + ":" + x[2]);
    }
    console.log("<<< end of current NODE list >>>");
}

module.exports = async(deployer, network) => {
    // exit early if we are NOT redeploying this contract
    if (retainCurrentRulesContract) {
        console.log("not deploying NodeRules because retain=" + retainCurrentRulesContract);
        logCurrentAllowlist(await Rules.deployed());
        return;
    }
    if (! nodeIngress) {
        // Only deploy if we haven't been provided a predeployed address
        await deployer.deploy(NodeIngress);
        console.log("   > Deployed NodeIngress contract to address = " + NodeIngress.address);
        nodeIngress = NodeIngress.address;

    }
    // If supplied an address, make sure there's something there
    const nodeIngressInstance = await NodeIngress.at(nodeIngress);
    try {
        const result = await nodeIngressInstance.getContractVersion();
        console.log("   > NodeIngress contract initialised at address = " + nodeIngress + " version=" + result);
    } catch (err) {
        console.log(err);
        console.error("   > Predeployed NodeIngress contract is not responding like an NodeIngress contract at address = " + nodeIngress);
    }

    const admin = await Admin.deployed()
    await nodeIngressInstance.setContractAddress(adminContractName, admin.address);
    console.log("   > Updated NodeIngress with Admin  address = " + admin.address);

    // STORAGE
    var storageInstance;
    if (! nodeStorage) {
        // Only deploy if we haven't been provided a pre-deployed address
        storageInstance = await deployer.deploy(NodeStorage, nodeIngress);
        console.log("   > Deployed NodeStorage contract to address = " + NodeStorage.address);
        nodeStorage = NodeStorage.address;
    } else {
        // is there a storage already deployed
        storageInstance = await NodeStorage.at(nodeStorage);
        console.log(">>> Using existing NodeStorage " + storageInstance.address);
        // TODO check that this contract is a storage contract eg call a method
    }

    // rules -> storage
    await deployer.deploy(Rules, nodeIngress, nodeStorage);
    console.log("   > Rules deployed with NodeIngress.address = " + nodeIngress + "\n   > and storageAddress = " + nodeStorage);
    console.log("   > Rules.address " + Rules.address);
    let nodeRulesContract = await Rules.deployed();
    
    // storage -> rules
    await storageInstance.upgradeVersion(Rules.address);
    console.log("   >>> Set storage owner to Rules.address " + Rules.address);

    await nodeIngressInstance.setContractAddress(rulesContractName, Rules.address);
    console.log("   > Updated NodeIngress contract with NodeRules address = " + Rules.address);

    if(AllowlistUtils.isInitialAllowlistedNodesAvailable()) {
        console.log("   > Adding Initial Allowlisted eNodes ...");
        let allowlistedNodes = AllowlistUtils.getInitialAllowlistedNodes();
        for (i = 0; i < allowlistedNodes.length; i++) {
            let enode = allowlistedNodes[i];
            const { enodeId, host, port } = AllowlistUtils.enodeToParams(enode);
            
            let result = await nodeRulesContract.addEnode(
                enodeId,
                host,    
                Web3Utils.toBN(port)
            );
            console.log("     > eNode added: " + enode );
        }
    }

    logCurrentAllowlist(nodeRulesContract);
}