@layr-labs/ecloud-cli 1.0.0-devep1 → 1.0.0-devep2

package/dist/commands/billing/__tests__/top-up.test.js

@@ -319,7 +319,8 @@ async function createBillingClient(flags) {
 // src/commands/billing/top-up.ts
 import { formatUnits } from "viem";
 import chalk2 from "chalk";
-import { input as input2 } from "@inquirer/prompts";
+import { input as input2, select as select2 } from "@inquirer/prompts";
+import open from "open";
 
 // src/telemetry.ts
 import {
@@ -383,12 +384,22 @@ async function withTelemetry(command, action) {
 var POLL_INTERVAL_MS = 5e3;
 var POLL_TIMEOUT_MS = 3 * 60 * 1e3;
 var BillingTopUp = class _BillingTopUp extends Command {
-  static description = "Purchase EigenCompute credits with USDC";
+  static description = "Purchase EigenCompute credits with USDC or credit card";
+  static examples = [
+    "<%= config.bin %> billing top-up",
+    "<%= config.bin %> billing top-up --method usdc --amount 50",
+    "<%= config.bin %> billing top-up --method card --amount 25"
+  ];
   static flags = {
     ...commonFlags,
+    method: Flags2.string({
+      required: false,
+      description: "Payment method: usdc (on-chain) or card (credit card)",
+      options: ["usdc", "card"]
+    }),
     amount: Flags2.string({
       required: false,
-      description: "Amount of USDC to spend (e.g., '50')"
+      description: "Amount to spend (USDC for on-chain, whole dollars for card)"
     }),
     account: Flags2.string({
       required: false,
@@ -424,101 +435,162 @@ ${chalk2.bold("Purchase EigenCompute credits")}`);
         const remaining = status.remainingCredits ?? 0;
         const applied = status.creditsApplied ?? 0;
         baselineTotal = remaining + applied;
-        if (status.remainingCredits !== void 0) {
-          this.log(`  ${chalk2.bold("Credits:")} ${chalk2.cyan(`$${status.remainingCredits.toFixed(2)}`)}`);
-        }
+        this.log(`  ${chalk2.bold("Credits:")} ${chalk2.cyan(`$${remaining.toFixed(2)}`)}`);
       } catch {
         this.debug("Could not fetch current credit balance");
       }
-      const onChainState = await billing.getTopUpInfo();
-      const { usdcBalance, minimumPurchase } = onChainState;
-      const balanceFormatted = formatUnits(usdcBalance, 6);
-      this.log(`  ${chalk2.bold("USDC:")}    ${balanceFormatted} USDC`);
-      if (usdcBalance === BigInt(0)) {
-        this.log(`
+      const method = flags.method ?? await select2({
+        message: "How would you like to pay?",
+        choices: [
+          { value: "card", name: "Credit card" },
+          { value: "usdc", name: "USDC (on-chain)" }
+        ]
+      });
+      if (method === "usdc") {
+        await this.handleUsdc(billing, flags, walletAddress, targetAccount, baselineTotal);
+      } else {
+        await this.handleCard(billing, flags, baselineTotal);
+      }
+    });
+  }
+  async handleUsdc(billing, flags, walletAddress, targetAccount, baselineTotal) {
+    const onChainState = await billing.getTopUpInfo();
+    const { usdcBalance, minimumPurchase } = onChainState;
+    const balanceFormatted = formatUnits(usdcBalance, 6);
+    this.log(`  ${chalk2.bold("USDC:")}    ${balanceFormatted} USDC`);
+    if (usdcBalance === BigInt(0)) {
+      this.log(`
 ${chalk2.yellow("  No USDC in wallet.")}`);
-        this.log(`  Send USDC on Sepolia to: ${chalk2.cyan(walletAddress)}`);
-        this.log(`  Then re-run: ${chalk2.cyan("ecloud billing top-up")}
+      this.log(`  Send USDC on Sepolia to: ${chalk2.cyan(walletAddress)}`);
+      this.log(`  Then re-run: ${chalk2.cyan("ecloud billing top-up")}
 `);
-        return;
+      return;
+    }
+    const minimumFormatted = formatUnits(minimumPurchase, 6);
+    const amountStr = flags.amount ?? await input2({
+      message: `How much USDC to spend on credits? (minimum: ${minimumFormatted})`,
+      validate: (val) => {
+        const n = parseFloat(val);
+        if (isNaN(n) || n <= 0) return "Enter a positive number";
+        const raw = BigInt(Math.round(n * 1e6));
+        if (raw < minimumPurchase)
+          return `Minimum purchase is ${minimumFormatted} USDC`;
+        if (raw > usdcBalance)
+          return `Insufficient balance. You have ${balanceFormatted} USDC`;
+        return true;
       }
-      const minimumFormatted = formatUnits(minimumPurchase, 6);
-      const amountStr = flags.amount ?? await input2({
-        message: `How much USDC to spend on credits? (minimum: ${minimumFormatted})`,
-        validate: (val) => {
-          const n = parseFloat(val);
-          if (isNaN(n) || n <= 0) return "Enter a positive number";
-          const raw = BigInt(Math.round(n * 1e6));
-          if (raw < minimumPurchase)
-            return `Minimum purchase is ${minimumFormatted} USDC`;
-          if (raw > usdcBalance)
-            return `Insufficient balance. You have ${balanceFormatted} USDC`;
-          return true;
-        }
-      });
-      const amountFloat = parseFloat(amountStr);
-      const amountRaw = BigInt(Math.round(amountFloat * 1e6));
-      if (amountRaw < minimumPurchase) {
-        this.error(`Minimum purchase is ${minimumFormatted} USDC`);
+    });
+    const amountFloat = parseFloat(amountStr);
+    const amountRaw = BigInt(Math.round(amountFloat * 1e6));
+    if (amountRaw < minimumPurchase) {
+      this.error(`Minimum purchase is ${minimumFormatted} USDC`);
+    }
+    if (amountRaw > usdcBalance) {
+      this.error(
+        `Insufficient USDC balance. You have ${balanceFormatted} USDC but requested ${amountFloat.toFixed(2)}`
+      );
+    }
+    this.log(`
+  Purchasing ${chalk2.bold(`$${amountFloat.toFixed(2)}`)} in credits...`);
+    const { txHash } = await billing.topUp({
+      amount: amountRaw,
+      account: targetAccount
+    });
+    this.log(`  ${chalk2.green("\u2713")} Transaction confirmed: ${txHash}`);
+    await this.pollForCredits(billing, flags, baselineTotal, amountFloat);
+  }
+  async handleCard(billing, flags, baselineTotal) {
+    const MINIMUM_DOLLARS = 5;
+    const amountStr = flags.amount ?? await input2({
+      message: `How many dollars of credits to purchase? (minimum: $${MINIMUM_DOLLARS})`,
+      validate: (val) => {
+        const n = parseInt(val, 10);
+        if (isNaN(n) || n <= 0) return "Enter a positive whole number";
+        if (n.toString() !== val.trim()) return "Enter a whole dollar amount (no cents)";
+        if (n < MINIMUM_DOLLARS) return `Minimum purchase is $${MINIMUM_DOLLARS}`;
+        return true;
       }
-      if (amountRaw > usdcBalance) {
-        this.error(
-          `Insufficient USDC balance. You have ${balanceFormatted} USDC but requested ${amountFloat.toFixed(2)}`
-        );
+    });
+    const dollars = parseInt(amountStr, 10);
+    if (isNaN(dollars) || dollars < MINIMUM_DOLLARS) {
+      this.error(`Minimum purchase is $${MINIMUM_DOLLARS}`);
+    }
+    const amountCents = dollars * 100;
+    const { paymentMethods } = await billing.getPaymentMethods();
+    let paymentMethodId;
+    if (paymentMethods.length > 0) {
+      const choices = paymentMethods.map((card) => ({
+        value: card.id,
+        name: `${card.brand.charAt(0).toUpperCase() + card.brand.slice(1)} ending in ${card.last4}`
+      }));
+      choices.push({ value: "new", name: "Add a new card" });
+      const selection = await select2({
+        message: "Which card would you like to use?",
+        choices
+      });
+      if (selection !== "new") {
+        paymentMethodId = selection;
       }
+    }
+    this.log(`
+  Purchasing ${chalk2.bold(`$${dollars}`)} in credits...`);
+    const result = await billing.purchaseCredits(amountCents, paymentMethodId);
+    if (result.checkoutUrl) {
       this.log(`
-  Purchasing ${chalk2.bold(`$${amountFloat.toFixed(2)}`)} in credits...`);
-      const { txHash } = await billing.topUp({
-        amount: amountRaw,
-        account: targetAccount
-      });
-      this.log(`  ${chalk2.green("\u2713")} Transaction confirmed: ${txHash}`);
-      this.log(chalk2.gray("\n  Waiting for credits to appear..."));
-      const startTime = Date.now();
-      while (Date.now() - startTime < POLL_TIMEOUT_MS) {
-        await new Promise((r) => setTimeout(r, POLL_INTERVAL_MS));
-        try {
-          const status = await billing.getStatus({
-            productId: flags.product
-          });
-          const remaining = status.remainingCredits ?? 0;
-          const applied = status.creditsApplied ?? 0;
-          const currentTotal = remaining + applied;
-          this.debug(`Poll: remaining=${remaining}, applied=${applied}, total=${currentTotal}, baseline=${baselineTotal}`);
-          if (baselineTotal === void 0 || currentTotal > baselineTotal) {
-            const creditsAdded = baselineTotal !== void 0 ? currentTotal - baselineTotal : void 0;
-            const isMatched = creditsAdded !== void 0 && Math.abs(creditsAdded - amountFloat * 2) < 0.01;
-            const appliedFromTopUp = creditsAdded !== void 0 ? creditsAdded - remaining : 0;
-            this.log(`
-  ${chalk2.green("\u2713")} Credits received: ${chalk2.cyan(`$${(creditsAdded ?? amountFloat).toFixed(2)}`)}`);
-            if (isMatched) {
-              this.log(`  ${chalk2.green("\u2713")} Includes $${amountFloat.toFixed(2)} match bonus!`);
-            }
-            if (remaining > 0) {
-              this.log(`  Remaining balance: ${chalk2.cyan(`$${remaining.toFixed(2)}`)}`);
-            }
-            if (appliedFromTopUp > 0) {
-              this.log(`  ${chalk2.gray(`$${appliedFromTopUp.toFixed(2)} applied to current bill`)}`);
-            }
-            this.log();
-            return;
+  ${chalk2.cyan(result.checkoutUrl)}`);
+      this.log(chalk2.gray("  Opening checkout in browser..."));
+      await open(result.checkoutUrl);
+    } else if (result.checkoutSessionId) {
+      this.error(
+        "Checkout session created but no URL was returned. Please contact support."
+      );
+    } else {
+      this.log(`  ${chalk2.green("\u2713")} Payment submitted`);
+    }
+    await this.pollForCredits(billing, flags, baselineTotal, dollars);
+  }
+  async pollForCredits(billing, flags, baselineTotal, amountPurchased) {
+    this.log(chalk2.gray("\n  Waiting for credits to appear..."));
+    const startTime = Date.now();
+    while (Date.now() - startTime < POLL_TIMEOUT_MS) {
+      await new Promise((r) => setTimeout(r, POLL_INTERVAL_MS));
+      try {
+        const status = await billing.getStatus({
+          productId: flags.product
+        });
+        const remaining = status.remainingCredits ?? 0;
+        const applied = status.creditsApplied ?? 0;
+        const currentTotal = remaining + applied;
+        this.debug(
+          `Poll: remaining=${remaining}, applied=${applied}, total=${currentTotal}, baseline=${baselineTotal}`
+        );
+        if (baselineTotal === void 0 || currentTotal > baselineTotal) {
+          const creditsAdded = baselineTotal !== void 0 ? currentTotal - baselineTotal : void 0;
+          this.log(
+            `
+  ${chalk2.green("\u2713")} Credits received: ${chalk2.cyan(`$${(creditsAdded ?? amountPurchased).toFixed(2)}`)}`
+          );
+          if (remaining > 0) {
+            this.log(`  Remaining balance: ${chalk2.cyan(`$${remaining.toFixed(2)}`)}`);
           }
-        } catch {
-          this.debug("Error polling for credit balance");
+          this.log();
+          return;
         }
+      } catch {
+        this.debug("Error polling for credit balance");
       }
-      this.log(
-        `
+    }
+    this.log(
+      `
   ${chalk2.yellow("\u26A0")} Credits haven't appeared yet. This can take a few minutes.`
-      );
-      this.log(`  ${chalk2.gray("Check your balance:")} ecloud billing status
+    );
+    this.log(`  ${chalk2.gray("Check your balance:")} ecloud billing status
 `);
-    });
   }
 };
 
 // src/commands/billing/__tests__/top-up.test.ts
-import { input as input3 } from "@inquirer/prompts";
+import { input as input3, select as select3 } from "@inquirer/prompts";
 vi.mock("../../../client", () => ({
   createBillingClient: vi.fn()
 }));
@@ -526,7 +598,11 @@ vi.mock("../../../telemetry", () => ({
   withTelemetry: vi.fn((_cmd, fn) => fn())
 }));
 vi.mock("@inquirer/prompts", () => ({
-  input: vi.fn()
+  input: vi.fn(),
+  select: vi.fn()
+}));
+vi.mock("open", () => ({
+  default: vi.fn()
 }));
 var WALLET_ADDRESS = "0x1234567890abcdef1234567890abcdef12345678";
 var TX_HASH = "0xabcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890";
@@ -541,7 +617,9 @@ describe("ecloud billing top-up", () => {
       address: WALLET_ADDRESS,
       getStatus: vi.fn(),
       getTopUpInfo: vi.fn(),
-      topUp: vi.fn()
+      topUp: vi.fn(),
+      getPaymentMethods: vi.fn(),
+      purchaseCredits: vi.fn()
     };
     createBillingClient.mockResolvedValue(mockBilling);
     input3.mockResolvedValue("50");
@@ -586,7 +664,7 @@ describe("ecloud billing top-up", () => {
     setupOnChainState();
     mockBilling.topUp.mockResolvedValue({ txHash: TX_HASH, walletAddress: WALLET_ADDRESS });
     mockBilling.getStatus.mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 10 }).mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 60 });
-    const cmd = createCommand({ amount: "50" });
+    const cmd = createCommand({ amount: "50", method: "usdc" });
     const promise = cmd.run();
     for (let i = 0; i < 10; i++) {
       await vi.advanceTimersByTimeAsync(5e3);
@@ -608,7 +686,7 @@ describe("ecloud billing top-up", () => {
   it("zero USDC balance: exits with fund wallet message", async () => {
     setupOnChainState({ usdcBalance: BigInt(0) });
     mockBilling.getStatus.mockResolvedValue({ subscriptionStatus: "inactive" });
-    const cmd = createCommand({ amount: "50" });
+    const cmd = createCommand({ amount: "50", method: "usdc" });
     await cmd.run();
     const fullOutput = logOutput.join("\n");
     expect(fullOutput).toContain("No USDC in wallet");
@@ -619,7 +697,7 @@ describe("ecloud billing top-up", () => {
   it("below minimum purchase: shows error", async () => {
     setupOnChainState({ minimumPurchase: BigInt(1e7) });
     mockBilling.getStatus.mockResolvedValue({ subscriptionStatus: "inactive" });
-    const cmd = createCommand({ amount: "5" });
+    const cmd = createCommand({ amount: "5", method: "usdc" });
     await expect(cmd.run()).rejects.toThrow("Minimum purchase is 10 USDC");
   });
   it("--account flag: passes different address to topUp", async () => {
@@ -627,7 +705,7 @@ describe("ecloud billing top-up", () => {
     setupOnChainState();
     mockBilling.topUp.mockResolvedValue({ txHash: TX_HASH, walletAddress: WALLET_ADDRESS });
     mockBilling.getStatus.mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 10 }).mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 60 });
-    const cmd = createCommand({ amount: "50", account: targetAccount });
+    const cmd = createCommand({ amount: "50", method: "usdc", account: targetAccount });
     const promise = cmd.run();
     for (let i = 0; i < 10; i++) {
       await vi.advanceTimersByTimeAsync(5e3);
@@ -647,7 +725,7 @@ describe("ecloud billing top-up", () => {
       subscriptionStatus: "active",
       remainingCredits: 10
     });
-    const cmd = createCommand({ amount: "50" });
+    const cmd = createCommand({ amount: "50", method: "usdc" });
     const promise = cmd.run();
     await vi.advanceTimersByTimeAsync(2e5);
     await promise;
@@ -659,7 +737,7 @@ describe("ecloud billing top-up", () => {
     setupOnChainState();
     mockBilling.topUp.mockResolvedValue({ txHash: TX_HASH, walletAddress: WALLET_ADDRESS });
     mockBilling.getStatus.mockResolvedValueOnce({ subscriptionStatus: "inactive" }).mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 100 });
-    const cmd = createCommand({ amount: "100" });
+    const cmd = createCommand({ amount: "100", method: "usdc" });
     const promise = cmd.run();
     for (let i = 0; i < 10; i++) {
       await vi.advanceTimersByTimeAsync(5e3);
@@ -671,7 +749,7 @@ describe("ecloud billing top-up", () => {
     setupOnChainState();
     mockBilling.topUp.mockResolvedValue({ txHash: TX_HASH, walletAddress: WALLET_ADDRESS });
     mockBilling.getStatus.mockRejectedValue(new Error("API unavailable"));
-    const cmd = createCommand({ amount: "50" });
+    const cmd = createCommand({ amount: "50", method: "usdc" });
     const promise = cmd.run();
     await vi.advanceTimersByTimeAsync(2e5);
     await promise;
@@ -680,5 +758,111 @@ describe("ecloud billing top-up", () => {
     expect(fullOutput).toContain("Transaction confirmed");
     expect(fullOutput).toContain("Credits haven't appeared yet");
   });
+  it("credit card: charges selected card on file", async () => {
+    mockBilling.getStatus.mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 10 }).mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 35 });
+    mockBilling.getPaymentMethods.mockResolvedValue({
+      paymentMethods: [
+        {
+          id: "029641fc-3e5c-11f1-986c-5601121cbf6d",
+          stripePaymentMethodId: "pm_1ABC1234",
+          brand: "visa",
+          last4: "1234",
+          createdAt: "2026-04-20T15:00:00Z"
+        },
+        {
+          id: "139752fd-4e6d-22f2-a97d-6712232dcg7e",
+          stripePaymentMethodId: "pm_2DEF5678",
+          brand: "mastercard",
+          last4: "5678",
+          createdAt: "2026-04-21T10:00:00Z"
+        }
+      ]
+    });
+    mockBilling.purchaseCredits.mockResolvedValue({
+      purchaseId: "a1b2c3d4",
+      amountCents: "2500"
+    });
+    select3.mockResolvedValue("029641fc-3e5c-11f1-986c-5601121cbf6d");
+    const cmd = createCommand({ amount: "25", method: "card" });
+    const promise = cmd.run();
+    for (let i = 0; i < 10; i++) {
+      await vi.advanceTimersByTimeAsync(5e3);
+    }
+    await promise;
+    const fullOutput = logOutput.join("\n");
+    expect(mockBilling.purchaseCredits).toHaveBeenCalledWith(2500, "029641fc-3e5c-11f1-986c-5601121cbf6d");
+    expect(fullOutput).toContain("Payment submitted");
+    expect(fullOutput).toContain("Credits received");
+  });
+  it("credit card: opens checkout when user selects add new card", async () => {
+    const openMock = (await import("open")).default;
+    mockBilling.getStatus.mockResolvedValue({ subscriptionStatus: "active", remainingCredits: 10 });
+    mockBilling.getPaymentMethods.mockResolvedValue({
+      paymentMethods: [
+        {
+          id: "029641fc-3e5c-11f1-986c-5601121cbf6d",
+          stripePaymentMethodId: "pm_1ABC1234",
+          brand: "visa",
+          last4: "1234",
+          createdAt: "2026-04-20T15:00:00Z"
+        }
+      ]
+    });
+    mockBilling.purchaseCredits.mockResolvedValue({
+      checkoutSessionId: "cs_test_abc123",
+      checkoutUrl: "https://checkout.stripe.com/test",
+      amountCents: "2500"
+    });
+    select3.mockResolvedValue("new");
+    const cmd = createCommand({ amount: "25", method: "card" });
+    const promise = cmd.run();
+    await vi.advanceTimersByTimeAsync(2e5);
+    await promise;
+    const fullOutput = logOutput.join("\n");
+    expect(mockBilling.purchaseCredits).toHaveBeenCalledWith(2500, void 0);
+    expect(openMock).toHaveBeenCalledWith("https://checkout.stripe.com/test");
+    expect(fullOutput).toContain("https://checkout.stripe.com/test");
+  });
+  it("credit card: opens checkout when no card on file", async () => {
+    const openMock = (await import("open")).default;
+    mockBilling.getStatus.mockResolvedValue({ subscriptionStatus: "active", remainingCredits: 10 });
+    mockBilling.getPaymentMethods.mockResolvedValue({ paymentMethods: [] });
+    mockBilling.purchaseCredits.mockResolvedValue({
+      checkoutSessionId: "cs_test_abc123",
+      checkoutUrl: "https://checkout.stripe.com/test",
+      amountCents: "5000"
+    });
+    const cmd = createCommand({ amount: "50", method: "card" });
+    const promise = cmd.run();
+    await vi.advanceTimersByTimeAsync(2e5);
+    await promise;
+    const fullOutput = logOutput.join("\n");
+    expect(select3).not.toHaveBeenCalled();
+    expect(mockBilling.purchaseCredits).toHaveBeenCalledWith(5e3, void 0);
+    expect(openMock).toHaveBeenCalledWith("https://checkout.stripe.com/test");
+    expect(fullOutput).toContain("https://checkout.stripe.com/test");
+  });
+  it("credit card: rejects amount below $5 minimum", async () => {
+    mockBilling.getStatus.mockResolvedValue({ subscriptionStatus: "active", remainingCredits: 10 });
+    const cmd = createCommand({ amount: "3", method: "card" });
+    await expect(cmd.run()).rejects.toThrow("Minimum purchase is $5");
+  });
+  it("credit card: --method and --amount flags skip prompts", async () => {
+    mockBilling.getStatus.mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 10 }).mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 60 });
+    mockBilling.getPaymentMethods.mockResolvedValue({ paymentMethods: [] });
+    mockBilling.purchaseCredits.mockResolvedValue({
+      checkoutSessionId: "cs_test_abc123",
+      checkoutUrl: "https://checkout.stripe.com/test",
+      amountCents: "5000"
+    });
+    const cmd = createCommand({ amount: "50", method: "card" });
+    const promise = cmd.run();
+    for (let i = 0; i < 10; i++) {
+      await vi.advanceTimersByTimeAsync(5e3);
+    }
+    await promise;
+    expect(select3).not.toHaveBeenCalled();
+    expect(input3).not.toHaveBeenCalled();
+  });
 });
 //# sourceMappingURL=top-up.test.js.map