icbauggw 1.2.3 → 1.2.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "icbauggw",
-  "version": "1.2.3",
+  "version": "1.2.5",
   "description": "CLI tool for rotating Augment session accounts",
   "main": "src/index.js",
   "bin": {

package/src/commands/switch.js

@@ -4,11 +4,12 @@ const { apiRequest } = require("../utils/api");
 const { requireAuth } = require("../utils/auth");
 const { getSessionPath, getClaudeSettingsPath } = require("../utils/paths");
 
-async function switchCommand() {
+async function switchCommand(account) {
   requireAuth();
 
   try {
-    const data = await apiRequest("GET", "/api/session/next");
+    const query = account ? `?account=${encodeURIComponent(account)}` : "";
+    const data = await apiRequest("GET", `/api/session/next${query}`);
 
     const sessionPath = getSessionPath();
     const sessionDir = path.dirname(sessionPath);
@@ -26,7 +27,7 @@ async function switchCommand() {
     };
     fs.writeFileSync(sessionPath, JSON.stringify(session, null, 2), "utf8");
 
-    // Write Claude settings file
+    // Write Claude settings file (silently)
     if (data.claudeSettings) {
       const claudeSettingsPath = getClaudeSettingsPath();
       const claudeDir = path.dirname(claudeSettingsPath);
@@ -38,15 +39,34 @@ async function switchCommand() {
         JSON.stringify(data.claudeSettings, null, 2),
         "utf8",
       );
-      console.log(`   Claude settings → ${claudeSettingsPath}`);
     }
 
     console.log("✅ Session switched!");
     console.log(`   Account: ${data.accountName}`);
-    console.log(`   Tenant:  ${data.tenantURL}`);
+
+    // Credit info
+    if (data.creditTotal) {
+      const remaining = data.creditRemaining || 0;
+      const total = data.creditTotal;
+      const used = Math.round(total - remaining);
+      const pct = Math.round((used / total) * 100);
+      const barWidth = 30;
+      const filled = Math.round((pct / 100) * barWidth);
+      const empty = barWidth - filled;
+      const color =
+        pct > 90 ? "\x1b[31m" : pct > 70 ? "\x1b[33m" : "\x1b[32m";
+      const reset = "\x1b[0m";
+      const bar = `${color}${"█".repeat(filled)}${"\x1b[90m"}${"░".repeat(empty)}${reset}`;
+      console.log(
+        `   Credit:  ${used.toLocaleString()} / ${total.toLocaleString()} (${pct}%)`,
+      );
+      console.log(`   ${bar} ${color}${pct}% used${reset}`);
+    }
   } catch (err) {
     if (err.status === 401) {
       console.error("❌ Session expired. Please login again: auggw login");
+    } else if (err.status === 404) {
+      console.error(`❌ ${err.message}`);
     } else if (err.status === 503) {
       console.error(`❌ ${err.message}`);
     } else {

package/src/index.js

@@ -26,8 +26,8 @@ program
   .action(logoutCommand);
 
 program
-  .command('switch')
-  .description('Switch to the next available Augment session')
+  .command('switch [account]')
+  .description('Switch to the next available Augment session (or specify account name)')
   .action(switchCommand);
 
 program