@hypurrquant/defi-cli 0.2.3 → 0.2.4

package/dist/index.js

@@ -8219,6 +8219,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");
@@ -8270,6 +8414,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");