@eminent337/aery 0.1.11 → 1.0.0

package/dist/main.js

@@ -1,589 +0,0 @@
-/**
- * Main entry point for the coding agent CLI.
- *
- * This file handles CLI argument parsing and translates them into
- * createAgentSession() options. The SDK does the heavy lifting.
- */
-import { resolve } from "node:path";
-import { createInterface } from "node:readline";
-import { modelsAreEqual, supportsXhigh } from "@aryee/aery-ai";
-import { ProcessTerminal, setKeybindings, TUI } from "@aryee/aery-tui";
-import chalk from "chalk";
-import { parseArgs, printHelp } from "./cli/args.js";
-import { processFileArguments } from "./cli/file-processor.js";
-import { buildInitialMessage } from "./cli/initial-message.js";
-import { listModels } from "./cli/list-models.js";
-import { selectSession } from "./cli/session-picker.js";
-import { getAgentDir, getModelsPath, VERSION } from "./config.js";
-import { createAgentSessionRuntime } from "./core/agent-session-runtime.js";
-import { createAgentSessionFromServices, createAgentSessionServices, } from "./core/agent-session-services.js";
-import { AuthStorage } from "./core/auth-storage.js";
-import { exportFromFile } from "./core/export-html/index.js";
-import { KeybindingsManager } from "./core/keybindings.js";
-import { resolveCliModel, resolveModelScope } from "./core/model-resolver.js";
-import { restoreStdout, takeOverStdout } from "./core/output-guard.js";
-import { formatMissingSessionCwdPrompt, getMissingSessionCwdIssue, MissingSessionCwdError, } from "./core/session-cwd.js";
-import { SessionManager } from "./core/session-manager.js";
-import { SettingsManager } from "./core/settings-manager.js";
-import { printTimings, resetTimings, time } from "./core/timings.js";
-import { runMigrations, showDeprecationWarnings } from "./migrations.js";
-import { InteractiveMode, runPrintMode, runRpcMode } from "./modes/index.js";
-import { ExtensionSelectorComponent } from "./modes/interactive/components/extension-selector.js";
-import { initTheme, stopThemeWatcher } from "./modes/interactive/theme/theme.js";
-import { handleConfigCommand, handlePackageCommand } from "./package-manager-cli.js";
-import { isLocalPath } from "./utils/paths.js";
-/**
- * Read all content from piped stdin.
- * Returns undefined if stdin is a TTY (interactive terminal).
- */
-async function readPipedStdin() {
-    // If stdin is a TTY, we're running interactively - don't read stdin
-    if (process.stdin.isTTY) {
-        return undefined;
-    }
-    return new Promise((resolve) => {
-        let data = "";
-        process.stdin.setEncoding("utf8");
-        process.stdin.on("data", (chunk) => {
-            data += chunk;
-        });
-        process.stdin.on("end", () => {
-            resolve(data.trim() || undefined);
-        });
-        process.stdin.resume();
-    });
-}
-function collectSettingsDiagnostics(settingsManager, context) {
-    return settingsManager.drainErrors().map(({ scope, error }) => ({
-        type: "warning",
-        message: `(${context}, ${scope} settings) ${error.message}`,
-    }));
-}
-function reportDiagnostics(diagnostics) {
-    for (const diagnostic of diagnostics) {
-        const color = diagnostic.type === "error" ? chalk.red : diagnostic.type === "warning" ? chalk.yellow : chalk.dim;
-        const prefix = diagnostic.type === "error" ? "Error: " : diagnostic.type === "warning" ? "Warning: " : "";
-        console.error(color(`${prefix}${diagnostic.message}`));
-    }
-}
-function isTruthyEnvFlag(value) {
-    if (!value)
-        return false;
-    return value === "1" || value.toLowerCase() === "true" || value.toLowerCase() === "yes";
-}
-function resolveAppMode(parsed, stdinIsTTY) {
-    if (parsed.mode === "rpc") {
-        return "rpc";
-    }
-    if (parsed.mode === "json") {
-        return "json";
-    }
-    if (parsed.print || !stdinIsTTY) {
-        return "print";
-    }
-    return "interactive";
-}
-function toPrintOutputMode(appMode) {
-    return appMode === "json" ? "json" : "text";
-}
-async function prepareInitialMessage(parsed, autoResizeImages, stdinContent) {
-    if (parsed.fileArgs.length === 0) {
-        return buildInitialMessage({ parsed, stdinContent });
-    }
-    const { text, images } = await processFileArguments(parsed.fileArgs, { autoResizeImages });
-    return buildInitialMessage({
-        parsed,
-        fileText: text,
-        fileImages: images,
-        stdinContent,
-    });
-}
-/**
- * Resolve a session argument to a file path.
- * If it looks like a path, use as-is. Otherwise try to match as session ID prefix.
- */
-async function resolveSessionPath(sessionArg, cwd, sessionDir) {
-    // If it looks like a file path, use as-is
-    if (sessionArg.includes("/") || sessionArg.includes("\\") || sessionArg.endsWith(".jsonl")) {
-        return { type: "path", path: sessionArg };
-    }
-    // Try to match as session ID in current project first
-    const localSessions = await SessionManager.list(cwd, sessionDir);
-    const localMatches = localSessions.filter((s) => s.id.startsWith(sessionArg));
-    if (localMatches.length >= 1) {
-        return { type: "local", path: localMatches[0].path };
-    }
-    // Try global search across all projects
-    const allSessions = await SessionManager.listAll();
-    const globalMatches = allSessions.filter((s) => s.id.startsWith(sessionArg));
-    if (globalMatches.length >= 1) {
-        const match = globalMatches[0];
-        return { type: "global", path: match.path, cwd: match.cwd };
-    }
-    // Not found anywhere
-    return { type: "not_found", arg: sessionArg };
-}
-/** Prompt user for yes/no confirmation */
-async function promptConfirm(message) {
-    return new Promise((resolve) => {
-        const rl = createInterface({
-            input: process.stdin,
-            output: process.stdout,
-        });
-        rl.question(`${message} [y/N] `, (answer) => {
-            rl.close();
-            resolve(answer.toLowerCase() === "y" || answer.toLowerCase() === "yes");
-        });
-    });
-}
-function validateForkFlags(parsed) {
-    if (!parsed.fork)
-        return;
-    const conflictingFlags = [
-        parsed.session ? "--session" : undefined,
-        parsed.continue ? "--continue" : undefined,
-        parsed.resume ? "--resume" : undefined,
-        parsed.noSession ? "--no-session" : undefined,
-    ].filter((flag) => flag !== undefined);
-    if (conflictingFlags.length > 0) {
-        console.error(chalk.red(`Error: --fork cannot be combined with ${conflictingFlags.join(", ")}`));
-        process.exit(1);
-    }
-}
-function forkSessionOrExit(sourcePath, cwd, sessionDir) {
-    try {
-        return SessionManager.forkFrom(sourcePath, cwd, sessionDir);
-    }
-    catch (error) {
-        const message = error instanceof Error ? error.message : String(error);
-        console.error(chalk.red(`Error: ${message}`));
-        process.exit(1);
-    }
-}
-async function createSessionManager(parsed, cwd, sessionDir, settingsManager) {
-    if (parsed.noSession) {
-        return SessionManager.inMemory();
-    }
-    if (parsed.fork) {
-        const resolved = await resolveSessionPath(parsed.fork, cwd, sessionDir);
-        switch (resolved.type) {
-            case "path":
-            case "local":
-            case "global":
-                return forkSessionOrExit(resolved.path, cwd, sessionDir);
-            case "not_found":
-                console.error(chalk.red(`No session found matching '${resolved.arg}'`));
-                process.exit(1);
-        }
-    }
-    if (parsed.session) {
-        const resolved = await resolveSessionPath(parsed.session, cwd, sessionDir);
-        switch (resolved.type) {
-            case "path":
-            case "local":
-                return SessionManager.open(resolved.path, sessionDir);
-            case "global": {
-                console.log(chalk.yellow(`Session found in different project: ${resolved.cwd}`));
-                const shouldFork = await promptConfirm("Fork this session into current directory?");
-                if (!shouldFork) {
-                    console.log(chalk.dim("Aborted."));
-                    process.exit(0);
-                }
-                return forkSessionOrExit(resolved.path, cwd, sessionDir);
-            }
-            case "not_found":
-                console.error(chalk.red(`No session found matching '${resolved.arg}'`));
-                process.exit(1);
-        }
-    }
-    if (parsed.resume) {
-        initTheme(settingsManager.getTheme(), true);
-        try {
-            const selectedPath = await selectSession((onProgress) => SessionManager.list(cwd, sessionDir, onProgress), SessionManager.listAll);
-            if (!selectedPath) {
-                console.log(chalk.dim("No session selected"));
-                process.exit(0);
-            }
-            return SessionManager.open(selectedPath, sessionDir);
-        }
-        finally {
-            stopThemeWatcher();
-        }
-    }
-    if (parsed.continue) {
-        return SessionManager.continueRecent(cwd, sessionDir);
-    }
-    return SessionManager.create(cwd, sessionDir);
-}
-function buildSessionOptions(parsed, scopedModels, hasExistingSession, modelRegistry, settingsManager) {
-    const options = {};
-    const diagnostics = [];
-    let cliThinkingFromModel = false;
-    // Model from CLI
-    // - supports --provider <name> --model <pattern>
-    // - supports --model <provider>/<pattern>
-    if (parsed.model) {
-        const resolved = resolveCliModel({
-            cliProvider: parsed.provider,
-            cliModel: parsed.model,
-            modelRegistry,
-        });
-        if (resolved.warning) {
-            diagnostics.push({ type: "warning", message: resolved.warning });
-        }
-        if (resolved.error) {
-            diagnostics.push({ type: "error", message: resolved.error });
-        }
-        if (resolved.model) {
-            options.model = resolved.model;
-            // Allow "--model <pattern>:<thinking>" as a shorthand.
-            // Explicit --thinking still takes precedence (applied later).
-            if (!parsed.thinking && resolved.thinkingLevel) {
-                options.thinkingLevel = resolved.thinkingLevel;
-                cliThinkingFromModel = true;
-            }
-        }
-    }
-    if (!options.model && scopedModels.length > 0 && !hasExistingSession) {
-        // Check if saved default is in scoped models - use it if so, otherwise first scoped model
-        const savedProvider = settingsManager.getDefaultProvider();
-        const savedModelId = settingsManager.getDefaultModel();
-        const savedModel = savedProvider && savedModelId ? modelRegistry.find(savedProvider, savedModelId) : undefined;
-        const savedInScope = savedModel ? scopedModels.find((sm) => modelsAreEqual(sm.model, savedModel)) : undefined;
-        if (savedInScope) {
-            options.model = savedInScope.model;
-            // Use thinking level from scoped model config if explicitly set
-            if (!parsed.thinking && savedInScope.thinkingLevel) {
-                options.thinkingLevel = savedInScope.thinkingLevel;
-            }
-        }
-        else {
-            options.model = scopedModels[0].model;
-            // Use thinking level from first scoped model if explicitly set
-            if (!parsed.thinking && scopedModels[0].thinkingLevel) {
-                options.thinkingLevel = scopedModels[0].thinkingLevel;
-            }
-        }
-    }
-    // Thinking level from CLI (takes precedence over scoped model thinking levels set above)
-    if (parsed.thinking) {
-        options.thinkingLevel = parsed.thinking;
-    }
-    // Scoped models for Ctrl+P cycling
-    // Keep thinking level undefined when not explicitly set in the model pattern.
-    // Undefined means "inherit current session thinking level" during cycling.
-    if (scopedModels.length > 0) {
-        options.scopedModels = scopedModels.map((sm) => ({
-            model: sm.model,
-            thinkingLevel: sm.thinkingLevel,
-        }));
-    }
-    // API key from CLI - set in authStorage
-    // (handled by caller before createAgentSession)
-    // Tools
-    if (parsed.noTools) {
-        // --no-tools: start with no built-in tools
-        // --tools can still add specific ones back, including extension tools.
-        options.tools = parsed.tools && parsed.tools.length > 0 ? [...parsed.tools] : [];
-    }
-    else if (parsed.tools) {
-        options.tools = [...parsed.tools];
-    }
-    return { options, cliThinkingFromModel, diagnostics };
-}
-function resolveCliPaths(cwd, paths) {
-    return paths?.map((value) => (isLocalPath(value) ? resolve(cwd, value) : value));
-}
-async function promptForMissingSessionCwd(issue, settingsManager) {
-    initTheme(settingsManager.getTheme());
-    setKeybindings(KeybindingsManager.create());
-    return new Promise((resolve) => {
-        const ui = new TUI(new ProcessTerminal(), settingsManager.getShowHardwareCursor());
-        ui.setClearOnShrink(settingsManager.getClearOnShrink());
-        let settled = false;
-        const finish = (result) => {
-            if (settled) {
-                return;
-            }
-            settled = true;
-            ui.stop();
-            resolve(result);
-        };
-        const selector = new ExtensionSelectorComponent(formatMissingSessionCwdPrompt(issue), ["Continue", "Cancel"], (option) => finish(option === "Continue" ? issue.fallbackCwd : undefined), () => finish(undefined), { tui: ui });
-        ui.addChild(selector);
-        ui.setFocus(selector);
-        ui.start();
-    });
-}
-export async function main(args, options) {
-    resetTimings();
-    const offlineMode = args.includes("--offline") || isTruthyEnvFlag(process.env.PI_OFFLINE);
-    if (offlineMode) {
-        process.env.PI_OFFLINE = "1";
-        process.env.PI_SKIP_VERSION_CHECK = "1";
-    }
-    if (await handlePackageCommand(args)) {
-        return;
-    }
-    if (await handleConfigCommand(args)) {
-        return;
-    }
-    const parsed = parseArgs(args);
-    if (parsed.diagnostics.length > 0) {
-        for (const d of parsed.diagnostics) {
-            const color = d.type === "error" ? chalk.red : chalk.yellow;
-            console.error(color(`${d.type === "error" ? "Error" : "Warning"}: ${d.message}`));
-        }
-        if (parsed.diagnostics.some((d) => d.type === "error")) {
-            process.exit(1);
-        }
-    }
-    time("parseArgs");
-    let appMode = resolveAppMode(parsed, process.stdin.isTTY);
-    const shouldTakeOverStdout = appMode !== "interactive";
-    if (shouldTakeOverStdout) {
-        takeOverStdout();
-    }
-    if (parsed.version) {
-        console.log(VERSION);
-        process.exit(0);
-    }
-    if (parsed.export) {
-        let result;
-        try {
-            const outputPath = parsed.messages.length > 0 ? parsed.messages[0] : undefined;
-            result = await exportFromFile(parsed.export, outputPath);
-        }
-        catch (error) {
-            const message = error instanceof Error ? error.message : "Failed to export session";
-            console.error(chalk.red(`Error: ${message}`));
-            process.exit(1);
-        }
-        console.log(`Exported to: ${result}`);
-        process.exit(0);
-    }
-    if (parsed.mode === "rpc" && parsed.fileArgs.length > 0) {
-        console.error(chalk.red("Error: @file arguments are not supported in RPC mode"));
-        process.exit(1);
-    }
-    validateForkFlags(parsed);
-    // Run migrations (pass cwd for project-local migrations)
-    const { migratedAuthProviders: migratedProviders, deprecationWarnings } = runMigrations(process.cwd());
-    time("runMigrations");
-    const cwd = process.cwd();
-    const agentDir = getAgentDir();
-    const startupSettingsManager = SettingsManager.create(cwd, agentDir);
-    reportDiagnostics(collectSettingsDiagnostics(startupSettingsManager, "startup session lookup"));
-    // Decide the final runtime cwd before creating cwd-bound runtime services.
-    // --session and --resume may select a session from another project, so project-local
-    // settings, resources, provider registrations, and models must be resolved only after
-    // the target session cwd is known. The startup-cwd settings manager is used only for
-    // sessionDir lookup during session selection.
-    const sessionDir = parsed.sessionDir ?? startupSettingsManager.getSessionDir();
-    let sessionManager = await createSessionManager(parsed, cwd, sessionDir, startupSettingsManager);
-    const missingSessionCwdIssue = getMissingSessionCwdIssue(sessionManager, cwd);
-    if (missingSessionCwdIssue) {
-        if (appMode === "interactive") {
-            const selectedCwd = await promptForMissingSessionCwd(missingSessionCwdIssue, startupSettingsManager);
-            if (!selectedCwd) {
-                process.exit(0);
-            }
-            sessionManager = SessionManager.open(missingSessionCwdIssue.sessionFile, sessionDir, selectedCwd);
-        }
-        else {
-            console.error(chalk.red(new MissingSessionCwdError(missingSessionCwdIssue).message));
-            process.exit(1);
-        }
-    }
-    time("createSessionManager");
-    const resolvedExtensionPaths = resolveCliPaths(cwd, parsed.extensions);
-    const resolvedSkillPaths = resolveCliPaths(cwd, parsed.skills);
-    const resolvedPromptTemplatePaths = resolveCliPaths(cwd, parsed.promptTemplates);
-    const resolvedThemePaths = resolveCliPaths(cwd, parsed.themes);
-    const authStorage = AuthStorage.create();
-    const createRuntime = async ({ cwd, agentDir, sessionManager, sessionStartEvent, }) => {
-        const services = await createAgentSessionServices({
-            cwd,
-            agentDir,
-            authStorage,
-            extensionFlagValues: parsed.unknownFlags,
-            resourceLoaderOptions: {
-                additionalExtensionPaths: resolvedExtensionPaths,
-                additionalSkillPaths: resolvedSkillPaths,
-                additionalPromptTemplatePaths: resolvedPromptTemplatePaths,
-                additionalThemePaths: resolvedThemePaths,
-                noExtensions: parsed.noExtensions,
-                noSkills: parsed.noSkills,
-                noPromptTemplates: parsed.noPromptTemplates,
-                noThemes: parsed.noThemes,
-                noContextFiles: parsed.noContextFiles,
-                systemPrompt: parsed.systemPrompt,
-                appendSystemPrompt: parsed.appendSystemPrompt,
-                extensionFactories: options?.extensionFactories,
-            },
-        });
-        const { settingsManager, modelRegistry, resourceLoader } = services;
-        const diagnostics = [
-            ...services.diagnostics,
-            ...collectSettingsDiagnostics(settingsManager, "runtime creation"),
-            ...resourceLoader.getExtensions().errors.map(({ path, error }) => ({
-                type: "error",
-                message: `Failed to load extension "${path}": ${error}`,
-            })),
-        ];
-        const modelPatterns = parsed.models ?? settingsManager.getEnabledModels();
-        const scopedModels = modelPatterns && modelPatterns.length > 0 ? await resolveModelScope(modelPatterns, modelRegistry) : [];
-        const { options: sessionOptions, cliThinkingFromModel, diagnostics: sessionOptionDiagnostics, } = buildSessionOptions(parsed, scopedModels, sessionManager.buildSessionContext().messages.length > 0, modelRegistry, settingsManager);
-        diagnostics.push(...sessionOptionDiagnostics);
-        if (parsed.apiKey) {
-            if (!sessionOptions.model) {
-                diagnostics.push({
-                    type: "error",
-                    message: "--api-key requires a model to be specified via --model, --provider/--model, or --models",
-                });
-            }
-            else {
-                authStorage.setRuntimeApiKey(sessionOptions.model.provider, parsed.apiKey);
-            }
-        }
-        const created = await createAgentSessionFromServices({
-            services,
-            sessionManager,
-            sessionStartEvent,
-            model: sessionOptions.model,
-            thinkingLevel: sessionOptions.thinkingLevel,
-            scopedModels: sessionOptions.scopedModels,
-            tools: sessionOptions.tools,
-            customTools: sessionOptions.customTools,
-        });
-        const cliThinkingOverride = parsed.thinking !== undefined || cliThinkingFromModel;
-        if (created.session.model && cliThinkingOverride) {
-            let effectiveThinking = created.session.thinkingLevel;
-            if (!created.session.model.reasoning) {
-                effectiveThinking = "off";
-            }
-            else if (effectiveThinking === "xhigh" && !supportsXhigh(created.session.model)) {
-                effectiveThinking = "high";
-            }
-            if (effectiveThinking !== created.session.thinkingLevel) {
-                created.session.setThinkingLevel(effectiveThinking);
-            }
-        }
-        return {
-            ...created,
-            services,
-            diagnostics,
-        };
-    };
-    time("createRuntime");
-    const runtime = await createAgentSessionRuntime(createRuntime, {
-        cwd: sessionManager.getCwd(),
-        agentDir,
-        sessionManager,
-    });
-    const { services, session, modelFallbackMessage } = runtime;
-    const { settingsManager, modelRegistry, resourceLoader } = services;
-    if (parsed.help) {
-        const extensionFlags = resourceLoader
-            .getExtensions()
-            .extensions.flatMap((extension) => Array.from(extension.flags.values()));
-        printHelp(extensionFlags);
-        process.exit(0);
-    }
-    if (parsed.listModels !== undefined) {
-        const searchPattern = typeof parsed.listModels === "string" ? parsed.listModels : undefined;
-        await listModels(modelRegistry, searchPattern);
-        process.exit(0);
-    }
-    // Read piped stdin content (if any) - skip for RPC mode which uses stdin for JSON-RPC
-    let stdinContent;
-    if (appMode !== "rpc") {
-        stdinContent = await readPipedStdin();
-        if (stdinContent !== undefined && appMode === "interactive") {
-            appMode = "print";
-        }
-    }
-    time("readPipedStdin");
-    const { initialMessage, initialImages } = await prepareInitialMessage(parsed, settingsManager.getImageAutoResize(), stdinContent);
-    time("prepareInitialMessage");
-    initTheme(settingsManager.getTheme(), appMode === "interactive");
-    time("initTheme");
-    // Show deprecation warnings in interactive mode
-    if (appMode === "interactive" && deprecationWarnings.length > 0) {
-        await showDeprecationWarnings(deprecationWarnings);
-    }
-    const scopedModels = [...session.scopedModels];
-    time("resolveModelScope");
-    reportDiagnostics(runtime.diagnostics);
-    if (runtime.diagnostics.some((diagnostic) => diagnostic.type === "error")) {
-        process.exit(1);
-    }
-    time("createAgentSession");
-    if (appMode !== "interactive" && !session.model) {
-        console.error(chalk.red("No models available."));
-        console.error(chalk.yellow("\nSet an API key environment variable:"));
-        console.error("  ANTHROPIC_API_KEY, OPENAI_API_KEY, GEMINI_API_KEY, etc.");
-        console.error(chalk.yellow(`\nOr create ${getModelsPath()}`));
-        process.exit(1);
-    }
-    const startupBenchmark = isTruthyEnvFlag(process.env.PI_STARTUP_BENCHMARK);
-    if (startupBenchmark && appMode !== "interactive") {
-        console.error(chalk.red("Error: PI_STARTUP_BENCHMARK only supports interactive mode"));
-        process.exit(1);
-    }
-    if (appMode === "rpc") {
-        printTimings();
-        await runRpcMode(runtime);
-    }
-    else if (appMode === "interactive") {
-        if (scopedModels.length > 0 && (parsed.verbose || !settingsManager.getQuietStartup())) {
-            const modelList = scopedModels
-                .map((sm) => {
-                const thinkingStr = sm.thinkingLevel ? `:${sm.thinkingLevel}` : "";
-                return `${sm.model.id}${thinkingStr}`;
-            })
-                .join(", ");
-            console.log(chalk.dim(`Model scope: ${modelList} ${chalk.gray("(Ctrl+P to cycle)")}`));
-        }
-        const interactiveMode = new InteractiveMode(runtime, {
-            migratedProviders,
-            modelFallbackMessage,
-            initialMessage,
-            initialImages,
-            initialMessages: parsed.messages,
-            verbose: parsed.verbose,
-        });
-        if (startupBenchmark) {
-            await interactiveMode.init();
-            time("interactiveMode.init");
-            printTimings();
-            interactiveMode.stop();
-            stopThemeWatcher();
-            if (process.stdout.writableLength > 0) {
-                await new Promise((resolve) => process.stdout.once("drain", resolve));
-            }
-            if (process.stderr.writableLength > 0) {
-                await new Promise((resolve) => process.stderr.once("drain", resolve));
-            }
-            return;
-        }
-        printTimings();
-        await interactiveMode.run();
-    }
-    else {
-        printTimings();
-        const exitCode = await runPrintMode(runtime, {
-            mode: toPrintOutputMode(appMode),
-            messages: parsed.messages,
-            initialMessage,
-            initialImages,
-        });
-        stopThemeWatcher();
-        restoreStdout();
-        if (exitCode !== 0) {
-            process.exitCode = exitCode;
-        }
-        return;
-    }
-}
-//# sourceMappingURL=main.js.map