polygram 0.6.0 → 0.6.3

package/lib/db.js

@@ -8,7 +8,7 @@ const fs = require('fs');
 const path = require('path');
 const Database = require('better-sqlite3');
 
-const SCHEMA_VERSION = 7;
+const SCHEMA_VERSION = 8;
 
 function open(dbPath) {
   const db = new Database(dbPath);
@@ -56,14 +56,16 @@ function runMigrations(db, migrationsDir) {
 }
 
 function wrap(db) {
+  // 0.6.1: attachments_json column dropped (migration 008). All attachment
+  // data lives in the per-attachment table now (see attachments stmts below).
   const insertMessageStmt = db.prepare(`
     INSERT INTO messages (
       chat_id, thread_id, msg_id, user, user_id, text, reply_to_id,
-      direction, source, bot_name, attachments_json, session_id,
+      direction, source, bot_name, session_id,
       model, effort, turn_id, status, error, cost_usd, ts
     ) VALUES (
       @chat_id, @thread_id, @msg_id, @user, @user_id, @text, @reply_to_id,
-      @direction, @source, @bot_name, @attachments_json, @session_id,
+      @direction, @source, @bot_name, @session_id,
       @model, @effort, @turn_id, @status, @error, @cost_usd, @ts
     )
     ON CONFLICT(chat_id, msg_id) DO UPDATE SET
@@ -121,8 +123,7 @@ function wrap(db) {
 
   const setMessageTextStmt = db.prepare(`
     UPDATE messages
-       SET text = @text,
-           attachments_json = COALESCE(@attachments_json, attachments_json)
+       SET text = @text
      WHERE chat_id = @chat_id AND msg_id = @msg_id
   `);
 
@@ -174,7 +175,6 @@ function wrap(db) {
         direction: row.direction || 'in',
         source: row.source || 'polygram',
         bot_name: row.bot_name || null,
-        attachments_json: row.attachments_json || null,
         session_id: row.session_id || null,
         model: row.model || null,
         effort: row.effort || null,
@@ -240,12 +240,11 @@ function wrap(db) {
       return getMessageStmt.get(String(chatId), msgId);
     },
 
-    setMessageText({ chat_id, msg_id, text, attachments_json = null }) {
+    setMessageText({ chat_id, msg_id, text }) {
       return setMessageTextStmt.run({
         chat_id: String(chat_id),
         msg_id,
         text: text ?? '',
-        attachments_json,
       });
     },
 
@@ -324,7 +323,7 @@ function wrap(db) {
       const placeholders = chatIds.map(() => '?').join(',');
       return db.prepare(`
         SELECT id, chat_id, thread_id, msg_id, user, user_id, text, reply_to_id,
-               attachments_json, ts, handler_status
+               ts, handler_status
           FROM messages
          WHERE direction = 'in'
            AND handler_status IN ('dispatched', 'processing', 'replay-pending')
@@ -464,6 +463,24 @@ function wrap(db) {
       return db.prepare(sql).all(...args);
     },
 
+    // Re-FK every attachment whose (chat_id, msg_id) is in `msg_ids` over
+    // to a single primary message row. Used when the media-group buffer
+    // coalesces N Telegram messages (each carrying one photo of an album)
+    // into one synthetic turn — siblings were recorded under their own
+    // msg_ids by recordInbound, but Claude needs to see them all under
+    // the primary message so handleMessage's per-message attachment
+    // lookup returns the full album.
+    reassignAttachmentsToMessage({ chat_id, msg_ids, target_message_id }) {
+      if (!Array.isArray(msg_ids) || msg_ids.length === 0) return { changes: 0 };
+      const placeholders = msg_ids.map(() => '?').join(',');
+      return db.prepare(`
+        UPDATE attachments
+           SET message_id = ?, msg_id = (SELECT msg_id FROM messages WHERE id = ?)
+         WHERE chat_id = ? AND msg_id IN (${placeholders})
+           AND message_id != ?
+      `).run(target_message_id, target_message_id, String(chat_id), ...msg_ids, target_message_id);
+    },
+
     // Look up the messages.id auto-pk for an inbound message. Used by
     // recordInbound to FK attachments to the just-inserted message even
     // when an ON-CONFLICT update happened (lastInsertRowid is 0 in that

package/migrations/008-drop-attachments-json.sql

@@ -0,0 +1,12 @@
+-- Drop the legacy messages.attachments_json column. 0.6.0's migration 007
+-- created the per-attachment table and backfilled from this column; the
+-- column was kept for one minor as a safety net. polygram 0.6.1 reads
+-- exclusively from the attachments table now, so the column is dead code
+-- on the schema side and can go.
+--
+-- SQLite supports ALTER TABLE DROP COLUMN since 3.35 (well below
+-- better-sqlite3's bundled SQLite). The op rewrites the table in place,
+-- which is fine — `messages` is small enough that a one-time rewrite at
+-- migration time is cheaper than carrying the column around forever.
+
+ALTER TABLE messages DROP COLUMN attachments_json;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "polygram",
-  "version": "0.6.0",
+  "version": "0.6.3",
   "description": "Telegram daemon for Claude Code that preserves the OpenClaw per-chat session model. Migration path for OpenClaw users moving to Claude Code.",
   "main": "lib/ipc-client.js",
   "bin": {

package/polygram.js

@@ -220,10 +220,6 @@ function recordInbound(msg) {
       direction: 'in',
       source: 'polygram',
       bot_name: BOT_NAME,
-      // attachments_json kept temporarily as a fallback during the 0.6.0
-      // migration window; per-attachment rows below are the source of
-      // truth. Will be dropped in a follow-up minor.
-      attachments_json: attachments.length ? JSON.stringify(attachments) : null,
       model: chatConfig?.model || null,
       effort: chatConfig?.effort || null,
       ts,
@@ -380,9 +376,7 @@ async function transcribeVoiceAttachments(downloaded, { chatId, msgId, label, bo
   //     parses it back when building the prompt.
   //   - Message-level: setMessageText updates messages.text with the
   //     combined transcript so FTS finds "what Maria said" via the
-  //     normal chat search path. attachments_json is left as-is (will
-  //     be dropped in a future minor; per-attachment row is the source
-  //     of truth).
+  //     normal chat search path.
   const successful = targets.filter((a) => a.transcription?.text);
   if (!successful.length) return;
   for (const a of successful) {
@@ -393,8 +387,7 @@ async function transcribeVoiceAttachments(downloaded, { chatId, msgId, label, bo
   }
   const combinedText = successful.map((a) => a.transcription.text).join(' ').trim();
   dbWrite(() => db.setMessageText({
-    chat_id: chatId, msg_id: msgId,
-    text: combinedText, attachments_json: null,
+    chat_id: chatId, msg_id: msgId, text: combinedText,
   }), 'persist voice transcription');
 }
 
@@ -1937,6 +1930,29 @@ function createBot(token) {
       // all-media-no-text groups.
       const primary = messages.find((m) => m.text || m.caption) || messages[0];
       const merged = messages.flatMap((m) => extractAttachments(m));
+
+      // 0.6.0 attachment-table regression fix: recordInbound (called per
+      // sibling on bot.on('message')) inserted each photo's row under its
+      // OWN msg_id. handleMessage looks up attachments via
+      // getAttachmentsByMessage(primary.message_id) — which only returns
+      // the primary's row. Without re-FK'ing the siblings we'd silently
+      // drop N-1 of N photos in any album, exactly the umi-assistant bug
+      // the user hit (saw 1 of 2 photos sent in a Telegram album).
+      const chatId = String(primary.chat.id);
+      const primaryDbId = db.getInboundMessageId({
+        chat_id: chatId, msg_id: primary.message_id,
+      });
+      const siblingMsgIds = messages
+        .filter((m) => m.message_id !== primary.message_id)
+        .map((m) => m.message_id);
+      if (primaryDbId && siblingMsgIds.length) {
+        dbWrite(() => db.reassignAttachmentsToMessage({
+          chat_id: chatId,
+          msg_ids: siblingMsgIds,
+          target_message_id: primaryDbId,
+        }), 'reassign media-group sibling attachments');
+      }
+
       const synthetic = { ...primary, _mergedAttachments: merged };
       // Carry the primary's text verbatim (dispatchRegularMessage re-cleans
       // the mention). Caption → text so downstream sees it uniformly.
@@ -2434,18 +2450,12 @@ async function main() {
         // Attach already-recorded attachments via the media-group shortcut
         // field so extractAttachments picks them up without re-parsing
         // grammy fields that don't exist on this reconstructed object.
-        // 0.6.0: read from the per-attachment table; fall back to the
-        // legacy attachments_json blob for rows inserted before migration
-        // 007 ran (covers the small window during the upgrade).
         const attRows = db.getAttachmentsByMessage(row.id);
         if (attRows.length) {
           reconstructed._mergedAttachments = attRows.map((a) => ({
             kind: a.kind, name: a.name, mime_type: a.mime_type,
             size: a.size_bytes, file_id: a.file_id, file_unique_id: a.file_unique_id,
           }));
-        } else if (row.attachments_json) {
-          try { reconstructed._mergedAttachments = JSON.parse(row.attachments_json); }
-          catch {}
         }
         const chatConfig = config.chats[row.chat_id];
         if (!chatConfig) { skipped += 1; continue; }

package/scripts/split-db.js

@@ -148,18 +148,39 @@ function copy(src, dst, bot, chatToBot) {
       const rows = src.raw.prepare(
         `SELECT * FROM messages WHERE chat_id IN (${ph(chatIds.length)}) OR bot_name = ?`,
       ).all(...chatIds, bot);
+      // 0.6.1: messages.attachments_json column was dropped (migration 008).
+      // Per-attachment rows live in the `attachments` table now and are
+      // copied separately below.
       const ins = dst.raw.prepare(`
         INSERT OR IGNORE INTO messages
           (id, chat_id, thread_id, msg_id, user, user_id, text, reply_to_id,
-           direction, source, bot_name, attachments_json, session_id,
+           direction, source, bot_name, session_id,
            model, effort, turn_id, status, error, cost_usd, ts, edited_ts)
         VALUES
           (@id, @chat_id, @thread_id, @msg_id, @user, @user_id, @text, @reply_to_id,
-           @direction, @source, @bot_name, @attachments_json, @session_id,
+           @direction, @source, @bot_name, @session_id,
            @model, @effort, @turn_id, @status, @error, @cost_usd, @ts, @edited_ts)
       `);
       for (const r of rows) { if (ins.run(r).changes) stats.messages++; }
 
+      // Copy per-attachment rows for the messages we just copied. FK target
+      // exists since messages were inserted in the same transaction.
+      const arows = src.raw.prepare(
+        `SELECT * FROM attachments WHERE chat_id IN (${ph(chatIds.length)})`,
+      ).all(...chatIds);
+      const aIns = dst.raw.prepare(`
+        INSERT OR IGNORE INTO attachments
+          (id, message_id, chat_id, msg_id, thread_id, bot_name,
+           file_id, file_unique_id, kind, name, mime_type, size_bytes,
+           local_path, download_status, download_error, transcription, ts)
+        VALUES
+          (@id, @message_id, @chat_id, @msg_id, @thread_id, @bot_name,
+           @file_id, @file_unique_id, @kind, @name, @mime_type, @size_bytes,
+           @local_path, @download_status, @download_error, @transcription, @ts)
+      `);
+      stats.attachments = 0;
+      for (const r of arows) { if (aIns.run(r).changes) stats.attachments++; }
+
       const srows = src.raw.prepare(`SELECT * FROM sessions WHERE chat_id IN (${ph(chatIds.length)})`).all(...chatIds);
       const sins = dst.raw.prepare(`
         INSERT OR REPLACE INTO sessions