httpcat-cli 0.3.0 → 0.3.1

package/dist/utils/privateKeyPrompt.js

@@ -1,12 +1,16 @@
-import inquirer from 'inquirer';
-import chalk from 'chalk';
-import { formatEther } from 'viem';
+import inquirer from "inquirer";
+import chalk from "chalk";
+import { formatEther } from "viem";
 // USDC balance checking using viem directly
-import { config } from '../config.js';
-import { getNetworkConfig } from '../utils/constants.js';
-import { printCat } from '../interactive/art.js';
-import { formatAddress, printWarning } from '../utils/formatting.js';
-import { generateSeedPhrase, validateSeedPhrase, seedPhraseToPrivateKey, getAddressFromPrivateKey, } from './wallet.js';
+import { config } from "../config.js";
+import { getNetworkConfig } from "../utils/constants.js";
+import { printCat } from "../interactive/art.js";
+import { formatAddress, printWarning } from "../utils/formatting.js";
+import { retryWithBackoff } from "./retry.js";
+import { generateSeedPhrase, validateSeedPhrase, seedPhraseToPrivateKey, getAddressFromPrivateKey, } from "./wallet.js";
+import { createMcpConfig } from "./mcp-config.js";
+import { detectInstalledLlms } from "./llm-cli-detector.js";
+import { configureLLM } from "./llm-cli-config.js";
 /**
  * Prompts the user to set up their wallet with three options:
  * 1. Generate new wallet (seed phrase)
@@ -19,47 +23,47 @@ import { generateSeedPhrase, validateSeedPhrase, seedPhraseToPrivateKey, getAddr
 export async function promptForPrivateKey() {
     console.log();
     // Show wizard cat
-    printCat('wizard');
-    console.log(chalk.bold.cyan('🔐 Let\'s set up your wallet...\n'));
+    printCat("wizard");
+    console.log(chalk.bold.cyan("🔐 Let's set up your wallet...\n"));
     // Security messaging - make it more playful and safe-feeling
-    console.log(chalk.dim('🔒 Security & Privacy:'));
-    console.log(chalk.dim('   • Your keys are stored locally on your machine only'));
-    console.log(chalk.dim('   • They are never sent to any server or third party'));
-    console.log(chalk.dim('   • They are only used locally to sign transactions'));
-    console.log(chalk.dim('   • Stored in:'), chalk.cyan(config.getConfigPath()));
+    console.log(chalk.dim("🔒 Security & Privacy:"));
+    console.log(chalk.dim("   • Your keys are stored locally on your machine only"));
+    console.log(chalk.dim("   • They are never sent to any server or third party"));
+    console.log(chalk.dim("   • They are only used locally to sign transactions"));
+    console.log(chalk.dim("   • Stored in:"), chalk.cyan(config.getConfigPath()));
     console.log();
     // Ask user what they want to do - make "generate" the default
     const setupChoice = await inquirer.prompt([
         {
-            type: 'list',
-            name: 'option',
-            message: 'How would you like to set up your wallet?',
+            type: "list",
+            name: "option",
+            message: "How would you like to set up your wallet?",
             choices: [
                 {
-                    name: '✨ Generate new wallet (seed phrase) - Recommended for new users',
-                    value: 'generate',
+                    name: "✨ Generate new wallet (seed phrase) - Recommended for new users",
+                    value: "generate",
                 },
                 {
-                    name: '📥 Import existing seed phrase',
-                    value: 'import_seed',
+                    name: "📥 Import existing seed phrase",
+                    value: "import_seed",
                 },
                 {
-                    name: '🔑 Import existing private key',
-                    value: 'import_key',
+                    name: "🔑 Import existing private key",
+                    value: "import_key",
                 },
             ],
-            default: 'generate', // Make generate the default!
+            default: "generate", // Make generate the default!
         },
     ]);
     let seedPhrase;
     let privateKey;
     let accountIndex = 0;
-    if (setupChoice.option === 'generate') {
+    if (setupChoice.option === "generate") {
         // Generate new seed phrase
         seedPhrase = await handleGenerateSeedPhrase();
         accountIndex = 1; // First seed-derived account
     }
-    else if (setupChoice.option === 'import_seed') {
+    else if (setupChoice.option === "import_seed") {
         // Import existing seed phrase
         seedPhrase = await handleImportSeedPhrase();
         accountIndex = 1; // First seed-derived account
@@ -87,41 +91,41 @@ export async function promptForPrivateKey() {
  */
 export async function handleGenerateSeedPhrase() {
     console.log();
-    console.log(chalk.bold.cyan('🌱 Generating new wallet for you...\n'));
+    console.log(chalk.bold.cyan("🌱 Generating new wallet for you...\n"));
     // Ask for word count
     const wordChoice = await inquirer.prompt([
         {
-            type: 'list',
-            name: 'wordCount',
-            message: 'How many words?',
+            type: "list",
+            name: "wordCount",
+            message: "How many words?",
             choices: [
-                { name: '12 words (recommended) 🎯', value: 12 },
-                { name: '24 words (more secure) 🔒', value: 24 },
+                { name: "12 words (recommended) 🎯", value: 12 },
+                { name: "24 words (more secure) 🔒", value: 24 },
             ],
             default: 12,
         },
     ]);
     const seedPhrase = generateSeedPhrase(wordChoice.wordCount);
     console.log();
-    console.log(chalk.bold.green('✨ Your new wallet is ready! ✨\n'));
-    console.log(chalk.yellow('⚠️  IMPORTANT: Write down your seed phrase and keep it safe!'));
-    console.log(chalk.yellow('   If you lose it, you will lose access to your wallet.\n'));
-    console.log(chalk.bold('Your seed phrase:'));
+    console.log(chalk.bold.green("✨ Your new wallet is ready! ✨\n"));
+    console.log(chalk.yellow("⚠️  IMPORTANT: Write down your seed phrase and keep it safe!"));
+    console.log(chalk.yellow("   If you lose it, you will lose access to your wallet.\n"));
+    console.log(chalk.bold("Your seed phrase:"));
     console.log(chalk.cyan.bold(seedPhrase));
     console.log();
     // Confirm they saved it
     const confirmed = await inquirer.prompt([
         {
-            type: 'confirm',
-            name: 'saved',
-            message: 'Have you written down your seed phrase?',
+            type: "confirm",
+            name: "saved",
+            message: "Have you written down your seed phrase?",
             default: false,
         },
     ]);
     if (!confirmed.saved) {
         console.log();
-        console.log(chalk.yellow('🐱 The cat wants you to be safe! Please write it down first.'));
-        throw new Error('You must save your seed phrase before continuing.');
+        console.log(chalk.yellow("🐱 The cat wants you to be safe! Please write it down first."));
+        throw new Error("You must save your seed phrase before continuing.");
     }
     return seedPhrase;
 }
@@ -130,23 +134,23 @@ export async function handleGenerateSeedPhrase() {
  */
 export async function handleImportSeedPhrase() {
     console.log();
-    console.log(chalk.bold.cyan('📥 Importing seed phrase...\n'));
+    console.log(chalk.bold.cyan("📥 Importing seed phrase...\n"));
     const answers = await inquirer.prompt([
         {
-            type: 'input',
-            name: 'seedPhrase',
-            message: 'Enter your seed phrase (12 or 24 words):',
+            type: "input",
+            name: "seedPhrase",
+            message: "Enter your seed phrase (12 or 24 words):",
             validate: (input) => {
                 const trimmed = input.trim();
                 if (!trimmed) {
-                    return 'Seed phrase is required';
+                    return "Seed phrase is required";
                 }
                 const words = trimmed.split(/\s+/);
                 if (words.length !== 12 && words.length !== 24) {
-                    return 'Seed phrase must be 12 or 24 words';
+                    return "Seed phrase must be 12 or 24 words";
                 }
                 if (!validateSeedPhrase(trimmed)) {
-                    return 'Invalid seed phrase. Please check your words.';
+                    return "Invalid seed phrase. Please check your words.";
                 }
                 return true;
             },
@@ -159,36 +163,46 @@ export async function handleImportSeedPhrase() {
  */
 async function handleImportPrivateKey() {
     console.log();
-    console.log(chalk.bold.cyan('🔑 Importing private key...\n'));
+    console.log(chalk.bold.cyan("🔑 Importing private key...\n"));
     // Custom mask function: show first 5 chars, then dots
     const maskFunction = (input) => {
         if (!input || input.length === 0)
-            return '';
+            return "";
         if (input.length <= 5)
             return input;
-        return input.substring(0, 5) + '•'.repeat(input.length - 5);
+        return input.substring(0, 5) + "•".repeat(input.length - 5);
     };
     const answers = await inquirer.prompt([
         {
-            type: 'password',
-            name: 'privateKey',
-            message: 'Enter your private key:',
+            type: "password",
+            name: "privateKey",
+            message: "Enter your private key:",
             mask: maskFunction,
             validate: (input) => {
                 if (!input) {
-                    return 'Private key is required';
+                    return "Private key is required";
                 }
-                if (!input.startsWith('0x')) {
-                    return 'Private key must start with 0x';
+                // Normalize: add 0x if missing for validation
+                const normalized = input.startsWith("0x") ? input : "0x" + input;
+                if (normalized.length !== 66) {
+                    return "Private key must be 66 characters long (0x + 64 hex chars)";
                 }
-                if (input.length !== 66) {
-                    return 'Private key must be 66 characters long (0x + 64 hex chars)';
+                // Validate hex characters
+                const hexPattern = /^0x[0-9a-fA-F]{64}$/;
+                if (!hexPattern.test(normalized)) {
+                    return "Private key must contain only hexadecimal characters (0-9, a-f, A-F)";
                 }
                 return true;
             },
         },
     ]);
-    return answers.privateKey;
+    // Auto-prepend 0x if missing
+    let inputPrivateKey = answers.privateKey.trim();
+    if (!inputPrivateKey.startsWith("0x")) {
+        console.log(chalk.yellow("Adding 0x prefix..."));
+        inputPrivateKey = "0x" + inputPrivateKey;
+    }
+    return inputPrivateKey;
 }
 /**
  * Prompt if user wants to add more accounts after importing private key
@@ -197,9 +211,9 @@ async function promptForAccountCreation() {
     console.log();
     const answers = await inquirer.prompt([
         {
-            type: 'confirm',
-            name: 'addMore',
-            message: 'Want to add more accounts? Generate seed phrase?',
+            type: "confirm",
+            name: "addMore",
+            message: "Want to add more accounts? Generate seed phrase?",
             default: false,
         },
     ]);
@@ -210,67 +224,78 @@ async function promptForAccountCreation() {
  */
 async function promptForPassword() {
     console.log();
-    console.log(chalk.dim('🔐 Password Protection (optional):'));
-    console.log(chalk.dim('   • Set a password to encrypt your wallet'));
-    console.log(chalk.dim('   • Leave blank to skip password protection'));
-    console.log(chalk.dim('   • You\'ll be prompted to unlock when needed\n'));
+    console.log(chalk.bold.cyan("🔐 Password Protection (Optional)"));
+    console.log();
+    console.log(chalk.dim("   • Password protection is optional"));
+    console.log(chalk.dim("   • Press Enter to skip password (your keys will still be encrypted)"));
+    console.log(chalk.dim("   • If you set a password, you'll need it to unlock your wallet"));
+    console.log(chalk.dim("     for transactions (session timeout: 15 minutes)\n"));
     const answers = await inquirer.prompt([
         {
-            type: 'password',
-            name: 'password',
-            message: 'Enter password (or press Enter to skip):',
-            mask: '•',
+            type: "password",
+            name: "password",
+            message: "Enter password (optional - press Enter to skip):",
+            mask: "•",
         },
         {
-            type: 'password',
-            name: 'confirmPassword',
-            message: 'Confirm password (or press Enter to skip):',
-            mask: '•',
-            when: (answers) => answers.password !== '',
+            type: "password",
+            name: "confirmPassword",
+            message: "Confirm password (or press Enter to skip):",
+            mask: "•",
+            when: (answers) => answers.password !== "",
         },
     ]);
     if (answers.password) {
         if (answers.password !== answers.confirmPassword) {
-            throw new Error('Passwords do not match');
+            throw new Error("Passwords do not match");
         }
         return answers.password;
     }
-    return '';
+    return "";
 }
 /**
  * Save wallet configuration
  */
 async function saveWalletConfig(seedPhrase, privateKey, password, activeAccountIndex) {
     console.log();
-    console.log(chalk.dim('💾 Saving configuration...\n'));
+    console.log(chalk.dim("💾 Saving configuration...\n"));
     // Set password
     config.setPassword(password);
-    // Save seed phrase if provided
+    // Save private key first if provided (Account 0)
+    // This must be done before seed phrase to ensure Account 0 exists first
+    if (privateKey) {
+        config.setEncryptedPrivateKey(privateKey, password);
+        const account0Address = getAddressFromPrivateKey(privateKey);
+        const existingAccounts = config.getAllAccounts();
+        const account0Exists = existingAccounts.some((acc) => acc.index === 0);
+        if (!account0Exists) {
+            config.addAccount({
+                type: "custom",
+                index: 0,
+                address: account0Address,
+            });
+        }
+        console.log(chalk.green("✓ Private key saved"));
+        console.log(chalk.dim(`   Account 0: ${formatAddress(account0Address, 12)}\n`));
+    }
+    // Save seed phrase if provided (Account 1)
     if (seedPhrase) {
         config.setEncryptedSeedPhrase(seedPhrase, password);
         // Create Account 1 (first seed-derived account)
         const account1PrivateKey = seedPhraseToPrivateKey(seedPhrase, 0);
         const account1Address = getAddressFromPrivateKey(account1PrivateKey);
-        config.addAccount({
-            type: 'seed',
-            index: 1,
-            address: account1Address,
-        });
-        console.log(chalk.green('✓ Seed phrase saved'));
+        const existingAccounts = config.getAllAccounts();
+        const account1Exists = existingAccounts.some((acc) => acc.index === 1);
+        if (!account1Exists) {
+            config.addAccount({
+                type: "seed",
+                index: 1,
+                address: account1Address,
+            });
+        }
+        console.log(chalk.green("✓ Seed phrase saved"));
         console.log(chalk.dim(`   Account 1: ${formatAddress(account1Address, 12)}\n`));
     }
-    // Save private key if provided (Account 0)
-    if (privateKey) {
-        config.setEncryptedPrivateKey(privateKey, password);
-        const account0Address = getAddressFromPrivateKey(privateKey);
-        config.addAccount({
-            type: 'custom',
-            index: 0,
-            address: account0Address,
-        });
-        console.log(chalk.green('✓ Private key saved'));
-        console.log(chalk.dim(`   Account 0: ${formatAddress(account0Address, 12)}\n`));
-    }
     // Set active account
     config.setActiveAccount(activeAccountIndex);
     // Show balances for active account
@@ -278,75 +303,229 @@ async function saveWalletConfig(seedPhrase, privateKey, password, activeAccountI
     if (activeAccount) {
         const activePrivateKey = config.getAccountPrivateKey(activeAccountIndex);
         await showAccountBalances(activeAccount.address, activePrivateKey);
+        console.log(chalk.dim("   Config saved to:"), chalk.cyan(config.getConfigPath()));
+        console.log(chalk.green("✓ Configuration saved!"));
+        console.log();
+        console.log(chalk.dim("🔐 Remember: Your keys stay on your machine."));
+        console.log(chalk.dim("   They are never transmitted to any server or third party.\n"));
+        // MCP Integration Step:
+        // After successfully saving the wallet configuration, we:
+        // 1. Create the base MCP config file (~/.config/httpcat/mcp.json)
+        // 2. Detect installed LLM CLI tools that support MCP
+        // 3. Prompt user to select which tools to configure
+        // 4. Configure selected tools with the private key
+        // Step 1: Create base MCP configuration file
+        // This file serves as a reference and can be used by MCP clients
+        // Note: No private key needed - MCP server reads from config file
+        try {
+            createMcpConfig();
+            console.log(chalk.green("✓ MCP configuration file created"));
+        }
+        catch (error) {
+            // Non-fatal error - continue with wizard even if MCP config creation fails
+            console.log(chalk.yellow("⚠️  Failed to create MCP config file"));
+        }
+        // Step 2 & 3: Detect and configure LLM CLI tools
+        // This will:
+        // - Scan for installed LLM CLI tools (Claude Desktop, Cursor, Codex, Gemini, etc.)
+        // - Show user a selection interface if tools are found
+        // - Configure selected tools with the httpcat MCP server
+        // Note: No private key needed - MCP server reads from config file
+        await promptForMcpIntegration();
+    }
+}
+/**
+ * Prompt for MCP integration with LLM CLI tools
+ *
+ * This function is called after wallet configuration is complete. It:
+ * 1. Detects installed LLM CLI tools that support MCP integration
+ * 2. Shows user a selection interface (checkbox) if tools are found
+ * 3. Configures each selected tool with the httpcat MCP server
+ * 4. Shows success/failure results for each configuration
+ *
+ * Flow:
+ * - If no LLM tools detected: Skip silently (no user interaction needed)
+ * - If tools detected: Show selection prompt
+ * - If user selects tools: Configure each one
+ * - Show results: Success (✓) or failure (✗) for each tool
+ *
+ * Error Handling:
+ * - All errors are caught and logged, but don't break the wizard
+ * - Individual tool failures don't stop other tools from being configured
+ * - If the entire step fails, it's skipped silently
+ *
+ * Note: No private key needed - MCP server reads from config file instead.
+ */
+async function promptForMcpIntegration() {
+    try {
+        // Step 1: Detect installed LLM CLI tools
+        // Scans system for: Claude Desktop, Cursor, Codex, Gemini, LLM Tools, MCP-Use, MCP Client
+        const detectedLlms = await detectInstalledLlms();
+        // If no LLM tools found, skip this step silently
+        // User can manually configure later if needed
+        if (detectedLlms.length === 0) {
+            return;
+        }
+        // Step 2: Show user selection interface
+        console.log();
+        console.log(chalk.cyan("🔌 LLM CLI Integration"));
+        console.log(chalk.dim(`   Found ${detectedLlms.length} LLM CLI tool(s) that support MCP integration.\n`));
+        // Present checkbox selection to user
+        // Each option shows: Tool name + config file path
+        const selection = await inquirer.prompt([
+            {
+                type: "checkbox",
+                name: "selectedLlms",
+                message: "Which LLM CLI tools would you like to configure with httpcat MCP?",
+                choices: detectedLlms.map((llm) => ({
+                    name: `${llm.name} ${chalk.dim(`(${llm.configPath})`)}`,
+                    value: llm.id,
+                })),
+            },
+        ]);
+        // If user selected nothing, skip configuration
+        if (!selection.selectedLlms || selection.selectedLlms.length === 0) {
+            console.log(chalk.dim("   Skipping LLM CLI integration.\n"));
+            return;
+        }
+        // Step 3: Configure selected LLM tools
+        console.log();
+        console.log(chalk.dim("⚙️  Configuring selected LLM CLI tools...\n"));
+        // Track results for each tool configuration
+        const results = [];
+        // Configure each selected tool
+        for (const llmId of selection.selectedLlms) {
+            const llm = detectedLlms.find((l) => l.id === llmId);
+            if (!llm)
+                continue; // Skip if LLM not found (shouldn't happen)
+            try {
+                // Call appropriate configuration function for this LLM tool
+                // Each function handles its own format (JSON, TOML, extension JSON, CLI commands)
+                // Note: No private key needed - MCP server reads from config file
+                const success = await configureLLM(llm);
+                results.push({
+                    name: llm.name,
+                    success,
+                });
+            }
+            catch (error) {
+                // Catch and log errors for individual tools
+                // Don't let one failure stop others from being configured
+                results.push({
+                    name: llm.name,
+                    success: false,
+                    error: error?.message || "Unknown error",
+                });
+            }
+        }
+        // Step 4: Show configuration results
+        console.log();
+        for (const result of results) {
+            if (result.success) {
+                console.log(chalk.green(`✓ ${result.name} configured successfully`));
+            }
+            else {
+                console.log(chalk.red(`✗ ${result.name} configuration failed`));
+                if (result.error) {
+                    console.log(chalk.dim(`   Error: ${result.error}`));
+                }
+            }
+        }
+        console.log();
+    }
+    catch (error) {
+        // If entire step fails, skip silently
+        // Don't break the setup wizard for MCP integration issues
+        console.log(chalk.dim("   MCP integration step skipped.\n"));
     }
-    console.log(chalk.dim('   Config saved to:'), chalk.cyan(config.getConfigPath()));
-    console.log(chalk.green('✓ Configuration saved!'));
-    console.log();
-    console.log(chalk.dim('🔐 Remember: Your keys stay on your machine.'));
-    console.log(chalk.dim('   They are never transmitted to any server or third party.\n'));
 }
 /**
  * Show account balances
  */
 async function showAccountBalances(address, privateKey) {
-    console.log(chalk.dim('💰 Checking balances on Base Sepolia...\n'));
+    // Get network configuration
+    const network = config.get("network");
+    const networkDisplayName = config.caip2ToNetwork(network);
+    // Format network name for display (base-sepolia -> Base Sepolia, base -> Base)
+    const networkName = networkDisplayName === "base-sepolia"
+        ? "Base Sepolia"
+        : networkDisplayName === "base"
+            ? "Base"
+            : networkDisplayName;
+    console.log(chalk.dim(`💰 Checking balances on ${networkName}...\n`));
     try {
-        // Get network configuration
-        const network = config.get("network");
         const { chain, usdcAddress } = await getNetworkConfig(network);
         // Create signer using viem for balance checks
-        const { createPublicClient, http, parseAbi, formatUnits } = await import("viem");
+        const { createPublicClient, parseAbi, formatUnits } = await import("viem");
+        const { createRpcTransport } = await import("./rpc-transport.js");
         const publicClient = createPublicClient({
             chain,
-            transport: http(),
+            transport: createRpcTransport({
+                chainId: chain.id,
+                // HttpcatClient not available in this context, will use public/Alchemy RPC
+            }),
+        });
+        // Check ETH balance with retry logic for network failures
+        const ethBalance = await retryWithBackoff(() => publicClient.getBalance({
+            address: address,
+        }), {
+            maxRetries: 3,
+            initialDelay: 500,
+            maxDelay: 3000,
         });
-        // Check ETH balance
-        const ethBalance = await publicClient.getBalance({ address: address });
         const ethFormatted = formatEther(ethBalance);
-        // Check USDC balance using ERC20 token contract
+        // Check USDC balance using ERC20 token contract with retry logic
         const ERC20_ABI = parseAbi([
             "function balanceOf(address owner) view returns (uint256)",
             "function decimals() view returns (uint8)",
         ]);
         const [usdcBalance, decimals] = await Promise.all([
-            publicClient.readContract({
+            retryWithBackoff(() => publicClient.readContract({
                 address: usdcAddress,
                 abi: ERC20_ABI,
                 functionName: "balanceOf",
                 args: [address],
+            }), {
+                maxRetries: 3,
+                initialDelay: 500,
+                maxDelay: 3000,
             }).catch(() => BigInt(0)),
-            publicClient.readContract({
+            retryWithBackoff(() => publicClient.readContract({
                 address: usdcAddress,
                 abi: ERC20_ABI,
                 functionName: "decimals",
+            }), {
+                maxRetries: 3,
+                initialDelay: 500,
+                maxDelay: 3000,
             }).catch(() => 6), // Default to 6 decimals for USDC
         ]);
-        const usdcFormatted = (Number(formatUnits(usdcBalance, decimals))).toFixed(2);
+        const usdcFormatted = Number(formatUnits(usdcBalance, decimals)).toFixed(2);
         // Display balances
-        console.log(chalk.dim('   ETH: '), chalk.yellow(`${ethFormatted} ETH`));
-        console.log(chalk.dim('   USDC:'), chalk.green(`$${usdcFormatted} USDC`));
+        console.log(chalk.dim("   ETH: "), chalk.yellow(`${ethFormatted} ETH`));
+        console.log(chalk.dim("   USDC:"), chalk.green(`$${usdcFormatted} USDC`));
         console.log();
         // Show warnings if balances are low
         const ethNum = Number(ethFormatted);
         const usdcNum = Number(usdcFormatted);
         if (ethNum < 0.001 && usdcNum < 1) {
-            printWarning('Low balances detected!');
-            console.log(chalk.dim('   You need Base Sepolia ETH for gas fees'));
-            console.log(chalk.dim('   You need Base Sepolia USDC for trading\n'));
+            printWarning("Low balances detected!");
+            console.log(chalk.dim(`   You need ${networkName} ETH for gas fees`));
+            console.log(chalk.dim(`   You need ${networkName} USDC for trading\n`));
         }
         else if (ethNum < 0.001) {
-            printWarning('Low ETH balance - you may not have enough for gas fees\n');
+            printWarning("Low ETH balance - you may not have enough for gas fees\n");
         }
         else if (usdcNum < 1) {
-            printWarning('Low USDC balance - you may not have enough for trades\n');
+            printWarning("Low USDC balance - you may not have enough for trades\n");
         }
-        console.log(chalk.yellow('⚠️  Important:'));
-        console.log(chalk.dim('   Ensure you have Base Sepolia ETH (for gas fees)'));
-        console.log(chalk.dim('   and Base Sepolia USDC (for trading) in this wallet!\n'));
+        console.log(chalk.yellow("⚠️  Important:"));
+        console.log(chalk.dim(`   Ensure you have ${networkName} ETH (for gas fees)`));
+        console.log(chalk.dim(`   and ${networkName} USDC (for trading) in this wallet!\n`));
     }
     catch (error) {
-        console.log(chalk.yellow('⚠️  Could not check balances (network may be slow)'));
-        console.log(chalk.dim('   Make sure you have Base Sepolia ETH and USDC!\n'));
+        console.log(chalk.yellow("⚠️  Could not check balances (network may be slow)"));
+        console.log(chalk.dim(`   Make sure you have ${networkName} ETH and USDC!\n`));
     }
 }
 /**
@@ -356,9 +535,9 @@ export function formatEthBalance(wei) {
     const eth = formatEther(wei);
     const num = parseFloat(eth);
     if (num === 0)
-        return '0 ETH';
+        return "0 ETH";
     if (num < 0.0001)
-        return '< 0.0001 ETH';
+        return "< 0.0001 ETH";
     return `${num.toFixed(4)} ETH`;
 }
 /**