index.js

import { stat, writeFile, mkdir } from "fs";
import { homedir, platform } from "os";
import { spawn, exec } from "child_process";
import { createWallet, getWalletInfo, loadWallet, start } from "./rpc/commands.js";
/**
 *
 * @SPEC create multipule instances of a regtest node.
 *
 *
 */

// todo: Promot in a cli for user options. auth
// ttl on node dirs. how to set?
// delete old nodes, how to keep clean/manage nodes?
const nodes = 3; // How many node to be created.

const ip = "127.0.0.1:";
const linux = "/.bitcoin/";
//const windows = "UsersYourUserNameAppdataRoamingBitcoin"; // (Vista and 7)
const mac = "/Library/Application Support/Bitcoin/";
const env = platform();
const home = homedir();
const paths = [];
const ports = [];
const startPort = [8333]; // startPort 8333 is where the port generation starts from.

// Add os check to determine where data directories should be created.
const init = (env, home) => {
  const pkgName = "ledgerd"; // Directory name to build network in.
  if (env === "linux") {
    console.log(env);
    const url = home + linux + pkgName;
    return url;
  }

  if (env === "darwin") {
    console.log(env);
    const url = home + mac + pkgName;
    return url;
  }
};

// Determine database location
const db = init(env, home);

for (let i = nodes; i--; ) {
  // Generate paths
  const url = db + "/node" + i + "/";
  paths.push(url);
  // Sequential port generation.
  const port = startPort[0]++; // 8333++
  ports.push({ port: port + nodes, rpc: port });
}

for (let i = paths.length; i--; ) {
  // Generate directories
  stat(paths[i], (err, stats) => {
    if (err) console.log(`Directory doesn't exist, making now...`);
    if (stats) {
      console.log("Directories ready:", true);
    } else {
      mkdir(paths[i], { recursive: true }, (err) => {
        if (err) {
          console.log("errrrr", err);
        }
      });
      console.log("created dir:", paths[i]);
    }
  });
}

// if (ports.length && paths.length > 0) {
//   console.log("auth handeling can go here");
// }

//       **Important in production code:**
// todo: need auth handeling: ./auth.js
// todo: Update bitcoind auth settings to latest standard: rpcauth=user:passwordhash
//       to use "rpcauth=" in regtest with multipule nodes requires unique hashes for each node.
// const rpcauth = {
//   user: "jamesdon",
//   password: ":01714fd71a8c04711b73191ebcd2c7bb$20860750b73dbc98d5a793cc92a91c89541dd3beceac51bcb91a3c4d9268e23a",
// };
// rpcauth=${auth.user + auth.password}
if (paths.length === 0) {
  console.log("ERROR: no paths found..");
}

for (let i = ports.length; i--; ) {
  const port = ports[i].port;
  const rpc = ports[i].rpc;
  // using unsecure auth settings "rpcuser" and "rpcpassword"
  const conf = `
        regtest=1
        debug=rpc
        server=1
        rpcuser=jamesdon
        rpcpassword=thisisnotapassword
        datadir=${db}/node${i}/
        [regtest]
        bind=127.0.0.1:${port}
        rpcport=${rpc}
        port=${port}

        `;
  writeFile(`${db}/node${i}/bitcoin.conf`, conf, (err) => {
    if (err) console.log(err);
    console.log("bitcoin.conf:", true);
  });
}

// Node env is set at this point.
const nodesReady = [];
for (let i = paths.length; i--; ) {
  // Build a "ready to run" node Object.
  const nodeName = paths[i].slice(58, 63);
  const node = {
    name: nodeName,
    key: nodeName.slice(4, 5),
    ports: ports[i],
    paths: paths[i],
    running: false,
    wallets: [],
    addresses: [],
    account: [],
  };

  nodesReady.push(node);
}
// Time to run the start bitcoind.
if (nodesReady.length > 0) {
  for (let i = nodesReady.length; i--; ) {
    start(db, nodesReady[i], nodesReady[i].key);
  }
}

// using setTimeout to "hang" the process allowing for time for start up.
setTimeout(() => {
  nodeController(nodesReady);
}, 2500);

/**
 *
 * @spec
 *
 */
const nodesActive = [];
const walletActivate = async () => {
  for (let i = nodesReady.length; i--; ) {
    const loaded = await loadWallet(db, nodesReady[i]);

    if (loaded === "new wallet!") {
      const wallet = await createWallet(db, nodesReady[i]);
    //  nodesActive.push(wallet);
    } 
    
      if (nodesActive.length <= nodes) {
       // console.log('nodesActive length:', nodesActive)
     //   nodesActive.push(loaded[0]);
      } 
    
    if (loaded[1] === 'wallet active') {
    const info = await getWalletInfo(db, nodesReady[i]);
    nodesActive.push(info)
   // console.log('info:::', info)
    }
  }
  console.log('nodessss', nodesActive)
};
const nodeController = (nodesReady) => {
  process.stdout.write(
    "1) Activate wallet \n2) Add new addresses. \n3) Add new addresses. : \n4) Exit program. : 5 \n"
  );
  process.stdin.resume();
  process.stdin.setEncoding("utf-8");

  let buffer = "";
  process.stdin.on("data", (data) => {
    buffer = data.trim();
    switch (buffer) {
      case "1":
        walletActivate();
        break;
      case "2":
        for (let i = nodesReady.length; i--; ) {
           // could be added to walletActivate func?
        };
        break;
      case "3":
        for (let i = nodesReady.length; i--; ) {
          // const start = spawn(
          //   "bitcoin-cli",
          //   [
          //     "-regtest",
          //     `-rpcport=${nodesReady[i].ports.rpc}`,
          //     "-datadir=" + db +  "/node" + nodesReady[i].key,
          //     "getnewaddress", // add address type.
          //   ],
          //   { encoding: "utf-8", stdio: "pipe" }
          // );
          // start.stderr.on("data", (data) => {
          //   console.log("Gen new address error:", data.toString());
          // });
          // start.stdout.on("data", (data) => {
          //   if (data) {
          //     const address = {
          //       key: nodesReady[i].key,
          //       node: nodesReady[i].name,
          //       address: data,
          //     };
          //     console.log("Address ready:", address);
          //     nodesReady[i].address.push(address);
          //   }
          // });
          // start.on("close", (data) => {
          //   console.log("Node started, closing processes. ", data);
          //   console.log("node started:", nodesReady[i].running);
          // });
        };

        break;
      case "4":
        for (let i = nodesReady.length; i--; ) {
          const start = spawn(
            "bitcoin-cli",
            [
              "-regtest",
              `-rpcport=${nodesReady[i].ports.rpc}`,
              "-datadir=" + db + "/node" + nodesReady[i].key,
              "getaccount",
            ],
            { encoding: "utf-8", stdio: "pipe" }
          );
          start.stderr.on("data", (data) => {
            console.log("Get account details error:", data.toString());
          });

          start.stdout.on("data", (data) => {
            if (data) {
              const address = {
                key: nodesReady[i].key,
                node: nodesReady[i].name,
                address: data,
              };
              console.log("Address ready:", address);
              nodesReady[i].address.push(address);
            }
          });

          start.on("close", (data) => {
            console.log("Node started, closing processes. ", data);
            console.log("node started:", nodesReady[i].running);
          });
        }

        break;
      case "5":
        console.log("Closing cli, bitcoind daemon still alive...");
        process.exit(0);
      case "6":
        console.log("Nodes ready:", nodesReady);
        break;

      default:
        console.log("invalid option.");
    }
  });
};

// "SIGINT" signal interupt which triggers node stop command.
// if the need to abort arisies use "control + c". (mac)
process.on("SIGINT", () => {
  console.log("\nInterupting process.");
  // Stop bitcoind
  for (let i = nodesReady.length; i--; ) {
    const bitcoinStop = `bitcoin-cli -rpcport=${nodesReady[i].ports.rpc}  -datadir="${db}/node${nodesReady[i].key}" stop`;
    //  console.log("stop commands here:", bitcoinStop);
    exec(bitcoinStop, (error, stdout, stderr) => {
      if (error) {
        console.log("error", error);
        process.exit(0);
      }
      if (stderr) {
        console.log("stderr", stderr);
        process.exit(0);
      }
      if (stdout) {
        console.log(stdout, nodesReady[i].key);
        nodesReady[i].running = false;
        process.exit(0);
      }
    });
  }
});