augteam 1.1.2

package/SETUP.md

@@ -0,0 +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
+  ```

package/bin/auggw.js

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

package/bin/augteam.js

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

package/bin/icbauggw.js

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+require("dotenv").config({ path: require("path").resolve(__dirname, "../../.env") });
+require("../src/index");

package/package.json

@@ -0,0 +1,25 @@
+{
+  "name": "augteam",
+  "version": "1.1.2",
+  "description": "CLI tool for rotating Augment session accounts",
+  "main": "src/index.js",
+  "bin": {
+    "augteam": "./bin/augteam.js"
+  },
+  "scripts": {},
+  "keywords": [
+    "augment",
+    "session",
+    "rotate",
+    "cli"
+  ],
+  "author": "",
+  "license": "ISC",
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "commander": "^13.1.0",
+    "dotenv": "^17.3.1"
+  }
+}

package/publish.js

@@ -0,0 +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.');

package/src/commands/fetch.js

@@ -0,0 +1,145 @@
+const fs = require("fs");
+const path = require("path");
+const os = require("os");
+const readline = require("readline");
+const { apiRequest } = require("../utils/api");
+const { requireAuth } = require("../utils/auth");
+
+function ask(question) {
+  return new Promise((resolve) => {
+    const rl = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout,
+    });
+    rl.question(question, (answer) => {
+      rl.close();
+      resolve(answer.trim());
+    });
+  });
+}
+
+function getAugmentDir() {
+  return path.join(os.homedir(), ".augment");
+}
+
+function getClaudeDir() {
+  return path.join(os.homedir(), ".claude");
+}
+
+function writeFile(filePath, content) {
+  const dir = path.dirname(filePath);
+  if (!fs.existsSync(dir)) {
+    fs.mkdirSync(dir, { recursive: true });
+  }
+  fs.writeFileSync(filePath, content, "utf8");
+}
+
+function installToAugment(data) {
+  const baseDir = getAugmentDir();
+  let count = 0;
+
+  for (const cmd of data.commands) {
+    writeFile(path.join(baseDir, "commands", cmd.name), cmd.content);
+    count++;
+  }
+
+  for (const skill of data.skills) {
+    writeFile(path.join(baseDir, "skills", skill.name, "SKILL.md"), skill.content);
+    count++;
+  }
+
+  for (const agent of data.agents || []) {
+    writeFile(path.join(baseDir, "agents", agent.name), agent.content);
+    count++;
+  }
+
+  return count;
+}
+
+function installToClaude(data) {
+  const baseDir = getClaudeDir();
+  let count = 0;
+
+  for (const cmd of data.commands) {
+    writeFile(path.join(baseDir, "commands", cmd.name), cmd.content);
+    count++;
+  }
+
+  for (const skill of data.skills) {
+    writeFile(path.join(baseDir, "commands", skill.name, "SKILL.md"), skill.content);
+    count++;
+  }
+
+  for (const agent of data.agents || []) {
+    writeFile(path.join(baseDir, "agents", agent.name), agent.content);
+    count++;
+  }
+
+  return count;
+}
+
+async function fetchPrompts() {
+  console.log("📦 Fetch prompts (commands & skills)\n");
+  console.log("  1) Augment only  (~/.augment/)");
+  console.log("  2) Claude only   (~/.claude/)");
+  console.log("  3) Both\n");
+
+  const choice = await ask("Chọn (1/2/3): ");
+
+  if (!["1", "2", "3"].includes(choice)) {
+    console.error("❌ Lựa chọn không hợp lệ. Vui lòng chọn 1, 2 hoặc 3.");
+    process.exit(1);
+  }
+
+  console.log("\n⏳ Fetching prompts from server...");
+  const data = await apiRequest("GET", "/api/prompts");
+
+  const totalItems = data.commands.length + data.skills.length + (data.agents || []).length;
+  if (totalItems === 0) {
+    console.log("ℹ️  Không có prompts nào trên server.");
+    return;
+  }
+
+  console.log(
+    `   Found: ${data.commands.length} command(s), ${data.skills.length} skill(s), ${(data.agents || []).length} agent(s)\n`,
+  );
+
+  if (choice === "1" || choice === "3") {
+    const count = installToAugment(data);
+    console.log(`✅ Augment: ${count} file(s) → ${getAugmentDir()}`);
+  }
+
+  if (choice === "2" || choice === "3") {
+    const count = installToClaude(data);
+    console.log(`✅ Claude:  ${count} file(s) → ${getClaudeDir()}`);
+  }
+
+  console.log("\n🎉 Done!");
+}
+
+async function fetchCommand(options) {
+  requireAuth();
+
+  if (!options.prompts) {
+    console.log("Usage: auggw fetch --prompts");
+    console.log("\nOptions:");
+    console.log("  --prompts  Fetch commands & skills to ~/.augment and/or ~/.claude");
+    return;
+  }
+
+  try {
+    if (options.prompts) {
+      await fetchPrompts();
+    }
+  } catch (err) {
+    if (err.status === 401) {
+      console.error("❌ Session expired. Please login again: auggw login");
+    } else {
+      console.error(`❌ Error: ${err.message}`);
+    }
+    process.exit(1);
+  }
+}
+
+module.exports = fetchCommand;
+

package/src/commands/login.js

@@ -0,0 +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;
+

package/src/commands/logout.js

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

package/src/commands/status.js

@@ -0,0 +1,57 @@
+const { apiRequest } = require("../utils/api");
+const { requireAuth } = require("../utils/auth");
+
+async function statusCommand() {
+  requireAuth();
+
+  try {
+    const data = await apiRequest("GET", "/api/session/status");
+
+    if (!data.current) {
+      console.log("ℹ️  No active session. Run: auggw switch");
+      console.log();
+    } else {
+      console.log("📋 Current Account");
+      console.log(`   Name:    ${data.current.name}`);
+      console.log(`   Token:   ...${data.current.accessToken}`);
+      console.log(
+        `   Status:  ${data.current.isLimited ? "🔴 Limited" : "🟢 Active"}`,
+      );
+      if (data.current.creditTotal) {
+        const remaining = data.current.creditRemaining || 0;
+        const total = data.current.creditTotal;
+        const used = 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:  ${remaining.toLocaleString()} / ${total.toLocaleString()} remaining`,
+        );
+        console.log(
+          `   ${bar} ${color}${pct}% used${reset}`,
+        );
+      } else {
+        console.log(`   Credit:  \x1b[90mN/A — account chưa login trên admin panel\x1b[0m`);
+      }
+      console.log();
+    }
+
+    console.log("📊 Pool Status");
+    console.log(`   Active:  ${data.pool.active} accounts`);
+    console.log(`   Limited: ${data.pool.limited} accounts`);
+    console.log(`   Total:   ${data.pool.total} accounts`);
+  } catch (err) {
+    if (err.status === 401) {
+      console.error("❌ Session expired. Please login again: auggw login");
+    } else {
+      console.error(`❌ Error: ${err.message}`);
+    }
+    process.exit(1);
+  }
+}
+
+module.exports = statusCommand;

package/src/commands/switch.js

@@ -0,0 +1,44 @@
+const fs = require("fs");
+const path = require("path");
+const { apiRequest } = require("../utils/api");
+const { requireAuth } = require("../utils/auth");
+const { getSessionPath } = require("../utils/paths");
+
+async function switchCommand() {
+  requireAuth();
+
+  try {
+    const data = await apiRequest("GET", "/api/session/next");
+
+    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");
+
+    console.log("✅ Session switched!");
+    console.log(`   Account: ${data.accountName}`);
+    console.log(`   Tenant:  ${data.tenantURL}`);
+  } catch (err) {
+    if (err.status === 401) {
+      console.error("❌ Session expired. Please login again: auggw login");
+    } 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

@@ -0,0 +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;

package/src/index.js

@@ -0,0 +1,50 @@
+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

@@ -0,0 +1,74 @@
+const os = require("os");
+const { loadToken } = require("./token");
+
+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;
+}
+
+function getLocalIP() {
+  const interfaces = os.networkInterfaces();
+  // Skip virtual/VPN adapters — prefer real Ethernet/Wi-Fi
+  const skipPatterns =
+    /tailscale|vpn|virtual|vethernet|bluetooth|loopback|docker|vbox|vmware|wsl/i;
+
+  // First pass: find IP from real adapters (Ethernet, Wi-Fi)
+  for (const [name, addrs] of Object.entries(interfaces)) {
+    if (skipPatterns.test(name)) continue;
+    for (const iface of addrs) {
+      if (iface.family === "IPv4" && !iface.internal) {
+        return iface.address;
+      }
+    }
+  }
+  // Fallback: any non-internal IPv4
+  for (const addrs of Object.values(interfaces)) {
+    for (const iface of addrs) {
+      if (iface.family === "IPv4" && !iface.internal) {
+        return iface.address;
+      }
+    }
+  }
+  return "127.0.0.1";
+}
+
+async function apiRequest(method, path, body = null) {
+  const url = `${getBaseURL()}${path}`;
+  const token = loadToken();
+
+  const headers = {
+    "Content-Type": "application/json",
+    "X-Client-IP": getLocalIP(),
+    "X-Client-Hostname": os.hostname(),
+  };
+  if (token) {
+    headers["Authorization"] = `Bearer ${token}`;
+  }
+
+  const options = { method, headers };
+  if (body) {
+    options.body = JSON.stringify(body);
+  }
+
+  try {
+    const res = await fetch(url, options);
+    const data = await res.json();
+
+    if (!res.ok) {
+      const err = new Error(data.error || `HTTP ${res.status}`);
+      err.status = res.status;
+      throw err;
+    }
+
+    return data;
+  } catch (err) {
+    if (err.cause && err.cause.code === "ECONNREFUSED") {
+      throw new Error(`Could not connect to server at ${getBaseURL()}`);
+    }
+    throw err;
+  }
+}
+
+module.exports = { apiRequest, getBaseURL };

package/src/utils/auth.js

@@ -0,0 +1,11 @@
+const { loadToken } = require("./token");
+
+function requireAuth() {
+  const token = loadToken();
+  if (!token) {
+    console.error("❌ Please login first: auggw login");
+    process.exit(1);
+  }
+}
+
+module.exports = { requireAuth };

package/src/utils/paths.js

@@ -0,0 +1,12 @@
+const os = require("os");
+const path = require("path");
+
+function getTokenPath() {
+  return path.join(os.homedir(), ".auggw", "token");
+}
+
+function getSessionPath() {
+  return path.join(os.homedir(), ".augment", "session.json");
+}
+
+module.exports = { getTokenPath, getSessionPath };

package/src/utils/token.js

@@ -0,0 +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 };
+

package/src/utils/update.js

@@ -0,0 +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 };