@iletai/nzb 1.2.6 → 1.3.1

package/dist/telegram/bot.js

@@ -1,5 +1,6 @@
 import { autoRetry } from "@grammyjs/auto-retry";
-import { Bot, InlineKeyboard } from "grammy";
+import { Menu } from "@grammyjs/menu";
+import { Bot } from "grammy";
 import { Agent as HttpsAgent } from "https";
 import { config, persistEnvVar, persistModel } from "../config.js";
 import { cancelCurrentMessage, getQueueSize, getWorkers, sendToOrchestrator } from "../copilot/orchestrator.js";
@@ -10,18 +11,94 @@ import { chunkMessage, formatToolSummaryExpandable, toTelegramMarkdown } from ".
 import { initLogChannel, logDebug, logError, logInfo } from "./log-channel.js";
 let bot;
 const startedAt = Date.now();
-// Inline keyboard menu for quick actions
-const mainMenu = new InlineKeyboard()
-    .text("📊 Status", "action:status")
-    .text("🤖 Model", "action:model")
+// Helper: build uptime string
+function getUptimeStr() {
+    const uptime = Math.floor((Date.now() - startedAt) / 1000);
+    const hours = Math.floor(uptime / 3600);
+    const minutes = Math.floor((uptime % 3600) / 60);
+    const seconds = uptime % 60;
+    return hours > 0 ? `${hours}h ${minutes}m ${seconds}s` : minutes > 0 ? `${minutes}m ${seconds}s` : `${seconds}s`;
+}
+// Settings sub-menu
+const settingsMenu = new Menu("settings-menu")
+    .text((ctx) => `${config.showReasoning ? "✅" : "❌"} Show Reasoning`, async (ctx) => {
+    config.showReasoning = !config.showReasoning;
+    persistEnvVar("SHOW_REASONING", config.showReasoning ? "true" : "false");
+    ctx.menu.update();
+    await ctx.answerCallbackQuery(`Reasoning ${config.showReasoning ? "ON" : "OFF"}`);
+})
+    .row()
+    .back("🔙 Back", async (ctx) => {
+    await ctx.editMessageText("NZB Menu:");
+});
+// Main interactive menu with navigation
+const mainMenu = new Menu("main-menu")
+    .text("📊 Status", async (ctx) => {
+    const workers = Array.from(getWorkers().values());
+    const lines = [
+        "📊 NZB Status",
+        `Model: ${config.copilotModel}`,
+        `Uptime: ${getUptimeStr()}`,
+        `Workers: ${workers.length} active`,
+        `Queue: ${getQueueSize()} pending`,
+    ];
+    await ctx.answerCallbackQuery();
+    await ctx.reply(lines.join("\n"));
+})
+    .text("🤖 Model", async (ctx) => {
+    await ctx.answerCallbackQuery();
+    await ctx.reply(`Current model: ${config.copilotModel}`);
+})
     .row()
-    .text("👥 Workers", "action:workers")
-    .text("🧠 Skills", "action:skills")
+    .text("👥 Workers", async (ctx) => {
+    await ctx.answerCallbackQuery();
+    const workers = Array.from(getWorkers().values());
+    if (workers.length === 0) {
+        await ctx.reply("No active worker sessions.");
+    }
+    else {
+        const lines = workers.map((w) => `• ${w.name} (${w.workingDir}) — ${w.status}`);
+        await ctx.reply(lines.join("\n"));
+    }
+})
+    .text("🧠 Skills", async (ctx) => {
+    await ctx.answerCallbackQuery();
+    const skills = listSkills();
+    if (skills.length === 0) {
+        await ctx.reply("No skills installed.");
+    }
+    else {
+        const lines = skills.map((s) => `• ${s.name} (${s.source}) — ${s.description}`);
+        await ctx.reply(lines.join("\n"));
+    }
+})
     .row()
-    .text("🗂 Memory", "action:memory")
-    .text("⚙️ Settings", "action:settings")
+    .text("🗂 Memory", async (ctx) => {
+    await ctx.answerCallbackQuery();
+    const memories = searchMemories(undefined, undefined, 50);
+    if (memories.length === 0) {
+        await ctx.reply("No memories stored.");
+    }
+    else {
+        const lines = memories.map((m) => `#${m.id} [${m.category}] ${m.content}`);
+        await ctx.reply(lines.join("\n") + `\n\n${memories.length} total`);
+    }
+})
+    .submenu("⚙️ Settings", "settings-menu", async (ctx) => {
+    await ctx.editMessageText("⚙️ Settings\n\n" +
+        `🔧 Show Reasoning: ${config.showReasoning ? "✅ ON" : "❌ OFF"}\n` +
+        `  └ Hiển thị tools đã dùng + thời gian cuối mỗi phản hồi\n\n` +
+        `🤖 Model: ${config.copilotModel}\n` +
+        `  └ Dùng /model <name> để đổi`);
+})
     .row()
-    .text("❌ Cancel", "action:cancel");
+    .text("❌ Cancel", async (ctx) => {
+    await ctx.answerCallbackQuery();
+    const cancelled = await cancelCurrentMessage();
+    await ctx.reply(cancelled ? "Cancelled." : "Nothing to cancel.");
+});
+// Register sub-menu as child
+mainMenu.register(settingsMenu);
 // Direct-connection HTTPS agent for Telegram API requests.
 // This bypasses corporate proxy (HTTP_PROXY/HTTPS_PROXY env vars) without
 // modifying process.env, so other services (Copilot SDK, MCP, npm) are unaffected.
@@ -53,6 +130,8 @@ export function createBot() {
         }
         await next();
     });
+    // Register interactive menu plugin
+    bot.use(mainMenu);
     // /start and /help — with inline menu
     bot.command("start", (ctx) => ctx.reply("NZB is online. Send me anything, or use the menu below:", { reply_markup: mainMenu }));
     bot.command("help", (ctx) => ctx.reply("I'm NZB, your AI daemon.\n\n" +
@@ -156,92 +235,12 @@ export function createBot() {
             });
         }, 500);
     });
-    // /settings — show toggleable settings with inline keyboard
-    const buildSettingsKeyboard = () => new InlineKeyboard()
-        .text(`${config.showReasoning ? "✅" : "❌"} Show Reasoning`, "setting:toggle:reasoning")
-        .row()
-        .text("🔙 Back to Menu", "action:menu");
-    const buildSettingsText = () => "⚙️ Settings\n\n" +
-        `🔧 Show Reasoning: ${config.showReasoning ? "✅ ON" : "❌ OFF"}\n` +
-        `  └ Hiển thị tools đã dùng + thời gian cuối mỗi phản hồi\n\n` +
-        `🤖 Model: ${config.copilotModel}\n` +
-        `  └ Dùng /model <name> để đổi`;
     bot.command("settings", async (ctx) => {
-        await ctx.reply(buildSettingsText(), { reply_markup: buildSettingsKeyboard() });
-    });
-    // Callback query handlers for inline menu buttons
-    bot.callbackQuery("action:status", async (ctx) => {
-        await ctx.answerCallbackQuery();
-        const uptime = Math.floor((Date.now() - startedAt) / 1000);
-        const hours = Math.floor(uptime / 3600);
-        const minutes = Math.floor((uptime % 3600) / 60);
-        const seconds = uptime % 60;
-        const uptimeStr = hours > 0 ? `${hours}h ${minutes}m ${seconds}s` : minutes > 0 ? `${minutes}m ${seconds}s` : `${seconds}s`;
-        const workers = Array.from(getWorkers().values());
-        const lines = [
-            "📊 NZB Status",
-            `Model: ${config.copilotModel}`,
-            `Uptime: ${uptimeStr}`,
-            `Workers: ${workers.length} active`,
-            `Queue: ${getQueueSize()} pending`,
-        ];
-        await ctx.reply(lines.join("\n"));
-    });
-    bot.callbackQuery("action:model", async (ctx) => {
-        await ctx.answerCallbackQuery();
-        await ctx.reply(`Current model: ${config.copilotModel}`);
-    });
-    bot.callbackQuery("action:workers", async (ctx) => {
-        await ctx.answerCallbackQuery();
-        const workers = Array.from(getWorkers().values());
-        if (workers.length === 0) {
-            await ctx.reply("No active worker sessions.");
-        }
-        else {
-            const lines = workers.map((w) => `• ${w.name} (${w.workingDir}) — ${w.status}`);
-            await ctx.reply(lines.join("\n"));
-        }
-    });
-    bot.callbackQuery("action:skills", async (ctx) => {
-        await ctx.answerCallbackQuery();
-        const skills = listSkills();
-        if (skills.length === 0) {
-            await ctx.reply("No skills installed.");
-        }
-        else {
-            const lines = skills.map((s) => `• ${s.name} (${s.source}) — ${s.description}`);
-            await ctx.reply(lines.join("\n"));
-        }
-    });
-    bot.callbackQuery("action:memory", async (ctx) => {
-        await ctx.answerCallbackQuery();
-        const memories = searchMemories(undefined, undefined, 50);
-        if (memories.length === 0) {
-            await ctx.reply("No memories stored.");
-        }
-        else {
-            const lines = memories.map((m) => `#${m.id} [${m.category}] ${m.content}`);
-            await ctx.reply(lines.join("\n") + `\n\n${memories.length} total`);
-        }
-    });
-    bot.callbackQuery("action:cancel", async (ctx) => {
-        await ctx.answerCallbackQuery();
-        const cancelled = await cancelCurrentMessage();
-        await ctx.reply(cancelled ? "Cancelled." : "Nothing to cancel.");
-    });
-    bot.callbackQuery("action:settings", async (ctx) => {
-        await ctx.answerCallbackQuery();
-        await ctx.reply(buildSettingsText(), { reply_markup: buildSettingsKeyboard() });
-    });
-    bot.callbackQuery("action:menu", async (ctx) => {
-        await ctx.answerCallbackQuery();
-        await ctx.editMessageText("NZB Menu:", { reply_markup: mainMenu });
-    });
-    bot.callbackQuery("setting:toggle:reasoning", async (ctx) => {
-        config.showReasoning = !config.showReasoning;
-        persistEnvVar("SHOW_REASONING", config.showReasoning ? "true" : "false");
-        await ctx.answerCallbackQuery(`Reasoning ${config.showReasoning ? "ON" : "OFF"}`);
-        await ctx.editMessageText(buildSettingsText(), { reply_markup: buildSettingsKeyboard() });
+        await ctx.reply("⚙️ Settings\n\n" +
+            `🔧 Show Reasoning: ${config.showReasoning ? "✅ ON" : "❌ OFF"}\n` +
+            `  └ Hiển thị tools đã dùng + thời gian cuối mỗi phản hồi\n\n` +
+            `🤖 Model: ${config.copilotModel}\n` +
+            `  └ Dùng /model <name> để đổi`, { reply_markup: settingsMenu });
     });
     // Handle all text messages — progressive streaming with tool event feedback
     bot.on("message:text", async (ctx) => {
@@ -532,6 +531,149 @@ export function createBot() {
             }
         }, onToolEvent, onUsage);
     });
+    // Handle photo messages — download and pass to AI as image description request
+    bot.on("message:photo", async (ctx) => {
+        const chatId = ctx.chat.id;
+        const userMessageId = ctx.message.message_id;
+        const caption = ctx.message.caption || "Describe this image and analyze what you see.";
+        void logInfo(`📸 Photo received: ${caption.slice(0, 80)}`);
+        try {
+            await ctx.react("👀");
+        }
+        catch { }
+        // Get the largest photo (last in array)
+        const photo = ctx.message.photo[ctx.message.photo.length - 1];
+        try {
+            const file = await ctx.api.getFile(photo.file_id);
+            const filePath = file.file_path;
+            if (!filePath) {
+                await ctx.reply("❌ Could not download photo.", { reply_parameters: { message_id: userMessageId } });
+                return;
+            }
+            const url = `https://api.telegram.org/file/bot${config.telegramBotToken}/${filePath}`;
+            // Download to temp file
+            const { mkdtempSync, writeFileSync } = await import("fs");
+            const { join } = await import("path");
+            const { tmpdir } = await import("os");
+            const tmpDir = mkdtempSync(join(tmpdir(), "nzb-photo-"));
+            const ext = filePath.split(".").pop() || "jpg";
+            const localPath = join(tmpDir, `photo.${ext}`);
+            const response = await fetch(url);
+            const buffer = Buffer.from(await response.arrayBuffer());
+            writeFileSync(localPath, buffer);
+            // Send to orchestrator with image context
+            const prompt = `[User sent a photo saved at: ${localPath}]\n\nCaption: ${caption}\n\nPlease analyze this image. The file is at ${localPath} — you can use bash to view it with tools if needed.`;
+            sendToOrchestrator(prompt, { type: "telegram", chatId, messageId: userMessageId }, (text, done) => {
+                if (done) {
+                    const formatted = toTelegramMarkdown(text);
+                    const chunks = chunkMessage(formatted);
+                    const fallbackChunks = chunkMessage(text);
+                    void (async () => {
+                        for (let i = 0; i < chunks.length; i++) {
+                            if (i > 0)
+                                await new Promise((r) => setTimeout(r, 300));
+                            const pageTag = chunks.length > 1 ? `📄 ${i + 1}/${chunks.length}\n` : "";
+                            try {
+                                await ctx.api.sendMessage(chatId, pageTag + chunks[i], {
+                                    parse_mode: "MarkdownV2",
+                                    reply_parameters: i === 0 ? { message_id: userMessageId } : undefined,
+                                });
+                            }
+                            catch {
+                                try {
+                                    await ctx.api.sendMessage(chatId, pageTag + (fallbackChunks[i] ?? chunks[i]), {
+                                        reply_parameters: i === 0 ? { message_id: userMessageId } : undefined,
+                                    });
+                                }
+                                catch { }
+                            }
+                        }
+                        try {
+                            await ctx.api.setMessageReaction(chatId, userMessageId, [{ type: "emoji", emoji: "👍" }]);
+                        }
+                        catch { }
+                    })();
+                }
+            });
+        }
+        catch (err) {
+            await ctx.reply(`❌ Error processing photo: ${err instanceof Error ? err.message : String(err)}`, {
+                reply_parameters: { message_id: userMessageId },
+            });
+        }
+    });
+    // Handle document/file messages — download and pass to AI
+    bot.on("message:document", async (ctx) => {
+        const chatId = ctx.chat.id;
+        const userMessageId = ctx.message.message_id;
+        const doc = ctx.message.document;
+        const caption = ctx.message.caption || `Analyze this file: ${doc.file_name || "unknown"}`;
+        void logInfo(`📄 Document received: ${doc.file_name || "unknown"} (${doc.file_size || 0} bytes)`);
+        try {
+            await ctx.react("👀");
+        }
+        catch { }
+        // Limit file size to 10MB
+        if (doc.file_size && doc.file_size > 10 * 1024 * 1024) {
+            await ctx.reply("❌ File too large (max 10MB).", { reply_parameters: { message_id: userMessageId } });
+            return;
+        }
+        try {
+            const file = await ctx.api.getFile(doc.file_id);
+            const filePath = file.file_path;
+            if (!filePath) {
+                await ctx.reply("❌ Could not download file.", { reply_parameters: { message_id: userMessageId } });
+                return;
+            }
+            const url = `https://api.telegram.org/file/bot${config.telegramBotToken}/${filePath}`;
+            const { mkdtempSync, writeFileSync } = await import("fs");
+            const { join } = await import("path");
+            const { tmpdir } = await import("os");
+            const tmpDir = mkdtempSync(join(tmpdir(), "nzb-doc-"));
+            const localPath = join(tmpDir, doc.file_name || "file");
+            const response = await fetch(url);
+            const buffer = Buffer.from(await response.arrayBuffer());
+            writeFileSync(localPath, buffer);
+            const prompt = `[User sent a file: ${doc.file_name || "unknown"} (${doc.file_size || 0} bytes), saved at: ${localPath}]\n\nCaption: ${caption}\n\nPlease analyze this file. You can read it with bash tools.`;
+            sendToOrchestrator(prompt, { type: "telegram", chatId, messageId: userMessageId }, (text, done) => {
+                if (done) {
+                    const formatted = toTelegramMarkdown(text);
+                    const chunks = chunkMessage(formatted);
+                    const fallbackChunks = chunkMessage(text);
+                    void (async () => {
+                        for (let i = 0; i < chunks.length; i++) {
+                            if (i > 0)
+                                await new Promise((r) => setTimeout(r, 300));
+                            const pageTag = chunks.length > 1 ? `📄 ${i + 1}/${chunks.length}\n` : "";
+                            try {
+                                await ctx.api.sendMessage(chatId, pageTag + chunks[i], {
+                                    parse_mode: "MarkdownV2",
+                                    reply_parameters: i === 0 ? { message_id: userMessageId } : undefined,
+                                });
+                            }
+                            catch {
+                                try {
+                                    await ctx.api.sendMessage(chatId, pageTag + (fallbackChunks[i] ?? chunks[i]), {
+                                        reply_parameters: i === 0 ? { message_id: userMessageId } : undefined,
+                                    });
+                                }
+                                catch { }
+                            }
+                        }
+                        try {
+                            await ctx.api.setMessageReaction(chatId, userMessageId, [{ type: "emoji", emoji: "👍" }]);
+                        }
+                        catch { }
+                    })();
+                }
+            });
+        }
+        catch (err) {
+            await ctx.reply(`❌ Error processing file: ${err instanceof Error ? err.message : String(err)}`, {
+                reply_parameters: { message_id: userMessageId },
+            });
+        }
+    });
     return bot;
 }
 export async function startBot() {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@iletai/nzb",
-	"version": "1.2.6",
+	"version": "1.3.1",
 	"description": "NZB — a personal AI assistant for developers, built on the GitHub Copilot SDK",
 	"bin": {
 		"nzb": "dist/cli.js"
@@ -48,6 +48,7 @@
 	"dependencies": {
 		"@github/copilot-sdk": "^0.1.26",
 		"@grammyjs/auto-retry": "^2.0.2",
+		"@grammyjs/menu": "^1.3.1",
 		"better-sqlite3": "^12.6.2",
 		"dotenv": "^17.3.1",
 		"express": "^5.2.1",