jackwener__opencli/clis/chatgpt/ask.ts

import { execSync, spawnSync } from 'node:child_process';
import { cli, Strategy } from '@jackwener/opencli/registry';
import { ConfigError } from '@jackwener/opencli/errors';
import type { IPage } from '@jackwener/opencli/types';
import { activateChatGPT, getVisibleChatMessages, selectModel, MODEL_CHOICES, isGenerating } from './ax.js';

export const askCommand = cli({
  site: 'chatgpt',
  name: 'ask',
  description: 'Send a prompt and wait for the AI response (send + wait + read)',
  domain: 'localhost',
  strategy: Strategy.PUBLIC,
  browser: false,
  args: [
    { name: 'text', required: true, positional: true, help: 'Prompt to send' },
    { name: 'model', required: false, help: 'Model/mode to use: auto, instant, thinking, 5.2-instant, 5.2-thinking', choices: MODEL_CHOICES },
    { name: 'timeout', required: false, help: 'Max seconds to wait for response (default: 30)', default: '30' },
  ],
  columns: ['Role', 'Text'],
  func: async (page: IPage | null, kwargs: any) => {
    if (process.platform !== 'darwin') {
      throw new ConfigError('ChatGPT Desktop integration requires macOS (osascript is not available on this platform)');
    }

    const text = kwargs.text as string;
    const model = kwargs.model as string | undefined;
    const timeout = parseInt(kwargs.timeout as string, 10) || 30;

    // Switch model before sending if requested
    if (model) {
      activateChatGPT();
      selectModel(model);
    }

    // Backup clipboard
    let clipBackup = '';
    try { clipBackup = execSync('pbpaste', { encoding: 'utf-8' }); } catch {}
    const messagesBefore = getVisibleChatMessages();

    // Send the message
    spawnSync('pbcopy', { input: text });
    activateChatGPT();

    const cmd = "osascript " +
                "-e 'tell application \"System Events\"' " +
                "-e 'keystroke \"v\" using command down' " +
                "-e 'delay 0.2' " +
                "-e 'keystroke return' " +
                "-e 'end tell'";
    execSync(cmd);

    // Restore clipboard after the prompt is sent.
    if (clipBackup) spawnSync('pbcopy', { input: clipBackup });

    // Wait for response: poll until ChatGPT stops generating ("Stop generating" button disappears),
    // then read the final response text.
    const pollInterval = 2;
    const maxPolls = Math.ceil(timeout / pollInterval);
    let response = '';
    let generationStarted = false;

    for (let i = 0; i < maxPolls; i++) {
      execSync(`sleep ${pollInterval}`);
      const generating = isGenerating();
      if (generating) {
        generationStarted = true;
        continue;
      }
      // Generation finished (or never started yet)
      if (!generationStarted && i < 3) continue; // give it a moment to start

      // Read final response
      activateChatGPT(0.3);
      const messagesNow = getVisibleChatMessages();
      if (messagesNow.length > messagesBefore.length) {
        const newMessages = messagesNow.slice(messagesBefore.length);
        const candidate = [...newMessages].reverse().find((message) => message !== text);
        if (candidate) response = candidate;
      }
      break;
    }

    if (!response) {
      return [
        { Role: 'User', Text: text },
        { Role: 'System', Text: `No response within ${timeout}s. ChatGPT may still be generating.` },
      ];
    }

    return [
      { Role: 'User', Text: text },
      { Role: 'Assistant', Text: response },
    ];
  },
});