augteam 1.1.2 → 1.1.4

package/package.json

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

package/src/commands/status.js

@@ -1,5 +1,6 @@
 const { apiRequest } = require("../utils/api");
 const { requireAuth } = require("../utils/auth");
+const { getCliName } = require("../utils/paths");
 
 async function statusCommand() {
   requireAuth();
@@ -8,7 +9,7 @@ async function statusCommand() {
     const data = await apiRequest("GET", "/api/session/status");
 
     if (!data.current) {
-      console.log("ℹ️  No active session. Run: auggw switch");
+      console.log(`ℹ️  No active session. Run: ${getCliName()} switch`);
       console.log();
     } else {
       console.log("📋 Current Account");
@@ -20,7 +21,7 @@ async function statusCommand() {
       if (data.current.creditTotal) {
         const remaining = data.current.creditRemaining || 0;
         const total = data.current.creditTotal;
-        const used = total - remaining;
+        const used = Math.round(total - remaining);
         const pct = Math.round((used / total) * 100);
         const barWidth = 30;
         const filled = Math.round((pct / 100) * barWidth);
@@ -29,7 +30,7 @@ async function statusCommand() {
         const reset = "\x1b[0m";
         const bar = `${color}${"█".repeat(filled)}${"\x1b[90m"}${"░".repeat(empty)}${reset}`;
         console.log(
-          `   Credit:  ${remaining.toLocaleString()} / ${total.toLocaleString()} remaining`,
+          `   Credit:  ${used.toLocaleString()} / ${total.toLocaleString()} (${pct}%)`,
         );
         console.log(
           `   ${bar} ${color}${pct}% used${reset}`,
@@ -46,7 +47,7 @@ async function statusCommand() {
     console.log(`   Total:   ${data.pool.total} accounts`);
   } catch (err) {
     if (err.status === 401) {
-      console.error("❌ Session expired. Please login again: auggw login");
+      console.error(`❌ Session expired. Please login again: ${getCliName()} login`);
     } else {
       console.error(`❌ Error: ${err.message}`);
     }

package/src/index.js

@@ -1,50 +1,51 @@
-const { Command } = require('commander');
-const loginCommand = require('./commands/login');
-const logoutCommand = require('./commands/logout');
-const switchCommand = require('./commands/switch');
-const statusCommand = require('./commands/status');
-const updateCommand = require('./commands/update');
-const fetchCommand = require('./commands/fetch');
-const { checkUpdate } = require('./utils/update');
-const pkg = require('../package.json');
-
-const program = new Command();
-
-program
-  .name(Object.keys(pkg.bin)[0] || 'auggw')
-  .description('CLI tool for rotating Augment session accounts')
-  .version(pkg.version);
-
-program
-  .command('login')
-  .description('Authenticate with the session rotator backend')
-  .action(loginCommand);
-
-program
-  .command('logout')
-  .description('Remove saved token and log out')
-  .action(logoutCommand);
-
-program
-  .command('switch')
-  .description('Switch to the next available Augment session')
-  .action(switchCommand);
-
-program
-  .command('status')
-  .description('View current account and pool status')
-  .action(statusCommand);
-
-program
-  .command('update')
-  .description('Update CLI to the latest version')
-  .action(updateCommand);
-
-program
-  .command('fetch')
-  .description('Fetch resources from server')
-  .option('--prompts', 'Fetch commands & skills to ~/.augment and/or ~/.claude')
-  .action(fetchCommand);
-
-program.parseAsync().then(() => checkUpdate());
-
+const path = require("path");
+require("dotenv").config({ path: path.join(__dirname, "..", ".env"), quiet: true });
+const { Command } = require("commander");
+const loginCommand = require("./commands/login");
+const logoutCommand = require("./commands/logout");
+const switchCommand = require("./commands/switch");
+const statusCommand = require("./commands/status");
+const updateCommand = require("./commands/update");
+const fetchCommand = require("./commands/fetch");
+const { checkUpdate } = require("./utils/update");
+const pkg = require("../package.json");
+
+const program = new Command();
+
+program
+  .name(Object.keys(pkg.bin)[0] || "auggw")
+  .description("CLI tool for rotating Augment session accounts")
+  .version(pkg.version);
+
+program
+  .command("login")
+  .description("Authenticate with the session rotator backend")
+  .action(loginCommand);
+
+program
+  .command("logout")
+  .description("Remove saved token and log out")
+  .action(logoutCommand);
+
+program
+  .command("switch")
+  .description("Switch to the next available Augment session")
+  .action(switchCommand);
+
+program
+  .command("status")
+  .description("View current account and pool status")
+  .action(statusCommand);
+
+program
+  .command("update")
+  .description("Update CLI to the latest version")
+  .action(updateCommand);
+
+program
+  .command("fetch")
+  .description("Fetch resources from server")
+  .option("--prompts", "Fetch commands & skills to ~/.augment and/or ~/.claude")
+  .action(fetchCommand);
+
+program.parseAsync().then(() => checkUpdate());

package/src/utils/api.js

@@ -1,11 +1,13 @@
 const os = require("os");
 const { loadToken } = require("./token");
+const { getCliName } = require("./paths");
 
 const DEFAULT_API_URL = "https://augteam.quangit.site";
 // const DEFAULT_API_URL = "http://localhost:3000";
 
 function getBaseURL() {
-  return process.env.auggw_API_URL || DEFAULT_API_URL;
+  const envKey = `${getCliName()}_API_URL`;
+  return process.env[envKey] || DEFAULT_API_URL;
 }
 
 function getLocalIP() {

package/src/utils/auth.js

@@ -1,9 +1,10 @@
 const { loadToken } = require("./token");
+const { getCliName } = require("./paths");
 
 function requireAuth() {
   const token = loadToken();
   if (!token) {
-    console.error("❌ Please login first: auggw login");
+    console.error(`❌ Please login first: ${getCliName()} login`);
     process.exit(1);
   }
 }

package/src/utils/paths.js

@@ -1,12 +1,19 @@
 const os = require("os");
 const path = require("path");
 
+const pkg = require("../../package.json");
+const CLI_NAME = Object.keys(pkg.bin || {})[0] || "auggw";
+
 function getTokenPath() {
-  return path.join(os.homedir(), ".auggw", "token");
+  return path.join(os.homedir(), `.${CLI_NAME}`, "token");
 }
 
 function getSessionPath() {
   return path.join(os.homedir(), ".augment", "session.json");
 }
 
-module.exports = { getTokenPath, getSessionPath };
+function getCliName() {
+  return CLI_NAME;
+}
+
+module.exports = { getTokenPath, getSessionPath, getCliName };