fix: allow custom user address

Author: janek26

index.ts

@@ -14,6 +14,7 @@ import { selectAction } from "./actions";
 import { unionWith } from "lodash";
 import { oraLog } from "./oraLog";
 import { showTransferAll } from "./actions/transferAll/ui";
+import { extraAccount } from "./ui/extraAccount";
 
 program
   .name("Argent X CLI")
@@ -36,16 +37,21 @@ program.parse();
 (async () => {
   const spinner = ora();
 
-  const { accounts, network, defaultPrivateKey } = await getAccountsAndNetwork(
+  let { accounts, network, defaultPrivateKey } = await getAccountsAndNetwork(
     spinner
   );
 
+  spinner.succeed("Found " + accounts.length + " wallets");
+
+  if (accounts.length === 0) {
+    accounts = await extraAccount(network, defaultPrivateKey);
+  }
+
   const accountInfos = await getAccountInfos(
     accounts.map((x) => x.address),
     network,
     spinner
   );
-  spinner.succeed("Found " + accounts.length + " wallets");
 
   const accountWithSigner: Account[] = accounts.map((account, i) => ({
     ...account,

ui/extraAccount.ts

@@ -0,0 +1,56 @@
+import { isString } from "lodash";
+import prompts from "prompts";
+import { ValidationError } from "yup";
+import { addressSchema } from "../schema";
+
+export async function extraAccount(
+  network: "mainnet-alpha" | "goerli-alpha",
+  privateKey: string
+): Promise<
+  {
+    address: string;
+    networkId: string;
+    privateKey: string;
+  }[]
+> {
+  const { extraWalletAddress } = await prompts(
+    [
+      {
+        type: "text",
+        name: "extraWalletAddress",
+        message: "Enter wallet address you want to use",
+        validate: async (value: string, prev) => {
+          try {
+            const address = value.replace(" ", "");
+            if (!address.length) {
+              return "You must select one account";
+            }
+            try {
+              await addressSchema.validate(address);
+            } catch (e) {
+              return `Invalid address '${address}'.${
+                e instanceof ValidationError ? ` Reason: ${e.message}` : ""
+              }`;
+            }
+          } catch (e) {
+            return "Invalid address";
+          }
+          return true;
+        },
+      },
+    ],
+    { onCancel: () => process.exit(1) }
+  );
+
+  if (!isString(extraWalletAddress)) {
+    throw new Error("Invalid address");
+  }
+
+  return [
+    {
+      address: extraWalletAddress,
+      networkId: network,
+      privateKey,
+    },
+  ];
+}