@first087/agys 0.1.0

package/README.md

@@ -0,0 +1,44 @@
+# agys
+
+Manage multiple antigravity-cli oauth token accounts with ease.
+
+## Features
+
+- **Add Accounts**: Easily add and name new OAuth token accounts.
+- **List Accounts**: View all managed accounts and see which one is currently active.
+- **Interactive Switch**: Switch between accounts using an interactive CLI menu.
+- **Persistent State**: Keeps track of the active account automatically.
+
+## Installation
+
+```bash
+npm install -g agys
+# or
+bun install -g agys
+```
+
+## Usage
+
+### Add a new account
+```bash
+agys add <name>
+```
+
+### List all accounts
+```bash
+agys list
+# or
+agys ls
+```
+
+### Switch account
+Run `agys` without any arguments to open the interactive selection menu:
+```bash
+agys
+# or explicitly
+agys switch
+```
+
+## License
+
+MIT

package/index.ts

@@ -0,0 +1,30 @@
+#!/usr/bin/env bun
+import { Command } from 'commander';
+import packageJson from './package.json' with { type: 'json' };
+import { addCommand } from './src/commands/add';
+import { listCommand, lsCommand } from './src/commands/list';
+import { switchCommand, handleSwitch } from './src/commands/switch';
+
+const program = new Command();
+
+program
+  .name('agys')
+  .description('Manage multiple antigravity-cli oauth token accounts')
+  .version(packageJson.version);
+
+program.addCommand(addCommand);
+program.addCommand(listCommand);
+program.addCommand(lsCommand);
+program.addCommand(switchCommand);
+
+// Set description for switch command explicitly
+switchCommand.description('Switch to a different account (run agys without arguments)');
+
+// Set default action if no command is provided
+program.action(async () => {
+  if (program.args.length === 0) {
+    await handleSwitch();
+  }
+});
+
+program.parse(process.argv);

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "@first087/agys",
+  "version": "0.1.0",
+  "description": "Manage multiple antigravity-cli oauth token accounts",
+  "author": "Artit Kiuwilai",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/first087/agy-switch.git"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "module": "index.ts",
+  "type": "module",
+  "bin": {
+    "agys": "index.ts"
+  },
+  "files": [
+    "src",
+    "index.ts"
+  ],
+  "scripts": {
+    "start": "bun run index.ts",
+    "test": "bun test",
+    "build": "rm -rf dist && bun build index.ts --outdir ./dist --target bun --minify",
+    "lint": "bunx tsc --noEmit",
+    "publish:npm": "bun run build && npm publish --access public"
+  },
+  "license": "MIT",
+  "keywords": [
+    "cli",
+    "antigravity",
+    "oauth",
+    "token-manager"
+  ],
+  "devDependencies": {
+    "@types/bun": "latest",
+    "@types/fs-extra": "^11.0.4"
+  },
+  "peerDependencies": {
+    "typescript": "^5"
+  },
+  "dependencies": {
+    "chalk": "^5.6.2",
+    "commander": "^15.0.0",
+    "fs-extra": "^11.3.5",
+    "inquirer": "^14.0.2"
+  }
+}

package/src/commands/add.test.ts

@@ -0,0 +1,15 @@
+import { expect, test, spyOn } from "bun:test";
+import * as fileOps from "../utils/fileOperations";
+
+// Mocking the utility function
+spyOn(fileOps, "copyTokenToAccount").mockImplementation(() => Promise.resolve('/home/user/.agys/test-account'));
+
+test("add command calls copyTokenToAccount with correct arguments", async () => {
+  // Logic to test will be added as we implement the command
+  const accountName = "test-account";
+  const sourcePath = "/home/user/.gemini/antigravity-cli/antigravity-oauth-token";
+  
+  await fileOps.copyTokenToAccount(accountName, sourcePath);
+  
+  expect(fileOps.copyTokenToAccount).toHaveBeenCalledWith(accountName, sourcePath);
+});

package/src/commands/add.ts

@@ -0,0 +1,22 @@
+import { Command } from "commander";
+import * as fileOps from "../utils/fileOperations";
+import chalk from "chalk";
+import os from "os";
+import path from "path";
+
+export const addCommand = new Command("add")
+  .description("Add a new account token")
+  .argument("[name]", "account name") // Change <name> to [name] to make it optional
+  .action(async (name) => {
+    if (!name) {
+      addCommand.outputHelp();
+      return;
+    }
+    const sourcePath = path.join(os.homedir(), ".gemini", "antigravity-cli", "antigravity-oauth-token");
+    try {
+      await fileOps.copyTokenToAccount(name, sourcePath);
+      console.log(chalk.green(`Account '${name}' added successfully.`));
+    } catch (error) {
+      console.error(chalk.red(`Failed to add account '${name}':`), error);
+    }
+  });

package/src/commands/list.test.ts

@@ -0,0 +1,15 @@
+import { expect, test, spyOn } from "bun:test";
+import * as fileOps from "../utils/fileOperations";
+
+// Mocking the utility functions
+spyOn(fileOps, "getAccounts").mockImplementation(() => Promise.resolve(['test', 'test2', 'test3']));
+spyOn(fileOps, "getActiveAccount").mockImplementation(() => Promise.resolve('test'));
+
+test("list command returns all accounts with active one marked", async () => {
+  const accounts = await fileOps.getAccounts();
+  const active = await fileOps.getActiveAccount();
+  
+  expect(accounts).toContain('test');
+  expect(accounts).toContain('test2');
+  expect(active).toBe('test');
+});

package/src/commands/list.ts

@@ -0,0 +1,35 @@
+import { Command } from "commander";
+import * as fileOps from "../utils/fileOperations";
+import chalk from "chalk";
+
+async function listAction() {
+  try {
+    const accounts = await fileOps.getAccounts();
+    const active = await fileOps.getActiveAccount();
+
+    if (accounts.length === 0) {
+      console.log(chalk.yellow("No accounts found. Use 'add' to create one."));
+      return;
+    }
+
+    console.log(chalk.blue("Accounts:"));
+    accounts.forEach((account: string) => {
+      if (account === active) {
+        console.log(chalk.green(`* ${account} (active)`));
+      } else {
+        console.log(`  ${account}`);
+      }
+    });
+  } catch (error) {
+    console.error(chalk.red("Failed to list accounts:"), error);
+  }
+}
+
+export const listCommand = new Command("list")
+  .description("List all accounts")
+  .action(listAction);
+
+export const lsCommand = new Command("ls")
+  .description("Alias for list")
+  .action(listAction);
+

package/src/commands/switch.test.ts

@@ -0,0 +1,20 @@
+import { expect, test, spyOn } from "bun:test";
+import * as fileOps from "../utils/fileOperations";
+import inquirer from 'inquirer';
+
+// Mocking the utility functions and inquirer
+spyOn(fileOps, "getAccounts").mockImplementation(() => Promise.resolve(['test', 'test2', 'test3']));
+// @ts-ignore
+spyOn(inquirer, 'prompt').mockImplementation(() => Promise.resolve({ selectedAccount: 'test2' }));
+
+test("switch interactive menu returns selected account", async () => {
+  const accounts = await fileOps.getAccounts();
+  const answers = await inquirer.prompt([{
+    type: 'list',
+    name: 'selectedAccount',
+    message: 'Select account:',
+    choices: accounts
+  }]);
+  
+  expect(answers.selectedAccount).toBe('test2');
+});

package/src/commands/switch.ts

@@ -0,0 +1,51 @@
+import { Command } from "commander";
+import * as fileOps from "../utils/fileOperations";
+import inquirer from 'inquirer';
+import chalk from 'chalk';
+import path from 'path';
+import os from 'os';
+import fs from 'fs-extra';
+
+export async function handleSwitch() {
+  try {
+    const accounts = await fileOps.getAccounts();
+    const active = await fileOps.getActiveAccount();
+    
+    if (accounts.length === 0) {
+      console.log(chalk.yellow("No accounts found. Use 'agys add' to create one."));
+      return;
+    }
+
+    const { selectedAccount } = await inquirer.prompt([{
+        type: 'select',
+        name: 'selectedAccount',
+        message: 'Select account (Press Ctrl+C to cancel):',
+        choices: accounts.map((account: string) => ({
+          name: account === active ? `${account} (active)` : account,
+          value: account
+        })),
+        default: active
+      }]).catch(() => {
+        console.log(chalk.yellow("\nSwitching cancelled."));
+        process.exit(0);
+      });
+
+    if (selectedAccount !== active) {
+      const sourcePath = path.join(os.homedir(), '.agys', selectedAccount);
+      const destPath = path.join(os.homedir(), '.gemini', 'antigravity-cli', 'antigravity-oauth-token');
+      
+      // Use fs.copy to copy from source to dest (token file)
+      await fs.copy(sourcePath, destPath);
+      await fileOps.setActiveAccount(selectedAccount);
+      console.log(chalk.green(`Switched to account '${selectedAccount}'`));
+    } else {
+      console.log(chalk.blue(`Already using account '${selectedAccount}'`));
+    }
+  } catch (error) {
+    console.error(chalk.red("Failed to switch account:"), error);
+  }
+}
+
+export const switchCommand = new Command("switch")
+  .description("Switch to a different account")
+  .action(handleSwitch);

package/src/utils/fileOperations.ts

@@ -0,0 +1,38 @@
+import fs from 'fs-extra';
+import path from 'path';
+import os from 'os';
+
+const CONFIG_DIR = path.join(os.homedir(), '.agys');
+const ACTIVE_FILE = path.join(CONFIG_DIR, '.active');
+
+export async function ensureConfigDir() {
+  await fs.ensureDir(CONFIG_DIR);
+}
+
+export async function copyTokenToAccount(accountName: string, sourcePath: string) {
+  const destPath = path.join(CONFIG_DIR, accountName);
+  await fs.ensureDir(CONFIG_DIR);
+  await fs.copy(sourcePath, destPath);
+  return destPath;
+}
+
+export async function getAccounts() {
+  await ensureConfigDir();
+  const files = await fs.readdir(CONFIG_DIR);
+  return files.filter((file: string) => file !== '.active');
+}
+
+export async function setActiveAccount(accountName: string) {
+  await ensureConfigDir();
+  await fs.writeFile(ACTIVE_FILE, accountName, 'utf-8');
+}
+
+export async function getActiveAccount() {
+  await ensureConfigDir();
+  
+  if (!fs.existsSync(ACTIVE_FILE)) {
+    return null;
+  }
+
+  return await fs.readFile(ACTIVE_FILE, 'utf-8');
+}