tg-agent 0.1.0 → 1.1.1

package/dist/index.js

@@ -1,954 +1,17 @@
-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 b from"node:path";import B from"node:fs/promises";import te from"node-telegram-bot-api";import{discoverAuthStorage as ne,discoverModels as oe}from"@mariozechner/pi-coding-agent";import{getOAuthProviders as re}from"@mariozechner/pi-ai";import{config as m,assertConfig as se}from"./config.js";import{resolveApiKeyForProvider as ie,resolveProxyInfo as ae}from"./auth.js";import{applyFetchProxy as ce}from"./proxy.js";import{runPiAgentPrompt as le}from"./piAgentRunner.js";import{readCodexOAuth as ue}from"./codexAuth.js";import{formatMcpTarget as de,freezeMcpCatalog as F,getMcpCatalog as q,loadMcpServers as me,loadMcpServersSync as fe,probeMcpServer as ge,refreshMcpCatalog as pe}from"./mcp.js";import{appendMessage as W,closeSession as we,createSession as L,deleteSessionFile as k,getActiveSession as x,listSessions as he,loadUserState as H,pruneExpiredSessions as X,resetSession as ye,saveUserState as w,setActiveSession as $e}from"./sessionStore.js";import{chunkText as ve,createQueueMap as Se,createSemaphore as Me,ensureDir as U,nowMs as C}from"./utils.js";se();const y=new te(m.telegramToken,{polling:!0}),_e=Se(),Ae=Me(m.maxConcurrent),O=new Map;function Pe(e,t){O.set(e,t)}function Te(e){O.delete(e)}function xe(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 Ee="image/jpeg";function Re(){return b.join(m.agentDir,"uploads")}function ke(e){return e.length===0?null:e.reduce((t,n)=>{const o=t.file_size??t.width*t.height;return(n.file_size??n.width*n.height)>o?n:t})}function Ce(e){return!!e?.startsWith("image/")}function De(e){switch(b.extname(e).toLowerCase()){case".png":return"image/png";case".jpg":case".jpeg":return"image/jpeg";case".webp":return"image/webp";case".gif":return"image/gif";default:return}}function be(e){if(!Number.isFinite(e)||e<=0)return"0 B";const t=["B","KB","MB","GB"];let n=e,o=0;for(;n>=1024&&o<t.length-1;)n/=1024,o+=1;return`${n.toFixed(n>=10||o===0?0:1)} ${t[o]}`}function Le(e){return e.length===0?"":e.map(t=>{const n=[];t.mimeType&&n.push(t.mimeType),t.bytes>0&&n.push(be(t.bytes));const o=n.length>0?` (${n.join(", ")})`:"";return`Attachment saved: ${t.path}${o}`}).join(`
+`)}function Ue(e){const t=[];if(e.photo&&e.photo.length>0){const n=ke(e.photo);n&&t.push({kind:"photo",fileId:n.file_id,fileSize:n.file_size})}return e.document&&t.push({kind:"document",fileId:e.document.file_id,fileName:e.document.file_name,mimeType:e.document.mime_type,fileSize:e.document.file_size}),t}async function Oe(e){if(e.length===0)return{resolved:[],images:[]};const t=Re();await U(t);const n=[],o=[];for(const s of e)try{const a=await y.downloadFile(s.fileId,t),r=await B.stat(a),c=s.fileName??b.basename(a),u=s.mimeType??De(a),i=s.kind==="photo"||Ce(u),d={path:a,name:c,mimeType:u,bytes:r.size,isImage:i};if(n.push(d),i){const f=await B.readFile(a);o.push({type:"image",data:f.toString("base64"),mimeType:u??Ee})}}catch(a){console.warn("[tg-agent] attachment download failed",a)}return{resolved:n,images:o}}const ze=new Set(["_","*","[","]","(",")","~","`",">","#","+","-","=","|","{","}",".","!"]);function E(e,t){let n=0;for(let o=0;o<e.length;o+=1){const s=e[o];if(s==="\\"){o+=1;continue}s===t&&(n+=1)}return n}function Ne(e){if(E(e,"*")%2!==0||E(e,"_")%2!==0||E(e,"`")%2!==0)return!1;const s=E(e,"["),a=E(e,"]");return s===a}function je(e){for(let t=0;t<e.length;t+=1){const n=e[t];if(n==="\\"){t+=1;continue}if(ze.has(n))return!1}return!0}function V(e){const t=xe(m.telegramParseMode);return t?[t]:je(e)?["MarkdownV2","Markdown"]:Ne(e)?["Markdown"]:[]}const Be=600*1e3,S=new Map,z=new Set,Fe=new Map([["codex","openai-codex"],["antigravity","google-antigravity"],["gemini","google-gemini-cli"],["gemini-cli","google-gemini-cli"]]),qe={"codex-mini-latest":"gpt-5.1-codex-mini","codex-max-latest":"gpt-5.1-codex-max","codex-latest":"gpt-5.2-codex"};function A(e){const t=e.trim().toLowerCase();return Fe.get(t)??t}function We(e,t){return e!=="openai-codex"?t:qe[t]??t}async function P(){await U(m.agentDir);const e=ne(m.agentDir),t=ue();t&&e.setRuntimeApiKey("openai-codex",t.accessToken);const n=oe(e,m.agentDir);return{authStorage:e,modelRegistry:n}}function K(e){const t=S.get(e);t&&(clearTimeout(t.timeoutId),S.delete(e))}function He(e,t){const n=S.get(e);n&&(clearTimeout(n.timeoutId),S.delete(e),n.reject(new Error(t)))}async function G(e,t,n){if(S.has(e))throw new Error("Login is already awaiting input.");const o=n.placeholder?`${n.message} (${n.placeholder})`:n.message;return await l(t,o),await new Promise((s,a)=>{const r=setTimeout(()=>{S.delete(e),a(new Error("Login prompt timed out."))},Be);S.set(e,{resolve:s,reject:a,timeoutId:r,chatId:t})})}async function Xe(e,t){const n=O.get(t);if(!n){await l(e,"No active request to stop.");return}n.cancelRequested=!0,n.abortRequested=!0,n.status&&n.status.update("Cancelled by user.",!0),n.abort&&n.abort(),(!n.status||n.chatId!==e)&&await l(e,"Stopping current request...")}function Ve(){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=${m.modelProvider} modelRef=${m.modelRef} sessionDir=${m.sessionDir} maxConcurrent=${m.maxConcurrent}`),console.log(`[tg-agent] agentDir=${m.agentDir} workspaceDir=${m.workspaceDir}`);const n=fe(m.agentDir),o=n.length>0?`on (${n.length})`:"off";console.log(`[tg-agent] tools fetchMaxBytes=${m.fetchMaxBytes} fetchTimeoutMs=${m.fetchTimeoutMs} mcp=${o}`),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:a}=ie(m.modelProvider);console.log(`[tg-agent] authSource=${a}`)}catch(a){const r=a instanceof Error?a.message:String(a);console.warn(`[tg-agent] authSource=missing (${r})`)}const s=ae();console.log(s?`[tg-agent] proxyUrl=${s.url} kind=${s.kind} source=${s.source}`:"[tg-agent] proxyUrl=(none)")}Ve(),U(m.agentDir).catch(e=>{console.warn(`[tg-agent] ensure agentDir failed: ${e instanceof Error?e.message:String(e)}`)}),(async()=>{try{await q(m.agentDir,{timeoutMs:3e3,maxBytes:m.fetchMaxBytes,maxChars:2e3}),F()}catch(e){const t=e instanceof Error?e.message:String(e);console.warn(`[tg-agent] mcp catalog warmup failed: ${t}`)}})();const D=ce();console.log(D?`[tg-agent] fetchProxy=${D.url} kind=${D.kind} source=${D.source}`:"[tg-agent] fetchProxy=(none)");function J(e){if(!e.startsWith("/"))return null;const t=e.trim(),[n,...o]=t.split(" ");return{command:n.split("@")[0].slice(1).toLowerCase(),args:o.join(" ").trim()}}async function Y(e,t){let n=x(e);return n||(n=L(e,""),await w(e),await l(t,`Created session ${n.id}.`)),n}function Ke(e,t){const n=e.getAll(),o=new Map;for(const a of n){const r=o.get(a.provider)??{count:0,auth:t.hasAuth(a.provider)};r.count+=1,o.set(a.provider,r)}return o.size===0?"No providers found.":["Providers:",...Array.from(o.entries()).sort((a,r)=>a[0].localeCompare(r[0])).map(([a,r])=>`- ${a} (models: ${r.count}, auth: ${r.auth?"ok":"missing"})`),"","Use /models <provider> to list models."].join(`
+`)}function Ge(e,t,n){const o=e.getAll().filter(a=>a.provider===t);if(o.length===0)return`No models found for provider ${t}.`;const s=o.map(a=>`- ${a.id} | ${a.name}`);return[`Models for ${t} (auth: ${n?"ok":"missing"}):`,...s].join(`
+`)}function Je(e){const t=e.trim();if(!t)return null;if(t.includes("/")){const[n,o]=t.split("/",2);return!n||!o?null:{provider:A(n),modelId:o.trim()}}return{modelId:t}}const Ye=6e4,Z=new Map;function Ze(e,t){const n=Date.now(),o=Z.get(e)??0;n-o<Ye||(Z.set(e,n),console.warn(`[tg-agent] unauthorized user=${e} chat=${t}`))}function Qe(e){const t=m.telegramAllowedUsers;return!t||t.size===0?!0:t.has(e)}function Ie(e){return e.length===0?"No sessions found.":["Sessions:",...e.map(n=>{const o=new Date(n.updatedAt).toISOString();return`- ${n.id} | ${n.title} | updated ${o}`})].join(`
+`)}async function Q(e,t,n){return n?await y.sendMessage(e,t,{parse_mode:n}):await y.sendMessage(e,t)}async function I(e,t,n,o){if(o){await y.editMessageText(n,{chat_id:e,message_id:t,parse_mode:o});return}await y.editMessageText(n,{chat_id:e,message_id:t})}async function ee(e,t){let n;for(const o of t)try{return await e(o)}catch(s){n=s;const a=s instanceof Error?s.message:String(s);console.warn(`[tg-agent] parse_mode ${o} failed, trying fallback: ${a}`)}try{return await e(void 0)}catch(o){if(n){const s=o instanceof Error?o.message:String(o);console.warn(`[tg-agent] parse_mode fallback to plain failed: ${s}`)}throw o}}async function et(e,t){return await ee(n=>Q(e,t,n),V(t))}async function tt(e,t,n){await ee(o=>I(e,t,n,o),V(n))}async function l(e,t){const n=ve(t,3900);for(const o of n)await et(e,o)}const nt=3900,ot=1200;function N(e,t=nt){return e.length<=t?{text:e,truncated:!1}:{text:`${e.slice(0,Math.max(0,t-15))}...
+
+[truncated]`,truncated:!0}}async function rt(e){const t="Working...",o=(await Q(e,t)).message_id;let s=t,a=0,r=Promise.resolve();const c=(i,d=!1)=>{const f=Date.now();!d&&f-a<ot||i!==s&&(s=i,a=f,r=r.then(async()=>{await I(e,o,i)}).catch(h=>{console.warn("[tg-agent] status edit failed",h)}))},u=i=>{s=i,a=Date.now(),r=r.then(async()=>{await tt(e,o,i)}).catch(d=>{console.warn("[tg-agent] status edit failed",d)})};return{update:(i,d=!1)=>{const f=N(i).text;c(f,d)},finalize:async i=>{const d=N(i).text;u(d),await r},fail:async i=>{const d=N(i).text;u(d),await r}}}function st(e){switch(e.type){case"agent_start":return"Working...";case"tool_start":return`Running tool: ${e.name}
+args: ${it(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 it(e){try{const t=new WeakSet,n=200,o=12,s=12,a=3,r=(i,d)=>{if(i==null)return i;if(typeof i=="string")return i.length>n?`${i.slice(0,n)}...`: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(d>=a)return"[truncated]";if(t.add(i),Array.isArray(i)){const M=i.slice(0,s).map(g=>r(g,d+1));return i.length>s&&M.push("[truncated]"),M}const f=Object.entries(i),h={};for(const[M,g]of f.slice(0,o))h[M]=r(g,d+1);return f.length>o&&(h._truncated=!0),h},c=r(e,0),u=JSON.stringify(c);return u?u.length>700?`${u.slice(0,700)}...`:u:"null"}catch{return"[unavailable]"}}function j(e){const t=e instanceof Error?e.message:String(e),n=e?.cause;return n instanceof Error?`${t} (${n.message})`:n?`${t} (${String(n)})`:t}function at(e,t=140){return e.length<=t?e:`${e.slice(0,t-3)}...`}async function ct(){const e=await me(m.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(o=>ge(o))),n=["MCP servers:"];for(let o=0;o<e.length;o+=1){const s=e[o],a=t[o],r=a.ok?"ok":`error: ${at(a.error??"unknown")}`;n.push(`- ${s.name} (${s.type}) ${de(s)} status=${r} (${a.durationMs}ms)`)}return n.join(`
+`)}async function lt(e,t,n,o){const s=await H(t),a=X(s);switch(a.length>0&&(await w(s),await Promise.all(a.map(r=>k(t,r).catch(c=>{console.warn(`[tg-agent] cleanup session file failed id=${r}`,c)})))),n){case"start":case"help":{const r=["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","/mcp refresh - reload MCP catalog","/status - show session model settings","/login <provider> - login to OAuth provider","/logout <provider> - logout from provider","/stop - stop the current running request","","Tips:","Send images or files to attach them to the prompt."].join(`
+`);await l(e,r);return}case"new":{try{const c=L(s,o||"");await w(s),await l(e,`Created session ${c.id} (${c.title}).`)}catch(r){await l(e,r.message)}return}case"list":{await w(s),await l(e,Ie(he(s)));return}case"use":{if(!o){await l(e,"Usage: /use <id>");return}if(!$e(s,o)){await l(e,`Session not found: ${o}`);return}await w(s),await l(e,`Active session set to ${o}.`);return}case"close":{const r=o||s.activeSessionId;if(!r){await l(e,"No active session to close.");return}if(!we(s,r)){await l(e,`Session not found: ${r}`);return}await w(s),await k(t,r).catch(u=>{console.warn(`[tg-agent] delete session file failed id=${r}`,u)}),await l(e,`Closed session ${r}.`);return}case"reset":{const r=x(s);if(!r){await l(e,"No active session.");return}ye(r),await w(s),await k(t,r.id).catch(c=>{console.warn(`[tg-agent] reset session file failed id=${r.id}`,c)}),await l(e,`Reset session ${r.id}.`);return}case"providers":{const{authStorage:r,modelRegistry:c}=await P(),u=c.getError(),i=Ke(c,r),d=u?`Warning: ${u}
+
+${i}`:i;await l(e,d);return}case"models":{const{authStorage:r,modelRegistry:c}=await P();let u=o?A(o):"";if(!u){const d=x(s);if(!d?.modelProvider){await l(e,"Usage: /models <provider>");return}u=d.modelProvider}const i=r.hasAuth(u);await l(e,Ge(c,u,i));return}case"provider":{if(!o){await l(e,"Usage: /provider <name>");return}const r=A(o),{modelRegistry:c}=await P();if(!c.getAll().some(d=>d.provider===r)){await l(e,`Unknown provider: ${r}`);return}const i=await Y(s,e);i.modelProvider=r,i.modelId=void 0,i.updatedAt=C(),await w(s),await l(e,`Session ${i.id} provider set to ${r}.`);return}case"model":{const r=Je(o);if(!r){await l(e,"Usage: /model <provider>/<model>");return}const c=await Y(s,e),u=A(r.provider??c.modelProvider??"");if(!u){await l(e,"Usage: /model <provider>/<model>");return}const{modelRegistry:i}=await P(),d=We(u,r.modelId);if(!i.find(u,d)){await l(e,`Model not found: ${u}/${d}`);return}c.modelProvider=u,c.modelId=d,c.updatedAt=C(),await w(s),await l(e,`Session ${c.id} model set to ${u}/${d}.`);return}case"status":{const r=x(s);if(!r){await l(e,"No active session.");return}const c=m.modelRef.includes("/")?m.modelRef:`${m.modelProvider}/${m.modelRef}`,u=r.modelProvider||r.modelId?`Session model: ${r.modelProvider??"?"}/${r.modelId??"?"}`:`Default model: ${c}`,i=[`Session: ${r.id} (${r.title})`,u].join(`
+`);await l(e,i);return}case"mcp":{if(o.trim().toLowerCase()==="refresh"){pe(),await q(m.agentDir,{timeoutMs:5e3,maxBytes:m.fetchMaxBytes,maxChars:3e3}),F(),await l(e,"MCP catalog refreshed.");return}const r=await ct();await l(e,r);return}case"login":{const r=re();if(!o){const i=r.map(d=>`- ${d.id} (${d.name})`);await l(e,["OAuth providers:",...i].join(`
+`));return}const c=A(o);if(z.has(t)){await l(e,"Login already in progress.");return}const u=r.find(i=>i.id===c);if(!u){await l(e,`Unknown OAuth provider: ${c}`);return}z.add(t);try{const{authStorage:i}=await P();await l(e,`Starting login for ${u.id}...`),await i.login(u.id,{onAuth:({url:d,instructions:f})=>{const h=["Open this URL in your browser:",d];f&&h.push("",f),l(e,h.join(`
+`))},onPrompt:d=>G(t,e,d),onProgress:d=>{l(e,d)},onManualCodeInput:()=>G(t,e,{message:"Paste the authorization code:"})}),await l(e,`Login completed for ${u.id}.`)}catch(i){const d=j(i);await l(e,`Login failed: ${d}`)}finally{z.delete(t),K(t)}return}case"logout":{if(!o){await l(e,"Usage: /logout <provider>");return}const r=A(o),{authStorage:c}=await P();c.logout(r),await l(e,`Logged out from ${r}.`);return}default:await l(e,`Unknown command: ${n}`)}}async function ut(e,t,n,o=[]){const s=J(n);if(s){await lt(e,t,s.command,s.args);return}const a=await H(t),r=X(a);r.length>0&&(await w(a),await Promise.all(r.map(p=>k(t,p).catch(_=>{console.warn(`[tg-agent] cleanup session file failed id=${p}`,_)}))));let c=x(a);if(!c)try{c=L(a,""),await w(a),await l(e,`Created session ${c.id}.`)}catch(p){await l(e,p.message);return}const{resolved:u,images:i}=await Oe(o);if(o.length>0&&u.length===0&&!n.trim()){await l(e,"Failed to download attachments.");return}const d=Le(u),f=[];n.trim()&&f.push(n.trim()),d&&f.push(d);const h=f.join(`
+
+`)||"Attachment received.",M={role:"user",content:h,ts:C()};W(c,M,m.maxHistoryMessages),await w(a),console.log(`[tg-agent] request user=${t} session=${c.id} messages=${c.messages.length} textLen=${h.length} attachments=${u.length} provider=${c.modelProvider??"-"} model=${c.modelId??"-"}`);let g=null;try{g=await rt(e)}catch(p){console.warn("[tg-agent] status message failed",p)}const T={sessionId:c.id,chatId:e,status:g,abortRequested:!1,cancelRequested:!1};Pe(t,T);try{const p=await Ae(async()=>{const R=await le({userId:t,sessionId:c.id,prompt:h,images:i,systemPrompt:m.systemPrompt,modelProvider:c.modelProvider,modelId:c.modelId,telegram:{chatId:e,sendPhoto:async($,v)=>{await y.sendPhoto(e,$,v?{caption:v}:void 0)},sendDocument:async($,v)=>{await y.sendDocument(e,$,v?{caption:v}:void 0)}},onAbortReady:$=>{T.abort=$,T.abortRequested&&$()},onStatus:g?$=>{const v=st($);v&&g?.update(v)}:void 0});return console.log(`[tg-agent] response session=${c.id} model=${R.modelProvider}/${R.modelId} file=${R.sessionFile}`),R.text});if(T.cancelRequested){g?await g.finalize("Cancelled by user."):await l(e,"Cancelled by user.");return}const _={role:"assistant",content:p,ts:C()};W(c,_,m.maxHistoryMessages),await w(a),g?await g.finalize(p):await l(e,p)}catch(p){if(console.error("[tg-agent] runModel error",p),T.cancelRequested){g?await g.fail("Cancelled by user."):await l(e,"Cancelled by user.");return}const _=j(p);g?await g.fail(`Error: ${_}`):await l(e,`Error: ${_}`)}finally{Te(t)}}y.on("message",e=>{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),n=e.chat.id;if(!Qe(t)){Ze(t,n);return}const o=(e.text??e.caption??"").trim(),s=Ue(e);if(!o&&s.length===0)return;const a=S.get(t);if(a){if(o==="/stop"||o==="/cancel"){He(t,"Login cancelled by user."),l(n,"Login cancelled.");return}if(!o){l(n,"Login expects a text response.");return}K(t),a.resolve(o);return}if(J(o)?.command==="stop"){Xe(n,t);return}_e(t,()=>ut(n,t,o,s)).catch(async c=>{console.error("[tg-agent] runModel error",c);const u=j(c);await l(n,`Error: ${u}`)})}),y.on("polling_error",e=>{console.error("Polling error",e)}),console.log("tg-agent started");