@layr-labs/ecloud-cli 0.3.4 → 0.4.0-dev.0

package/dist/commands/billing/__tests__/status.test.js

@@ -0,0 +1,552 @@
+#!/usr/bin/env node
+var __defProp = Object.defineProperty;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __esm = (fn, res) => function __init() {
+  return fn && (res = (0, fn[__getOwnPropNames(fn)[0]])(fn = 0)), res;
+};
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+
+// src/utils/viemClients.ts
+import {
+  createPublicClient,
+  http
+} from "viem";
+import { privateKeyToAccount } from "viem/accounts";
+import {
+  getEnvironmentConfig,
+  addHexPrefix,
+  createViemClients as sdkCreateViemClients,
+  getChainFromID
+} from "@layr-labs/ecloud-sdk";
+var init_viemClients = __esm({
+  "src/utils/viemClients.ts"() {
+    "use strict";
+  }
+});
+
+// src/utils/globalConfig.ts
+import * as fs from "fs";
+import * as path from "path";
+import * as os from "os";
+import { load as loadYaml, dump as dumpYaml } from "js-yaml";
+import { getBuildType } from "@layr-labs/ecloud-sdk";
+import * as crypto from "crypto";
+function getGlobalConfigDir() {
+  const configHome = process.env.XDG_CONFIG_HOME;
+  let baseDir;
+  if (configHome && path.isAbsolute(configHome)) {
+    baseDir = configHome;
+  } else {
+    baseDir = path.join(os.homedir(), ".config");
+  }
+  const buildType = getBuildType();
+  const buildSuffix = buildType === "dev" ? "-dev" : "";
+  const configDirName = `ecloud${buildSuffix}`;
+  return path.join(baseDir, configDirName);
+}
+function getGlobalConfigPath() {
+  return path.join(getGlobalConfigDir(), GLOBAL_CONFIG_FILE);
+}
+function loadGlobalConfig() {
+  const configPath = getGlobalConfigPath();
+  if (!fs.existsSync(configPath)) {
+    return {
+      first_run: true
+    };
+  }
+  try {
+    const content = fs.readFileSync(configPath, "utf-8");
+    const config = loadYaml(content);
+    return config || { first_run: true };
+  } catch {
+    return {
+      first_run: true
+    };
+  }
+}
+function saveGlobalConfig(config) {
+  const configPath = getGlobalConfigPath();
+  const configDir = path.dirname(configPath);
+  fs.mkdirSync(configDir, { recursive: true, mode: 493 });
+  const content = dumpYaml(config, { lineWidth: -1 });
+  fs.writeFileSync(configPath, content, { mode: 420 });
+}
+function getDefaultEnvironment() {
+  const config = loadGlobalConfig();
+  return config.default_environment;
+}
+function getGlobalTelemetryPreference() {
+  const config = loadGlobalConfig();
+  return config.telemetry_enabled;
+}
+function getOrCreateUserUUID() {
+  const config = loadGlobalConfig();
+  if (config.user_uuid) {
+    return config.user_uuid;
+  }
+  const uuid = generateUUID();
+  config.user_uuid = uuid;
+  config.first_run = false;
+  saveGlobalConfig(config);
+  return uuid;
+}
+function generateUUID() {
+  const bytes = crypto.randomBytes(16);
+  bytes[6] = bytes[6] & 15 | 64;
+  bytes[8] = bytes[8] & 63 | 128;
+  const hex = Array.from(bytes, (b) => b.toString(16).padStart(2, "0"));
+  return hex.slice(0, 4).join("") + hex.slice(4, 6).join("") + "-" + hex.slice(6, 8).join("") + "-" + hex.slice(8, 10).join("") + "-" + hex.slice(10, 12).join("") + "-" + hex.slice(12, 16).join("");
+}
+var GLOBAL_CONFIG_FILE, PROFILE_CACHE_TTL_MS;
+var init_globalConfig = __esm({
+  "src/utils/globalConfig.ts"() {
+    "use strict";
+    GLOBAL_CONFIG_FILE = "config.yaml";
+    PROFILE_CACHE_TTL_MS = 24 * 60 * 60 * 1e3;
+  }
+});
+
+// src/utils/appNames.ts
+import * as fs2 from "fs";
+import * as path2 from "path";
+import * as os2 from "os";
+import { load as loadYaml2, dump as dumpYaml2 } from "js-yaml";
+var CONFIG_DIR, APPS_DIR;
+var init_appNames = __esm({
+  "src/utils/appNames.ts"() {
+    "use strict";
+    CONFIG_DIR = path2.join(os2.homedir(), ".eigenx");
+    APPS_DIR = path2.join(CONFIG_DIR, "apps");
+  }
+});
+
+// src/utils/version.ts
+var init_version = __esm({
+  "src/utils/version.ts"() {
+    "use strict";
+  }
+});
+
+// src/utils/appResolver.ts
+import { isAddress } from "viem";
+import { privateKeyToAccount as privateKeyToAccount2 } from "viem/accounts";
+import {
+  UserApiClient,
+  getAllAppsByDeveloper
+} from "@layr-labs/ecloud-sdk";
+var init_appResolver = __esm({
+  "src/utils/appResolver.ts"() {
+    "use strict";
+    init_viemClients();
+    init_globalConfig();
+    init_appNames();
+    init_version();
+  }
+});
+
+// src/utils/prompts.ts
+import { input, select, password, confirm as inquirerConfirm } from "@inquirer/prompts";
+import chalk from "chalk";
+import fs3 from "fs";
+import path3 from "path";
+import os3 from "os";
+import { isAddress as isAddress2 } from "viem";
+import { privateKeyToAccount as privateKeyToAccount3 } from "viem/accounts";
+import {
+  getEnvironmentConfig as getEnvironmentConfig2,
+  getAvailableEnvironments,
+  isEnvironmentAvailable,
+  getAllAppsByDeveloper as getAllAppsByDeveloper2,
+  getCategoryDescriptions,
+  fetchTemplateCatalog,
+  PRIMARY_LANGUAGES,
+  validateAppName,
+  validateImageReference,
+  validateFilePath,
+  validatePrivateKeyFormat,
+  extractAppNameFromImage,
+  UserApiClient as UserApiClient2
+} from "@layr-labs/ecloud-sdk";
+async function getPrivateKeyInteractive(privateKey) {
+  if (privateKey) {
+    if (!validatePrivateKeyFormat(privateKey)) {
+      throw new Error("Invalid private key format");
+    }
+    return privateKey;
+  }
+  const { getPrivateKeyWithSource: getPrivateKeyWithSource2 } = await import("@layr-labs/ecloud-sdk");
+  const result = await getPrivateKeyWithSource2({ privateKey: void 0 });
+  if (result) {
+    return result.key;
+  }
+  const key = await password({
+    message: "Enter private key:",
+    mask: true,
+    validate: (value) => {
+      if (!value.trim()) {
+        return "Private key is required";
+      }
+      if (!validatePrivateKeyFormat(value)) {
+        return "Invalid private key format (must be 64 hex characters, optionally prefixed with 0x)";
+      }
+      return true;
+    }
+  });
+  return key.trim();
+}
+var MAX_IMAGE_SIZE;
+var init_prompts = __esm({
+  "src/utils/prompts.ts"() {
+    "use strict";
+    init_appResolver();
+    init_viemClients();
+    init_globalConfig();
+    init_appNames();
+    init_version();
+    MAX_IMAGE_SIZE = 4 * 1024 * 1024;
+  }
+});
+
+// src/flags.ts
+import { Flags } from "@oclif/core";
+import { getBuildType as getBuildType2 } from "@layr-labs/ecloud-sdk";
+var commonFlags;
+var init_flags = __esm({
+  "src/flags.ts"() {
+    "use strict";
+    init_prompts();
+    init_globalConfig();
+    commonFlags = {
+      environment: Flags.string({
+        required: false,
+        description: "Deployment environment to use",
+        env: "ECLOUD_ENV",
+        default: async () => getDefaultEnvironment() || (getBuildType2() === "dev" ? "sepolia-dev" : "sepolia")
+      }),
+      "private-key": Flags.string({
+        required: false,
+        description: "Private key for signing transactions",
+        env: "ECLOUD_PRIVATE_KEY"
+      }),
+      "rpc-url": Flags.string({
+        required: false,
+        description: "RPC URL to connect to blockchain",
+        env: "ECLOUD_RPC_URL"
+      }),
+      verbose: Flags.boolean({
+        required: false,
+        description: "Enable verbose logging (default: false)",
+        default: false
+      })
+    };
+  }
+});
+
+// src/client.ts
+import {
+  createComputeModule,
+  createBillingModule,
+  createBuildModule,
+  getEnvironmentConfig as getEnvironmentConfig3,
+  requirePrivateKey,
+  getPrivateKeyWithSource,
+  addHexPrefix as addHexPrefix2
+} from "@layr-labs/ecloud-sdk";
+import { createWalletClient, custom } from "viem";
+import { privateKeyToAccount as privateKeyToAccount4 } from "viem/accounts";
+async function createBillingClient(flags) {
+  const result = await getPrivateKeyWithSource({
+    privateKey: flags["private-key"]
+  });
+  const privateKey = await getPrivateKeyInteractive(result?.key);
+  const account = privateKeyToAccount4(addHexPrefix2(privateKey));
+  const walletClient = createWalletClient({
+    account,
+    transport: custom({
+      async request() {
+        throw new Error("RPC not available - billing uses local signing only");
+      }
+    })
+  });
+  return createBillingModule({
+    verbose: flags.verbose ?? false,
+    walletClient,
+    skipTelemetry: true
+    // CLI already has telemetry, skip SDK telemetry
+  });
+}
+var init_client = __esm({
+  "src/client.ts"() {
+    "use strict";
+    init_flags();
+    init_prompts();
+    init_version();
+    init_viemClients();
+  }
+});
+
+// src/telemetry.ts
+import {
+  createTelemetryClient,
+  createAppEnvironment,
+  createMetricsContext,
+  addMetric,
+  addMetricWithDimensions,
+  emitMetrics,
+  getBuildType as getBuildType3
+} from "@layr-labs/ecloud-sdk";
+function createCLITelemetryClient() {
+  const userUUID = getOrCreateUserUUID();
+  const environment = createAppEnvironment(userUUID);
+  const telemetryEnabled = getGlobalTelemetryPreference();
+  return createTelemetryClient(environment, "ecloud-cli", {
+    telemetryEnabled: telemetryEnabled !== false
+    // Enabled by default, disabled only if explicitly set to false
+  });
+}
+async function withTelemetry(command, action) {
+  const client = createCLITelemetryClient();
+  const metrics = createMetricsContext();
+  metrics.properties["source"] = "ecloud-cli";
+  metrics.properties["command"] = command.id || command.constructor.name;
+  const environment = getDefaultEnvironment() || "sepolia";
+  metrics.properties["environment"] = environment;
+  const buildType = getBuildType3() || "prod";
+  metrics.properties["build_type"] = buildType;
+  const cliVersion = command.config.version;
+  if (cliVersion) {
+    metrics.properties["cli_version"] = cliVersion;
+  }
+  addMetric(metrics, "Count", 1);
+  let actionError;
+  let result;
+  try {
+    result = await action();
+    return result;
+  } catch (err) {
+    actionError = err instanceof Error ? err : new Error(String(err));
+    throw err;
+  } finally {
+    const resultValue = actionError ? "Failure" : "Success";
+    const dimensions = {};
+    if (actionError) {
+      dimensions["error"] = actionError.message;
+    }
+    addMetricWithDimensions(metrics, resultValue, 1, dimensions);
+    const duration = Date.now() - metrics.startTime.getTime();
+    addMetric(metrics, "DurationMilliseconds", duration);
+    try {
+      await emitMetrics(client, metrics);
+      await client.close();
+    } catch {
+    }
+  }
+}
+var init_telemetry = __esm({
+  "src/telemetry.ts"() {
+    "use strict";
+    init_globalConfig();
+  }
+});
+
+// src/commands/billing/status.ts
+var status_exports = {};
+__export(status_exports, {
+  default: () => BillingStatus
+});
+import { Command, Flags as Flags2 } from "@oclif/core";
+import chalk2 from "chalk";
+var BillingStatus;
+var init_status = __esm({
+  "src/commands/billing/status.ts"() {
+    "use strict";
+    init_client();
+    init_flags();
+    init_telemetry();
+    BillingStatus = class _BillingStatus extends Command {
+      static description = "Show subscription status";
+      static flags = {
+        "private-key": commonFlags["private-key"],
+        verbose: commonFlags.verbose,
+        product: Flags2.string({
+          required: false,
+          description: "Product ID",
+          default: "compute",
+          options: ["compute"],
+          env: "ECLOUD_PRODUCT_ID"
+        })
+      };
+      async run() {
+        return withTelemetry(this, async () => {
+          const { flags } = await this.parse(_BillingStatus);
+          const billing = await createBillingClient(flags);
+          const result = await billing.getStatus({
+            productId: flags.product
+          });
+          const formatExpiry = (timestamp) => timestamp ? ` (expires ${new Date(timestamp * 1e3).toLocaleDateString()})` : "";
+          const formatStatus = (status) => {
+            switch (status) {
+              case "active":
+                return `${chalk2.green("\u2713 Active")}`;
+              case "trialing":
+                return `${chalk2.green("\u2713 Trial")}`;
+              case "past_due":
+                return `${chalk2.yellow("\u26A0 Past Due")}`;
+              case "canceled":
+                return `${chalk2.red("\u2717 Canceled")}`;
+              case "inactive":
+                return `${chalk2.gray("\u2717 Inactive")}`;
+              case "incomplete":
+                return `${chalk2.yellow("\u26A0 Incomplete")}`;
+              case "incomplete_expired":
+                return `${chalk2.red("\u2717 Expired")}`;
+              case "unpaid":
+                return `${chalk2.yellow("\u26A0 Unpaid")}`;
+              case "paused":
+                return `${chalk2.yellow("\u26A0 Paused")}`;
+              default:
+                return status;
+            }
+          };
+          this.log(`
+${chalk2.bold("Subscription Status:")}`);
+          this.log(`  Wallet: ${billing.address}`);
+          this.log(`  Status: ${formatStatus(result.subscriptionStatus)}`);
+          this.log(`  Product: ${result.productId}`);
+          if (result.currentPeriodStart && result.currentPeriodEnd) {
+            const startDate = new Date(result.currentPeriodStart).toLocaleDateString();
+            const endDate = new Date(result.currentPeriodEnd).toLocaleDateString();
+            this.log(`  Current Period: ${startDate} - ${endDate}`);
+          }
+          if (result.lineItems && result.lineItems.length > 0) {
+            this.log(`
+${chalk2.bold("  Line Items:")}`);
+            for (const item of result.lineItems) {
+              const product = `${flags.product.charAt(0).toUpperCase()}${flags.product.slice(1)}`;
+              const chain = item.description.toLowerCase().includes("sepolia") ? "Sepolia" : "Mainnet";
+              this.log(
+                `    \u2022 ${product} (${chain}): $${item.subtotal.toFixed(2)} (${item.quantity} vCPU hours \xD7 $${item.price.toFixed(3)}/vCPU hour)`
+              );
+            }
+          }
+          if (result.creditsApplied !== void 0 && result.creditsApplied > 0) {
+            this.log(`
+${chalk2.bold("  Invoice Summary:")}`);
+            const subtotal = result.upcomingInvoiceSubtotal ?? result.upcomingInvoiceTotal ?? 0;
+            this.log(`    Subtotal:         $${subtotal.toFixed(2)}`);
+            this.log(`    Credits Applied: ${chalk2.green(`-$${result.creditsApplied.toFixed(2)}`)}`);
+            this.log(`    ${"\u2500".repeat(21)}`);
+            this.log(`    Total Due:        $${(result.upcomingInvoiceTotal ?? 0).toFixed(2)}`);
+            if (result.remainingCredits !== void 0) {
+              this.log(
+                `
+  ${chalk2.bold("Remaining Credits:")} ${chalk2.cyan(`$${result.remainingCredits.toFixed(2)}`)}${formatExpiry(result.nextCreditExpiry)}`
+              );
+            }
+          } else if (result.upcomingInvoiceTotal !== void 0) {
+            this.log(`
+  Upcoming Invoice: $${result.upcomingInvoiceTotal.toFixed(2)}`);
+            if (result.remainingCredits !== void 0 && result.remainingCredits > 0) {
+              this.log(
+                `  ${chalk2.bold("Available Credits:")} ${chalk2.cyan(`$${result.remainingCredits.toFixed(2)}`)}${formatExpiry(result.nextCreditExpiry)}`
+              );
+            }
+          }
+          if (result.cancelAtPeriodEnd) {
+            this.log(`
+  ${chalk2.yellow("\u26A0 Subscription will cancel at period end")}`);
+          }
+          if (result.canceledAt) {
+            const cancelDate = new Date(result.canceledAt).toLocaleDateString();
+            this.log(`  Canceled On: ${cancelDate}`);
+          }
+          if (result.portalUrl) {
+            this.log(`
+${chalk2.bold("Payment & Invoices:")}`);
+            this.log(`  ${chalk2.cyan(result.portalUrl)}`);
+          }
+          if (result.subscriptionStatus === "inactive" || result.remainingCredits !== void 0 && result.remainingCredits < 10) {
+            this.log(`
+${chalk2.bold("Need more credits?")}`);
+            this.log(`  Run ${chalk2.cyan("ecloud billing top-up")} to purchase credits.`);
+          }
+          this.log();
+        });
+      }
+    };
+  }
+});
+
+// src/commands/billing/__tests__/status.test.ts
+init_client();
+import { describe, it, expect, vi, beforeEach } from "vitest";
+vi.mock("../../../client", () => ({
+  createBillingClient: vi.fn()
+}));
+vi.mock("../../../telemetry", () => ({
+  withTelemetry: vi.fn((_cmd, fn) => fn())
+}));
+describe("ecloud billing status \u2014 top-up hint", () => {
+  let logOutput;
+  let mockBilling;
+  beforeEach(() => {
+    logOutput = [];
+    mockBilling = {
+      address: "0xabcdef1234567890abcdef1234567890abcdef12",
+      getStatus: vi.fn()
+    };
+    createBillingClient.mockResolvedValue(mockBilling);
+  });
+  async function runStatusCommand(statusResult) {
+    const { default: BillingStatus2 } = await Promise.resolve().then(() => (init_status(), status_exports));
+    mockBilling.getStatus.mockResolvedValue(statusResult);
+    const cmd = new BillingStatus2([], {});
+    cmd.parse = vi.fn().mockResolvedValue({
+      flags: { product: "compute", verbose: false }
+    });
+    cmd.log = vi.fn((...args) => logOutput.push(args.join(" ")));
+    cmd.debug = vi.fn();
+    await cmd.run();
+    return logOutput;
+  }
+  it("shows top-up hint when subscription is inactive", async () => {
+    const output = await runStatusCommand({
+      subscriptionStatus: "inactive",
+      productId: "compute"
+    });
+    const fullOutput = output.join("\n");
+    expect(fullOutput).toContain("ecloud billing top-up");
+    expect(fullOutput).toContain("Need more credits?");
+  });
+  it("shows top-up hint when credits are low (< $10)", async () => {
+    const output = await runStatusCommand({
+      subscriptionStatus: "active",
+      productId: "compute",
+      remainingCredits: 5
+    });
+    const fullOutput = output.join("\n");
+    expect(fullOutput).toContain("ecloud billing top-up");
+  });
+  it("does NOT show top-up hint when credits are healthy", async () => {
+    const output = await runStatusCommand({
+      subscriptionStatus: "active",
+      productId: "compute",
+      remainingCredits: 50,
+      upcomingInvoiceTotal: 12
+    });
+    const fullOutput = output.join("\n");
+    expect(fullOutput).not.toContain("Need more credits?");
+  });
+  it("does NOT show top-up hint when subscription is active with no credit info", async () => {
+    const output = await runStatusCommand({
+      subscriptionStatus: "active",
+      productId: "compute"
+    });
+    const fullOutput = output.join("\n");
+    expect(fullOutput).not.toContain("Need more credits?");
+  });
+});
+//# sourceMappingURL=status.test.js.map