opencode-gemini-business 1.0.0

package/bin/cli.js

@@ -0,0 +1,18 @@
+#!/usr/bin/env node
+
+/**
+ * CLI entry point for opencode-gemini-business
+ */
+
+import('../dist/index.js')
+  .then(module => {
+    if (module.cli) {
+      return module.cli();
+    } else {
+      throw new Error('CLI function not found in module exports');
+    }
+  })
+  .catch(error => {
+    console.error('Failed to load CLI:', error);
+    process.exit(1);
+  });

package/dist/index.d.ts

@@ -0,0 +1,69 @@
+#!/usr/bin/env node
+/**
+ * OpenCode Gemini Business Plugin
+ * Multi-account Gemini Business pool with automatic rotation
+ *
+ * Follows OpenCode plugin patterns from:
+ * - opencode-gemini-auth
+ * - opencode-openai-codex-auth
+ * - opencode-antigravity-auth
+ */
+import { AccountManager } from './src/account-manager.js';
+import { GeminiBusinessAPI } from './src/gemini-business-api.js';
+/**
+ * OpenCode Plugin Entry Point
+ *
+ * This plugin doesn't use traditional OAuth - it uses cookie-based auth
+ * with Gemini Business accounts configured via CLI.
+ */
+export declare const GeminiBusinessPlugin: (ctx: any) => Promise<{
+    auth: {
+        provider: string;
+        /**
+         * Loader function - called by OpenCode to initialize the provider
+         */
+        loader(getAuth: any, provider: any): Promise<{
+            accountCount?: undefined;
+            strategy?: undefined;
+            /**
+             * Custom fetch implementation for Gemini Business API
+             *
+             * This intercepts all OpenCode API calls and routes them through
+             * the Gemini Business API with account rotation.
+             */
+            fetch?: undefined;
+        } | {
+            accountCount: number;
+            strategy: "round-robin" | "least-used" | "random";
+            /**
+             * Custom fetch implementation for Gemini Business API
+             *
+             * This intercepts all OpenCode API calls and routes them through
+             * the Gemini Business API with account rotation.
+             */
+            fetch(input: any, options: any): Promise<any>;
+        }>;
+        /**
+         * Authentication methods
+         *
+         * For Gemini Business, we don't support OAuth - accounts are configured
+         * via CLI with cookie extraction from browser.
+         */
+        methods: {
+            type: "cookie";
+            label: string;
+            authorize: () => Promise<{
+                type: "instructions";
+                instructions: string;
+            }>;
+        }[];
+    };
+}>;
+export default GeminiBusinessPlugin;
+export { AccountManager, GeminiBusinessAPI };
+/**
+ * CLI entry point - handles account management commands
+ */
+declare function cli(): Promise<void>;
+export { cli };
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":";AAEA;;;;;;;;GAQG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAOjE;;;;;GAKG;AACH,eAAO,MAAM,oBAAoB,GAAU,KAAK,GAAG;;;QAK7C;;WAEG;wBACmB,GAAG,YAAY,GAAG;;;YA0BpC;;;;;eAKG;;;;;YALH;;;;;eAKG;yBACgB,GAAG,WAAW,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;;QAsDvD;;;;;WAKG;;;;;;;;;;EAyBR,CAAC;AAGF,eAAe,oBAAoB,CAAC;AAGpC,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,CAAC;AAM7C;;GAEG;AACH,iBAAe,GAAG,kBAkCjB;AAsJD,OAAO,EAAE,GAAG,EAAE,CAAC"}

package/dist/index.js

@@ -0,0 +1,306 @@
+#!/usr/bin/env node
+/**
+ * OpenCode Gemini Business Plugin
+ * Multi-account Gemini Business pool with automatic rotation
+ *
+ * Follows OpenCode plugin patterns from:
+ * - opencode-gemini-auth
+ * - opencode-openai-codex-auth
+ * - opencode-antigravity-auth
+ */
+import { AccountManager } from './src/account-manager.js';
+import { GeminiBusinessAPI } from './src/gemini-business-api.js';
+// ============================================================================
+// OpenCode Plugin Export
+// ============================================================================
+/**
+ * OpenCode Plugin Entry Point
+ *
+ * This plugin doesn't use traditional OAuth - it uses cookie-based auth
+ * with Gemini Business accounts configured via CLI.
+ */
+export const GeminiBusinessPlugin = async (ctx) => {
+    return {
+        auth: {
+            provider: 'gemini-business',
+            /**
+             * Loader function - called by OpenCode to initialize the provider
+             */
+            async loader(getAuth, provider) {
+                const accountManager = new AccountManager();
+                await accountManager.loadAccounts();
+                const accounts = accountManager.getAccounts();
+                if (accounts.length === 0) {
+                    console.error('\n❌ No Gemini Business accounts configured!\n' +
+                        '\nTo add an account, run:\n' +
+                        '  opencode-gemini-business add-account\n' +
+                        '\nFor help, run:\n' +
+                        '  opencode-gemini-business help\n');
+                    return {};
+                }
+                console.log(`\n✅ Gemini Business: Loaded ${accounts.length} account(s)\n` +
+                    `   Strategy: ${accountManager.getConfig().rotation_strategy}\n`);
+                return {
+                    // Return empty config - we handle everything in custom fetch
+                    accountCount: accounts.length,
+                    strategy: accountManager.getConfig().rotation_strategy,
+                    /**
+                     * Custom fetch implementation for Gemini Business API
+                     *
+                     * This intercepts all OpenCode API calls and routes them through
+                     * the Gemini Business API with account rotation.
+                     */
+                    async fetch(input, options) {
+                        try {
+                            // Get next available account based on rotation strategy
+                            const account = accountManager.getNextAccount();
+                            if (!account) {
+                                throw new Error('No available accounts');
+                            }
+                            // Create API client for this account
+                            const api = new GeminiBusinessAPI(account);
+                            // Refresh session if needed
+                            if (api.needsSessionRefresh()) {
+                                console.log(`   ⟳ Refreshing session for: ${account.name}`);
+                                await api.refreshSession();
+                                await accountManager.updateSession(account.id, account.session_id, 50 * 60 * 1000 // 50 minutes
+                                );
+                            }
+                            // Parse request body
+                            const requestBody = options?.body ? JSON.parse(options.body) : {};
+                            // Make API call
+                            const response = await api.chatCompletion(requestBody);
+                            // Success! Reset error count
+                            accountManager.resetAccountErrors(account.id);
+                            // Convert to Response object for OpenCode
+                            return new Response(JSON.stringify(response), {
+                                status: 200,
+                                headers: {
+                                    'Content-Type': 'application/json',
+                                },
+                            });
+                        }
+                        catch (error) {
+                            const errorMessage = error instanceof Error ? error.message : String(error);
+                            // Mark account error
+                            const account = accountManager.getNextAccount();
+                            if (account) {
+                                accountManager.markAccountError(account.id, errorMessage);
+                            }
+                            // Rethrow for OpenCode to handle
+                            throw new Error(`Gemini Business API Error: ${errorMessage}`);
+                        }
+                    },
+                };
+            },
+            /**
+             * Authentication methods
+             *
+             * For Gemini Business, we don't support OAuth - accounts are configured
+             * via CLI with cookie extraction from browser.
+             */
+            methods: [
+                {
+                    type: 'cookie',
+                    label: 'Gemini Business Cookie Auth',
+                    authorize: async () => ({
+                        type: 'instructions',
+                        instructions: '📋 To configure Gemini Business accounts:\n\n' +
+                            '1. Login to https://business.gemini.google\n' +
+                            '2. Open DevTools (F12) → Network tab\n' +
+                            '3. Make any API request (refresh page or send a prompt)\n' +
+                            '4. Find request headers and copy:\n' +
+                            '   • team_id (from X-Goog-Team-Id header)\n' +
+                            '   • __Secure-c_ses cookie\n' +
+                            '   • __Host-c_oses cookie\n' +
+                            '   • csesidx value\n\n' +
+                            '5. Run: opencode-gemini-business add-account\n' +
+                            '   Or use environment variables (see docs)\n\n' +
+                            'For more help: opencode-gemini-business help',
+                    }),
+                },
+            ],
+        },
+    };
+};
+// Export as default for CommonJS compatibility
+export default GeminiBusinessPlugin;
+// Export legacy functions for backward compatibility (if needed)
+export { AccountManager, GeminiBusinessAPI };
+// ============================================================================
+// CLI Commands
+// ============================================================================
+/**
+ * CLI entry point - handles account management commands
+ */
+async function cli() {
+    const args = process.argv.slice(2);
+    const command = args[0];
+    const accountManager = new AccountManager();
+    try {
+        await accountManager.loadAccounts();
+    }
+    catch {
+        // Config file might not exist yet - that's ok
+    }
+    switch (command) {
+        case 'add-account':
+            await addAccountCommand(accountManager, args.slice(1));
+            break;
+        case 'list-accounts':
+            await listAccountsCommand(accountManager);
+            break;
+        case 'remove-account':
+            await removeAccountCommand(accountManager, args[1]);
+            break;
+        case 'test-account':
+            await testAccountCommand(accountManager, args[1]);
+            break;
+        case 'help':
+        default:
+            printHelp();
+            break;
+    }
+}
+/**
+ * Add account command
+ */
+async function addAccountCommand(manager, args) {
+    console.log('Add Gemini Business Account');
+    console.log('----------------------------');
+    // Read from command line args or env vars
+    const account = {
+        name: args[0] || process.env.GEMINI_ACCOUNT_NAME || `account-${Date.now()}`,
+        team_id: args[1] || process.env.GEMINI_TEAM_ID || '',
+        cookies: {
+            secure_c_ses: args[2] || process.env.GEMINI_SECURE_C_SES || '',
+            host_c_oses: args[3] || process.env.GEMINI_HOST_C_OSES || '',
+        },
+        csesidx: args[4] || process.env.GEMINI_CSESIDX || '',
+        user_agent: args[5] || process.env.GEMINI_USER_AGENT,
+        enabled: true,
+    };
+    // Validate required fields
+    if (!account.team_id || !account.cookies.secure_c_ses || !account.cookies.host_c_oses || !account.csesidx) {
+        console.error('Error: Missing required fields!');
+        console.error('\nUsage:');
+        console.error('  opencode-gemini-business add-account <name> <team_id> <secure_c_ses> <host_c_oses> <csesidx> [user_agent]');
+        console.error('\nOr set environment variables:');
+        console.error('  GEMINI_ACCOUNT_NAME, GEMINI_TEAM_ID, GEMINI_SECURE_C_SES, GEMINI_HOST_C_OSES, GEMINI_CSESIDX, GEMINI_USER_AGENT');
+        process.exit(1);
+    }
+    const id = await manager.addAccount(account);
+    console.log(`✅ Account added successfully! ID: ${id}`);
+    console.log(`   Name: ${account.name}`);
+    console.log(`   Team ID: ${account.team_id}`);
+}
+/**
+ * List accounts command
+ */
+async function listAccountsCommand(manager) {
+    const accounts = manager.getAccounts();
+    if (accounts.length === 0) {
+        console.log('No accounts configured.');
+        return;
+    }
+    console.log(`Found ${accounts.length} account(s):\n`);
+    accounts.forEach((account, index) => {
+        console.log(`[${index + 1}] ${account.name} (${account.id})`);
+        console.log(`    Team ID: ${account.team_id}`);
+        console.log(`    Status: ${account.enabled ? '✅ Enabled' : '❌ Disabled'}`);
+        console.log(`    Errors: ${account.error_count || 0}`);
+        if (account.last_error) {
+            console.log(`    Last Error: ${account.last_error}`);
+        }
+        if (account.last_used) {
+            const lastUsed = new Date(account.last_used);
+            console.log(`    Last Used: ${lastUsed.toISOString()}`);
+        }
+        console.log('');
+    });
+}
+/**
+ * Remove account command
+ */
+async function removeAccountCommand(manager, accountId) {
+    if (!accountId) {
+        console.error('Error: Account ID required');
+        console.error('Usage: opencode-gemini-business remove-account <account_id>');
+        process.exit(1);
+    }
+    const removed = await manager.removeAccount(accountId);
+    if (removed) {
+        console.log(`✅ Account ${accountId} removed successfully`);
+    }
+    else {
+        console.error(`❌ Account ${accountId} not found`);
+        process.exit(1);
+    }
+}
+/**
+ * Test account command
+ */
+async function testAccountCommand(manager, accountId) {
+    if (!accountId) {
+        console.error('Error: Account ID required');
+        console.error('Usage: opencode-gemini-business test-account <account_id>');
+        process.exit(1);
+    }
+    const accounts = manager.getAccounts();
+    const account = accounts.find(acc => acc.id === accountId);
+    if (!account) {
+        console.error(`❌ Account ${accountId} not found`);
+        process.exit(1);
+    }
+    console.log(`Testing account: ${account.name} (${account.id})`);
+    console.log('Please wait...\n');
+    const api = new GeminiBusinessAPI(account);
+    const result = await api.testAccount();
+    if (result.success) {
+        console.log('✅ Account test successful!');
+        console.log('   XSRF Token: OK');
+        console.log('   Session: OK');
+        console.log('   Authentication: OK');
+    }
+    else {
+        console.log('❌ Account test failed!');
+        console.log(`   Error: ${result.error}`);
+        process.exit(1);
+    }
+}
+/**
+ * Print help
+ */
+function printHelp() {
+    console.log('OpenCode Gemini Business Plugin');
+    console.log('================================\n');
+    console.log('Multi-account Gemini Business pool with automatic rotation\n');
+    console.log('Commands:');
+    console.log('  add-account <name> <team_id> <secure_c_ses> <host_c_oses> <csesidx> [user_agent]');
+    console.log('    Add a new Gemini Business account\n');
+    console.log('  list-accounts');
+    console.log('    List all configured accounts\n');
+    console.log('  remove-account <account_id>');
+    console.log('    Remove an account\n');
+    console.log('  test-account <account_id>');
+    console.log('    Test account credentials\n');
+    console.log('  help');
+    console.log('    Show this help message\n');
+    console.log('Environment Variables:');
+    console.log('  GEMINI_ACCOUNT_NAME    - Account name');
+    console.log('  GEMINI_TEAM_ID         - Team ID');
+    console.log('  GEMINI_SECURE_C_SES    - __Secure-c_ses cookie');
+    console.log('  GEMINI_HOST_C_OSES     - __Host-c_oses cookie');
+    console.log('  GEMINI_CSESIDX         - csesidx value');
+    console.log('  GEMINI_USER_AGENT      - Custom user agent (optional)\n');
+}
+// Export CLI function
+export { cli };
+// Run CLI if executed directly
+if (import.meta.url === `file://${process.argv[1]}`) {
+    cli().catch(error => {
+        console.error('Error:', error.message);
+        process.exit(1);
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":";AAEA;;;;;;;;GAQG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAGjE,+EAA+E;AAC/E,yBAAyB;AACzB,+EAA+E;AAE/E;;;;;GAKG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,KAAK,EAAE,GAAQ,EAAE,EAAE;IACrD,OAAO;QACL,IAAI,EAAE;YACJ,QAAQ,EAAE,iBAAiB;YAE3B;;eAEG;YACH,KAAK,CAAC,MAAM,CAAC,OAAY,EAAE,QAAa;gBACtC,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;gBAC5C,MAAM,cAAc,CAAC,YAAY,EAAE,CAAC;gBAEpC,MAAM,QAAQ,GAAG,cAAc,CAAC,WAAW,EAAE,CAAC;gBAC9C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC1B,OAAO,CAAC,KAAK,CACX,+CAA+C;wBAC/C,6BAA6B;wBAC7B,0CAA0C;wBAC1C,oBAAoB;wBACpB,mCAAmC,CACpC,CAAC;oBACF,OAAO,EAAE,CAAC;gBACZ,CAAC;gBAED,OAAO,CAAC,GAAG,CACT,+BAA+B,QAAQ,CAAC,MAAM,eAAe;oBAC7D,gBAAgB,cAAc,CAAC,SAAS,EAAE,CAAC,iBAAiB,IAAI,CACjE,CAAC;gBAEF,OAAO;oBACL,6DAA6D;oBAC7D,YAAY,EAAE,QAAQ,CAAC,MAAM;oBAC7B,QAAQ,EAAE,cAAc,CAAC,SAAS,EAAE,CAAC,iBAAiB;oBAEtD;;;;;uBAKG;oBACH,KAAK,CAAC,KAAK,CAAC,KAAU,EAAE,OAAY;wBAClC,IAAI,CAAC;4BACH,wDAAwD;4BACxD,MAAM,OAAO,GAAG,cAAc,CAAC,cAAc,EAAE,CAAC;4BAChD,IAAI,CAAC,OAAO,EAAE,CAAC;gCACb,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;4BAC3C,CAAC;4BAED,qCAAqC;4BACrC,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;4BAE3C,4BAA4B;4BAC5B,IAAI,GAAG,CAAC,mBAAmB,EAAE,EAAE,CAAC;gCAC9B,OAAO,CAAC,GAAG,CAAC,gCAAgC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gCAC5D,MAAM,GAAG,CAAC,cAAc,EAAE,CAAC;gCAC3B,MAAM,cAAc,CAAC,aAAa,CAChC,OAAO,CAAC,EAAE,EACV,OAAO,CAAC,UAAW,EACnB,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,aAAa;iCAC7B,CAAC;4BACJ,CAAC;4BAED,qBAAqB;4BACrB,MAAM,WAAW,GAAG,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAc,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;4BAE5E,gBAAgB;4BAChB,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;4BAEvD,6BAA6B;4BAC7B,cAAc,CAAC,kBAAkB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;4BAE9C,0CAA0C;4BAC1C,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE;gCAC5C,MAAM,EAAE,GAAG;gCACX,OAAO,EAAE;oCACP,cAAc,EAAE,kBAAkB;iCACnC;6BACF,CAAC,CAAC;wBACL,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;4BAE5E,qBAAqB;4BACrB,MAAM,OAAO,GAAG,cAAc,CAAC,cAAc,EAAE,CAAC;4BAChD,IAAI,OAAO,EAAE,CAAC;gCACZ,cAAc,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;4BAC5D,CAAC;4BAED,iCAAiC;4BACjC,MAAM,IAAI,KAAK,CAAC,8BAA8B,YAAY,EAAE,CAAC,CAAC;wBAChE,CAAC;oBACH,CAAC;iBACF,CAAC;YACJ,CAAC;YAED;;;;;eAKG;YACH,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,QAAiB;oBACvB,KAAK,EAAE,6BAA6B;oBACpC,SAAS,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;wBACtB,IAAI,EAAE,cAAuB;wBAC7B,YAAY,EACV,+CAA+C;4BAC/C,8CAA8C;4BAC9C,wCAAwC;4BACxC,2DAA2D;4BAC3D,qCAAqC;4BACrC,6CAA6C;4BAC7C,8BAA8B;4BAC9B,6BAA6B;4BAC7B,wBAAwB;4BACxB,gDAAgD;4BAChD,gDAAgD;4BAChD,8CAA8C;qBACjD,CAAC;iBACH;aACF;SACF;KACF,CAAC;AACJ,CAAC,CAAC;AAEF,+CAA+C;AAC/C,eAAe,oBAAoB,CAAC;AAEpC,iEAAiE;AACjE,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,CAAC;AAE7C,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,KAAK,UAAU,GAAG;IAChB,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IAExB,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;IAE5C,IAAI,CAAC;QACH,MAAM,cAAc,CAAC,YAAY,EAAE,CAAC;IACtC,CAAC;IAAC,MAAM,CAAC;QACP,8CAA8C;IAChD,CAAC;IAED,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,aAAa;YAChB,MAAM,iBAAiB,CAAC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACvD,MAAM;QAER,KAAK,eAAe;YAClB,MAAM,mBAAmB,CAAC,cAAc,CAAC,CAAC;YAC1C,MAAM;QAER,KAAK,gBAAgB;YACnB,MAAM,oBAAoB,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YACpD,MAAM;QAER,KAAK,cAAc;YACjB,MAAM,kBAAkB,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM;QAER,KAAK,MAAM,CAAC;QACZ;YACE,SAAS,EAAE,CAAC;YACZ,MAAM;IACV,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,iBAAiB,CAAC,OAAuB,EAAE,IAAc;IACtE,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAC3C,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAE5C,0CAA0C;IAC1C,MAAM,OAAO,GAAsC;QACjD,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,WAAW,IAAI,CAAC,GAAG,EAAE,EAAE;QAC3E,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE;QACpD,OAAO,EAAE;YACP,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,EAAE;YAC9D,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,EAAE;SAC7D;QACD,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE;QACpD,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB;QACpD,OAAO,EAAE,IAAI;KACd,CAAC;IAEF,2BAA2B;IAC3B,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAC1G,OAAO,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACjD,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC1B,OAAO,CAAC,KAAK,CAAC,6GAA6G,CAAC,CAAC;QAC7H,OAAO,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACjD,OAAO,CAAC,KAAK,CAAC,mHAAmH,CAAC,CAAC;QACnI,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,qCAAqC,EAAE,EAAE,CAAC,CAAC;IACvD,OAAO,CAAC,GAAG,CAAC,YAAY,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,eAAe,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,mBAAmB,CAAC,OAAuB;IACxD,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAEvC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QACvC,OAAO;IACT,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,SAAS,QAAQ,CAAC,MAAM,gBAAgB,CAAC,CAAC;IAEtD,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;QAClC,OAAO,CAAC,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,KAAK,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,EAAE,GAAG,CAAC,CAAC;QAC9D,OAAO,CAAC,GAAG,CAAC,gBAAgB,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/C,OAAO,CAAC,GAAG,CAAC,eAAe,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC;QAC3E,OAAO,CAAC,GAAG,CAAC,eAAe,OAAO,CAAC,WAAW,IAAI,CAAC,EAAE,CAAC,CAAC;QACvD,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,OAAO,CAAC,GAAG,CAAC,mBAAmB,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACtB,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC7C,OAAO,CAAC,GAAG,CAAC,kBAAkB,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC1D,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,oBAAoB,CAAC,OAAuB,EAAE,SAAiB;IAC5E,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAC5C,OAAO,CAAC,KAAK,CAAC,6DAA6D,CAAC,CAAC;QAC7E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;IACvD,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,CAAC,GAAG,CAAC,aAAa,SAAS,uBAAuB,CAAC,CAAC;IAC7D,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,KAAK,CAAC,aAAa,SAAS,YAAY,CAAC,CAAC;QAClD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,kBAAkB,CAAC,OAAuB,EAAE,SAAiB;IAC1E,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAC5C,OAAO,CAAC,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC3E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IACvC,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC;IAE3D,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,aAAa,SAAS,YAAY,CAAC,CAAC;QAClD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,oBAAoB,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,EAAE,GAAG,CAAC,CAAC;IAChE,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;IAEhC,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAC3C,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC;IAEvC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAC1C,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC9B,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,aAAa,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QACzC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,SAAS;IAChB,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IAC/C,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,8DAA8D,CAAC,CAAC;IAC5E,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACzB,OAAO,CAAC,GAAG,CAAC,oFAAoF,CAAC,CAAC;IAClG,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;IACvD,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IAC/B,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IACvC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAC3C,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;IAC9C,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACtB,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;IACtC,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;IACvD,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;IAChE,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;IAC/D,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;IACxD,OAAO,CAAC,GAAG,CAAC,2DAA2D,CAAC,CAAC;AAC3E,CAAC;AAED,sBAAsB;AACtB,OAAO,EAAE,GAAG,EAAE,CAAC;AAEf,+BAA+B;AAC/B,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,UAAU,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IACpD,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;QAClB,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QACvC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/src/account-manager.d.ts

@@ -0,0 +1,58 @@
+/**
+ * Account Manager for multi-account rotation
+ */
+import { GeminiBusinessAccount, PoolConfig } from './types.js';
+export declare class AccountManager {
+    private config;
+    private currentIndex;
+    constructor(config?: Partial<PoolConfig>);
+    /**
+     * Load accounts from config file
+     */
+    loadAccounts(): Promise<void>;
+    /**
+     * Save accounts to config file
+     */
+    saveAccounts(): Promise<void>;
+    /**
+     * Add a new account
+     */
+    addAccount(account: Omit<GeminiBusinessAccount, 'id'>): Promise<string>;
+    /**
+     * Remove an account
+     */
+    removeAccount(id: string): Promise<boolean>;
+    /**
+     * Get next available account based on rotation strategy
+     */
+    getNextAccount(): GeminiBusinessAccount | null;
+    /**
+     * Mark account as failed
+     */
+    markAccountError(accountId: string, error: string): void;
+    /**
+     * Reset account errors (e.g., after successful request)
+     */
+    resetAccountErrors(accountId: string): void;
+    /**
+     * Check if session needs refresh
+     */
+    needsSessionRefresh(account: GeminiBusinessAccount): boolean;
+    /**
+     * Update account session
+     */
+    updateSession(accountId: string, sessionId: string, expiresIn: number): Promise<void>;
+    /**
+     * Update XSRF token
+     */
+    updateXSRFToken(accountId: string, token: string, expiresIn: number): Promise<void>;
+    /**
+     * Get all accounts
+     */
+    getAccounts(): GeminiBusinessAccount[];
+    /**
+     * Get config
+     */
+    getConfig(): PoolConfig;
+}
+//# sourceMappingURL=account-manager.d.ts.map

package/dist/src/account-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"account-manager.d.ts","sourceRoot":"","sources":["../../src/account-manager.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,qBAAqB,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAS/D,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,YAAY,CAAa;gBAErB,MAAM,CAAC,EAAE,OAAO,CAAC,UAAU,CAAC;IAYxC;;OAEG;IACG,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAsBnC;;OAEG;IACG,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAWnC;;OAEG;IACG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,qBAAqB,EAAE,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAc7E;;OAEG;IACG,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IASjD;;OAEG;IACH,cAAc,IAAI,qBAAqB,GAAG,IAAI;IAqC9C;;OAEG;IACH,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAcxD;;OAEG;IACH,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAQ3C;;OAEG;IACH,mBAAmB,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO;IAS5D;;OAEG;IACG,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAS3F;;OAEG;IACG,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASzF;;OAEG;IACH,WAAW,IAAI,qBAAqB,EAAE;IAItC;;OAEG;IACH,SAAS,IAAI,UAAU;CAGxB"}

package/dist/src/account-manager.js

@@ -0,0 +1,190 @@
+/**
+ * Account Manager for multi-account rotation
+ */
+import { readFile, writeFile } from 'fs/promises';
+import { existsSync } from 'fs';
+import { homedir } from 'os';
+import { join } from 'path';
+const CONFIG_DIR = join(homedir(), '.config', 'opencode');
+const ACCOUNTS_FILE = join(CONFIG_DIR, 'gemini-business-accounts.json');
+export class AccountManager {
+    config;
+    currentIndex = 0;
+    constructor(config) {
+        this.config = {
+            accounts: [],
+            rotation_strategy: 'round-robin',
+            max_retries: 3,
+            retry_delay: 1000,
+            session_refresh_threshold: 300, // 5 minutes
+            error_threshold: 3,
+            ...config,
+        };
+    }
+    /**
+     * Load accounts from config file
+     */
+    async loadAccounts() {
+        if (!existsSync(ACCOUNTS_FILE)) {
+            console.warn(`No accounts file found at ${ACCOUNTS_FILE}`);
+            return;
+        }
+        try {
+            const data = await readFile(ACCOUNTS_FILE, 'utf-8');
+            const saved = JSON.parse(data);
+            this.config.accounts = saved.accounts || [];
+            // Merge other config options if they exist
+            if (saved.rotation_strategy)
+                this.config.rotation_strategy = saved.rotation_strategy;
+            if (saved.max_retries !== undefined)
+                this.config.max_retries = saved.max_retries;
+            console.log(`Loaded ${this.config.accounts.length} accounts`);
+        }
+        catch (error) {
+            console.error('Failed to load accounts:', error);
+            throw new Error(`Failed to load accounts: ${error}`);
+        }
+    }
+    /**
+     * Save accounts to config file
+     */
+    async saveAccounts() {
+        try {
+            const data = JSON.stringify(this.config, null, 2);
+            await writeFile(ACCOUNTS_FILE, data, 'utf-8');
+            console.log(`Saved ${this.config.accounts.length} accounts to ${ACCOUNTS_FILE}`);
+        }
+        catch (error) {
+            console.error('Failed to save accounts:', error);
+            throw new Error(`Failed to save accounts: ${error}`);
+        }
+    }
+    /**
+     * Add a new account
+     */
+    async addAccount(account) {
+        const id = `account-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
+        const newAccount = {
+            ...account,
+            id,
+            enabled: true,
+            error_count: 0,
+        };
+        this.config.accounts.push(newAccount);
+        await this.saveAccounts();
+        return id;
+    }
+    /**
+     * Remove an account
+     */
+    async removeAccount(id) {
+        const index = this.config.accounts.findIndex(acc => acc.id === id);
+        if (index === -1)
+            return false;
+        this.config.accounts.splice(index, 1);
+        await this.saveAccounts();
+        return true;
+    }
+    /**
+     * Get next available account based on rotation strategy
+     */
+    getNextAccount() {
+        const enabledAccounts = this.config.accounts.filter(acc => acc.enabled);
+        if (enabledAccounts.length === 0) {
+            console.error('No enabled accounts available');
+            return null;
+        }
+        let account;
+        switch (this.config.rotation_strategy) {
+            case 'round-robin':
+                account = enabledAccounts[this.currentIndex % enabledAccounts.length];
+                this.currentIndex++;
+                break;
+            case 'least-used':
+                account = enabledAccounts.reduce((least, current) => {
+                    const leastUsed = least.last_used || 0;
+                    const currentUsed = current.last_used || 0;
+                    return currentUsed < leastUsed ? current : least;
+                });
+                break;
+            case 'random':
+                account = enabledAccounts[Math.floor(Math.random() * enabledAccounts.length)];
+                break;
+            default:
+                account = enabledAccounts[0];
+        }
+        // Update last used timestamp
+        account.last_used = Date.now();
+        return account;
+    }
+    /**
+     * Mark account as failed
+     */
+    markAccountError(accountId, error) {
+        const account = this.config.accounts.find(acc => acc.id === accountId);
+        if (!account)
+            return;
+        account.error_count = (account.error_count || 0) + 1;
+        account.last_error = error;
+        // Disable account if error threshold reached
+        if (account.error_count >= this.config.error_threshold) {
+            account.enabled = false;
+            console.warn(`Account ${account.name} (${accountId}) disabled after ${account.error_count} errors`);
+        }
+    }
+    /**
+     * Reset account errors (e.g., after successful request)
+     */
+    resetAccountErrors(accountId) {
+        const account = this.config.accounts.find(acc => acc.id === accountId);
+        if (!account)
+            return;
+        account.error_count = 0;
+        account.last_error = undefined;
+    }
+    /**
+     * Check if session needs refresh
+     */
+    needsSessionRefresh(account) {
+        if (!account.session_id || !account.session_expires)
+            return true;
+        const now = Date.now();
+        const expiresIn = account.session_expires - now;
+        return expiresIn < this.config.session_refresh_threshold * 1000;
+    }
+    /**
+     * Update account session
+     */
+    async updateSession(accountId, sessionId, expiresIn) {
+        const account = this.config.accounts.find(acc => acc.id === accountId);
+        if (!account)
+            return;
+        account.session_id = sessionId;
+        account.session_expires = Date.now() + expiresIn;
+        await this.saveAccounts();
+    }
+    /**
+     * Update XSRF token
+     */
+    async updateXSRFToken(accountId, token, expiresIn) {
+        const account = this.config.accounts.find(acc => acc.id === accountId);
+        if (!account)
+            return;
+        account.xsrf_token = token;
+        account.xsrf_expires = Date.now() + expiresIn;
+        await this.saveAccounts();
+    }
+    /**
+     * Get all accounts
+     */
+    getAccounts() {
+        return this.config.accounts;
+    }
+    /**
+     * Get config
+     */
+    getConfig() {
+        return this.config;
+    }
+}
+//# sourceMappingURL=account-manager.js.map