icbauggw 1.2.5 → 1.2.7

package/SETUP.md

@@ -1,62 +1,62 @@
-# Hướng dẫn cài đặt CLI - Augment Session Rotator
-
-## Yêu cầu
-
-- Node.js >= 18
-
-## Cài đặt
-
-```bash
-npm install -g @itd2902/auggw
-```
-
-## Chạy từ source (local dev)
-
-```bash
-cd cli
-npm link
-```
-
-Sau đó dùng `auggw` như bình thường. Mọi thay đổi trong source sẽ có hiệu lực ngay.
-
-Gỡ link khi không cần:
-
-```bash
-cd cli
-npm unlink -g
-```
-
-## Đăng nhập
-
-```bash
-auggw login
-```
-
-Nhập username và password được cấp. Token sẽ lưu tại `~/gg/token`.
-
-## Sử dụng
-
-### Chuyển session
-
-```bash
-auggw switch
-```
-
-Lệnh này lấy account tiếp theo từ pool và ghi vào `~/.augment/session.json`. Chạy lại mỗi khi cần đổi account.
-
-### Xem trạng thái
-
-```bash
-auggw status
-```
-
-Hiển thị account đang dùng, credit usage và số lượng account trong pool.
-
-## Lưu ý
-
-- Sau khi `switch`, restart lại Augment extension để nó đọc session mới.
-- Nếu gặp lỗi `Session expired`, chạy lại `auggw login`.
-- Nếu cần trỏ sang server khác (dev/test), set env:
-  ```bash
-  auggw_API_URL=http://localhost:3000 auggw login
-  ```
+# Hướng dẫn cài đặt CLI - Augment Session Rotator
+
+## Yêu cầu
+
+- Node.js >= 18
+
+## Cài đặt
+
+```bash
+npm install -g @itd2902/auggw
+```
+
+## Chạy từ source (local dev)
+
+```bash
+cd cli
+npm link
+```
+
+Sau đó dùng `auggw` như bình thường. Mọi thay đổi trong source sẽ có hiệu lực ngay.
+
+Gỡ link khi không cần:
+
+```bash
+cd cli
+npm unlink -g
+```
+
+## Đăng nhập
+
+```bash
+auggw login
+```
+
+Nhập username và password được cấp. Token sẽ lưu tại `~/gg/token`.
+
+## Sử dụng
+
+### Chuyển session
+
+```bash
+auggw switch
+```
+
+Lệnh này lấy account tiếp theo từ pool và ghi vào `~/.augment/session.json`. Chạy lại mỗi khi cần đổi account.
+
+### Xem trạng thái
+
+```bash
+auggw status
+```
+
+Hiển thị account đang dùng, credit usage và số lượng account trong pool.
+
+## Lưu ý
+
+- Sau khi `switch`, restart lại Augment extension để nó đọc session mới.
+- Nếu gặp lỗi `Session expired`, chạy lại `auggw login`.
+- Nếu cần trỏ sang server khác (dev/test), set env:
+  ```bash
+  auggw_API_URL=http://localhost:3000 auggw login
+  ```

package/bin/auggw.js

@@ -1,2 +1,2 @@
-#!/usr/bin/env node
-require("../src/index");
+#!/usr/bin/env node
+require("../src/index");

package/bin/augteam.js

@@ -1,3 +1,3 @@
-#!/usr/bin/env node
-require("../src/index");
-
+#!/usr/bin/env node
+require("../src/index");
+

package/package.json

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

package/publish.js

@@ -1,69 +1,69 @@
-#!/usr/bin/env node
-const fs = require('fs');
-const path = require('path');
-const { execSync } = require('child_process');
-
-const CLI_DIR = path.join(__dirname);
-const PKG_PATH = path.join(CLI_DIR, 'package.json');
-const API_PATH = path.join(CLI_DIR, 'src', 'utils', 'api.js');
-
-const packages = [
-  {
-    name: '@itd2902/auggw',
-    bin: { auggw: './bin/auggw.js' },
-    access: '--access public',
-    apiUrl: 'https://auggw.quangit.site',
-  },
-  {
-    name: 'icbauggw',
-    bin: { icbauggw: './bin/icbauggw.js' },
-    access: '',
-    apiUrl: 'https://icbauggw.quangit.site',
-  },
-  {
-    name: 'augteam',
-    bin: { augteam: './bin/augteam.js' },
-    access: '',
-    apiUrl: 'https://augteam.quangit.site',
-  },
-];
-
-const version = process.argv[2];
-if (!version) {
-  console.error('Usage: node publish.js <version>');
-  console.error('Example: node publish.js 1.0.2');
-  process.exit(1);
-}
-
-const originalPkg = fs.readFileSync(PKG_PATH, 'utf8');
-const originalApi = fs.readFileSync(API_PATH, 'utf8');
-
-for (const pkg of packages) {
-  const json = JSON.parse(originalPkg);
-  json.name = pkg.name;
-  json.version = version;
-  json.bin = pkg.bin;
-
-  fs.writeFileSync(PKG_PATH, JSON.stringify(json, null, 2) + '\n');
-
-  // Swap DEFAULT_API_URL
-  const apiContent = originalApi.replace(
-    /const DEFAULT_API_URL = ".*?";/,
-    `const DEFAULT_API_URL = "${pkg.apiUrl}";`
-  );
-  fs.writeFileSync(API_PATH, apiContent);
-
-  console.log(`\nPublishing ${pkg.name}@${version} → ${pkg.apiUrl}`);
-
-  try {
-    execSync(`npm publish ${pkg.access}`.trim(), { cwd: CLI_DIR, stdio: 'inherit' });
-    console.log(`${pkg.name}@${version} published.`);
-  } catch (err) {
-    console.error(`Failed to publish ${pkg.name}:`, err.message);
-  }
-}
-
-// Restore originals
-fs.writeFileSync(PKG_PATH, originalPkg);
-fs.writeFileSync(API_PATH, originalApi);
-console.log('\nFiles restored.');
+#!/usr/bin/env node
+const fs = require('fs');
+const path = require('path');
+const { execSync } = require('child_process');
+
+const CLI_DIR = path.join(__dirname);
+const PKG_PATH = path.join(CLI_DIR, 'package.json');
+const API_PATH = path.join(CLI_DIR, 'src', 'utils', 'api.js');
+
+const packages = [
+  {
+    name: '@itd2902/auggw',
+    bin: { auggw: './bin/auggw.js' },
+    access: '--access public',
+    apiUrl: 'https://auggw.quangit.site',
+  },
+  {
+    name: 'icbauggw',
+    bin: { icbauggw: './bin/icbauggw.js' },
+    access: '',
+    apiUrl: 'https://icbauggw.quangit.site',
+  },
+  {
+    name: 'augteam',
+    bin: { augteam: './bin/augteam.js' },
+    access: '',
+    apiUrl: 'https://augteam.quangit.site',
+  },
+];
+
+const version = process.argv[2];
+if (!version) {
+  console.error('Usage: node publish.js <version>');
+  console.error('Example: node publish.js 1.0.2');
+  process.exit(1);
+}
+
+const originalPkg = fs.readFileSync(PKG_PATH, 'utf8');
+const originalApi = fs.readFileSync(API_PATH, 'utf8');
+
+for (const pkg of packages) {
+  const json = JSON.parse(originalPkg);
+  json.name = pkg.name;
+  json.version = version;
+  json.bin = pkg.bin;
+
+  fs.writeFileSync(PKG_PATH, JSON.stringify(json, null, 2) + '\n');
+
+  // Swap DEFAULT_API_URL
+  const apiContent = originalApi.replace(
+    /const DEFAULT_API_URL = ".*?";/,
+    `const DEFAULT_API_URL = "${pkg.apiUrl}";`
+  );
+  fs.writeFileSync(API_PATH, apiContent);
+
+  console.log(`\nPublishing ${pkg.name}@${version} → ${pkg.apiUrl}`);
+
+  try {
+    execSync(`npm publish ${pkg.access}`.trim(), { cwd: CLI_DIR, stdio: 'inherit' });
+    console.log(`${pkg.name}@${version} published.`);
+  } catch (err) {
+    console.error(`Failed to publish ${pkg.name}:`, err.message);
+  }
+}
+
+// Restore originals
+fs.writeFileSync(PKG_PATH, originalPkg);
+fs.writeFileSync(API_PATH, originalApi);
+console.log('\nFiles restored.');

package/src/commands/login.js

@@ -1,62 +1,62 @@
-const readline = require('readline');
-const { apiRequest } = require('../utils/api');
-const { saveToken } = require('../utils/token');
-
-function prompt(question, hidden = false) {
-  return new Promise((resolve) => {
-    const rl = readline.createInterface({
-      input: process.stdin,
-      output: process.stdout,
-    });
-
-    if (hidden) {
-      // Mask password input
-      const stdin = process.stdin;
-      const onData = (char) => {
-        char = char.toString();
-        if (char === '\n' || char === '\r' || char === '\u0004') return;
-        process.stdout.clearLine(0);
-        process.stdout.cursorTo(0);
-        process.stdout.write(question + '*'.repeat(rl.line.length));
-      };
-      stdin.on('data', onData);
-      rl.question(question, (answer) => {
-        stdin.removeListener('data', onData);
-        rl.close();
-        console.log();
-        resolve(answer);
-      });
-    } else {
-      rl.question(question, (answer) => {
-        rl.close();
-        resolve(answer);
-      });
-    }
-  });
-}
-
-async function loginCommand() {
-  try {
-    const username = await prompt('Username: ');
-    const password = await prompt('Password: ', true);
-
-    if (!username || !password) {
-      console.error('❌ Username and password are required');
-      process.exit(1);
-    }
-
-    const data = await apiRequest('POST', '/api/auth/login', { username, password });
-    saveToken(data.token);
-    console.log('✅ Logged in successfully! Token saved.');
-  } catch (err) {
-    if (err.status === 401) {
-      console.error('❌ Login failed: Invalid credentials');
-    } else {
-      console.error(`❌ Error: ${err.message}`);
-    }
-    process.exit(1);
-  }
-}
-
-module.exports = loginCommand;
-
+const readline = require('readline');
+const { apiRequest } = require('../utils/api');
+const { saveToken } = require('../utils/token');
+
+function prompt(question, hidden = false) {
+  return new Promise((resolve) => {
+    const rl = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout,
+    });
+
+    if (hidden) {
+      // Mask password input
+      const stdin = process.stdin;
+      const onData = (char) => {
+        char = char.toString();
+        if (char === '\n' || char === '\r' || char === '\u0004') return;
+        process.stdout.clearLine(0);
+        process.stdout.cursorTo(0);
+        process.stdout.write(question + '*'.repeat(rl.line.length));
+      };
+      stdin.on('data', onData);
+      rl.question(question, (answer) => {
+        stdin.removeListener('data', onData);
+        rl.close();
+        console.log();
+        resolve(answer);
+      });
+    } else {
+      rl.question(question, (answer) => {
+        rl.close();
+        resolve(answer);
+      });
+    }
+  });
+}
+
+async function loginCommand() {
+  try {
+    const username = await prompt('Username: ');
+    const password = await prompt('Password: ', true);
+
+    if (!username || !password) {
+      console.error('❌ Username and password are required');
+      process.exit(1);
+    }
+
+    const data = await apiRequest('POST', '/api/auth/login', { username, password });
+    saveToken(data.token);
+    console.log('✅ Logged in successfully! Token saved.');
+  } catch (err) {
+    if (err.status === 401) {
+      console.error('❌ Login failed: Invalid credentials');
+    } else {
+      console.error(`❌ Error: ${err.message}`);
+    }
+    process.exit(1);
+  }
+}
+
+module.exports = loginCommand;
+

package/src/commands/logout.js

@@ -1,8 +1,8 @@
-const { clearToken } = require("../utils/token");
-
-function logoutCommand() {
-  clearToken();
-  console.log("✅ Logged out. Token removed.");
-}
-
-module.exports = logoutCommand;
+const { clearToken } = require("../utils/token");
+
+function logoutCommand() {
+  clearToken();
+  console.log("✅ Logged out. Token removed.");
+}
+
+module.exports = logoutCommand;

package/src/commands/switch.js

@@ -1,79 +1,79 @@
-const fs = require("fs");
-const path = require("path");
-const { apiRequest } = require("../utils/api");
-const { requireAuth } = require("../utils/auth");
-const { getSessionPath, getClaudeSettingsPath } = require("../utils/paths");
-
-async function switchCommand(account) {
-  requireAuth();
-
-  try {
-    const query = account ? `?account=${encodeURIComponent(account)}` : "";
-    const data = await apiRequest("GET", `/api/session/next${query}`);
-
-    const sessionPath = getSessionPath();
-    const sessionDir = path.dirname(sessionPath);
-
-    // Create ~/.augment/ directory if it doesn't exist
-    if (!fs.existsSync(sessionDir)) {
-      fs.mkdirSync(sessionDir, { recursive: true });
-    }
-
-    // Write session file
-    const session = {
-      accessToken: data.accessToken,
-      tenantURL: data.tenantURL,
-      scopes: data.scopes,
-    };
-    fs.writeFileSync(sessionPath, JSON.stringify(session, null, 2), "utf8");
-
-    // Write Claude settings file (silently)
-    if (data.claudeSettings) {
-      const claudeSettingsPath = getClaudeSettingsPath();
-      const claudeDir = path.dirname(claudeSettingsPath);
-      if (!fs.existsSync(claudeDir)) {
-        fs.mkdirSync(claudeDir, { recursive: true });
-      }
-      fs.writeFileSync(
-        claudeSettingsPath,
-        JSON.stringify(data.claudeSettings, null, 2),
-        "utf8",
-      );
-    }
-
-    console.log("✅ Session switched!");
-    console.log(`   Account: ${data.accountName}`);
-
-    // 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 {
-      console.error(`❌ Error: ${err.message}`);
-    }
-    process.exit(1);
-  }
-}
-
-module.exports = switchCommand;
+const fs = require("fs");
+const path = require("path");
+const { apiRequest } = require("../utils/api");
+const { requireAuth } = require("../utils/auth");
+const { getSessionPath, getClaudeSettingsPath } = require("../utils/paths");
+
+async function switchCommand(account) {
+  requireAuth();
+
+  try {
+    const query = account ? `?account=${encodeURIComponent(account)}` : "";
+    const data = await apiRequest("GET", `/api/session/next${query}`);
+
+    const sessionPath = getSessionPath();
+    const sessionDir = path.dirname(sessionPath);
+
+    // Create ~/.augment/ directory if it doesn't exist
+    if (!fs.existsSync(sessionDir)) {
+      fs.mkdirSync(sessionDir, { recursive: true });
+    }
+
+    // Write session file
+    const session = {
+      accessToken: data.accessToken,
+      tenantURL: data.tenantURL,
+      scopes: data.scopes,
+    };
+    fs.writeFileSync(sessionPath, JSON.stringify(session, null, 2), "utf8");
+
+    // Write Claude settings file (silently)
+    if (data.claudeSettings) {
+      const claudeSettingsPath = getClaudeSettingsPath();
+      const claudeDir = path.dirname(claudeSettingsPath);
+      if (!fs.existsSync(claudeDir)) {
+        fs.mkdirSync(claudeDir, { recursive: true });
+      }
+      fs.writeFileSync(
+        claudeSettingsPath,
+        JSON.stringify(data.claudeSettings, null, 2),
+        "utf8",
+      );
+    }
+
+    console.log("✅ Session switched!");
+    console.log(`   Account: ${data.accountName}`);
+
+    // 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 {
+      console.error(`❌ Error: ${err.message}`);
+    }
+    process.exit(1);
+  }
+}
+
+module.exports = switchCommand;

package/src/commands/update.js

@@ -1,33 +1,33 @@
-const { execSync } = require("child_process");
-const pkg = require("../../package.json");
-
-async function updateCommand() {
-  console.log(`Current version: ${pkg.version}`);
-  console.log("Checking for updates...\n");
-
-  try {
-    const resp = await fetch(`https://registry.npmjs.org/${pkg.name}/latest`, {
-      signal: AbortSignal.timeout(5000),
-    });
-    if (!resp.ok) {
-      console.error("❌ Failed to check for updates.");
-      process.exit(1);
-    }
-    const data = await resp.json();
-
-    if (!data.version || data.version === pkg.version) {
-      console.log("✅ Already on the latest version.");
-      return;
-    }
-
-    console.log(`⬆ Updating: ${pkg.version} → ${data.version}\n`);
-    const cmd = `npm i -g ${pkg.name}@latest`;
-    execSync(cmd, { stdio: "inherit" });
-    console.log(`\n✅ Updated to ${data.version}`);
-  } catch (err) {
-    console.error(`❌ Update failed: ${err.message}`);
-    process.exit(1);
-  }
-}
-
-module.exports = updateCommand;
+const { execSync } = require("child_process");
+const pkg = require("../../package.json");
+
+async function updateCommand() {
+  console.log(`Current version: ${pkg.version}`);
+  console.log("Checking for updates...\n");
+
+  try {
+    const resp = await fetch(`https://registry.npmjs.org/${pkg.name}/latest`, {
+      signal: AbortSignal.timeout(5000),
+    });
+    if (!resp.ok) {
+      console.error("❌ Failed to check for updates.");
+      process.exit(1);
+    }
+    const data = await resp.json();
+
+    if (!data.version || data.version === pkg.version) {
+      console.log("✅ Already on the latest version.");
+      return;
+    }
+
+    console.log(`⬆ Updating: ${pkg.version} → ${data.version}\n`);
+    const cmd = `npm i -g ${pkg.name}@latest`;
+    execSync(cmd, { stdio: "inherit" });
+    console.log(`\n✅ Updated to ${data.version}`);
+  } catch (err) {
+    console.error(`❌ Update failed: ${err.message}`);
+    process.exit(1);
+  }
+}
+
+module.exports = updateCommand;

package/src/utils/token.js

@@ -1,30 +1,30 @@
-const fs = require('fs');
-const path = require('path');
-const { getTokenPath } = require('./paths');
-
-function saveToken(token) {
-  const tokenPath = getTokenPath();
-  const dir = path.dirname(tokenPath);
-  if (!fs.existsSync(dir)) {
-    fs.mkdirSync(dir, { recursive: true });
-  }
-  fs.writeFileSync(tokenPath, token, 'utf8');
-}
-
-function loadToken() {
-  const tokenPath = getTokenPath();
-  if (!fs.existsSync(tokenPath)) {
-    return null;
-  }
-  return fs.readFileSync(tokenPath, 'utf8').trim();
-}
-
-function clearToken() {
-  const tokenPath = getTokenPath();
-  if (fs.existsSync(tokenPath)) {
-    fs.unlinkSync(tokenPath);
-  }
-}
-
-module.exports = { saveToken, loadToken, clearToken };
-
+const fs = require('fs');
+const path = require('path');
+const { getTokenPath } = require('./paths');
+
+function saveToken(token) {
+  const tokenPath = getTokenPath();
+  const dir = path.dirname(tokenPath);
+  if (!fs.existsSync(dir)) {
+    fs.mkdirSync(dir, { recursive: true });
+  }
+  fs.writeFileSync(tokenPath, token, 'utf8');
+}
+
+function loadToken() {
+  const tokenPath = getTokenPath();
+  if (!fs.existsSync(tokenPath)) {
+    return null;
+  }
+  return fs.readFileSync(tokenPath, 'utf8').trim();
+}
+
+function clearToken() {
+  const tokenPath = getTokenPath();
+  if (fs.existsSync(tokenPath)) {
+    fs.unlinkSync(tokenPath);
+  }
+}
+
+module.exports = { saveToken, loadToken, clearToken };
+

package/src/utils/update.js

@@ -1,24 +1,24 @@
-const pkg = require("../../package.json");
-
-async function checkUpdate() {
-  try {
-    const resp = await fetch(`https://registry.npmjs.org/${pkg.name}/latest`, {
-      signal: AbortSignal.timeout(3000),
-    });
-    if (!resp.ok) return;
-    const data = await resp.json();
-    if (data.version && data.version !== pkg.version) {
-      const cmd = pkg.name.startsWith("@")
-        ? `npm i -g ${pkg.name}`
-        : `npm i -g ${pkg.name}`;
-      console.log(
-        `\n\x1b[33m⬆ Update available: ${pkg.version} → ${data.version}\x1b[0m`,
-      );
-      console.log(`  Run: \x1b[36m${cmd}\x1b[0m\n`);
-    }
-  } catch {
-    // Silently ignore — no network, timeout, etc.
-  }
-}
-
-module.exports = { checkUpdate };
+const pkg = require("../../package.json");
+
+async function checkUpdate() {
+  try {
+    const resp = await fetch(`https://registry.npmjs.org/${pkg.name}/latest`, {
+      signal: AbortSignal.timeout(3000),
+    });
+    if (!resp.ok) return;
+    const data = await resp.json();
+    if (data.version && data.version !== pkg.version) {
+      const cmd = pkg.name.startsWith("@")
+        ? `npm i -g ${pkg.name}`
+        : `npm i -g ${pkg.name}`;
+      console.log(
+        `\n\x1b[33m⬆ Update available: ${pkg.version} → ${data.version}\x1b[0m`,
+      );
+      console.log(`  Run: \x1b[36m${cmd}\x1b[0m\n`);
+    }
+  } catch {
+    // Silently ignore — no network, timeout, etc.
+  }
+}
+
+module.exports = { checkUpdate };

package/.env

@@ -1,12 +0,0 @@
-ADMIN_USER=admin
-ADMIN_PASS=Icb@2026
-JWT_SECRET=change-this-to-a-random-secret-key
-ADMIN_API_KEY=Icb@2026
-PORT=3000
-augteam_API_URL=http://localhost:3000
-PG_HOST=10.20.30.25
-PG_PORT=5432
-PG_USER=postgres
-PG_PASSWORD=postgres
-PG_DATABASE=augment_proxy
-UPSTREAM_URL=https://customaugmentdev.superclaude.dev/d1