@steipete/oracle 1.1.0 → 1.2.0

package/dist/src/cli/tui/index.js

@@ -0,0 +1,436 @@
+import chalk from 'chalk';
+import inquirer from 'inquirer';
+import kleur from 'kleur';
+import path from 'node:path';
+import os from 'node:os';
+import fs from 'node:fs/promises';
+import { MODEL_CONFIGS } from '../../oracle.js';
+import { renderMarkdownAnsi } from '../markdownRenderer.js';
+import { createSessionLogWriter, getSessionPaths, initializeSession, listSessionsMetadata, readSessionLog, readSessionMetadata, readSessionRequest, ensureSessionStorage, } from '../../sessionManager.js';
+import { performSessionRun } from '../sessionRunner.js';
+import { MAX_RENDER_BYTES, trimBeforeFirstAnswer } from '../sessionDisplay.js';
+import { buildBrowserConfig, resolveBrowserModelLabel } from '../browserConfig.js';
+import { resolveNotificationSettings } from '../notifier.js';
+import { loadUserConfig } from '../../config.js';
+const isTty = () => Boolean(process.stdout.isTTY && chalk.level > 0);
+const dim = (text) => (isTty() ? kleur.dim(text) : text);
+const RECENT_WINDOW_HOURS = 24;
+const PAGE_SIZE = 10;
+export async function launchTui({ version }) {
+    const userConfig = (await loadUserConfig()).config;
+    console.log(chalk.bold(`🧿 oracle v${version}`), dim('— Whispering your tokens to the silicon sage'));
+    console.log('');
+    let olderOffset = 0;
+    for (;;) {
+        const { recent, older, hasMoreOlder } = await fetchSessionBuckets(olderOffset);
+        const choices = [];
+        if (recent.length > 0) {
+            choices.push(new inquirer.Separator());
+            choices.push(new inquirer.Separator('Status  Model      Mode    Timestamp           Chars  Cost  Slug'));
+            choices.push(...recent.map(toSessionChoice));
+        }
+        else if (older.length > 0 && olderOffset === 0) {
+            choices.push(new inquirer.Separator());
+            choices.push(new inquirer.Separator('Status  Model      Mode    Timestamp           Chars  Cost  Slug'));
+            choices.push(...older.slice(0, PAGE_SIZE).map(toSessionChoice));
+        }
+        if (older.length > 0 && olderOffset > 0) {
+            choices.push(new inquirer.Separator());
+            choices.push(new inquirer.Separator('Status  Model      Mode    Timestamp           Chars  Cost  Slug'));
+            choices.push(...older.slice(0, PAGE_SIZE).map(toSessionChoice));
+        }
+        choices.push(new inquirer.Separator());
+        choices.push(new inquirer.Separator('Actions'));
+        choices.push({ name: chalk.bold.green('ask oracle'), value: '__ask__' });
+        if (hasMoreOlder) {
+            choices.push({ name: 'Load older', value: '__more__' });
+        }
+        if (olderOffset > 0) {
+            choices.push({ name: 'Back to recent', value: '__reset__' });
+        }
+        choices.push({ name: 'Exit', value: '__exit__' });
+        const { selection } = await inquirer.prompt([
+            {
+                name: 'selection',
+                type: 'list',
+                message: 'Select a session or action',
+                choices,
+                pageSize: 16,
+            },
+        ]);
+        if (selection === '__exit__') {
+            console.log(chalk.green('🧿 Closing the book. See you next prompt.'));
+            return;
+        }
+        if (selection === '__more__') {
+            olderOffset += PAGE_SIZE;
+            continue;
+        }
+        if (selection === '__reset__') {
+            olderOffset = 0;
+            continue;
+        }
+        if (selection === '__ask__') {
+            await askOracleFlow(version, userConfig);
+            continue;
+        }
+        await showSessionDetail(selection);
+    }
+}
+async function fetchSessionBuckets(olderOffset) {
+    const all = await listSessionsMetadata();
+    const cutoff = Date.now() - RECENT_WINDOW_HOURS * 60 * 60 * 1000;
+    const recent = all.filter((meta) => new Date(meta.createdAt).getTime() >= cutoff).slice(0, PAGE_SIZE);
+    const olderAll = all.filter((meta) => new Date(meta.createdAt).getTime() < cutoff);
+    const older = olderAll.slice(olderOffset, olderOffset + PAGE_SIZE);
+    const hasMoreOlder = olderAll.length > olderOffset + PAGE_SIZE;
+    if (recent.length === 0 && older.length === 0 && olderAll.length > 0) {
+        // No recent entries; fall back to top 10 overall.
+        return { recent: olderAll.slice(0, PAGE_SIZE), older: [], hasMoreOlder: olderAll.length > PAGE_SIZE };
+    }
+    return { recent, older, hasMoreOlder };
+}
+function toSessionChoice(meta) {
+    return {
+        name: formatSessionLabel(meta),
+        value: meta.id,
+    };
+}
+function formatSessionLabel(meta) {
+    const status = colorStatus(meta.status ?? 'unknown');
+    const created = formatTimestampAligned(meta.createdAt);
+    const model = meta.model ?? 'n/a';
+    const mode = meta.mode ?? meta.options?.mode ?? 'api';
+    const slug = meta.id;
+    const chars = meta.options?.prompt?.length ?? meta.promptPreview?.length ?? 0;
+    const charLabel = chars > 0 ? chalk.gray(String(chars).padStart(5)) : chalk.gray('    -');
+    const cost = mode === 'browser' ? null : resolveCost(meta);
+    const costLabel = cost != null ? chalk.gray(formatCostTable(cost)) : chalk.gray('      -');
+    return `${status} ${chalk.white(model.padEnd(10))} ${chalk.gray(mode.padEnd(7))} ${chalk.gray(created)} ${charLabel} ${costLabel}  ${chalk.cyan(slug)}`;
+}
+function resolveCost(meta) {
+    if (meta.usage?.cost != null) {
+        return meta.usage.cost;
+    }
+    if (!meta.model || !meta.usage) {
+        return null;
+    }
+    const pricing = MODEL_CONFIGS[meta.model]?.pricing;
+    if (!pricing)
+        return null;
+    const input = meta.usage.inputTokens ?? 0;
+    const output = meta.usage.outputTokens ?? 0;
+    const cost = input * pricing.inputPerToken + output * pricing.outputPerToken;
+    return cost > 0 ? cost : null;
+}
+function formatCostTable(cost) {
+    return `$${cost.toFixed(3)}`.padStart(7);
+}
+function formatTimestampAligned(iso) {
+    const date = new Date(iso);
+    const locale = 'en-US';
+    const opts = {
+        year: 'numeric',
+        month: '2-digit',
+        day: '2-digit',
+        hour: 'numeric',
+        minute: '2-digit',
+        second: undefined,
+        hour12: true,
+    };
+    const formatted = date.toLocaleString(locale, opts);
+    // Insert a leading space when hour is a single digit to align AM/PM column.
+    // Example: "11/18/2025, 1:07:05 AM" -> "11/18/2025,  1:07:05 AM"
+    return formatted.replace(/(, )(\d:)/, '$1 $2');
+}
+function colorStatus(status) {
+    const padded = status.padEnd(9);
+    switch (status) {
+        case 'completed':
+            return chalk.green(padded);
+        case 'error':
+            return chalk.red(padded);
+        case 'running':
+            return chalk.yellow(padded);
+        default:
+            return padded;
+    }
+}
+async function showSessionDetail(sessionId) {
+    for (;;) {
+        const meta = await readSessionMetadataSafe(sessionId);
+        if (!meta) {
+            console.log(chalk.red(`No session found with ID ${sessionId}`));
+            return;
+        }
+        console.clear();
+        printSessionHeader(meta);
+        const prompt = await readStoredPrompt(sessionId);
+        if (prompt) {
+            console.log(chalk.bold('Prompt:'));
+            console.log(renderMarkdownAnsi(prompt));
+            console.log(dim('---'));
+        }
+        const logPath = await getSessionLogPath(sessionId);
+        if (logPath) {
+            console.log(dim(`Log file: ${logPath}`));
+        }
+        console.log('');
+        await renderSessionLog(sessionId);
+        const isRunning = meta.status === 'running';
+        const actions = [
+            ...(isRunning ? [{ name: 'Refresh', value: 'refresh' }] : []),
+            { name: 'Back', value: 'back' },
+        ];
+        const { next } = await inquirer.prompt([
+            {
+                name: 'next',
+                type: 'list',
+                message: 'Actions',
+                choices: actions,
+            },
+        ]);
+        if (next === 'back') {
+            return;
+        }
+        // refresh loops
+    }
+}
+async function renderSessionLog(sessionId) {
+    const raw = await readSessionLog(sessionId);
+    const text = trimBeforeFirstAnswer(raw);
+    const size = Buffer.byteLength(text, 'utf8');
+    if (size > MAX_RENDER_BYTES) {
+        console.log(chalk.yellow(`Log is large (${size.toLocaleString()} bytes). Rendering raw text; open the log file for full context.`));
+        process.stdout.write(text);
+        console.log('');
+        return;
+    }
+    process.stdout.write(renderMarkdownAnsi(text));
+    console.log('');
+}
+async function getSessionLogPath(sessionId) {
+    try {
+        const paths = await getSessionPaths(sessionId);
+        return paths.log;
+    }
+    catch {
+        return null;
+    }
+}
+function printSessionHeader(meta) {
+    console.log(chalk.bold(`Session ${chalk.cyan(meta.id)}`));
+    console.log(`${chalk.white('Status:')} ${meta.status}`);
+    console.log(`${chalk.white('Created:')} ${meta.createdAt}`);
+    if (meta.model) {
+        console.log(`${chalk.white('Model:')} ${meta.model}`);
+    }
+    const mode = meta.mode ?? meta.options?.mode;
+    if (mode) {
+        console.log(`${chalk.white('Mode:')} ${mode}`);
+    }
+    if (meta.errorMessage) {
+        console.log(chalk.red(`Error: ${meta.errorMessage}`));
+    }
+}
+async function askOracleFlow(version, userConfig) {
+    const modelChoices = Object.keys(MODEL_CONFIGS);
+    const hasApiKey = Boolean(process.env.OPENAI_API_KEY);
+    const initialMode = hasApiKey ? 'api' : 'browser';
+    const preferredMode = userConfig.engine ?? initialMode;
+    const answers = await inquirer.prompt([
+        {
+            name: 'promptInput',
+            type: 'input',
+            message: 'Paste your prompt text or a path to a file (leave blank to cancel):',
+        },
+        ...(hasApiKey
+            ? [
+                {
+                    name: 'mode',
+                    type: 'list',
+                    message: 'Engine',
+                    default: preferredMode,
+                    choices: [
+                        { name: 'API', value: 'api' },
+                        { name: 'Browser', value: 'browser' },
+                    ],
+                },
+            ]
+            : [
+                {
+                    name: 'mode',
+                    type: 'list',
+                    message: 'Engine',
+                    default: preferredMode,
+                    choices: [{ name: 'Browser', value: 'browser' }],
+                },
+            ]),
+        {
+            name: 'slug',
+            type: 'input',
+            message: 'Optional slug (3–5 words, leave blank for auto):',
+        },
+        {
+            name: 'model',
+            type: 'list',
+            message: 'Model',
+            default: 'gpt-5-pro',
+            choices: modelChoices,
+        },
+        {
+            name: 'files',
+            type: 'input',
+            message: 'Files or globs to attach (comma-separated, optional):',
+            filter: (value) => value
+                .split(',')
+                .map((entry) => entry.trim())
+                .filter(Boolean),
+        },
+        {
+            name: 'chromeProfile',
+            type: 'input',
+            message: 'Chrome profile to reuse cookies from:',
+            default: 'Default',
+            when: (ans) => ans.mode === 'browser',
+        },
+        {
+            name: 'headless',
+            type: 'confirm',
+            message: 'Run Chrome headless?',
+            default: false,
+            when: (ans) => ans.mode === 'browser',
+        },
+        {
+            name: 'hideWindow',
+            type: 'confirm',
+            message: 'Hide Chrome window (macOS headful only)?',
+            default: false,
+            when: (ans) => ans.mode === 'browser',
+        },
+        {
+            name: 'keepBrowser',
+            type: 'confirm',
+            message: 'Keep browser open after completion?',
+            default: false,
+            when: (ans) => ans.mode === 'browser',
+        },
+    ]);
+    const mode = (answers.mode ?? initialMode);
+    const prompt = await resolvePromptInput(answers.promptInput);
+    if (!prompt.trim()) {
+        console.log(chalk.yellow('Cancelled.'));
+        return;
+    }
+    const promptWithSuffix = userConfig.promptSuffix ? `${prompt.trim()}\n${userConfig.promptSuffix}` : prompt;
+    await ensureSessionStorage();
+    const runOptions = {
+        prompt: promptWithSuffix,
+        model: answers.model,
+        file: answers.files,
+        slug: answers.slug,
+        filesReport: false,
+        maxInput: undefined,
+        maxOutput: undefined,
+        system: undefined,
+        silent: false,
+        search: undefined,
+        preview: false,
+        previewMode: undefined,
+        apiKey: undefined,
+        sessionId: undefined,
+        verbose: false,
+        heartbeatIntervalMs: undefined,
+        browserInlineFiles: false,
+        browserBundleFiles: false,
+        background: undefined,
+    };
+    const browserConfig = mode === 'browser'
+        ? buildBrowserConfig({
+            browserChromeProfile: answers.chromeProfile,
+            browserHeadless: answers.headless,
+            browserHideWindow: answers.hideWindow,
+            browserKeepBrowser: answers.keepBrowser,
+            browserModelLabel: resolveBrowserModelLabel(undefined, answers.model),
+            model: answers.model,
+        })
+        : undefined;
+    const notifications = resolveNotificationSettings({
+        cliNotify: undefined,
+        cliNotifySound: undefined,
+        env: process.env,
+        config: userConfig.notify,
+    });
+    const sessionMeta = await initializeSession({
+        ...runOptions,
+        mode,
+        browserConfig,
+    }, process.cwd(), notifications);
+    const { logLine, writeChunk, stream } = createSessionLogWriter(sessionMeta.id);
+    const combinedLog = (message) => {
+        if (message) {
+            console.log(message);
+            logLine(message);
+        }
+    };
+    const combinedWrite = (chunk) => {
+        writeChunk(chunk);
+        return process.stdout.write(chunk);
+    };
+    console.log(chalk.bold(`Session ${sessionMeta.id} starting...`));
+    console.log(dim(`Log path: ${path.join(os.homedir(), '.oracle', 'sessions', sessionMeta.id, 'output.log')}`));
+    try {
+        await performSessionRun({
+            sessionMeta,
+            runOptions: { ...runOptions, sessionId: sessionMeta.id },
+            mode,
+            browserConfig,
+            cwd: process.cwd(),
+            log: combinedLog,
+            write: combinedWrite,
+            version,
+            notifications,
+        });
+        console.log(chalk.green(`Session ${sessionMeta.id} completed.`));
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        console.log(chalk.red(`Session ${sessionMeta.id} failed: ${message}`));
+    }
+    finally {
+        stream.end();
+    }
+}
+const readSessionMetadataSafe = (sessionId) => readSessionMetadata(sessionId);
+async function resolvePromptInput(rawInput) {
+    const trimmed = rawInput.trim();
+    if (!trimmed) {
+        return trimmed;
+    }
+    const asPath = path.resolve(process.cwd(), trimmed);
+    try {
+        const stats = await fs.stat(asPath);
+        if (stats.isFile()) {
+            const contents = await fs.readFile(asPath, 'utf8');
+            return contents;
+        }
+    }
+    catch {
+        // not a file; fall through
+    }
+    return trimmed;
+}
+async function readStoredPrompt(sessionId) {
+    const request = await readSessionRequest(sessionId);
+    if (request?.prompt && request.prompt.trim().length > 0) {
+        return request.prompt;
+    }
+    const meta = await readSessionMetadata(sessionId);
+    if (meta?.options?.prompt && meta.options.prompt.trim().length > 0) {
+        return meta.options.prompt;
+    }
+    return null;
+}
+// Exported for testing
+export { askOracleFlow, showSessionDetail };
+export { resolveCost };

package/dist/src/config.js

@@ -0,0 +1,27 @@
+import fs from 'node:fs/promises';
+import os from 'node:os';
+import path from 'node:path';
+import JSON5 from 'json5';
+function resolveConfigPath() {
+    const oracleHome = process.env.ORACLE_HOME_DIR ?? path.join(os.homedir(), '.oracle');
+    return path.join(oracleHome, 'config.json');
+}
+export async function loadUserConfig() {
+    const CONFIG_PATH = resolveConfigPath();
+    try {
+        const raw = await fs.readFile(CONFIG_PATH, 'utf8');
+        const parsed = JSON5.parse(raw);
+        return { config: parsed ?? {}, path: CONFIG_PATH, loaded: true };
+    }
+    catch (error) {
+        const code = error.code;
+        if (code === 'ENOENT') {
+            return { config: {}, path: CONFIG_PATH, loaded: false };
+        }
+        console.warn(`Failed to read ${CONFIG_PATH}: ${error instanceof Error ? error.message : String(error)}`);
+        return { config: {}, path: CONFIG_PATH, loaded: false };
+    }
+}
+export function configPath() {
+    return resolveConfigPath();
+}

package/dist/src/mcp/server.js

@@ -0,0 +1,36 @@
+#!/usr/bin/env node
+import 'dotenv/config';
+import process from 'node:process';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { getCliVersion } from '../version.js';
+import { registerConsultTool } from './tools/consult.js';
+import { registerSessionsTool } from './tools/sessions.js';
+import { registerSessionResources } from './tools/sessionResources.js';
+export async function startMcpServer() {
+    const server = new McpServer({
+        name: 'oracle-mcp',
+        version: getCliVersion(),
+    }, {
+        capabilities: {
+            logging: {},
+        },
+    });
+    registerConsultTool(server);
+    registerSessionsTool(server);
+    registerSessionResources(server);
+    const transport = new StdioServerTransport();
+    transport.onerror = (error) => {
+        console.error('MCP transport error:', error);
+    };
+    transport.onclose = () => {
+        // Keep quiet on normal close; caller owns lifecycle.
+    };
+    await server.connect(transport);
+}
+if (import.meta.url === `file://${process.argv[1]}` || process.argv[1]?.endsWith('oracle-mcp')) {
+    startMcpServer().catch((error) => {
+        console.error('Failed to start oracle-mcp:', error);
+        process.exitCode = 1;
+    });
+}

package/dist/src/mcp/tools/consult.js

@@ -0,0 +1,158 @@
+import { z } from 'zod';
+import { getCliVersion } from '../../version.js';
+import { LoggingMessageNotificationParamsSchema } from '@modelcontextprotocol/sdk/types.js';
+import { ensureBrowserAvailable, mapConsultToRunOptions } from '../utils.js';
+import { createSessionLogWriter, initializeSession, readSessionMetadata, readSessionLog, } from '../../sessionManager.js';
+async function readSessionLogTail(sessionId, maxBytes) {
+    try {
+        const log = await readSessionLog(sessionId);
+        if (log.length <= maxBytes) {
+            return log;
+        }
+        return log.slice(-maxBytes);
+    }
+    catch {
+        return null;
+    }
+}
+import { performSessionRun } from '../../cli/sessionRunner.js';
+import { CHATGPT_URL } from '../../browser/constants.js';
+import { consultInputSchema } from '../types.js';
+import { loadUserConfig } from '../../config.js';
+import { resolveNotificationSettings } from '../../cli/notifier.js';
+import { mapModelToBrowserLabel, resolveBrowserModelLabel } from '../../cli/browserConfig.js';
+// Use raw shapes so the MCP SDK (with its bundled Zod) wraps them and emits valid JSON Schema.
+const consultInputShape = {
+    prompt: z.string().min(1, 'Prompt is required.'),
+    files: z.array(z.string()).default([]),
+    model: z.string().optional(),
+    engine: z.enum(['api', 'browser']).optional(),
+    browserModelLabel: z.string().optional(),
+    search: z.boolean().optional(),
+    slug: z.string().optional(),
+};
+const consultOutputShape = {
+    sessionId: z.string(),
+    status: z.string(),
+    output: z.string(),
+};
+export function registerConsultTool(server) {
+    server.registerTool('consult', {
+        title: 'Run an oracle session',
+        description: 'Run a one-shot Oracle session (API or browser). Attach files/dirs for context, optional model/engine overrides, and an optional slug. Background handling follows the CLI defaults; browser runs only start when Chrome is available.',
+        // Cast to any to satisfy SDK typings across differing Zod versions.
+        inputSchema: consultInputShape,
+        outputSchema: consultOutputShape,
+    }, async (input) => {
+        const textContent = (text) => [{ type: 'text', text }];
+        const { prompt, files, model, engine, search, browserModelLabel, slug } = consultInputSchema.parse(input);
+        const { config: userConfig } = await loadUserConfig();
+        const { runOptions, resolvedEngine } = mapConsultToRunOptions({
+            prompt,
+            files: files ?? [],
+            model,
+            engine,
+            search,
+            userConfig,
+            env: process.env,
+        });
+        const cwd = process.cwd();
+        const browserGuard = ensureBrowserAvailable(resolvedEngine);
+        if (resolvedEngine === 'browser' &&
+            (browserGuard ||
+                (process.platform === 'linux' && !process.env.DISPLAY && !process.env.CHROME_PATH))) {
+            return {
+                isError: true,
+                content: textContent(browserGuard ?? 'Browser engine unavailable: set DISPLAY or CHROME_PATH.'),
+            };
+        }
+        let browserConfig;
+        if (resolvedEngine === 'browser') {
+            const preferredLabel = (browserModelLabel ?? model)?.trim();
+            const desiredModelLabel = resolveBrowserModelLabel(preferredLabel, runOptions.model);
+            // Keep the browser path minimal; only forward a desired model label for the ChatGPT picker.
+            browserConfig = {
+                url: CHATGPT_URL,
+                cookieSync: true,
+                headless: false,
+                hideWindow: false,
+                keepBrowser: false,
+                desiredModel: desiredModelLabel || mapModelToBrowserLabel(runOptions.model),
+            };
+        }
+        const notifications = resolveNotificationSettings({
+            cliNotify: undefined,
+            cliNotifySound: undefined,
+            env: process.env,
+            config: userConfig.notify,
+        });
+        const sessionMeta = await initializeSession({
+            ...runOptions,
+            mode: resolvedEngine,
+            slug,
+            browserConfig,
+        }, cwd, notifications);
+        const logWriter = createSessionLogWriter(sessionMeta.id);
+        // Best-effort: emit MCP logging notifications for live chunks but never block the run.
+        const sendLog = (text, level = 'info') => server.server
+            .sendLoggingMessage(LoggingMessageNotificationParamsSchema.parse({
+            level,
+            data: { text, bytes: Buffer.byteLength(text, 'utf8') },
+        }))
+            .catch(() => { });
+        // Stream logs to both the session log and MCP logging notifications, but avoid buffering in memory
+        const log = (line) => {
+            logWriter.logLine(line);
+            if (line !== undefined) {
+                sendLog(line);
+            }
+        };
+        const write = (chunk) => {
+            logWriter.writeChunk(chunk);
+            sendLog(chunk, 'debug');
+            return true;
+        };
+        try {
+            await performSessionRun({
+                sessionMeta,
+                runOptions,
+                mode: resolvedEngine,
+                browserConfig,
+                cwd,
+                log,
+                write,
+                version: getCliVersion(),
+                notifications,
+            });
+        }
+        catch (error) {
+            log(`Run failed: ${error instanceof Error ? error.message : String(error)}`);
+            return {
+                isError: true,
+                content: textContent(`Session ${sessionMeta.id} failed: ${error instanceof Error ? error.message : String(error)}`),
+            };
+        }
+        finally {
+            logWriter.stream.end();
+        }
+        try {
+            const finalMeta = (await readSessionMetadata(sessionMeta.id)) ?? sessionMeta;
+            const summary = `Session ${sessionMeta.id} (${finalMeta.status})`;
+            const logTail = await readSessionLogTail(sessionMeta.id, 4000);
+            return {
+                content: textContent([summary, logTail || '(log empty)'].join('\n').trim()),
+                structuredContent: {
+                    sessionId: sessionMeta.id,
+                    status: finalMeta.status,
+                    output: logTail ?? '',
+                },
+            };
+        }
+        catch (error) {
+            return {
+                isError: true,
+                content: textContent(`Session completed but metadata fetch failed: ${error instanceof Error ? error.message : String(error)}`),
+            };
+        }
+    });
+}