feat: choose random source

activity-generator/bin/index.js

@@ -7,7 +7,7 @@ import { program } from 'commander';
 import { select, input, confirm } from '@inquirer/prompts';
 import { v4 } from 'uuid';
 import { parse } from 'json2csv';
-import { getFrequency, getAmountInSats } from './validation/inputGetters.js';
+import { getFrequency, getAmountInSats, verifyPubKey } from './validation/inputGetters.js';
 program.requiredOption('--config <file>');
 program.option('--csv');
 program.parse();
@@ -54,6 +54,7 @@ async function buildControlNodes() {
                 console.log(`Node: ${node.alias} has no graph`)
                 return console.error("Please check that controlled nodes have open channels to other nodes")
             }
+
             //dump graph information
             nodeObj[current_node.identity_pubkey] = current_node;
             nodeObj[current_node.identity_pubkey].graph = nodeGraph;
@@ -87,17 +88,29 @@ async function promptForActivities() {
     let activity = {};
     activity.uuid = v4();
 
+    let sourceArray = [{
+        name: '(choose random)',
+        value: Object.values(nodeObj)[Math.floor(Math.random() * Object.values(nodeObj).length)].identity_pubkey
+    }, {
+        name: '(input pubkey)',
+        value: false
+    }]
+
     activity.src = await select({
         message: "Choose a source? \n",
-        choices: Object.keys(nodeObj).map(key => {
+        choices: sourceArray.concat(Object.keys(nodeObj).map(key => {
             let node = nodeObj[key];
             return {
                 name: `${node.alias}:  (${node.identity_pubkey})`,
                 value: node.identity_pubkey
             }
-        })
+        }))
     })
 
+    if (!activity.src) {
+        activity.src = await input({ message: 'Enter pubkey:' });
+    }
+
     let destArray = [{
         name: `(choose random)`,
         value: nodeObj[activity.src].possible_dests[Math.floor(Math.random() * nodeObj[activity.src].possible_dests.length)].pub_key
@@ -118,7 +131,14 @@ async function promptForActivities() {
     })
 
     if (!activity.dest) {
-        activity.dest = await input({ message: 'Enter pubkey:' });
+        // console.log(nodeObj[Object.keys(nodeObj)[0]].graph.nodes)
+        const singleNodeGraph = Object.values(nodeObj).find((node) => {
+            return node.graph.nodes
+        }).graph
+
+        const allPossibleNodes = singleNodeGraph.nodes.map((node) => node.pub_key)
+        activity.dest = await verifyPubKey(allPossibleNodes)
+        // activity.dest = await input({ message: 'Enter pubkey:' });
     }
 
     activity.action = await input({ message: 'What action?', default: "keysend" });

activity-generator/bin/validation/err_config.js

@@ -1,3 +1,4 @@
 export const ErrorConfig = {
-  "POSITIVE_INTEGER": "must be a positive interger"
+  "POSITIVE_INTEGER": "must be a positive interger",
+  "PUBKEY_NOT_FOUND": "pubkey does not exist on the network"
 }

activity-generator/bin/validation/inputGetters.js

@@ -32,5 +32,20 @@ const getAmountInSats = async () => {
   }
 }
 
+const verifyPubKey = async (allPossibleNodes) => {
+  try {
+    const pubkey = await input({ message: 'Enter pubkey:' });
+    const isValidPubkey = allPossibleNodes.find((_pubkey) => pubkey === _pubkey )
+    if (!isValidPubkey) {
+      throw new Error(ErrorConfig.PUBKEY_NOT_FOUND)
+    }
+    return pubkey
+  } catch (err) {
+    console.error(err.message)
+    if (err.message === ErrorConfig.PUBKEY_NOT_FOUND) {
+      return await verifyPubKey(allPossibleNodes)
+    }
+  }
+}
 
-export { getFrequency, getAmountInSats }
+export { getFrequency, getAmountInSats, verifyPubKey }