agent-tank 0.9.5 → 0.9.7

package/src/agents/gemini.js

@@ -1,412 +0,0 @@
-const { BaseAgent } = require('./base.js');
-const { calculatePace } = require('../pace-evaluator.js');
-const { CYCLE_DURATIONS } = require('../usage-formatters.js');
-const logger = require('../logger.js');
-
-class GeminiAgent extends BaseAgent {
-  constructor() {
-    super('gemini', 'gemini');
-    this._aboutSent = false;
-  }
-
-  getTimeout() {
-    return 35000; // 35 seconds - Gemini can take 10+ seconds to start
-  }
-
-  getEnv() {
-    // Use xterm-256color for proper terminal support
-    // Gemini CLI needs good terminal emulation for its TUI
-    return {
-      ...process.env,
-      TERM: 'xterm-256color',
-      COLORTERM: 'truecolor',
-    };
-  }
-
-  _hasReadyPrompt(output) {
-    const clean = this.stripAnsi(output);
-    return clean.includes('Type your message')
-      || clean.includes('gemini>')
-      || /Ready\s*\(/.test(clean);
-  }
-
-  isReadyForCommands(output) {
-    // Don't consider ready if authentication is in progress
-    if (this._isAuthenticating(output)) {
-      return false;
-    }
-    return this._hasReadyPrompt(output);
-  }
-
-  _isAuthenticating(output) {
-    // Detect OAuth/authentication screens where Escape would cancel the flow
-    const clean = this.stripAnsi(output);
-    const hasAuthFlow = /waiting for authentication/i.test(clean)
-      || /press esc.*to cancel/i.test(clean)
-      || /authenticating\.\.\./i.test(clean);
-
-    if (!hasAuthFlow) {
-      return false;
-    }
-
-    // Newer Gemini builds can render footer hints like "? for shortcuts"
-    // before the actual input prompt is available. Treat auth as complete
-    // only when the real prompt has returned.
-    return !this._hasReadyPrompt(clean);
-  }
-
-  detectAuthenticationState(output) {
-    const clean = this.stripAnsi(output);
-    if (!clean) return null;
-
-    const hasAuthMenu = /how would you like to authenticate for this project\?/i.test(clean)
-      || /get started/i.test(clean);
-    const hasAuthFailure = /failed to sign in\.\s*message:\s*([^\n]+)/i.test(clean)
-      || /please run gemini cli in an interactive terminal to authenticate/i.test(clean)
-      || /authentication consent could not be obtained/i.test(clean);
-
-    if (!hasAuthMenu && !hasAuthFailure) {
-      return null;
-    }
-
-    const failureMatch = clean.match(/failed to sign in\.\s*message:\s*([^\n]+)/i);
-    const detail = failureMatch
-      ? `Failed to sign in: ${failureMatch[1].trim()}`
-      : 'Gemini CLI is waiting for authentication.';
-    const action = /NO_BROWSER=true/i.test(clean)
-      ? 'Run Gemini CLI in an interactive terminal to authenticate, or use NO_BROWSER=true for manual authentication.'
-      : 'Run Gemini CLI in an interactive terminal to authenticate.';
-
-    return {
-      authenticated: false,
-      status: 'unauthenticated',
-      message: 'Authentication required',
-      detail,
-      action,
-    };
-  }
-
-  handleTrustPrompt(shell, output) {
-    // Check for trust prompt with various wordings
-    const trustPatterns = [
-      'Do you trust',
-      'trust the files',
-      'trust this folder',
-      'Trust this workspace',
-      'allow access',
-      'grant access',
-      'grant full access',
-    ];
-
-    if (trustPatterns.some(pattern => output.toLowerCase().includes(pattern.toLowerCase()))) {
-      logger.agent(this.name, 'Detected trust prompt, auto-accepting...');
-      // v0.35+: numbered menu with "1. Trust folder" pre-selected — Enter accepts it
-      shell.write('1\r');
-
-      // v0.35+ does an internal restart after trust — send Escape after a delay
-      // to dismiss any transitional UI before the process becomes ready again
-      setTimeout(() => {
-        if (!shell) return;
-        logger.agent(this.name, 'Post-trust: sending Escape to clear restart UI...');
-        shell.write('\x1b');
-      }, 3000);
-
-      return true;
-    }
-    return false;
-  }
-
-  hasCompleteOutput(output) {
-    // Gemini shows "Usage limits" after the model usage table (v0.24.x)
-    if (output.includes('Usage limits span all sessions')) return true;
-    const clean = this.stripAnsi(output);
-    if (this.detectAuthenticationState(clean)) return true;
-    const hasModel = /gemini-[\w.-]+/i.test(clean);
-    const hasPercent = /\d+(?:\.\d+)?%/i.test(clean);
-    // v0.24: "Resets in 3h 26m", v0.35+: "(23h 21m)" after time
-    const hasResets = /resets?\s+in/i.test(clean) || /\(\d+[hmd]\s+\d+[hmd]?\)/i.test(clean);
-    return hasModel && hasPercent && hasResets;
-  }
-
-  // Convert duration string like "3h 26m" or "5d 2h" to seconds
-  parseDurationToSeconds(durationStr) {
-    if (!durationStr) return null;
-
-    let totalSeconds = 0;
-
-    // Match days
-    const daysMatch = durationStr.match(/(\d+)\s*d/i);
-    if (daysMatch) {
-      totalSeconds += parseInt(daysMatch[1]) * 24 * 60 * 60;
-    }
-
-    // Match hours
-    const hoursMatch = durationStr.match(/(\d+)\s*h/i);
-    if (hoursMatch) {
-      totalSeconds += parseInt(hoursMatch[1]) * 60 * 60;
-    }
-
-    // Match minutes
-    const minsMatch = durationStr.match(/(\d+)\s*m/i);
-    if (minsMatch) {
-      totalSeconds += parseInt(minsMatch[1]) * 60;
-    }
-
-    return totalSeconds > 0 ? totalSeconds : null;
-  }
-
-  sendCommands(shell, output) {
-    // Check if authentication is still in progress - don't send commands that could interfere
-    if (this._isAuthenticating(output) || this.detectAuthenticationState(output)) {
-      logger.agent(this.name, 'Authentication required or in progress, cannot send /stats command');
-      // Don't retry - let the timeout handle it. Sending Escape would cancel auth.
-      return;
-    }
-    logger.agent(this.name, 'Sending /stats command...');
-    // Newer Gemini builds use Esc for "rewind", which can derail the command flow.
-    // Submit /stats directly once the prompt is ready.
-    setTimeout(() => shell.write('/stats'), 200);
-    setTimeout(() => shell.write('\r'), 600);
-  }
-
-  // Calculate pace data for a model entry (Gemini uses 24h sessions)
-  _addPaceData(entry, resetsInSeconds) {
-    const cycleDuration = CYCLE_DURATIONS.sessionGemini;
-    if (!cycleDuration || resetsInSeconds == null) return;
-    const paceData = calculatePace({ usagePercent: entry.percentUsed, resetsInSeconds, cycleDurationSeconds: cycleDuration });
-    if (paceData) entry.pace = paceData;
-  }
-
-  parseOutput(output) {
-    const clean = this.stripAnsi(output);
-    const usage = {
-      models: [],
-    };
-
-    // Parse model usage lines
-    // Format: "gemini-2.5-flash   -90.2% (Resets in 3h 26m)" or "gemini-2.5-flash   90.2% (Resets in 3h 26m)"
-    const lines = clean.split('\n');
-
-    for (const line of lines) {
-      // v0.24: "gemini-2.5-flash   90.2% (Resets in 3h 26m)" — percentage is REMAINING
-      const matchOld = line.match(/(gemini-[\w.-]+)\s+[-\s]*(\d+(?:\.\d+)?)\s*%\s*\(Resets in\s*([^)]+)\)/i);
-      // v0.35+: "gemini-2.5-flash   -   ▬▬▬   6%  10:46 PM (23h 21m)" — percentage is USED
-      const matchNew = !matchOld && line.match(/(gemini-[\w.-]+)\s+.*?(\d+(?:\.\d+)?)\s*%.*?\((\d+[hmd]\s*\d*[hmd]?)\)/i);
-      const match = matchOld || matchNew;
-      if (match) {
-        const model = match[1];
-        const rawPercent = parseFloat(match[2]);
-        const resetsIn = match[3].trim();
-        const percentUsed = matchOld ? parseFloat((100 - rawPercent).toFixed(1)) : rawPercent;
-        const resetsInSeconds = this.parseDurationToSeconds(resetsIn);
-
-        // Avoid duplicates
-        if (!usage.models.find(m => m.model === model)) {
-          const usageLeft = matchOld ? rawPercent : parseFloat((100 - rawPercent).toFixed(1));
-          const modelEntry = { model, usageLeft, resetsIn, percentUsed, resetsInSeconds };
-          this._addPaceData(modelEntry, resetsInSeconds);
-          usage.models.push(modelEntry);
-        }
-      }
-    }
-
-    // Detect version update notification in output
-    // Gemini CLI shows: "Update available! 0.24.5 → 0.32.1" or similar
-    const updateMatch = clean.match(/[Uu]pdate\s+available[^\d]*([\d.]+)\s*(?:→|->|to)\s*([\d.]+)/);
-    if (updateMatch) {
-      usage.version = { current: updateMatch[1], latest: updateMatch[2] };
-    }
-
-    return usage;
-  }
-
-  // Fetch metadata by sending /about command once on first refresh
-  async fetchMetadata() {
-    if (this._aboutSent) {
-      return this.metadata;
-    }
-
-    // Ensure process is spawned and ready
-    if (!this.shell || !this.processReady) {
-      await this.spawnProcess();
-    }
-
-    if (this.detectAuthenticationState(this.output)) {
-      return null;
-    }
-
-    return new Promise((resolve, _reject) => {
-      let aboutOutput = '';
-      let completed = false;
-
-      const finish = (result) => {
-        if (completed) return;
-        completed = true;
-        this._aboutSent = true;
-        this._onDataCallback = null;
-        this._commandInFlight = false;
-        clearTimeout(timer);
-        resolve(result);
-      };
-
-      const timer = setTimeout(() => {
-        logger.agent(this.name, '/about timeout, using partial output');
-        finish(this._parseAboutOutput(aboutOutput));
-      }, 10000); // 10 second timeout for /about
-
-      this._commandInFlight = true;
-      let trustHandled = false;
-      this._onDataCallback = () => {
-        aboutOutput = this.output;
-        // Handle trust prompts that appear during command execution
-        if (!trustHandled && this.handleTrustPrompt(this.shell, aboutOutput)) {
-          trustHandled = true;
-          aboutOutput = '';
-          this.output = '';
-          return;
-        }
-        // Check if we have complete /about output
-        if (this._hasCompleteAboutOutput(aboutOutput)) {
-          logger.agent(this.name, 'Complete /about output detected');
-          setTimeout(() => finish(this._parseAboutOutput(aboutOutput)), 100);
-        }
-      };
-
-      logger.agent(this.name, 'Sending /about command for metadata...');
-      this.output = '';
-
-      // Wait for authentication to complete before sending commands
-      // Auth can start after spawn if token refresh is needed
-      // Gemini can take 10+ seconds to fully start up
-      let authWaitCount = 0;
-      const maxAuthWait = 15; // Max 15 seconds waiting for auth
-
-      const waitForAuth = () => {
-        if (this.detectAuthenticationState(this.output)) {
-          logger.agent(this.name, 'Authentication required, skipping /about metadata fetch');
-          finish(null);
-          return;
-        }
-        if (this._isAuthenticating(this.output)) {
-          authWaitCount++;
-          if (authWaitCount >= maxAuthWait) {
-            logger.agent(this.name, 'Authentication timeout - Gemini may be stuck in auth flow');
-            // Don't send commands that would interfere with stuck auth
-            // Let the overall timeout handle this
-            return;
-          }
-          logger.agent(this.name, 'Authentication in progress, waiting...');
-          setTimeout(waitForAuth, 1000);
-          return;
-        }
-        // Auth complete (or not happening), send commands
-        // Guard against shell being nullified if process exits during setTimeout delays
-        // Don't send Escape first - it can cancel auth flows that started between checks
-        setTimeout(() => this.shell?.write('/about\r'), 500);
-        setTimeout(() => this.shell?.write('\r'), 1000);
-      };
-
-      setTimeout(waitForAuth, 100);
-    });
-  }
-
-  _hasCompleteAboutOutput(output) {
-    const clean = this.stripAnsi(output);
-    // Require actual /about content (not just status bar text)
-    const hasAboutContent = /about\s+gemini/i.test(clean) || /CLI\s+Version/i.test(clean);
-    // Require prompt to have returned after the about box
-    const hasPrompt = clean.includes('Type your message');
-    return hasAboutContent && hasPrompt;
-  }
-
-  _parseAboutOutput(output) {
-    const clean = this.stripAnsi(output);
-
-    // Guard: verify we have actual /about output, not just status bar noise.
-    // The status bar contains "no sandbox (see /docs) Auto (Gemini 2.5) /model"
-    // which can false-positive match loose regexes.
-    const hasAboutContent = /about\s+gemini/i.test(clean) || /CLI\s+Version/i.test(clean);
-    if (!hasAboutContent) {
-      logger.agent(this.name, '/about output missing expected header, dumping for debug');
-      require('fs').writeFileSync(`/tmp/${this.name}-about-output.txt`, output);
-      logger.agent(this.name, 'Stripped output preview:', logger.dim(clean.substring(0, 500)));
-      return null;
-    }
-
-    const metadata = {};
-
-    // Gemini /about uses whitespace-separated key-value pairs (no colons):
-    //   CLI Version    0.24.5
-    //   OS             linux
-    //   Auth Method    OAuth
-    //   User Email     livecart@gmail.com
-    // After stripAnsi collapses multi-spaces, these become single-space separated.
-
-    // Extract version: "CLI Version 0.24.5"
-    const versionMatch = clean.match(/CLI\s+Version[:\s]+v?([\d.]+)/i);
-    if (versionMatch) {
-      metadata.version = this.stripBoxChars(versionMatch[1]);
-    }
-
-    // Extract email: "User Email livecart@gmail.com"
-    const emailMatch = clean.match(/User\s+Email[:\s]+(\S+@\S+)/i);
-    if (emailMatch) {
-      metadata.email = this.stripBoxChars(emailMatch[1]);
-    }
-
-    // Extract auth method: "Auth Method OAuth"
-    const authMatch = clean.match(/Auth\s*Method[:\s]+(\w+)/i);
-    if (authMatch) {
-      metadata.authMethod = this.stripBoxChars(authMatch[1]);
-    }
-
-    // Extract model: "Model auto-gemini-2.5" (but NOT "/model" from the status bar)
-    const modelMatch = clean.match(/(?<!\/)Model[:\s]+([\w.-]+)/i);
-    if (modelMatch) {
-      metadata.model = this.stripBoxChars(modelMatch[1]);
-    }
-
-    // Extract OS: "OS linux"
-    const osMatch = clean.match(/\bOS[:\s]+(\w+)/i);
-    if (osMatch) {
-      metadata.os = this.stripBoxChars(osMatch[1]);
-    }
-
-    // Detect version update notification
-    const updateMatch = clean.match(/[Uu]pdate\s+available[^\d]*([\d.]+)\s*(?:→|->|to)\s*([\d.]+)/);
-    if (updateMatch) {
-      metadata.updateAvailable = { current: updateMatch[1], latest: updateMatch[2] };
-    }
-
-    return Object.keys(metadata).length > 0 ? metadata : null;
-  }
-
-  /**
-   * Lightweight keepalive to prevent session expiration.
-   * Sends escape key to maintain the PTY session without triggering API calls.
-   *
-   * @returns {Promise<boolean>} True if keepalive succeeded
-   */
-  async keepalive() {
-    if (this.freshProcess) {
-      console.log(`[${this.name}] Keepalive skipped (fresh process mode)`);
-      return true;
-    }
-
-    if (!this.shell || !this.processReady) {
-      console.log(`[${this.name}] Keepalive: spawning process...`);
-      await this.spawnProcess();
-    }
-
-    if (this.shell) {
-      console.log(`[${this.name}] Keepalive: sending ping...`);
-      // Send escape key to dismiss any UI and trigger activity
-      this.shell.write('\x1b');
-      return true;
-    }
-
-    return false;
-  }
-}
-
-module.exports = { GeminiAgent };