@aeon-ai-pay/aigateway 0.1.0

package/src/commands/wallet-topup.mjs

@@ -0,0 +1,221 @@
+/**
+ * topup：充值 + 一次性 approve facilitator
+ *
+ * 流程：
+ *   1. 校验 session key 已存在（来自 aigateway wallet-init）
+ *   2. 检查 USDT 余额 + facilitator allowance
+ *   3. 如果 USDT < LOW_BALANCE_THRESHOLD（1 USDT）或 allowance == 0，触发 WalletConnect 充值
+ *      - 充值 amount: TTY 模式交互选择 presets [5, 10, 20, 50] 或自定义；
+ *                    非 TTY 模式需要传 --amount <usdt>，否则报 TOPUP_REQUIRED
+ *      - 同时按需转 0.0003 BNB 用作 approve gas
+ *   4. session key 自己广播 ERC20.approve(facilitator, MaxUint256)
+ *   5. 重新查余额 / allowance，返回最终状态
+ */
+import { resolve } from "../config.mjs";
+import { getWalletBalance, getAllowance } from "../balance.mjs";
+import {
+  fundSessionKey,
+  approveFacilitator,
+  promptTopupAmount,
+  LOW_BALANCE_THRESHOLD,
+  MIN_TOPUP_USDT,
+  TOPUP_PRESETS,
+} from "../funding.mjs";
+import { WalletConnectError } from "../walletconnect.mjs";
+import { emitOk, emitErr, logInfo } from "../output.mjs";
+
+export async function topup(opts) {
+  logInfo("Topping up wallet: verifying readiness...");
+  const privateKey = resolve(opts.privateKey, "EVM_PRIVATE_KEY", "privateKey");
+  const appId = opts.appId;
+
+  if (!privateKey) {
+    emitErr("wallet-topup", "WALLET_NOT_CONFIGURED", {
+      message: "Wallet not configured. Run: aigateway wallet-init",
+      appId,
+    });
+    return;
+  }
+
+  let address, usdt, bnb, bnbRaw;
+  try {
+    ({ address, usdt, bnb, bnbRaw } = await getWalletBalance(privateKey));
+  } catch (e) {
+    emitErr("wallet-topup", "BALANCE_CHECK_FAILED", {
+      message: `Balance check failed: ${e.message}`,
+      appId,
+    });
+    return;
+  }
+  const usdtNum = parseFloat(usdt);
+  logInfo(`Wallet:    ${address}`);
+  logInfo(`Balance:   ${usdt} USDT, ${bnb} BNB`);
+  logInfo(`App ID:    ${appId}`);
+
+  logInfo("Checking facilitator allowance...");
+  let allowance;
+  try {
+    allowance = await getAllowance(address);
+  } catch (e) {
+    emitErr("wallet-topup", "ALLOWANCE_CHECK_FAILED", {
+      message: `Allowance check failed: ${e.message}`,
+      appId,
+    });
+    return;
+  }
+  logInfo(`Allowance: ${allowance === 0n ? "0 (approve required)" : "already approved"}`);
+
+  const explicitTopup = opts.amount != null && String(opts.amount).trim() !== "";
+  const balanceLow = usdtNum < LOW_BALANCE_THRESHOLD;
+  const needTopup = balanceLow || explicitTopup;
+  const needApprove = allowance === 0n;
+  const needGas = needApprove && bnbRaw === 0n;
+
+  // 已就绪：余额够 + 已 approve
+  if (!needTopup && !needApprove) {
+    logInfo("Wallet already prepared (balance ≥ minimum, facilitator approved).");
+    const data = {
+      ready: true,
+      appId,
+      address,
+      initialUsdt: usdt,
+      usdt,
+      bnb,
+      allowance: allowance.toString(),
+      topup: null,
+      approveTx: null,
+    };
+    emitOk("wallet-topup", data, { ...data, success: true });
+    return;
+  }
+
+  // 决定充值金额
+  let topupAmount = null;
+  if (needTopup) {
+    if (balanceLow) {
+      logInfo(`USDT balance ${usdtNum} < ${LOW_BALANCE_THRESHOLD} USDT threshold; a top-up of ≥ ${MIN_TOPUP_USDT} USDT is required.`);
+    } else {
+      logInfo(`Explicit top-up requested via --amount (current balance ${usdtNum} USDT).`);
+    }
+    if (explicitTopup) {
+      const amt = Number(opts.amount);
+      if (!Number.isFinite(amt) || amt <= 0) {
+        emitErr("wallet-topup", "AMOUNT_INVALID", {
+          message: `Invalid --amount: ${opts.amount}`,
+          appId,
+        });
+        return;
+      }
+      if (amt < MIN_TOPUP_USDT) {
+        emitErr("wallet-topup", "TOPUP_AMOUNT_TOO_SMALL", {
+          message: `--amount ${amt} USDT is below the ${MIN_TOPUP_USDT} USDT minimum.`,
+          minTopup: MIN_TOPUP_USDT,
+          appId,
+        });
+        return;
+      }
+      topupAmount = String(opts.amount);
+      logInfo(`Using --amount: ${topupAmount} USDT`);
+    } else if (process.stdin.isTTY) {
+      topupAmount = await promptTopupAmount(MIN_TOPUP_USDT);
+      logInfo(`Selected top-up amount: ${topupAmount} USDT`);
+    } else {
+      emitErr("wallet-topup", "TOPUP_REQUIRED", {
+        message: `USDT balance ${usdt} is below the ${LOW_BALANCE_THRESHOLD} USDT threshold. Choose a top-up amount and rerun with --amount <usdt>.`,
+        threshold: LOW_BALANCE_THRESHOLD,
+        minTopup: MIN_TOPUP_USDT,
+        currentBalance: usdt,
+        address,
+        appId,
+        presets: TOPUP_PRESETS,
+        hint: "Rerun: aigateway wallet-topup --amount <usdt> --app-id <appId>",
+      });
+      return;
+    }
+  }
+
+  // WalletConnect 充值（USDT + 按需 BNB gas）
+  if (needTopup || needGas) {
+    const willTransfer = [];
+    if (needTopup) willTransfer.push(`${topupAmount} USDT`);
+    if (needGas) willTransfer.push("0.0003 BNB (approve gas)");
+    logInfo(`Funding flow triggered (${willTransfer.join(" + ")})...`);
+    logInfo("Opening WalletConnect QR — please scan with your wallet app.");
+    try {
+      await fundSessionKey({
+        sessionAddress: address,
+        usdtAmount: needTopup ? topupAmount : null,
+        needGas,
+      });
+    } catch (e) {
+      if (e instanceof WalletConnectError) {
+        emitErr("wallet-topup", e.code, { message: e.message, address, appId });
+      } else {
+        emitErr("wallet-topup", "FUNDING_FAILED", {
+          message: `Funding failed: ${e.message}`,
+          address,
+          appId,
+        });
+      }
+      return;
+    }
+  }
+
+  // session key 一次性 approve facilitator
+  let approveTx = null;
+  if (needApprove) {
+    let postBnbRaw = bnbRaw;
+    if (needGas) {
+      try {
+        const post = await getWalletBalance(privateKey);
+        postBnbRaw = post.bnbRaw;
+      } catch (e) {
+        logInfo(`Post-funding balance re-check failed: ${e.message}`);
+      }
+    }
+    if (postBnbRaw === 0n) {
+      emitErr("wallet-topup", "INSUFFICIENT_BNB", {
+        message: "No BNB available for approve transaction. Run 'aigateway wallet-gas' to add BNB manually.",
+        address,
+        appId,
+      });
+      return;
+    }
+    try {
+      approveTx = await approveFacilitator(privateKey);
+    } catch (e) {
+      emitErr("wallet-topup", "APPROVE_FAILED", {
+        message: `Pre-authorize failed: ${e.message}`,
+        address,
+        appId,
+      });
+      return;
+    }
+  }
+
+  // 最终查余额 + allowance
+  let finalUsdt = usdt;
+  let finalBnb = bnb;
+  let finalAllowance = allowance;
+  try {
+    const final = await getWalletBalance(privateKey);
+    finalUsdt = final.usdt;
+    finalBnb = final.bnb;
+    finalAllowance = await getAllowance(address);
+  } catch (e) {
+    logInfo(`Final balance/allowance check failed: ${e.message}`);
+  }
+
+  const data = {
+    ready: true,
+    appId,
+    address,
+    initialUsdt: usdt,
+    usdt: finalUsdt,
+    bnb: finalBnb,
+    allowance: finalAllowance.toString(),
+    topup: topupAmount,
+    approveTx,
+  };
+  emitOk("wallet-topup", data, { ...data, success: true });
+}

package/src/commands/wallet-withdraw.mjs

@@ -0,0 +1,183 @@
+/**
+ * withdraw 命令：将 session key 中的资金转回主钱包（USDT + BNB）
+ */
+import { createPublicClient, createWalletClient, http, parseUnits, formatUnits, encodeFunctionData } from "viem";
+import { privateKeyToAccount } from "viem/accounts";
+import { bsc } from "viem/chains";
+import { loadConfig } from "../config.mjs";
+import { getBalanceByAddress } from "../balance.mjs";
+import { BSC_RPC_URL, USDT_BSC, ERC20_TRANSFER_ABI } from "../constants.mjs";
+import { emitOk, emitErr, logInfo } from "../output.mjs";
+
+const BNB_TRANSFER_GAS = 21000n;
+
+export async function withdraw(opts) {
+  logInfo("Reclaiming funds...");
+  const config = loadConfig();
+  const { appId } = opts;
+
+  if (!config.privateKey || !config.address) {
+    emitErr("wallet-withdraw", "WALLET_NOT_CONFIGURED", {
+      message: "No session key found. Nothing to withdraw.",
+      appId,
+    });
+    return;
+  }
+
+  const mainWallet = opts.to || config.mainWallet;
+  if (!mainWallet) {
+    emitErr("wallet-withdraw", "NO_MAIN_WALLET", {
+      message: "No main wallet address found. Use --to <address> to specify.",
+      appId,
+    });
+    return;
+  }
+
+  const sessionAddress = config.address;
+  const account = privateKeyToAccount(config.privateKey);
+
+  const publicClient = createPublicClient({
+    chain: bsc,
+    transport: http(BSC_RPC_URL, { timeout: 15000, retryCount: 2 }),
+  });
+
+  const walletClient = createWalletClient({
+    account,
+    chain: bsc,
+    transport: http(BSC_RPC_URL),
+  });
+
+  const balance = await getBalanceByAddress(sessionAddress);
+  logInfo(`Session key: ${sessionAddress}`);
+  logInfo(`Balance: ${balance.usdt} USDT, ${balance.bnb} BNB`);
+  logInfo(`Withdraw to: ${mainWallet}`);
+
+  const isWithdrawAll = !opts.amount;
+
+  // 无任何资金
+  if (balance.usdtRaw === 0n && balance.bnbRaw === 0n) {
+    emitErr("wallet-withdraw", "NO_FUNDS", { message: "No funds to withdraw.", appId });
+    return;
+  }
+
+  let usdtTxHash = null;
+  let bnbTxHash = null;
+
+  // 1. 赎回 USDT（有 USDT 才执行）
+  if (balance.usdtRaw > 0n) {
+    // USDT 转账需要 BNB 作 gas
+    if (balance.bnbRaw === 0n) {
+      emitErr("wallet-withdraw", "INSUFFICIENT_BNB", {
+        message: "No BNB for gas. Withdraw is a normal on-chain transfer and requires BNB to pay gas.",
+        address: sessionAddress,
+        appId,
+        hint: "Run 'aigateway wallet-gas' to top up BNB via WalletConnect, then retry.",
+      });
+      return;
+    }
+
+    let withdrawAmount = balance.usdtRaw;
+    if (opts.amount) {
+      const requested = parseUnits(opts.amount, 18);
+      if (requested > balance.usdtRaw) {
+        emitErr("wallet-withdraw", "AMOUNT_EXCEEDS_BALANCE", {
+          message: `Requested ${opts.amount} USDT but only ${balance.usdt} available.`,
+          requested: opts.amount,
+          available: balance.usdt,
+          appId,
+        });
+        return;
+      }
+      withdrawAmount = requested;
+    }
+
+    try {
+      const data = encodeFunctionData({
+        abi: ERC20_TRANSFER_ABI,
+        functionName: "transfer",
+        args: [mainWallet, withdrawAmount],
+      });
+
+      logInfo(`\nTransferring ${formatUnits(withdrawAmount, 18)} USDT → ${mainWallet}...`);
+      usdtTxHash = await walletClient.sendTransaction({ to: USDT_BSC, data });
+      logInfo(`USDT tx: ${usdtTxHash}`);
+
+      const receipt = await publicClient.waitForTransactionReceipt({
+        hash: usdtTxHash,
+        timeout: 60_000,
+      });
+      if (receipt.status !== "success") {
+        throw new Error("USDT transfer reverted");
+      }
+      logInfo("USDT reclaimed.");
+    } catch (error) {
+      emitErr("wallet-withdraw", "WITHDRAW_FAILED", {
+        message: `USDT withdraw failed: ${error.message}`,
+        appId,
+      });
+      return;
+    }
+  }
+
+  // 2. 赎回剩余 BNB（仅赎回全部时）
+  if (isWithdrawAll) {
+    const freshBalance = balance.usdtRaw > 0n
+      ? await getBalanceByAddress(sessionAddress)
+      : balance;
+
+    if (freshBalance.bnbRaw > 0n) {
+      try {
+        const gasPrice = await publicClient.getGasPrice();
+        // 预留 20% buffer 应对 gas price 波动
+        const gasCost = BNB_TRANSFER_GAS * (gasPrice * 120n / 100n);
+        const sendable = freshBalance.bnbRaw - gasCost;
+
+        if (sendable > 0n) {
+          logInfo(`Transferring ${formatUnits(sendable, 18)} BNB → ${mainWallet}...`);
+          bnbTxHash = await walletClient.sendTransaction({
+            to: mainWallet,
+            value: sendable,
+            gas: BNB_TRANSFER_GAS,
+            gasPrice,
+          });
+          logInfo(`BNB tx: ${bnbTxHash}`);
+
+          const receipt = await publicClient.waitForTransactionReceipt({
+            hash: bnbTxHash,
+            timeout: 60_000,
+          });
+          if (receipt.status !== "success") {
+            throw new Error("BNB transfer reverted");
+          }
+          logInfo("BNB reclaimed.");
+        } else {
+          logInfo("BNB balance too small to cover transfer gas, skipping.");
+        }
+      } catch (error) {
+        logInfo(`Warning: BNB reclaim failed (${error.message}).`);
+      }
+    }
+  }
+
+  // 查询最终余额
+  let finalBalance;
+  try {
+    finalBalance = await getBalanceByAddress(sessionAddress);
+  } catch {
+    finalBalance = { usdt: "unknown", bnb: "unknown" };
+  }
+
+  const data = {
+    appId,
+    to: mainWallet,
+    transactions: {
+      usdt: usdtTxHash,
+      bnb: bnbTxHash,
+    },
+    remaining: {
+      usdt: finalBalance.usdt,
+      bnb: finalBalance.bnb,
+    },
+  };
+  emitOk("wallet-withdraw", data, { success: true, ...data });
+}

package/src/config.mjs

@@ -0,0 +1,50 @@
+/**
+ * 配置管理：~/.aigateway/config.json
+ * 优先级：CLI 参数 > 环境变量 > config.json
+ *
+ * AEON AI Gateway 统一使用同一个 x402 服务端（ai-api.aeon.xyz），
+ * 不同能力（虚拟卡 / Skill Boss 调用）走不同的路径前缀。
+ */
+import { readFileSync, writeFileSync, mkdirSync, chmodSync } from "fs";
+import { join } from "path";
+import { homedir } from "os";
+
+const CONFIG_DIR = join(homedir(), ".aigateway");
+const CONFIG_FILE = join(CONFIG_DIR, "config.json");
+
+const DEFAULTS = {
+  serviceUrl: "https://ai-api.aeon.xyz",
+};
+
+export function loadConfig() {
+  try {
+    return { ...DEFAULTS, ...JSON.parse(readFileSync(CONFIG_FILE, "utf-8")) };
+  } catch {
+    return { ...DEFAULTS };
+  }
+}
+
+export function saveConfig(config) {
+  mkdirSync(CONFIG_DIR, { recursive: true });
+  writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2), { mode: 0o600 });
+  chmodSync(CONFIG_FILE, 0o600);
+}
+
+/**
+ * 解析配置值，优先级：cliValue > envKey > config[configKey]
+ */
+export function resolve(cliValue, envKey, configKey) {
+  if (cliValue) return cliValue;
+  if (process.env[envKey]) return process.env[envKey];
+  const cfg = loadConfig();
+  return cfg[configKey] || undefined;
+}
+
+export function getConfigPath() {
+  return CONFIG_FILE;
+}
+
+export function isSessionKeyMode() {
+  const config = loadConfig();
+  return config.mode === "session-key";
+}

package/src/constants.mjs

@@ -0,0 +1,22 @@
+export const MIN_AMOUNT = 0.6;
+export const MAX_AMOUNT = 800;
+export const POLL_INTERVAL = 5000;
+export const MAX_POLLS = 42;
+export const BSC_RPC_URL = "https://few-boldest-spring.bsc.quiknode.pro/ec468d8a1ea2c310457b2e2f4eea257e62ba3b1e/";
+export const USDT_BSC = "0x55d398326f99059fF775485246999027B3197955";
+export const FACILITATOR_ADDRESS = "0x555e3311a9893c9B17444C1Ff0d88192a57Ef13e";
+export const DEFAULT_WC_PROJECT_ID = "1c5e29cd4b466f52393cd39d05ec265c";
+export const WC_CONNECT_TIMEOUT_MS = 5 * 60 * 1000;
+
+export const ERC20_TRANSFER_ABI = [
+  {
+    name: "transfer",
+    type: "function",
+    inputs: [
+      { name: "to", type: "address" },
+      { name: "amount", type: "uint256" },
+    ],
+    outputs: [{ name: "success", type: "bool" }],
+    stateMutability: "nonpayable",
+  },
+];

package/src/error-codes.mjs

@@ -0,0 +1,50 @@
+/**
+ * 错误码常量表 —— CLI 与文档的单一事实源
+ *
+ * 退出码语义：
+ *   0 成功
+ *   1 用户错误（参数、余额、配置、用户拒绝）
+ *   2 超时（轮询、WalletConnect、签名、链上）
+ *   3 服务端 / 网络
+ *   4 内部错误
+ */
+
+export const ERROR_CODES = {
+  // ===== 用户错误（exit 1）=====
+  WALLET_NOT_CONFIGURED:  { exit: 1, message: "Wallet not configured. Run: aigateway wallet-init" },
+  SERVICE_URL_MISSING:    { exit: 1, message: "Service URL not configured." },
+  AMOUNT_INVALID:         { exit: 1, message: "Invalid amount." },
+  AMOUNT_OUT_OF_RANGE:    { exit: 1, message: "Amount is outside the allowed range." },
+  AMOUNT_EXCEEDS_BALANCE: { exit: 1, message: "Requested amount exceeds available balance." },
+  INSUFFICIENT_USDT:      { exit: 1, message: "Insufficient USDT balance." },
+  INSUFFICIENT_BNB:       { exit: 1, message: "Insufficient BNB for gas." },
+  NO_FUNDS:               { exit: 1, message: "No funds available." },
+  NO_MAIN_WALLET:         { exit: 1, message: "No main wallet address configured. Use --to <address>." },
+  MISSING_PROMPT:         { exit: 1, message: "Missing --prompt. Provide a non-empty image prompt." },
+  TOPUP_REQUIRED:         { exit: 1, message: "Wallet top-up required. Choose an amount and rerun with --topup-amount <usdt>." },
+  TOPUP_AMOUNT_TOO_SMALL: { exit: 1, message: "Top-up amount is below the minimum." },
+  PAYMENT_REJECTED:       { exit: 1, message: "Payment approval was rejected. Please try again if you'd like to proceed." },
+
+  // ===== 超时（exit 2）=====
+  PAYMENT_TIMEOUT:        { exit: 2, message: "Payment approval timed out. Please try again." },
+  WC_SESSION_EXPIRED:     { exit: 2, message: "WalletConnect session expired." },
+  POLL_TIMEOUT:           { exit: 2, message: "Polling timed out. Card may still be provisioning." },
+  TX_TIMEOUT:             { exit: 2, message: "On-chain transaction timed out." },
+
+  // ===== 服务/网络（exit 3）=====
+  SERVICE_UNAVAILABLE:    { exit: 3, message: "Service unavailable or network error." },
+  PAYMENT_FETCH_FAILED:   { exit: 3, message: "Failed to fetch payment requirements." },
+  BALANCE_CHECK_FAILED:   { exit: 3, message: "Failed to check balance." },
+  ALLOWANCE_CHECK_FAILED: { exit: 3, message: "Failed to check allowance." },
+  TX_REVERTED:            { exit: 3, message: "On-chain transaction reverted." },
+  WITHDRAW_FAILED:        { exit: 3, message: "Withdraw transaction failed." },
+  APPROVE_FAILED:         { exit: 3, message: "Pre-authorization (approve) failed." },
+  INVALID_PAYMENT_AMOUNT: { exit: 3, message: "Server returned invalid payment amount." },
+  PAYMENT_FAILED:         { exit: 3, message: "Payment request failed." },
+  IMAGE_DOWNLOAD_FAILED:  { exit: 3, message: "Image download failed." },
+  FUNDING_FAILED:         { exit: 3, message: "Funding flow failed." },
+
+  // ===== 内部（exit 4）=====
+  INTERNAL_ERROR:         { exit: 4, message: "Internal error." },
+  WALLET_ERROR:           { exit: 1, message: "Wallet operation failed." },
+};