@aion0/forge 0.2.1 → 0.2.3

package/lib/telegram-bot.ts

@@ -39,6 +39,9 @@ const chatListMode = new Map<number, 'tasks' | 'projects' | 'sessions' | 'task-c
 // Pending task creation: waiting for prompt text
 const pendingTaskProject = new Map<number, { name: string; path: string }>();  // chatId → project
 
+// Pending note: waiting for content
+const pendingNote = new Set<number>(); // chatIds waiting for note content
+
 // Buffer for streaming logs
 const logBuffers = new Map<string, { entries: string[]; timer: ReturnType<typeof setTimeout> | null }>();
 
@@ -112,10 +115,25 @@ async function poll() {
 
 async function handleMessage(msg: any) {
   const chatId = msg.chat.id;
+
+  // Whitelist check — only allow configured chat IDs, block all if not configured
+  const settings = loadSettings();
+  const allowedIds = settings.telegramChatId.split(',').map((s: string) => s.trim()).filter(Boolean);
+  if (allowedIds.length === 0 || !allowedIds.includes(String(chatId))) {
+    return;
+  }
+
   // Message received (logged silently)
   const text: string = msg.text.trim();
   const replyTo = msg.reply_to_message?.message_id;
 
+  // Check if waiting for note content
+  if (pendingNote.has(chatId) && !text.startsWith('/')) {
+    pendingNote.delete(chatId);
+    await sendNoteToDocsClaude(chatId, text);
+    return;
+  }
+
   // Check if waiting for task prompt
   const pending = pendingTaskProject.get(chatId);
   if (pending && !text.startsWith('/')) {
@@ -186,6 +204,7 @@ async function handleMessage(msg: any) {
   if (text.startsWith('/')) {
     // Any new command cancels pending states
     pendingTaskProject.delete(chatId);
+    pendingNote.delete(chatId);
 
     const [cmd, ...args] = text.split(/\s+/);
     switch (cmd) {
@@ -238,6 +257,10 @@ async function handleMessage(msg: any) {
       case '/doc':
         await handleDocs(chatId, args.join(' '));
         break;
+      case '/note':
+      case '/docs_write':
+        await handleDocsWrite(chatId, args.join(' '));
+        break;
       case '/cancel':
         await handleCancel(chatId, args[0]);
         break;
@@ -293,7 +316,8 @@ async function sendHelp(chatId: number) {
     `📝 Submit task:\nproject-name: your instructions\n\n` +
     `👀 /peek [project] [sessionId] — session summary\n` +
     `📖 /docs — docs session summary\n` +
-    `/docs <filename> — view doc file\n\n` +
+    `/docs <filename> — view doc file\n` +
+    `📝 /note — quick note to docs claude\n\n` +
     `🔧 /cancel <id>  /retry <id>\n` +
     `/projects — list projects\n\n` +
     `🌐 /tunnel — tunnel status\n` +
@@ -1151,6 +1175,79 @@ async function handleDocs(chatId: number, input: string) {
   }
 }
 
+// ─── Docs Write (Quick Notes) ────────────────────────────────
+
+async function handleDocsWrite(chatId: number, content: string) {
+  const settings = loadSettings();
+  if (String(chatId) !== settings.telegramChatId) { await send(chatId, '⛔ Unauthorized'); return; }
+
+  if (!content) {
+    pendingNote.add(chatId);
+    await send(chatId, '📝 Send your note content:');
+    return;
+  }
+
+  await sendNoteToDocsClaude(chatId, content);
+}
+
+async function sendNoteToDocsClaude(chatId: number, content: string) {
+  const settings = loadSettings();
+  const docRoots = (settings.docRoots || []).map((r: string) => r.replace(/^~/, require('os').homedir()));
+
+  if (docRoots.length === 0) {
+    await send(chatId, '⚠️ No document directories configured.');
+    return;
+  }
+
+  const { execSync, spawnSync } = require('child_process');
+  const { writeFileSync, unlinkSync } = require('fs');
+  const { join } = require('path');
+  const { homedir } = require('os');
+  const SESSION_NAME = 'mw-docs-claude';
+
+  // Check if the docs tmux session exists
+  let sessionExists = false;
+  try {
+    execSync(`tmux has-session -t ${SESSION_NAME} 2>/dev/null`);
+    sessionExists = true;
+  } catch {}
+
+  if (!sessionExists) {
+    await send(chatId, '⚠️ Docs Claude session not running. Open the Docs tab first to start it.');
+    return;
+  }
+
+  // Check if Claude is the active process (not shell)
+  let paneCmd = '';
+  try {
+    paneCmd = execSync(`tmux display-message -p -t ${SESSION_NAME} '#{pane_current_command}'`, { encoding: 'utf-8', timeout: 2000 }).trim();
+  } catch {}
+
+  if (paneCmd === 'zsh' || paneCmd === 'bash' || paneCmd === 'fish' || !paneCmd) {
+    await send(chatId, '⚠️ Claude is not running in the Docs session. Open the Docs tab and start Claude first.');
+    return;
+  }
+
+  // Write content to a temp file, then use tmux to send a prompt referencing it
+  const tmpFile = join(homedir(), '.forge', '.note-tmp.txt');
+  try {
+    writeFileSync(tmpFile, content, 'utf-8');
+
+    // Send a single-line prompt to Claude via tmux send-keys using the temp file
+    const prompt = `Please read the file ${tmpFile} and save its content as a note in the appropriate location in my docs. Analyze the content to determine the best file and location. After saving, delete the temp file.`;
+
+    // Use tmux send-keys with literal flag to avoid interpretation issues
+    spawnSync('tmux', ['send-keys', '-t', SESSION_NAME, '-l', prompt], { timeout: 5000 });
+    // Send Enter separately
+    spawnSync('tmux', ['send-keys', '-t', SESSION_NAME, 'Enter'], { timeout: 2000 });
+
+    await send(chatId, `📝 Note sent to Docs Claude:\n\n${content.slice(0, 200)}${content.length > 200 ? '...' : ''}`);
+  } catch (err) {
+    try { unlinkSync(tmpFile); } catch {}
+    await send(chatId, '❌ Failed to send note to Claude session');
+  }
+}
+
 // ─── Real-time Streaming ─────────────────────────────────────
 
 function bufferLogEntry(taskId: string, chatId: number, entry: TaskLogEntry) {
@@ -1283,6 +1380,7 @@ async function setBotCommands(token: string) {
           { command: 'tunnel_password', description: 'Get login password' },
           { command: 'peek', description: 'Session summary (AI + recent)' },
           { command: 'docs', description: 'Docs session summary / view file' },
+          { command: 'note', description: 'Quick note to docs Claude' },
           { command: 'watch', description: 'Monitor session' },
           { command: 'watchers', description: 'List watchers' },
           { command: 'help', description: 'Show help' },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aion0/forge",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "Unified AI workflow platform — multi-model task orchestration, persistent sessions, web terminal, remote access",
   "type": "module",
   "scripts": {
@@ -34,6 +34,7 @@
     "@auth/core": "^0.34.3",
     "@xterm/addon-fit": "^0.11.0",
     "@xterm/xterm": "^6.0.0",
+    "@xyflow/react": "^12.10.1",
     "ai": "^6.0.116",
     "better-sqlite3": "^12.6.2",
     "next": "^16.1.6",