icbauggw 1.0.0

package/bin/icbauggw.js

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

package/package.json

@@ -0,0 +1,24 @@
+{
+  "name": "icbauggw",
+  "version": "1.0.0",
+  "description": "CLI tool for rotating Augment session accounts",
+  "main": "src/index.js",
+  "bin": {
+    "icbauggw": "./bin/icbauggw.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: icbauggw 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}%)`);
+      }
+      if (data.current.clients && data.current.clients.length > 0) {
+        console.log(`   Clients: ${data.current.clients.length} user(s)`);
+      }
+      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: icbauggw login');
+    } else {
+      console.error(`❌ Error: ${err.message}`);
+    }
+    process.exit(1);
+  }
+}
+
+module.exports = statusCommand;
+

package/src/commands/switch.js

@@ -0,0 +1,45 @@
+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: icbauggw 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('icbauggw')
+  .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,73 @@
+const os = require('os');
+const { loadToken } = require('./token');
+
+const DEFAULT_API_URL = 'http://localhost:3000';
+
+function getBaseURL() {
+  return process.env.ICBAUGGW_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,12 @@
+const { loadToken } = require('./token');
+
+function requireAuth() {
+  const token = loadToken();
+  if (!token) {
+    console.error('❌ Please login first: icbauggw login');
+    process.exit(1);
+  }
+}
+
+module.exports = { requireAuth };
+

package/src/utils/paths.js

@@ -0,0 +1,13 @@
+const os = require('os');
+const path = require('path');
+
+function getTokenPath() {
+  return path.join(os.homedir(), '.icb-aug', '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 };
+