feat: display wallet address in billing commands (#55)

Author: solimander

packages/cli/src/commands/billing/cancel.ts

@@ -32,7 +32,7 @@ export default class BillingCancel extends Command {
       const billing = await createBillingClient(flags);
 
       // Check subscription status first
-      this.log(`\nChecking subscription status for ${flags.product}...`);
+      this.debug(`\nChecking subscription status for ${flags.product}...`);
       const status = await billing.getStatus({
         productId: flags.product as "compute",
       });
@@ -47,7 +47,7 @@ export default class BillingCancel extends Command {
       // Confirm cancellation unless --force flag is used
       if (!flags.force) {
         const confirmed = await confirm({
-          message: `${chalk.yellow("Warning:")} This will cancel your ${flags.product} subscription. Continue?`,
+          message: `${chalk.yellow("Warning:")} This will cancel the ${flags.product} subscription for wallet ${chalk.bold(billing.address)}. Continue?`,
         });
         if (!confirmed) {
           this.log(chalk.gray("\nCancellation aborted."));

packages/cli/src/commands/billing/status.ts

@@ -58,6 +58,7 @@ export default class BillingStatus extends Command {
       };
 
       this.log(`\n${chalk.bold("Subscription Status:")}`);
+      this.log(`  Wallet: ${billing.address}`);
       this.log(`  Status: ${formatStatus(result.subscriptionStatus)}`);
       this.log(`  Product: ${result.productId}`);
 

packages/cli/src/commands/billing/subscribe.ts

@@ -37,7 +37,7 @@ export default class BillingSubscribe extends Command {
 
       // Handle already active subscription
       if (result.type === "already_active") {
-        this.log(`\n${chalk.green("✓")} You're already subscribed to ${flags.product}.`);
+        this.log(`\n${chalk.green("✓")} Wallet ${chalk.bold(billing.address)} is already subscribed to ${flags.product}.`);
         this.log(chalk.gray("Run 'ecloud billing status' for details."));
         return;
       }
@@ -57,7 +57,7 @@ export default class BillingSubscribe extends Command {
       }
 
       // Open checkout URL in browser
-      this.log(`\n${chalk.gray("Opening checkout in your browser...")}`);
+      this.log(`\nOpening checkout for wallet ${chalk.bold(billing.address)}...`);
       this.log(chalk.gray(`\nURL: ${result.checkoutUrl}`));
       await open(result.checkoutUrl);
 

packages/sdk/src/client/modules/billing/index.ts

@@ -5,9 +5,10 @@
 import { BillingApiClient } from "../../common/utils/billingapi";
 import { getBillingEnvironmentConfig, getBuildType } from "../../common/config/environment";
 import { getLogger, isSubscriptionActive, addHexPrefix } from "../../common/utils";
+import { getAddressFromPrivateKey } from "../../common/auth";
 import { withSDKTelemetry } from "../../common/telemetry/wrapper";
 
-import type { Hex } from "viem";
+import type { Address, Hex } from "viem";
 import type {
   ProductID,
   SubscriptionOpts,
@@ -17,6 +18,7 @@ import type {
 } from "../../common/types";
 
 export interface BillingModule {
+  address: Address;
   subscribe: (opts?: SubscriptionOpts) => Promise<SubscribeResponse>;
   getStatus: (opts?: SubscriptionOpts) => Promise<ProductSubscriptionResponse>;
   cancel: (opts?: SubscriptionOpts) => Promise<CancelResponse>;
@@ -31,6 +33,7 @@ export interface BillingModuleConfig {
 export function createBillingModule(config: BillingModuleConfig): BillingModule {
   const { verbose = false, skipTelemetry = false } = config;
   const privateKey = addHexPrefix(config.privateKey);
+  const address = getAddressFromPrivateKey(privateKey) as Address;
 
   const logger = getLogger(verbose);
 
@@ -41,6 +44,7 @@ export function createBillingModule(config: BillingModuleConfig): BillingModule
   const billingApi = new BillingApiClient(billingEnvConfig, privateKey);
 
   return {
+    address,
     async subscribe(opts) {
       return withSDKTelemetry(
         {