@hypurrquant/defi-cli 0.2.3 → 0.2.4

package/dist/main.js

@@ -1,5 +1,9 @@
 #!/usr/bin/env node
 
+// src/main.ts
+import { config } from "dotenv";
+import { resolve as resolve4 } from "path";
+
 // src/cli.ts
 import { Command } from "commander";
 import { createRequire } from "module";
@@ -8223,6 +8227,150 @@ function registerFarm(parent, getOpts, makeExecutor2) {
   });
 }
 
+// src/commands/setup.ts
+import pc2 from "picocolors";
+import { createInterface } from "readline";
+import { existsSync as existsSync3, mkdirSync as mkdirSync2, readFileSync as readFileSync3, writeFileSync as writeFileSync2 } from "fs";
+import { resolve as resolve3 } from "path";
+var DEFI_DIR = resolve3(process.env.HOME || "~", ".defi");
+var ENV_FILE = resolve3(DEFI_DIR, ".env");
+function ensureDefiDir() {
+  if (!existsSync3(DEFI_DIR)) mkdirSync2(DEFI_DIR, { recursive: true, mode: 448 });
+}
+function loadEnvFile() {
+  if (!existsSync3(ENV_FILE)) return {};
+  const lines = readFileSync3(ENV_FILE, "utf-8").split("\n");
+  const env = {};
+  for (const line of lines) {
+    const trimmed = line.trim();
+    if (!trimmed || trimmed.startsWith("#")) continue;
+    const eqIdx = trimmed.indexOf("=");
+    if (eqIdx > 0) {
+      env[trimmed.slice(0, eqIdx)] = trimmed.slice(eqIdx + 1);
+    }
+  }
+  return env;
+}
+function writeEnvFile(env) {
+  ensureDefiDir();
+  const lines = [
+    "# defi-cli configuration",
+    "# Generated by 'defi setup' \u2014 edit freely",
+    ""
+  ];
+  for (const [key, value] of Object.entries(env)) {
+    lines.push(`${key}=${value}`);
+  }
+  lines.push("");
+  writeFileSync2(ENV_FILE, lines.join("\n"), { mode: 384 });
+}
+function ask(rl, question) {
+  return new Promise((res) => rl.question(question, (answer) => res(answer.trim())));
+}
+function isValidAddress(s) {
+  return /^0x[0-9a-fA-F]{40}$/.test(s);
+}
+function isValidPrivateKey(s) {
+  return /^0x[0-9a-fA-F]{64}$/.test(s);
+}
+async function deriveAddress(privateKey) {
+  try {
+    const { privateKeyToAccount: privateKeyToAccount3 } = await import("viem/accounts");
+    const account = privateKeyToAccount3(privateKey);
+    return account.address;
+  } catch {
+    return null;
+  }
+}
+function registerSetup(program2) {
+  program2.command("setup").alias("init").description("Interactive setup wizard \u2014 configure wallet & RPC URLs").action(async () => {
+    const rl = createInterface({ input: process.stdin, output: process.stdout });
+    try {
+      console.log(pc2.cyan(pc2.bold("\n  defi-cli Setup Wizard\n")));
+      const existing = loadEnvFile();
+      if (Object.keys(existing).length > 0) {
+        console.log(pc2.white("  Current configuration:"));
+        for (const [key, value] of Object.entries(existing)) {
+          const masked = key.toLowerCase().includes("key") ? value.slice(0, 6) + "..." + value.slice(-4) : value;
+          console.log(`    ${pc2.cyan(key.padEnd(24))} ${pc2.gray(masked)}`);
+        }
+        console.log();
+        const overwrite = await ask(rl, "  Overwrite existing config? (y/N): ");
+        if (overwrite.toLowerCase() !== "y" && overwrite.toLowerCase() !== "yes") {
+          console.log(pc2.gray("\n  Keeping existing configuration.\n"));
+          rl.close();
+          return;
+        }
+        console.log();
+      }
+      const newEnv = {};
+      console.log(pc2.cyan(pc2.bold("  Wallet")));
+      const privateKey = await ask(rl, "  Private key (optional, for --broadcast, 0x...): ");
+      if (privateKey) {
+        const normalized = privateKey.startsWith("0x") ? privateKey : `0x${privateKey}`;
+        if (!isValidPrivateKey(normalized)) {
+          console.log(pc2.red("  Invalid private key (must be 0x + 64 hex chars). Skipped."));
+        } else {
+          newEnv.DEFI_PRIVATE_KEY = normalized;
+          const derived = await deriveAddress(normalized);
+          if (derived) {
+            newEnv.DEFI_WALLET_ADDRESS = derived;
+            console.log(`  ${pc2.green("OK")} derived address: ${pc2.gray(derived)}`);
+          }
+        }
+      }
+      if (!newEnv.DEFI_WALLET_ADDRESS) {
+        const address = await ask(rl, "  Wallet address (0x...): ");
+        if (address) {
+          if (!isValidAddress(address)) {
+            console.log(pc2.yellow("  Invalid address format. Skipping."));
+          } else {
+            newEnv.DEFI_WALLET_ADDRESS = address;
+            console.log(`  ${pc2.green("OK")} ${pc2.gray(address)}`);
+          }
+        }
+      }
+      console.log(pc2.cyan(pc2.bold("\n  RPC URLs")) + pc2.gray(" (press Enter to use public defaults)"));
+      const hyperevmRpc = await ask(rl, "  HyperEVM RPC URL: ");
+      if (hyperevmRpc) {
+        newEnv.HYPEREVM_RPC_URL = hyperevmRpc;
+        console.log(`  ${pc2.green("OK")} HyperEVM RPC set`);
+      }
+      const mantleRpc = await ask(rl, "  Mantle RPC URL: ");
+      if (mantleRpc) {
+        newEnv.MANTLE_RPC_URL = mantleRpc;
+        console.log(`  ${pc2.green("OK")} Mantle RPC set`);
+      }
+      const finalEnv = { ...existing, ...newEnv };
+      writeEnvFile(finalEnv);
+      console.log(pc2.cyan(pc2.bold("\n  Setup Complete!\n")));
+      console.log(`  Config:  ${pc2.gray(ENV_FILE)}`);
+      if (finalEnv.DEFI_WALLET_ADDRESS) {
+        console.log(`  Wallet:  ${pc2.gray(finalEnv.DEFI_WALLET_ADDRESS)}`);
+      }
+      if (finalEnv.DEFI_PRIVATE_KEY) {
+        console.log(`  Key:     ${pc2.green("configured")}`);
+      }
+      if (finalEnv.HYPEREVM_RPC_URL) {
+        console.log(`  HyperEVM RPC: ${pc2.gray(finalEnv.HYPEREVM_RPC_URL)}`);
+      }
+      if (finalEnv.MANTLE_RPC_URL) {
+        console.log(`  Mantle RPC:   ${pc2.gray(finalEnv.MANTLE_RPC_URL)}`);
+      }
+      console.log(pc2.bold(pc2.white("\n  Next steps:")));
+      console.log(`    ${pc2.green("defi portfolio")}          view balances & positions`);
+      console.log(`    ${pc2.green("defi scan")}               scan for exploits`);
+      console.log(`    ${pc2.green("defi dex quote")}          get a swap quote`);
+      console.log(`    ${pc2.green("defi --help")}             browse all commands
+`);
+      rl.close();
+    } catch (err) {
+      rl.close();
+      throw err;
+    }
+  });
+}
+
 // src/cli.ts
 var _require = createRequire(import.meta.url);
 var _pkg = _require("../package.json");
@@ -8274,6 +8422,7 @@ registerSwap(program, getOutputMode, makeExecutor);
 registerBridge(program, getOutputMode);
 registerNft(program, getOutputMode);
 registerFarm(program, getOutputMode, makeExecutor);
+registerSetup(program);
 program.command("agent").description("Agent mode: read JSON commands from stdin (for AI agents)").action(async () => {
   const executor = makeExecutor();
   process.stderr.write("Agent mode: reading JSON commands from stdin...\n");
@@ -8282,7 +8431,7 @@ program.command("agent").description("Agent mode: read JSON commands from stdin
 });
 
 // src/landing.ts
-import pc2 from "picocolors";
+import pc3 from "picocolors";
 import { encodeFunctionData as encodeFunctionData28, parseAbi as parseAbi31, formatUnits } from "viem";
 var HYPEREVM_DISPLAY = ["HYPE", "WHYPE", "USDC", "USDT0", "USDe", "kHYPE", "wstHYPE"];
 var MANTLE_DISPLAY = ["MNT", "WMNT", "USDC", "USDT", "WETH", "mETH"];
@@ -8376,20 +8525,20 @@ async function showLandingPage(isJson) {
   const version = pkg.version;
   if (!wallet) {
     console.log("");
-    console.log(pc2.bold(pc2.cyan("  DeFi CLI v" + version)));
+    console.log(pc3.bold(pc3.cyan("  DeFi CLI v" + version)));
     console.log("");
-    console.log(pc2.yellow("  Wallet not configured."));
+    console.log(pc3.yellow("  Wallet not configured."));
     console.log("  Set DEFI_WALLET_ADDRESS to see your balances:");
     console.log("");
-    console.log(pc2.dim("    export DEFI_WALLET_ADDRESS=0x..."));
+    console.log(pc3.dim("    export DEFI_WALLET_ADDRESS=0x..."));
     console.log("");
     console.log("  Commands:");
-    console.log(pc2.dim("    defi status              Protocol overview"));
-    console.log(pc2.dim("    defi lending rates       Compare lending APYs"));
-    console.log(pc2.dim("    defi dex quote           Get swap quotes"));
-    console.log(pc2.dim("    defi portfolio           View all positions"));
-    console.log(pc2.dim("    defi scan                Exploit detection"));
-    console.log(pc2.dim("    defi --help              Full command list"));
+    console.log(pc3.dim("    defi status              Protocol overview"));
+    console.log(pc3.dim("    defi lending rates       Compare lending APYs"));
+    console.log(pc3.dim("    defi dex quote           Get swap quotes"));
+    console.log(pc3.dim("    defi portfolio           View all positions"));
+    console.log(pc3.dim("    defi scan                Exploit detection"));
+    console.log(pc3.dim("    defi --help              Full command list"));
     console.log("");
     return;
   }
@@ -8411,20 +8560,20 @@ async function showLandingPage(isJson) {
   const divider = "\u2500".repeat(colWidth - 2);
   console.log("");
   console.log(
-    pc2.bold(pc2.cyan("  DeFi CLI v" + version)) + pc2.dim("  \u2014  ") + pc2.bold(heChain.name) + pc2.dim(" \xB7 ") + pc2.bold(mantleChain.name)
+    pc3.bold(pc3.cyan("  DeFi CLI v" + version)) + pc3.dim("  \u2014  ") + pc3.bold(heChain.name) + pc3.dim(" \xB7 ") + pc3.bold(mantleChain.name)
   );
   console.log("");
-  console.log("  Wallet: " + pc2.yellow(shortenAddress(wallet)));
+  console.log("  Wallet: " + pc3.yellow(shortenAddress(wallet)));
   console.log("");
   const heHeader = padRight(
-    "  " + pc2.bold(heChain.name),
+    "  " + pc3.bold(heChain.name),
     colWidth + 10
     /* account for ANSI */
   );
-  const mantleHeader = pc2.bold(mantleChain.name);
+  const mantleHeader = pc3.bold(mantleChain.name);
   console.log(heHeader + "  " + mantleHeader);
-  const heDivider = padRight("  " + pc2.dim(divider), colWidth + 10);
-  const mantleDivider = pc2.dim(divider);
+  const heDivider = padRight("  " + pc3.dim(divider), colWidth + 10);
+  const mantleDivider = pc3.dim(divider);
   console.log(heDivider + "  " + mantleDivider);
   const maxRows = Math.max(heBalances.length, mantleBalances.length);
   for (let i = 0; i < maxRows; i++) {
@@ -8432,25 +8581,27 @@ async function showLandingPage(isJson) {
     const mantleEntry = mantleBalances[i];
     const heText = heEntry ? formatBalanceLine(heEntry.symbol, heEntry.balance) : "";
     const mantleText = mantleEntry ? formatBalanceLine(mantleEntry.symbol, mantleEntry.balance) : "";
-    const heColored = heEntry ? heEntry.balance === "0.00" || heEntry.balance === "?" ? pc2.dim(heText) : heText : "";
-    const mantleColored = mantleEntry ? mantleEntry.balance === "0.00" || mantleEntry.balance === "?" ? pc2.dim(mantleText) : mantleText : "";
+    const heColored = heEntry ? heEntry.balance === "0.00" || heEntry.balance === "?" ? pc3.dim(heText) : heText : "";
+    const mantleColored = mantleEntry ? mantleEntry.balance === "0.00" || mantleEntry.balance === "?" ? pc3.dim(mantleText) : mantleText : "";
     const visibleLen = heText.length;
     const padNeeded = colWidth - visibleLen;
     const paddedHe = heColored + (padNeeded > 0 ? " ".repeat(padNeeded) : "");
     console.log(paddedHe + "  " + mantleColored);
   }
   console.log("");
-  console.log("  " + pc2.bold("Commands:"));
-  console.log("    " + pc2.cyan("defi status") + "              Protocol overview");
-  console.log("    " + pc2.cyan("defi lending rates") + "       Compare lending APYs");
-  console.log("    " + pc2.cyan("defi dex quote") + "           Get swap quotes");
-  console.log("    " + pc2.cyan("defi portfolio") + "           View all positions");
-  console.log("    " + pc2.cyan("defi scan") + "                Exploit detection");
-  console.log("    " + pc2.cyan("defi --help") + "              Full command list");
+  console.log("  " + pc3.bold("Commands:"));
+  console.log("    " + pc3.cyan("defi status") + "              Protocol overview");
+  console.log("    " + pc3.cyan("defi lending rates") + "       Compare lending APYs");
+  console.log("    " + pc3.cyan("defi dex quote") + "           Get swap quotes");
+  console.log("    " + pc3.cyan("defi portfolio") + "           View all positions");
+  console.log("    " + pc3.cyan("defi scan") + "                Exploit detection");
+  console.log("    " + pc3.cyan("defi --help") + "              Full command list");
   console.log("");
 }
 
 // src/main.ts
+config({ path: resolve4(process.env.HOME || "~", ".defi", ".env"), quiet: true });
+config({ quiet: true });
 async function main() {
   try {
     const rawArgs = process.argv.slice(2);
@@ -8479,7 +8630,9 @@ async function main() {
       "bridge",
       "nft",
       "farm",
-      "agent"
+      "agent",
+      "setup",
+      "init"
     ]);
     const hasSubcommand = rawArgs.some((a) => !a.startsWith("-") && knownSubcommands.has(a));
     const isJson = rawArgs.includes("--json") || rawArgs.includes("--ndjson");