@itd2902/auggw 1.0.1

package/SETUP.md

@@ -0,0 +1,46 @@
+# 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 auggw
+```
+
+## Đă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/package.json

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

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/status.js

@@ -0,0 +1,46 @@
+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(`   Tenant:  ${data.current.tenantURL}`);
+      console.log(`   Token:   ...${data.current.accessToken}`);
+      console.log(
+        `   Status:  ${data.current.isLimited ? "🔴 Limited" : "🟢 Active"}`,
+      );
+      if (data.current.creditTotal) {
+        const used = data.current.creditRemaining || 0;
+        const total = data.current.creditTotal;
+        const pct = Math.round((used / total) * 100);
+        console.log(
+          `   Credit:  ${used.toLocaleString()} / ${total.toLocaleString()} (${pct}%)`,
+        );
+      }
+      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/index.js

@@ -0,0 +1,29 @@
+const { Command } = require('commander');
+const loginCommand = require('./commands/login');
+const switchCommand = require('./commands/switch');
+const statusCommand = require('./commands/status');
+
+const program = new Command();
+
+program
+  .name('auggw')
+  .description('CLI tool for rotating Augment session accounts')
+  .version('1.0.0');
+
+program
+  .command('login')
+  .description('Authenticate with the session rotator backend')
+  .action(loginCommand);
+
+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.parse();
+

package/src/utils/api.js

@@ -0,0 +1,74 @@
+const os = require("os");
+const { loadToken } = require("./token");
+
+const DEFAULT_API_URL = "https://auggw.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 };
+