tg-agent 0.1.0 → 1.0.1

package/dist/index.js

@@ -1,954 +1,14 @@
-import TelegramBot from "node-telegram-bot-api";
-import { discoverAuthStorage, discoverModels } from "@mariozechner/pi-coding-agent";
-import { getOAuthProviders } from "@mariozechner/pi-ai";
-import { config, assertConfig } from "./config.js";
-import { resolveApiKeyForProvider, resolveProxyInfo } from "./auth.js";
-import { applyFetchProxy } from "./proxy.js";
-import { runPiAgentPrompt } from "./piAgentRunner.js";
-import { readCodexOAuth } from "./codexAuth.js";
-import { formatMcpTarget, loadMcpServers, loadMcpServersSync, probeMcpServer } from "./mcp.js";
-import { appendMessage, closeSession, createSession, deleteSessionFile, getActiveSession, listSessions, loadUserState, pruneExpiredSessions, resetSession, saveUserState, setActiveSession, } from "./sessionStore.js";
-import { chunkText, createQueueMap, createSemaphore, ensureDir, nowMs } from "./utils.js";
-assertConfig();
-const bot = new TelegramBot(config.telegramToken, { polling: true });
-const enqueueUser = createQueueMap();
-const withSemaphore = createSemaphore(config.maxConcurrent);
-const activeRuns = new Map();
-function registerActiveRun(userId, run) {
-    activeRuns.set(userId, run);
-}
-function clearActiveRun(userId) {
-    activeRuns.delete(userId);
-}
-function resolveTelegramParseMode(value) {
-    const trimmed = value.trim();
-    if (!trimmed) {
-        return undefined;
-    }
-    if (trimmed === "Markdown" || trimmed === "MarkdownV2" || trimmed === "HTML") {
-        return trimmed;
-    }
-    console.warn(`[tg-agent] invalid TELEGRAM_PARSE_MODE=${trimmed}, ignoring`);
-    return undefined;
-}
-const MARKDOWN_V2_RESERVED = new Set([
-    "_",
-    "*",
-    "[",
-    "]",
-    "(",
-    ")",
-    "~",
-    "`",
-    ">",
-    "#",
-    "+",
-    "-",
-    "=",
-    "|",
-    "{",
-    "}",
-    ".",
-    "!",
-]);
-function countUnescaped(text, target) {
-    let count = 0;
-    for (let i = 0; i < text.length; i += 1) {
-        const char = text[i];
-        if (char === "\\") {
-            i += 1;
-            continue;
-        }
-        if (char === target) {
-            count += 1;
-        }
-    }
-    return count;
-}
-function isMarkdownSafe(text) {
-    const stars = countUnescaped(text, "*");
-    if (stars % 2 !== 0)
-        return false;
-    const underscores = countUnescaped(text, "_");
-    if (underscores % 2 !== 0)
-        return false;
-    const backticks = countUnescaped(text, "`");
-    if (backticks % 2 !== 0)
-        return false;
-    const left = countUnescaped(text, "[");
-    const right = countUnescaped(text, "]");
-    if (left !== right)
-        return false;
-    return true;
-}
-function isMarkdownV2Safe(text) {
-    for (let i = 0; i < text.length; i += 1) {
-        const char = text[i];
-        if (char === "\\") {
-            i += 1;
-            continue;
-        }
-        if (MARKDOWN_V2_RESERVED.has(char)) {
-            return false;
-        }
-    }
-    return true;
-}
-function resolveBestParseModes(text) {
-    const configured = resolveTelegramParseMode(config.telegramParseMode);
-    if (configured) {
-        return [configured];
-    }
-    if (isMarkdownV2Safe(text)) {
-        return ["MarkdownV2", "Markdown"];
-    }
-    if (isMarkdownSafe(text)) {
-        return ["Markdown"];
-    }
-    return [];
-}
-const AUTH_PROMPT_TIMEOUT_MS = 10 * 60 * 1000;
-const authPromptState = new Map();
-const authInFlight = new Set();
-const PROVIDER_ALIASES = new Map([
-    ["codex", "openai-codex"],
-    ["antigravity", "google-antigravity"],
-    ["gemini", "google-gemini-cli"],
-    ["gemini-cli", "google-gemini-cli"],
-]);
-const CODEX_MODEL_ALIASES = {
-    "codex-mini-latest": "gpt-5.1-codex-mini",
-    "codex-max-latest": "gpt-5.1-codex-max",
-    "codex-latest": "gpt-5.2-codex",
-};
-function normalizeProviderId(input) {
-    const trimmed = input.trim().toLowerCase();
-    return PROVIDER_ALIASES.get(trimmed) ?? trimmed;
-}
-function normalizeModelId(provider, modelId) {
-    if (provider !== "openai-codex") {
-        return modelId;
-    }
-    return CODEX_MODEL_ALIASES[modelId] ?? modelId;
-}
-async function loadModelRegistry() {
-    await ensureDir(config.agentDir);
-    const authStorage = discoverAuthStorage(config.agentDir);
-    const codex = readCodexOAuth();
-    if (codex) {
-        authStorage.setRuntimeApiKey("openai-codex", codex.accessToken);
-    }
-    const modelRegistry = discoverModels(authStorage, config.agentDir);
-    return { authStorage, modelRegistry };
-}
-function clearAuthPrompt(userId) {
-    const pending = authPromptState.get(userId);
-    if (!pending)
-        return;
-    clearTimeout(pending.timeoutId);
-    authPromptState.delete(userId);
-}
-function cancelAuthPrompt(userId, reason) {
-    const pending = authPromptState.get(userId);
-    if (!pending)
-        return;
-    clearTimeout(pending.timeoutId);
-    authPromptState.delete(userId);
-    pending.reject(new Error(reason));
-}
-async function promptForAuthInput(userId, chatId, prompt) {
-    if (authPromptState.has(userId)) {
-        throw new Error("Login is already awaiting input.");
-    }
-    const text = prompt.placeholder ? `${prompt.message} (${prompt.placeholder})` : prompt.message;
-    await sendLongMessage(chatId, text);
-    return await new Promise((resolve, reject) => {
-        const timeoutId = setTimeout(() => {
-            authPromptState.delete(userId);
-            reject(new Error("Login prompt timed out."));
-        }, AUTH_PROMPT_TIMEOUT_MS);
-        authPromptState.set(userId, { resolve, reject, timeoutId, chatId });
-    });
-}
-async function handleStopCommand(chatId, userId) {
-    const active = activeRuns.get(userId);
-    if (!active) {
-        await sendLongMessage(chatId, "No active request to stop.");
-        return;
-    }
-    active.cancelRequested = true;
-    active.abortRequested = true;
-    if (active.status) {
-        active.status.update("Cancelled by user.", true);
-    }
-    if (active.abort) {
-        active.abort();
-    }
-    if (!active.status || active.chatId !== chatId) {
-        await sendLongMessage(chatId, "Stopping current request...");
-    }
-}
-function logStartupInfo() {
-    const proxyVars = {
-        https_proxy: process.env.https_proxy ?? process.env.HTTPS_PROXY ?? "",
-        http_proxy: process.env.http_proxy ?? process.env.HTTP_PROXY ?? "",
-        all_proxy: process.env.all_proxy ?? process.env.ALL_PROXY ?? "",
-        no_proxy: process.env.no_proxy ?? process.env.NO_PROXY ?? "",
-    };
-    const tlsVars = {
-        NODE_EXTRA_CA_CERTS: process.env.NODE_EXTRA_CA_CERTS ?? "",
-        NODE_TLS_REJECT_UNAUTHORIZED: process.env.NODE_TLS_REJECT_UNAUTHORIZED ?? "",
-    };
-    console.log(`[tg-agent] modelProvider=${config.modelProvider} modelRef=${config.modelRef} sessionDir=${config.sessionDir} maxConcurrent=${config.maxConcurrent}`);
-    console.log(`[tg-agent] agentDir=${config.agentDir} workspaceDir=${config.workspaceDir}`);
-    const mcpServers = loadMcpServersSync(config.agentDir);
-    const mcpStatus = mcpServers.length > 0 ? `on (${mcpServers.length})` : "off";
-    console.log(`[tg-agent] tools fetchMaxBytes=${config.fetchMaxBytes} fetchTimeoutMs=${config.fetchTimeoutMs} mcp=${mcpStatus}`);
-    console.log(`[tg-agent] proxy https=${proxyVars.https_proxy || "(empty)"} http=${proxyVars.http_proxy || "(empty)"} all=${proxyVars.all_proxy || "(empty)"} no=${proxyVars.no_proxy || "(empty)"}`);
-    console.log(`[tg-agent] tls extra_ca=${tlsVars.NODE_EXTRA_CA_CERTS || "(empty)"} reject_unauthorized=${tlsVars.NODE_TLS_REJECT_UNAUTHORIZED || "(empty)"}`);
-    try {
-        const { source } = resolveApiKeyForProvider(config.modelProvider);
-        console.log(`[tg-agent] authSource=${source}`);
-    }
-    catch (error) {
-        const message = error instanceof Error ? error.message : String(error);
-        console.warn(`[tg-agent] authSource=missing (${message})`);
-    }
-    const proxyInfo = resolveProxyInfo();
-    if (proxyInfo) {
-        console.log(`[tg-agent] proxyUrl=${proxyInfo.url} kind=${proxyInfo.kind} source=${proxyInfo.source}`);
-    }
-    else {
-        console.log("[tg-agent] proxyUrl=(none)");
-    }
-}
-logStartupInfo();
-void ensureDir(config.agentDir).catch((error) => {
-    console.warn(`[tg-agent] ensure agentDir failed: ${error instanceof Error ? error.message : String(error)}`);
-});
-const appliedProxy = applyFetchProxy();
-if (appliedProxy) {
-    console.log(`[tg-agent] fetchProxy=${appliedProxy.url} kind=${appliedProxy.kind} source=${appliedProxy.source}`);
-}
-else {
-    console.log("[tg-agent] fetchProxy=(none)");
-}
-function parseCommand(text) {
-    if (!text.startsWith("/")) {
-        return null;
-    }
-    const trimmed = text.trim();
-    const [rawCommand, ...rest] = trimmed.split(" ");
-    const command = rawCommand.split("@")[0].slice(1).toLowerCase();
-    return { command, args: rest.join(" ").trim() };
-}
-async function ensureActiveSessionForCommand(state, chatId) {
-    let session = getActiveSession(state);
-    if (!session) {
-        session = createSession(state, "");
-        await saveUserState(state);
-        await sendLongMessage(chatId, `Created session ${session.id}.`);
-    }
-    return session;
-}
-function formatProviders(modelRegistry, authStorage) {
-    const models = modelRegistry.getAll();
-    const providerStats = new Map();
-    for (const model of models) {
-        const entry = providerStats.get(model.provider) ?? { count: 0, auth: authStorage.hasAuth(model.provider) };
-        entry.count += 1;
-        providerStats.set(model.provider, entry);
-    }
-    if (providerStats.size === 0) {
-        return "No providers found.";
-    }
-    const lines = Array.from(providerStats.entries())
-        .sort((a, b) => a[0].localeCompare(b[0]))
-        .map(([provider, info]) => `- ${provider} (models: ${info.count}, auth: ${info.auth ? "ok" : "missing"})`);
-    return ["Providers:", ...lines, "", "Use /models <provider> to list models."].join("\n");
-}
-function formatModelsForProvider(modelRegistry, provider, authOk) {
-    const models = modelRegistry.getAll().filter((model) => model.provider === provider);
-    if (models.length === 0) {
-        return `No models found for provider ${provider}.`;
-    }
-    const lines = models.map((model) => `- ${model.id} | ${model.name}`);
-    return [`Models for ${provider} (auth: ${authOk ? "ok" : "missing"}):`, ...lines].join("\n");
-}
-function parseModelSelector(raw) {
-    const trimmed = raw.trim();
-    if (!trimmed)
-        return null;
-    if (trimmed.includes("/")) {
-        const [provider, modelId] = trimmed.split("/", 2);
-        if (!provider || !modelId)
-            return null;
-        return { provider: normalizeProviderId(provider), modelId: modelId.trim() };
-    }
-    return { modelId: trimmed };
-}
-const unauthorizedLogWindowMs = 60_000;
-const unauthorizedLogState = new Map();
-function logUnauthorizedUser(userId, chatId) {
-    const now = Date.now();
-    const lastLoggedAt = unauthorizedLogState.get(userId) ?? 0;
-    if (now - lastLoggedAt < unauthorizedLogWindowMs) {
-        return;
-    }
-    unauthorizedLogState.set(userId, now);
-    console.warn(`[tg-agent] unauthorized user=${userId} chat=${chatId}`);
-}
-function isUserAllowed(userId) {
-    const allowed = config.telegramAllowedUsers;
-    if (!allowed || allowed.size === 0) {
-        return true;
-    }
-    return allowed.has(userId);
-}
-function formatSessions(stateSessions) {
-    if (stateSessions.length === 0) {
-        return "No sessions found.";
-    }
-    const lines = stateSessions.map((session) => {
-        const updated = new Date(session.updatedAt).toISOString();
-        return `- ${session.id} | ${session.title} | updated ${updated}`;
-    });
-    return ["Sessions:", ...lines].join("\n");
-}
-async function sendTelegramMessage(chatId, text, parseMode) {
-    if (parseMode) {
-        return await bot.sendMessage(chatId, text, { parse_mode: parseMode });
-    }
-    return await bot.sendMessage(chatId, text);
-}
-async function editTelegramMessage(chatId, messageId, text, parseMode) {
-    if (parseMode) {
-        await bot.editMessageText(text, { chat_id: chatId, message_id: messageId, parse_mode: parseMode });
-        return;
-    }
-    await bot.editMessageText(text, { chat_id: chatId, message_id: messageId });
-}
-async function sendWithParseModes(attempt, parseModes) {
-    let lastError;
-    for (const mode of parseModes) {
-        try {
-            return await attempt(mode);
-        }
-        catch (error) {
-            lastError = error;
-            const message = error instanceof Error ? error.message : String(error);
-            console.warn(`[tg-agent] parse_mode ${mode} failed, trying fallback: ${message}`);
-        }
-    }
-    try {
-        return await attempt(undefined);
-    }
-    catch (error) {
-        if (lastError) {
-            const message = error instanceof Error ? error.message : String(error);
-            console.warn(`[tg-agent] parse_mode fallback to plain failed: ${message}`);
-        }
-        throw error;
-    }
-}
-async function sendTelegramMessageBest(chatId, text) {
-    return await sendWithParseModes((parseMode) => sendTelegramMessage(chatId, text, parseMode), resolveBestParseModes(text));
-}
-async function editTelegramMessageBest(chatId, messageId, text) {
-    await sendWithParseModes((parseMode) => editTelegramMessage(chatId, messageId, text, parseMode), resolveBestParseModes(text));
-}
-async function sendLongMessage(chatId, text) {
-    const chunks = chunkText(text, 3900);
-    for (const chunk of chunks) {
-        await sendTelegramMessageBest(chatId, chunk);
-    }
-}
-const TELEGRAM_MAX_LEN = 3900;
-const STATUS_UPDATE_MIN_MS = 1200;
-function truncateForTelegram(text, maxLen = TELEGRAM_MAX_LEN) {
-    if (text.length <= maxLen) {
-        return { text, truncated: false };
-    }
-    const slice = text.slice(0, Math.max(0, maxLen - 15));
-    return { text: `${slice}...\n\n[truncated]`, truncated: true };
-}
-async function createStatusReporter(chatId) {
-    const initial = "Working...";
-    const sent = await sendTelegramMessage(chatId, initial);
-    const messageId = sent.message_id;
-    let lastText = initial;
-    let lastSentAt = 0;
-    let chain = Promise.resolve();
-    const enqueueEdit = (text, force = false) => {
-        const now = Date.now();
-        if (!force && now - lastSentAt < STATUS_UPDATE_MIN_MS)
-            return;
-        if (text === lastText)
-            return;
-        lastText = text;
-        lastSentAt = now;
-        chain = chain
-            .then(async () => {
-            await editTelegramMessage(chatId, messageId, text);
-        })
-            .catch((error) => {
-            console.warn("[tg-agent] status edit failed", error);
-        });
-    };
-    const enqueueEditBest = (text) => {
-        lastText = text;
-        lastSentAt = Date.now();
-        chain = chain
-            .then(async () => {
-            await editTelegramMessageBest(chatId, messageId, text);
-        })
-            .catch((error) => {
-            console.warn("[tg-agent] status edit failed", error);
-        });
-    };
-    return {
-        update: (text, force = false) => {
-            const trimmed = truncateForTelegram(text).text;
-            enqueueEdit(trimmed, force);
-        },
-        finalize: async (text) => {
-            const trimmed = truncateForTelegram(text).text;
-            enqueueEditBest(trimmed);
-            await chain;
-        },
-        fail: async (text) => {
-            const trimmed = truncateForTelegram(text).text;
-            enqueueEditBest(trimmed);
-            await chain;
-        },
-    };
-}
-function formatStatusEvent(event) {
-    switch (event.type) {
-        case "agent_start":
-            return "Working...";
-        case "tool_start":
-            return `Running tool: ${event.name}\nargs: ${summarizeArgs(event.args)}`;
-        case "tool_end":
-            return event.ok
-                ? `Tool finished: ${event.name} (${event.durationMs}ms)`
-                : `Tool failed: ${event.name} (${event.durationMs}ms)`;
-        case "message_start":
-            return event.role === "assistant" ? "Generating reply..." : null;
-        case "message_end":
-            return event.role === "assistant" ? "Finalizing reply..." : null;
-        case "heartbeat": {
-            const seconds = Math.max(1, Math.round(event.elapsedMs / 1000));
-            return `Working... ${seconds}s`;
-        }
-        default:
-            return null;
-    }
-}
-function summarizeArgs(args) {
-    try {
-        const seen = new WeakSet();
-        const maxString = 200;
-        const maxKeys = 12;
-        const maxArray = 12;
-        const maxDepth = 3;
-        const normalize = (value, depth) => {
-            if (value === null || value === undefined)
-                return value;
-            if (typeof value === "string") {
-                return value.length > maxString ? `${value.slice(0, maxString)}...` : value;
-            }
-            if (typeof value === "number" || typeof value === "boolean")
-                return value;
-            if (typeof value === "bigint")
-                return value.toString();
-            if (typeof value === "function")
-                return "[function]";
-            if (typeof value !== "object")
-                return String(value);
-            if (seen.has(value))
-                return "[circular]";
-            if (depth >= maxDepth)
-                return "[truncated]";
-            seen.add(value);
-            if (Array.isArray(value)) {
-                const items = value.slice(0, maxArray).map((item) => normalize(item, depth + 1));
-                if (value.length > maxArray)
-                    items.push("[truncated]");
-                return items;
-            }
-            const entries = Object.entries(value);
-            const result = {};
-            for (const [key, val] of entries.slice(0, maxKeys)) {
-                result[key] = normalize(val, depth + 1);
-            }
-            if (entries.length > maxKeys) {
-                result._truncated = true;
-            }
-            return result;
-        };
-        const normalized = normalize(args, 0);
-        const json = JSON.stringify(normalized);
-        if (!json)
-            return "null";
-        if (json.length > 700) {
-            return `${json.slice(0, 700)}...`;
-        }
-        return json;
-    }
-    catch (error) {
-        return "[unavailable]";
-    }
-}
-function formatErrorMessage(error) {
-    const base = error instanceof Error ? error.message : String(error);
-    const cause = error?.cause;
-    if (cause instanceof Error) {
-        return `${base} (${cause.message})`;
-    }
-    if (cause) {
-        return `${base} (${String(cause)})`;
-    }
-    return base;
-}
-function truncateLine(text, maxLen = 140) {
-    if (text.length <= maxLen) {
-        return text;
-    }
-    return `${text.slice(0, maxLen - 3)}...`;
-}
-async function formatMcpStatus() {
-    const servers = await loadMcpServers(config.agentDir);
-    if (servers.length === 0) {
-        return "No MCP servers configured. Add [mcp_servers.*] to ~/.tg-agent/config.toml.";
-    }
-    const probes = await Promise.all(servers.map((server) => probeMcpServer(server)));
-    const lines = ["MCP servers:"];
-    for (let i = 0; i < servers.length; i += 1) {
-        const server = servers[i];
-        const probe = probes[i];
-        const status = probe.ok ? "ok" : `error: ${truncateLine(probe.error ?? "unknown")}`;
-        lines.push(`- ${server.name} (${server.type}) ${formatMcpTarget(server)} status=${status} (${probe.durationMs}ms)`);
-    }
-    return lines.join("\n");
-}
-async function handleCommand(chatId, userId, command, args) {
-    const state = await loadUserState(userId);
-    const removed = pruneExpiredSessions(state);
-    if (removed.length > 0) {
-        await saveUserState(state);
-        await Promise.all(removed.map((sessionId) => deleteSessionFile(userId, sessionId).catch((error) => {
-            console.warn(`[tg-agent] cleanup session file failed id=${sessionId}`, error);
-        })));
-    }
-    switch (command) {
-        case "start":
-        case "help": {
-            const helpText = [
-                "Commands:",
-                "/new [title] - create a new session",
-                "/list - list sessions",
-                "/use <id> - switch active session",
-                "/close [id] - close a session (default: active)",
-                "/reset - clear active session history",
-                "/providers - list available providers",
-                "/models [provider] - list models for provider",
-                "/provider <name> - set provider for current session",
-                "/model <provider>/<model> - set model for current session",
-                "/mcp - list configured MCP servers",
-                "/status - show session model settings",
-                "/login <provider> - login to OAuth provider",
-                "/logout <provider> - logout from provider",
-                "/stop - stop the current running request",
-            ].join("\n");
-            await sendLongMessage(chatId, helpText);
-            return;
-        }
-        case "new": {
-            try {
-                const title = args || "";
-                const session = createSession(state, title);
-                await saveUserState(state);
-                await sendLongMessage(chatId, `Created session ${session.id} (${session.title}).`);
-            }
-            catch (error) {
-                await sendLongMessage(chatId, error.message);
-            }
-            return;
-        }
-        case "list": {
-            await saveUserState(state);
-            await sendLongMessage(chatId, formatSessions(listSessions(state)));
-            return;
-        }
-        case "use": {
-            if (!args) {
-                await sendLongMessage(chatId, "Usage: /use <id>");
-                return;
-            }
-            const ok = setActiveSession(state, args);
-            if (!ok) {
-                await sendLongMessage(chatId, `Session not found: ${args}`);
-                return;
-            }
-            await saveUserState(state);
-            await sendLongMessage(chatId, `Active session set to ${args}.`);
-            return;
-        }
-        case "close": {
-            const target = args || state.activeSessionId;
-            if (!target) {
-                await sendLongMessage(chatId, "No active session to close.");
-                return;
-            }
-            const ok = closeSession(state, target);
-            if (!ok) {
-                await sendLongMessage(chatId, `Session not found: ${target}`);
-                return;
-            }
-            await saveUserState(state);
-            await deleteSessionFile(userId, target).catch((error) => {
-                console.warn(`[tg-agent] delete session file failed id=${target}`, error);
-            });
-            await sendLongMessage(chatId, `Closed session ${target}.`);
-            return;
-        }
-        case "reset": {
-            const active = getActiveSession(state);
-            if (!active) {
-                await sendLongMessage(chatId, "No active session.");
-                return;
-            }
-            resetSession(active);
-            await saveUserState(state);
-            await deleteSessionFile(userId, active.id).catch((error) => {
-                console.warn(`[tg-agent] reset session file failed id=${active.id}`, error);
-            });
-            await sendLongMessage(chatId, `Reset session ${active.id}.`);
-            return;
-        }
-        case "providers": {
-            const { authStorage, modelRegistry } = await loadModelRegistry();
-            const error = modelRegistry.getError();
-            const text = formatProviders(modelRegistry, authStorage);
-            const message = error ? `Warning: ${error}\n\n${text}` : text;
-            await sendLongMessage(chatId, message);
-            return;
-        }
-        case "models": {
-            const { authStorage, modelRegistry } = await loadModelRegistry();
-            let provider = args ? normalizeProviderId(args) : "";
-            if (!provider) {
-                const active = getActiveSession(state);
-                if (!active?.modelProvider) {
-                    await sendLongMessage(chatId, "Usage: /models <provider>");
-                    return;
-                }
-                provider = active.modelProvider;
-            }
-            const authOk = authStorage.hasAuth(provider);
-            await sendLongMessage(chatId, formatModelsForProvider(modelRegistry, provider, authOk));
-            return;
-        }
-        case "provider": {
-            if (!args) {
-                await sendLongMessage(chatId, "Usage: /provider <name>");
-                return;
-            }
-            const provider = normalizeProviderId(args);
-            const { modelRegistry } = await loadModelRegistry();
-            const exists = modelRegistry.getAll().some((model) => model.provider === provider);
-            if (!exists) {
-                await sendLongMessage(chatId, `Unknown provider: ${provider}`);
-                return;
-            }
-            const session = await ensureActiveSessionForCommand(state, chatId);
-            session.modelProvider = provider;
-            session.modelId = undefined;
-            session.updatedAt = nowMs();
-            await saveUserState(state);
-            await sendLongMessage(chatId, `Session ${session.id} provider set to ${provider}.`);
-            return;
-        }
-        case "model": {
-            const selector = parseModelSelector(args);
-            if (!selector) {
-                await sendLongMessage(chatId, "Usage: /model <provider>/<model>");
-                return;
-            }
-            const session = await ensureActiveSessionForCommand(state, chatId);
-            const provider = normalizeProviderId(selector.provider ?? session.modelProvider ?? "");
-            if (!provider) {
-                await sendLongMessage(chatId, "Usage: /model <provider>/<model>");
-                return;
-            }
-            const { modelRegistry } = await loadModelRegistry();
-            const normalizedModelId = normalizeModelId(provider, selector.modelId);
-            const model = modelRegistry.find(provider, normalizedModelId);
-            if (!model) {
-                await sendLongMessage(chatId, `Model not found: ${provider}/${normalizedModelId}`);
-                return;
-            }
-            session.modelProvider = provider;
-            session.modelId = normalizedModelId;
-            session.updatedAt = nowMs();
-            await saveUserState(state);
-            await sendLongMessage(chatId, `Session ${session.id} model set to ${provider}/${normalizedModelId}.`);
-            return;
-        }
-        case "status": {
-            const active = getActiveSession(state);
-            if (!active) {
-                await sendLongMessage(chatId, "No active session.");
-                return;
-            }
-            const defaultModel = config.modelRef.includes("/")
-                ? config.modelRef
-                : `${config.modelProvider}/${config.modelRef}`;
-            const overrides = active.modelProvider || active.modelId
-                ? `Session model: ${active.modelProvider ?? "?"}/${active.modelId ?? "?"}`
-                : `Default model: ${defaultModel}`;
-            const message = [
-                `Session: ${active.id} (${active.title})`,
-                overrides,
-            ].join("\n");
-            await sendLongMessage(chatId, message);
-            return;
-        }
-        case "mcp": {
-            const message = await formatMcpStatus();
-            await sendLongMessage(chatId, message);
-            return;
-        }
-        case "login": {
-            const providers = getOAuthProviders();
-            if (!args) {
-                const lines = providers.map((p) => `- ${p.id} (${p.name})`);
-                await sendLongMessage(chatId, ["OAuth providers:", ...lines].join("\n"));
-                return;
-            }
-            const provider = normalizeProviderId(args);
-            if (authInFlight.has(userId)) {
-                await sendLongMessage(chatId, "Login already in progress.");
-                return;
-            }
-            const providerInfo = providers.find((p) => p.id === provider);
-            if (!providerInfo) {
-                await sendLongMessage(chatId, `Unknown OAuth provider: ${provider}`);
-                return;
-            }
-            authInFlight.add(userId);
-            try {
-                const { authStorage } = await loadModelRegistry();
-                await sendLongMessage(chatId, `Starting login for ${providerInfo.id}...`);
-                await authStorage.login(providerInfo.id, {
-                    onAuth: ({ url, instructions }) => {
-                        const lines = [
-                            "Open this URL in your browser:",
-                            url,
-                        ];
-                        if (instructions) {
-                            lines.push("", instructions);
-                        }
-                        void sendLongMessage(chatId, lines.join("\n"));
-                    },
-                    onPrompt: (prompt) => promptForAuthInput(userId, chatId, prompt),
-                    onProgress: (message) => {
-                        void sendLongMessage(chatId, message);
-                    },
-                    onManualCodeInput: () => promptForAuthInput(userId, chatId, { message: "Paste the authorization code:" }),
-                });
-                await sendLongMessage(chatId, `Login completed for ${providerInfo.id}.`);
-            }
-            catch (error) {
-                const message = formatErrorMessage(error);
-                await sendLongMessage(chatId, `Login failed: ${message}`);
-            }
-            finally {
-                authInFlight.delete(userId);
-                clearAuthPrompt(userId);
-            }
-            return;
-        }
-        case "logout": {
-            if (!args) {
-                await sendLongMessage(chatId, "Usage: /logout <provider>");
-                return;
-            }
-            const provider = normalizeProviderId(args);
-            const { authStorage } = await loadModelRegistry();
-            authStorage.logout(provider);
-            await sendLongMessage(chatId, `Logged out from ${provider}.`);
-            return;
-        }
-        default: {
-            await sendLongMessage(chatId, `Unknown command: ${command}`);
-        }
-    }
-}
-async function handleMessage(chatId, userId, text) {
-    const command = parseCommand(text);
-    if (command) {
-        await handleCommand(chatId, userId, command.command, command.args);
-        return;
-    }
-    const state = await loadUserState(userId);
-    const expired = pruneExpiredSessions(state);
-    if (expired.length > 0) {
-        await saveUserState(state);
-        await Promise.all(expired.map((sessionId) => deleteSessionFile(userId, sessionId).catch((error) => {
-            console.warn(`[tg-agent] cleanup session file failed id=${sessionId}`, error);
-        })));
-    }
-    let session = getActiveSession(state);
-    if (!session) {
-        try {
-            session = createSession(state, "");
-            await saveUserState(state);
-            await sendLongMessage(chatId, `Created session ${session.id}.`);
-        }
-        catch (error) {
-            await sendLongMessage(chatId, error.message);
-            return;
-        }
-    }
-    const userMessage = {
-        role: "user",
-        content: text,
-        ts: nowMs(),
-    };
-    appendMessage(session, userMessage, config.maxHistoryMessages);
-    await saveUserState(state);
-    console.log(`[tg-agent] request user=${userId} session=${session.id} messages=${session.messages.length} textLen=${text.length} provider=${session.modelProvider ?? "-"} model=${session.modelId ?? "-"}`);
-    let statusReporter = null;
-    try {
-        statusReporter = await createStatusReporter(chatId);
-    }
-    catch (error) {
-        console.warn("[tg-agent] status message failed", error);
-    }
-    const activeRun = {
-        sessionId: session.id,
-        chatId,
-        status: statusReporter,
-        abortRequested: false,
-        cancelRequested: false,
-    };
-    registerActiveRun(userId, activeRun);
-    try {
-        const assistantText = await withSemaphore(async () => {
-            const result = await runPiAgentPrompt({
-                userId,
-                sessionId: session.id,
-                prompt: text,
-                systemPrompt: config.systemPrompt,
-                modelProvider: session.modelProvider,
-                modelId: session.modelId,
-                onAbortReady: (abort) => {
-                    activeRun.abort = abort;
-                    if (activeRun.abortRequested) {
-                        abort();
-                    }
-                },
-                onStatus: statusReporter
-                    ? (event) => {
-                        const statusText = formatStatusEvent(event);
-                        if (statusText) {
-                            statusReporter?.update(statusText);
-                        }
-                    }
-                    : undefined,
-            });
-            console.log(`[tg-agent] response session=${session.id} model=${result.modelProvider}/${result.modelId} file=${result.sessionFile}`);
-            return result.text;
-        });
-        if (activeRun.cancelRequested) {
-            if (statusReporter) {
-                await statusReporter.finalize("Cancelled by user.");
-            }
-            else {
-                await sendLongMessage(chatId, "Cancelled by user.");
-            }
-            return;
-        }
-        const assistantMessage = {
-            role: "assistant",
-            content: assistantText,
-            ts: nowMs(),
-        };
-        appendMessage(session, assistantMessage, config.maxHistoryMessages);
-        await saveUserState(state);
-        if (statusReporter) {
-            await statusReporter.finalize(assistantText);
-        }
-        else {
-            await sendLongMessage(chatId, assistantText);
-        }
-    }
-    catch (error) {
-        console.error("[tg-agent] runModel error", error);
-        if (activeRun.cancelRequested) {
-            if (statusReporter) {
-                await statusReporter.fail("Cancelled by user.");
-            }
-            else {
-                await sendLongMessage(chatId, "Cancelled by user.");
-            }
-            return;
-        }
-        const message = formatErrorMessage(error);
-        if (statusReporter) {
-            await statusReporter.fail(`Error: ${message}`);
-        }
-        else {
-            await sendLongMessage(chatId, `Error: ${message}`);
-        }
-    }
-    finally {
-        clearActiveRun(userId);
-    }
-}
-bot.on("message", (msg) => {
-    if (!msg.text) {
-        return;
-    }
-    if (msg.chat.type !== "private") {
-        void bot.sendMessage(msg.chat.id, "This bot only supports direct messages.");
-        return;
-    }
-    if (!msg.from?.id) {
-        return;
-    }
-    const userId = String(msg.from.id);
-    const chatId = msg.chat.id;
-    if (!isUserAllowed(userId)) {
-        logUnauthorizedUser(userId, chatId);
-        return;
-    }
-    const text = msg.text.trim();
-    const pendingAuth = authPromptState.get(userId);
-    if (pendingAuth) {
-        if (text === "/stop" || text === "/cancel") {
-            cancelAuthPrompt(userId, "Login cancelled by user.");
-            void sendLongMessage(chatId, "Login cancelled.");
-            return;
-        }
-        clearAuthPrompt(userId);
-        pendingAuth.resolve(text);
-        return;
-    }
-    const command = parseCommand(text);
-    if (command?.command === "stop") {
-        void handleStopCommand(chatId, userId);
-        return;
-    }
-    enqueueUser(userId, () => handleMessage(chatId, userId, text)).catch(async (error) => {
-        console.error("[tg-agent] runModel error", error);
-        const message = formatErrorMessage(error);
-        await sendLongMessage(chatId, `Error: ${message}`);
-    });
-});
-bot.on("polling_error", (error) => {
-    console.error("Polling error", error);
-});
-console.log("tg-agent started");
+import V from"node-telegram-bot-api";import{discoverAuthStorage as B,discoverModels as J}from"@mariozechner/pi-coding-agent";import{getOAuthProviders as K}from"@mariozechner/pi-ai";import{config as f,assertConfig as Y}from"./config.js";import{resolveApiKeyForProvider as G,resolveProxyInfo as Z}from"./auth.js";import{applyFetchProxy as Q}from"./proxy.js";import{runPiAgentPrompt as I}from"./piAgentRunner.js";import{readCodexOAuth as ee}from"./codexAuth.js";import{formatMcpTarget as te,loadMcpServers as oe,loadMcpServersSync as ne,probeMcpServer as re}from"./mcp.js";import{appendMessage as D,closeSession as se,createSession as R,deleteSessionFile as A,getActiveSession as M,listSessions as ie,loadUserState as b,pruneExpiredSessions as O,resetSession as ae,saveUserState as p,setActiveSession as ce}from"./sessionStore.js";import{chunkText as le,createQueueMap as ue,createSemaphore as de,ensureDir as U,nowMs as P}from"./utils.js";Y();const y=new V(f.telegramToken,{polling:!0}),fe=ue(),me=de(f.maxConcurrent),T=new Map;function ge(e,t){T.set(e,t)}function pe(e){T.delete(e)}function we(e){const t=e.trim();if(t){if(t==="Markdown"||t==="MarkdownV2"||t==="HTML")return t;console.warn(`[tg-agent] invalid TELEGRAM_PARSE_MODE=${t}, ignoring`)}}const $e=new Set(["_","*","[","]","(",")","~","`",">","#","+","-","=","|","{","}",".","!"]);function _(e,t){let o=0;for(let r=0;r<e.length;r+=1){const s=e[r];if(s==="\\"){r+=1;continue}s===t&&(o+=1)}return o}function ye(e){if(_(e,"*")%2!==0||_(e,"_")%2!==0||_(e,"`")%2!==0)return!1;const s=_(e,"["),u=_(e,"]");return s===u}function ve(e){for(let t=0;t<e.length;t+=1){const o=e[t];if(o==="\\"){t+=1;continue}if($e.has(o))return!1}return!0}function L(e){const t=we(f.telegramParseMode);return t?[t]:ve(e)?["MarkdownV2","Markdown"]:ye(e)?["Markdown"]:[]}const he=600*1e3,$=new Map,x=new Set,Se=new Map([["codex","openai-codex"],["antigravity","google-antigravity"],["gemini","google-gemini-cli"],["gemini-cli","google-gemini-cli"]]),Me={"codex-mini-latest":"gpt-5.1-codex-mini","codex-max-latest":"gpt-5.1-codex-max","codex-latest":"gpt-5.2-codex"};function h(e){const t=e.trim().toLowerCase();return Se.get(t)??t}function _e(e,t){return e!=="openai-codex"?t:Me[t]??t}async function S(){await U(f.agentDir);const e=B(f.agentDir),t=ee();t&&e.setRuntimeApiKey("openai-codex",t.accessToken);const o=J(e,f.agentDir);return{authStorage:e,modelRegistry:o}}function N(e){const t=$.get(e);t&&(clearTimeout(t.timeoutId),$.delete(e))}function Ae(e,t){const o=$.get(e);o&&(clearTimeout(o.timeoutId),$.delete(e),o.reject(new Error(t)))}async function z(e,t,o){if($.has(e))throw new Error("Login is already awaiting input.");const r=o.placeholder?`${o.message} (${o.placeholder})`:o.message;return await c(t,r),await new Promise((s,u)=>{const n=setTimeout(()=>{$.delete(e),u(new Error("Login prompt timed out."))},he);$.set(e,{resolve:s,reject:u,timeoutId:n,chatId:t})})}async function Pe(e,t){const o=T.get(t);if(!o){await c(e,"No active request to stop.");return}o.cancelRequested=!0,o.abortRequested=!0,o.status&&o.status.update("Cancelled by user.",!0),o.abort&&o.abort(),(!o.status||o.chatId!==e)&&await c(e,"Stopping current request...")}function Ee(){const e={https_proxy:process.env.https_proxy??process.env.HTTPS_PROXY??"",http_proxy:process.env.http_proxy??process.env.HTTP_PROXY??"",all_proxy:process.env.all_proxy??process.env.ALL_PROXY??"",no_proxy:process.env.no_proxy??process.env.NO_PROXY??""},t={NODE_EXTRA_CA_CERTS:process.env.NODE_EXTRA_CA_CERTS??"",NODE_TLS_REJECT_UNAUTHORIZED:process.env.NODE_TLS_REJECT_UNAUTHORIZED??""};console.log(`[tg-agent] modelProvider=${f.modelProvider} modelRef=${f.modelRef} sessionDir=${f.sessionDir} maxConcurrent=${f.maxConcurrent}`),console.log(`[tg-agent] agentDir=${f.agentDir} workspaceDir=${f.workspaceDir}`);const o=ne(f.agentDir),r=o.length>0?`on (${o.length})`:"off";console.log(`[tg-agent] tools fetchMaxBytes=${f.fetchMaxBytes} fetchTimeoutMs=${f.fetchTimeoutMs} mcp=${r}`),console.log(`[tg-agent] proxy https=${e.https_proxy||"(empty)"} http=${e.http_proxy||"(empty)"} all=${e.all_proxy||"(empty)"} no=${e.no_proxy||"(empty)"}`),console.log(`[tg-agent] tls extra_ca=${t.NODE_EXTRA_CA_CERTS||"(empty)"} reject_unauthorized=${t.NODE_TLS_REJECT_UNAUTHORIZED||"(empty)"}`);try{const{source:u}=G(f.modelProvider);console.log(`[tg-agent] authSource=${u}`)}catch(u){const n=u instanceof Error?u.message:String(u);console.warn(`[tg-agent] authSource=missing (${n})`)}const s=Z();console.log(s?`[tg-agent] proxyUrl=${s.url} kind=${s.kind} source=${s.source}`:"[tg-agent] proxyUrl=(none)")}Ee(),U(f.agentDir).catch(e=>{console.warn(`[tg-agent] ensure agentDir failed: ${e instanceof Error?e.message:String(e)}`)});const E=Q();console.log(E?`[tg-agent] fetchProxy=${E.url} kind=${E.kind} source=${E.source}`:"[tg-agent] fetchProxy=(none)");function q(e){if(!e.startsWith("/"))return null;const t=e.trim(),[o,...r]=t.split(" ");return{command:o.split("@")[0].slice(1).toLowerCase(),args:r.join(" ").trim()}}async function j(e,t){let o=M(e);return o||(o=R(e,""),await p(e),await c(t,`Created session ${o.id}.`)),o}function Re(e,t){const o=e.getAll(),r=new Map;for(const u of o){const n=r.get(u.provider)??{count:0,auth:t.hasAuth(u.provider)};n.count+=1,r.set(u.provider,n)}return r.size===0?"No providers found.":["Providers:",...Array.from(r.entries()).sort((u,n)=>u[0].localeCompare(n[0])).map(([u,n])=>`- ${u} (models: ${n.count}, auth: ${n.auth?"ok":"missing"})`),"","Use /models <provider> to list models."].join(`
+`)}function Te(e,t,o){const r=e.getAll().filter(u=>u.provider===t);if(r.length===0)return`No models found for provider ${t}.`;const s=r.map(u=>`- ${u.id} | ${u.name}`);return[`Models for ${t} (auth: ${o?"ok":"missing"}):`,...s].join(`
+`)}function xe(e){const t=e.trim();if(!t)return null;if(t.includes("/")){const[o,r]=t.split("/",2);return!o||!r?null:{provider:h(o),modelId:r.trim()}}return{modelId:t}}const ke=6e4,F=new Map;function Ce(e,t){const o=Date.now(),r=F.get(e)??0;o-r<ke||(F.set(e,o),console.warn(`[tg-agent] unauthorized user=${e} chat=${t}`))}function De(e){const t=f.telegramAllowedUsers;return!t||t.size===0?!0:t.has(e)}function be(e){return e.length===0?"No sessions found.":["Sessions:",...e.map(o=>{const r=new Date(o.updatedAt).toISOString();return`- ${o.id} | ${o.title} | updated ${r}`})].join(`
+`)}async function H(e,t,o){return o?await y.sendMessage(e,t,{parse_mode:o}):await y.sendMessage(e,t)}async function W(e,t,o,r){if(r){await y.editMessageText(o,{chat_id:e,message_id:t,parse_mode:r});return}await y.editMessageText(o,{chat_id:e,message_id:t})}async function X(e,t){let o;for(const r of t)try{return await e(r)}catch(s){o=s;const u=s instanceof Error?s.message:String(s);console.warn(`[tg-agent] parse_mode ${r} failed, trying fallback: ${u}`)}try{return await e(void 0)}catch(r){if(o){const s=r instanceof Error?r.message:String(r);console.warn(`[tg-agent] parse_mode fallback to plain failed: ${s}`)}throw r}}async function Oe(e,t){return await X(o=>H(e,t,o),L(t))}async function Ue(e,t,o){await X(r=>W(e,t,o,r),L(o))}async function c(e,t){const o=le(t,3900);for(const r of o)await Oe(e,r)}const Le=3900,Ne=1200;function k(e,t=Le){return e.length<=t?{text:e,truncated:!1}:{text:`${e.slice(0,Math.max(0,t-15))}...
+
+[truncated]`,truncated:!0}}async function ze(e){const t="Working...",r=(await H(e,t)).message_id;let s=t,u=0,n=Promise.resolve();const d=(i,a=!1)=>{const m=Date.now();!a&&m-u<Ne||i!==s&&(s=i,u=m,n=n.then(async()=>{await W(e,r,i)}).catch(g=>{console.warn("[tg-agent] status edit failed",g)}))},l=i=>{s=i,u=Date.now(),n=n.then(async()=>{await Ue(e,r,i)}).catch(a=>{console.warn("[tg-agent] status edit failed",a)})};return{update:(i,a=!1)=>{const m=k(i).text;d(m,a)},finalize:async i=>{const a=k(i).text;l(a),await n},fail:async i=>{const a=k(i).text;l(a),await n}}}function qe(e){switch(e.type){case"agent_start":return"Working...";case"tool_start":return`Running tool: ${e.name}
+args: ${je(e.args)}`;case"tool_end":return e.ok?`Tool finished: ${e.name} (${e.durationMs}ms)`:`Tool failed: ${e.name} (${e.durationMs}ms)`;case"message_start":return e.role==="assistant"?"Generating reply...":null;case"message_end":return e.role==="assistant"?"Finalizing reply...":null;case"heartbeat":return`Working... ${Math.max(1,Math.round(e.elapsedMs/1e3))}s`;default:return null}}function je(e){try{const t=new WeakSet,o=200,r=12,s=12,u=3,n=(i,a)=>{if(i==null)return i;if(typeof i=="string")return i.length>o?`${i.slice(0,o)}...`:i;if(typeof i=="number"||typeof i=="boolean")return i;if(typeof i=="bigint")return i.toString();if(typeof i=="function")return"[function]";if(typeof i!="object")return String(i);if(t.has(i))return"[circular]";if(a>=u)return"[truncated]";if(t.add(i),Array.isArray(i)){const w=i.slice(0,s).map(v=>n(v,a+1));return i.length>s&&w.push("[truncated]"),w}const m=Object.entries(i),g={};for(const[w,v]of m.slice(0,r))g[w]=n(v,a+1);return m.length>r&&(g._truncated=!0),g},d=n(e,0),l=JSON.stringify(d);return l?l.length>700?`${l.slice(0,700)}...`:l:"null"}catch{return"[unavailable]"}}function C(e){const t=e instanceof Error?e.message:String(e),o=e?.cause;return o instanceof Error?`${t} (${o.message})`:o?`${t} (${String(o)})`:t}function Fe(e,t=140){return e.length<=t?e:`${e.slice(0,t-3)}...`}async function He(){const e=await oe(f.agentDir);if(e.length===0)return"No MCP servers configured. Add [mcp_servers.*] to ~/.tg-agent/config.toml.";const t=await Promise.all(e.map(r=>re(r))),o=["MCP servers:"];for(let r=0;r<e.length;r+=1){const s=e[r],u=t[r],n=u.ok?"ok":`error: ${Fe(u.error??"unknown")}`;o.push(`- ${s.name} (${s.type}) ${te(s)} status=${n} (${u.durationMs}ms)`)}return o.join(`
+`)}async function We(e,t,o,r){const s=await b(t),u=O(s);switch(u.length>0&&(await p(s),await Promise.all(u.map(n=>A(t,n).catch(d=>{console.warn(`[tg-agent] cleanup session file failed id=${n}`,d)})))),o){case"start":case"help":{const n=["Commands:","/new [title] - create a new session","/list - list sessions","/use <id> - switch active session","/close [id] - close a session (default: active)","/reset - clear active session history","/providers - list available providers","/models [provider] - list models for provider","/provider <name> - set provider for current session","/model <provider>/<model> - set model for current session","/mcp - list configured MCP servers","/status - show session model settings","/login <provider> - login to OAuth provider","/logout <provider> - logout from provider","/stop - stop the current running request"].join(`
+`);await c(e,n);return}case"new":{try{const d=R(s,r||"");await p(s),await c(e,`Created session ${d.id} (${d.title}).`)}catch(n){await c(e,n.message)}return}case"list":{await p(s),await c(e,be(ie(s)));return}case"use":{if(!r){await c(e,"Usage: /use <id>");return}if(!ce(s,r)){await c(e,`Session not found: ${r}`);return}await p(s),await c(e,`Active session set to ${r}.`);return}case"close":{const n=r||s.activeSessionId;if(!n){await c(e,"No active session to close.");return}if(!se(s,n)){await c(e,`Session not found: ${n}`);return}await p(s),await A(t,n).catch(l=>{console.warn(`[tg-agent] delete session file failed id=${n}`,l)}),await c(e,`Closed session ${n}.`);return}case"reset":{const n=M(s);if(!n){await c(e,"No active session.");return}ae(n),await p(s),await A(t,n.id).catch(d=>{console.warn(`[tg-agent] reset session file failed id=${n.id}`,d)}),await c(e,`Reset session ${n.id}.`);return}case"providers":{const{authStorage:n,modelRegistry:d}=await S(),l=d.getError(),i=Re(d,n),a=l?`Warning: ${l}
+
+${i}`:i;await c(e,a);return}case"models":{const{authStorage:n,modelRegistry:d}=await S();let l=r?h(r):"";if(!l){const a=M(s);if(!a?.modelProvider){await c(e,"Usage: /models <provider>");return}l=a.modelProvider}const i=n.hasAuth(l);await c(e,Te(d,l,i));return}case"provider":{if(!r){await c(e,"Usage: /provider <name>");return}const n=h(r),{modelRegistry:d}=await S();if(!d.getAll().some(a=>a.provider===n)){await c(e,`Unknown provider: ${n}`);return}const i=await j(s,e);i.modelProvider=n,i.modelId=void 0,i.updatedAt=P(),await p(s),await c(e,`Session ${i.id} provider set to ${n}.`);return}case"model":{const n=xe(r);if(!n){await c(e,"Usage: /model <provider>/<model>");return}const d=await j(s,e),l=h(n.provider??d.modelProvider??"");if(!l){await c(e,"Usage: /model <provider>/<model>");return}const{modelRegistry:i}=await S(),a=_e(l,n.modelId);if(!i.find(l,a)){await c(e,`Model not found: ${l}/${a}`);return}d.modelProvider=l,d.modelId=a,d.updatedAt=P(),await p(s),await c(e,`Session ${d.id} model set to ${l}/${a}.`);return}case"status":{const n=M(s);if(!n){await c(e,"No active session.");return}const d=f.modelRef.includes("/")?f.modelRef:`${f.modelProvider}/${f.modelRef}`,l=n.modelProvider||n.modelId?`Session model: ${n.modelProvider??"?"}/${n.modelId??"?"}`:`Default model: ${d}`,i=[`Session: ${n.id} (${n.title})`,l].join(`
+`);await c(e,i);return}case"mcp":{const n=await He();await c(e,n);return}case"login":{const n=K();if(!r){const i=n.map(a=>`- ${a.id} (${a.name})`);await c(e,["OAuth providers:",...i].join(`
+`));return}const d=h(r);if(x.has(t)){await c(e,"Login already in progress.");return}const l=n.find(i=>i.id===d);if(!l){await c(e,`Unknown OAuth provider: ${d}`);return}x.add(t);try{const{authStorage:i}=await S();await c(e,`Starting login for ${l.id}...`),await i.login(l.id,{onAuth:({url:a,instructions:m})=>{const g=["Open this URL in your browser:",a];m&&g.push("",m),c(e,g.join(`
+`))},onPrompt:a=>z(t,e,a),onProgress:a=>{c(e,a)},onManualCodeInput:()=>z(t,e,{message:"Paste the authorization code:"})}),await c(e,`Login completed for ${l.id}.`)}catch(i){const a=C(i);await c(e,`Login failed: ${a}`)}finally{x.delete(t),N(t)}return}case"logout":{if(!r){await c(e,"Usage: /logout <provider>");return}const n=h(r),{authStorage:d}=await S();d.logout(n),await c(e,`Logged out from ${n}.`);return}default:await c(e,`Unknown command: ${o}`)}}async function Xe(e,t,o){const r=q(o);if(r){await We(e,t,r.command,r.args);return}const s=await b(t),u=O(s);u.length>0&&(await p(s),await Promise.all(u.map(a=>A(t,a).catch(m=>{console.warn(`[tg-agent] cleanup session file failed id=${a}`,m)}))));let n=M(s);if(!n)try{n=R(s,""),await p(s),await c(e,`Created session ${n.id}.`)}catch(a){await c(e,a.message);return}const d={role:"user",content:o,ts:P()};D(n,d,f.maxHistoryMessages),await p(s),console.log(`[tg-agent] request user=${t} session=${n.id} messages=${n.messages.length} textLen=${o.length} provider=${n.modelProvider??"-"} model=${n.modelId??"-"}`);let l=null;try{l=await ze(e)}catch(a){console.warn("[tg-agent] status message failed",a)}const i={sessionId:n.id,chatId:e,status:l,abortRequested:!1,cancelRequested:!1};ge(t,i);try{const a=await me(async()=>{const g=await I({userId:t,sessionId:n.id,prompt:o,systemPrompt:f.systemPrompt,modelProvider:n.modelProvider,modelId:n.modelId,onAbortReady:w=>{i.abort=w,i.abortRequested&&w()},onStatus:l?w=>{const v=qe(w);v&&l?.update(v)}:void 0});return console.log(`[tg-agent] response session=${n.id} model=${g.modelProvider}/${g.modelId} file=${g.sessionFile}`),g.text});if(i.cancelRequested){l?await l.finalize("Cancelled by user."):await c(e,"Cancelled by user.");return}const m={role:"assistant",content:a,ts:P()};D(n,m,f.maxHistoryMessages),await p(s),l?await l.finalize(a):await c(e,a)}catch(a){if(console.error("[tg-agent] runModel error",a),i.cancelRequested){l?await l.fail("Cancelled by user."):await c(e,"Cancelled by user.");return}const m=C(a);l?await l.fail(`Error: ${m}`):await c(e,`Error: ${m}`)}finally{pe(t)}}y.on("message",e=>{if(!e.text)return;if(e.chat.type!=="private"){y.sendMessage(e.chat.id,"This bot only supports direct messages.");return}if(!e.from?.id)return;const t=String(e.from.id),o=e.chat.id;if(!De(t)){Ce(t,o);return}const r=e.text.trim(),s=$.get(t);if(s){if(r==="/stop"||r==="/cancel"){Ae(t,"Login cancelled by user."),c(o,"Login cancelled.");return}N(t),s.resolve(r);return}if(q(r)?.command==="stop"){Pe(o,t);return}fe(t,()=>Xe(o,t,r)).catch(async n=>{console.error("[tg-agent] runModel error",n);const d=C(n);await c(o,`Error: ${d}`)})}),y.on("polling_error",e=>{console.error("Polling error",e)}),console.log("tg-agent started");