paperkite 0.1.0

package/dist/packages/archive/src/storage/postgres.js

@@ -0,0 +1,1021 @@
+import { Pool } from "pg";
+import { albumEntryOf, buildContextEntries, entitiesJson, groupKey, mediaKey, normalizeBlockword, normalizeContextLimit, normalizeLimit, normalizeOffset, normalizeRecordId, normalizeTimeMode, normalizeUserId, paramPlaceholders, parseEntities, splitTerms, toIsoDate, uniqueGroupKeys } from "./model.js";
+const SCHEMA_PATTERN = /^[A-Za-z_][A-Za-z0-9_]*$/;
+export class PostgresArchiveStore {
+    url;
+    pool;
+    schemaName;
+    schema;
+    /** 屏蔽词内存缓存（小写归一）；读路径条件固定为 blocked = FALSE，与词数无关。 */
+    blockwords = [];
+    blockwordsVersion = 0;
+    /** 屏蔽用户内存缓存（sender_id 精确匹配）。 */
+    blockedUsers = [];
+    blockedUsersVersion = 0;
+    constructor(url, schema = "public") {
+        this.url = url;
+        if (!SCHEMA_PATTERN.test(schema))
+            throw new Error(`invalid schema name: ${schema}`);
+        this.schemaName = schema;
+        this.schema = quoteIdentifier(schema);
+        this.pool = new Pool({ connectionString: url });
+    }
+    async init() {
+        const client = await this.pool.connect();
+        try {
+            await client.query(`CREATE SCHEMA IF NOT EXISTS ${this.schema}`);
+            await client.query(`
+        CREATE TABLE IF NOT EXISTS ${this.table("messages")} (
+          id BIGINT GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY,
+          message_id BIGINT NOT NULL,
+          chat_id TEXT NOT NULL,
+          grouped_id TEXT,
+          chat_title TEXT,
+          sender_id TEXT,
+          sender_username TEXT,
+          sender_first_name TEXT,
+          sender_last_name TEXT,
+          date TIMESTAMPTZ NOT NULL,
+          text TEXT,
+          message_type TEXT DEFAULT 'text',
+          reply_to_msg_id BIGINT,
+          forward_from_id TEXT,
+          forward_from_name TEXT,
+          entities TEXT,
+          has_media BOOLEAN DEFAULT FALSE,
+          media_type TEXT,
+          media_path TEXT,
+          created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP,
+          UNIQUE(message_id, chat_id)
+        );
+        CREATE INDEX IF NOT EXISTS idx_messages_chat_date
+          ON ${this.table("messages")} (chat_id, date);
+        CREATE INDEX IF NOT EXISTS idx_messages_chat_date_id
+          ON ${this.table("messages")} (chat_id, date, id);
+        CREATE INDEX IF NOT EXISTS idx_messages_chat_grouped
+          ON ${this.table("messages")} (chat_id, grouped_id);
+        CREATE INDEX IF NOT EXISTS idx_messages_chat_grouped_date_id
+          ON ${this.table("messages")} (chat_id, grouped_id, date, id);
+        CREATE INDEX IF NOT EXISTS idx_messages_date_id
+          ON ${this.table("messages")} (date, id);
+        CREATE INDEX IF NOT EXISTS idx_messages_sender
+          ON ${this.table("messages")} (sender_id);
+        CREATE INDEX IF NOT EXISTS idx_messages_chat
+          ON ${this.table("messages")} (chat_id);
+        CREATE INDEX IF NOT EXISTS idx_messages_chat_blocked
+          ON ${this.table("messages")} (chat_id, blocked);
+        CREATE INDEX IF NOT EXISTS idx_messages_chat_grouped_blocked
+          ON ${this.table("messages")} (chat_id, grouped_id, blocked);
+        CREATE INDEX IF NOT EXISTS idx_messages_chat_reply
+          ON ${this.table("messages")} (chat_id, reply_to_msg_id);
+        CREATE INDEX IF NOT EXISTS idx_messages_forward_from
+          ON ${this.table("messages")} (forward_from_id);
+        CREATE TABLE IF NOT EXISTS ${this.table("media_files")} (
+          id BIGINT GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY,
+          message_id BIGINT NOT NULL,
+          chat_id TEXT NOT NULL,
+          media_type TEXT NOT NULL,
+          file_name TEXT NOT NULL,
+          file_path TEXT NOT NULL,
+          file_size BIGINT,
+          mime_type TEXT,
+          created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP,
+          FOREIGN KEY (message_id, chat_id)
+            REFERENCES ${this.table("messages")} (message_id, chat_id)
+            ON DELETE CASCADE
+        );
+        CREATE INDEX IF NOT EXISTS idx_media_message
+          ON ${this.table("media_files")} (message_id, chat_id);
+        CREATE TABLE IF NOT EXISTS ${this.table("chats")} (
+          id BIGINT GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY,
+          chat_id TEXT UNIQUE NOT NULL,
+          title TEXT,
+          username TEXT,
+          type TEXT,
+          description TEXT,
+          members_count BIGINT,
+          last_sync TIMESTAMPTZ,
+          created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP
+        );
+        CREATE TABLE IF NOT EXISTS ${this.table("sync_history")} (
+          id BIGINT GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY,
+          chat_id TEXT NOT NULL,
+          start_date TIMESTAMPTZ,
+          end_date TIMESTAMPTZ,
+          messages_count BIGINT,
+          media_count BIGINT,
+          sync_started_at TIMESTAMPTZ,
+          sync_completed_at TIMESTAMPTZ,
+          status TEXT DEFAULT 'running'
+        );
+        CREATE TABLE IF NOT EXISTS ${this.table("blockwords")} (
+          word TEXT PRIMARY KEY,
+          created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP
+        );
+        CREATE TABLE IF NOT EXISTS ${this.table("blocked_users")} (
+          user_id TEXT PRIMARY KEY,
+          name TEXT,
+          username TEXT,
+          created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP
+        );
+      `);
+            await this.ensureLegacyColumns(client);
+            await this.ensureBlockedIndexes(client);
+            const words = await client.query(`SELECT word FROM ${this.table("blockwords")} ORDER BY word`);
+            this.blockwords = words.rows.map((row) => String(row.word));
+            const users = await client.query(`SELECT user_id, name, username FROM ${this.table("blocked_users")} ORDER BY user_id`);
+            this.blockedUsers = users.rows.map((row) => toBlockedUserInfo(row));
+        }
+        finally {
+            client.release();
+        }
+    }
+    /** 老库迁移：messages 补 blocked 标志列与 entities 实体列。 */
+    async ensureLegacyColumns(client) {
+        const exists = await client.query(`SELECT column_name FROM information_schema.columns
+        WHERE table_schema = $1 AND table_name = 'messages'`, [this.schemaName]);
+        const names = new Set(exists.rows.map((row) => String(row.column_name)));
+        if (!names.has("blocked")) {
+            await client.query(`ALTER TABLE ${this.table("messages")} ADD COLUMN blocked BOOLEAN NOT NULL DEFAULT FALSE`);
+        }
+        if (!names.has("entities")) {
+            await client.query(`ALTER TABLE ${this.table("messages")} ADD COLUMN entities TEXT`);
+        }
+    }
+    /** blocked 标志维护的部分索引：置位只扫未屏蔽行，解锁只扫已屏蔽行。 */
+    async ensureBlockedIndexes(client) {
+        await client.query(`CREATE INDEX IF NOT EXISTS idx_messages_blocked_zero
+         ON ${this.table("messages")} (blocked) WHERE blocked = FALSE`);
+        await client.query(`CREATE INDEX IF NOT EXISTS idx_messages_blocked_one
+         ON ${this.table("messages")} (blocked) WHERE blocked = TRUE`);
+    }
+    async close() {
+        await this.pool.end();
+    }
+    async saveChat(chat) {
+        await this.pool.query(`INSERT INTO ${this.table("chats")}
+        (chat_id, title, username, type, description, members_count, last_sync)
+       VALUES ($1, $2, $3, $4, $5, $6, $7)
+       ON CONFLICT (chat_id) DO UPDATE SET
+         title = EXCLUDED.title,
+         username = EXCLUDED.username,
+         type = EXCLUDED.type,
+         description = EXCLUDED.description,
+         members_count = EXCLUDED.members_count,
+         last_sync = EXCLUDED.last_sync`, [chat.chatId, chat.title ?? null, chat.username ?? null, chat.type ?? null,
+            chat.description ?? null, chat.membersCount ?? null, new Date().toISOString()]);
+        return true;
+    }
+    async startSyncSession(chatId, startDate, endDate) {
+        const result = await this.pool.query(`INSERT INTO ${this.table("sync_history")}
+        (chat_id, start_date, end_date, sync_started_at, status)
+       VALUES ($1, $2::timestamptz, $3::timestamptz, $4, 'running')
+       RETURNING id`, [chatId, startDate, endDate, new Date().toISOString()]);
+        return Number(result.rows[0]?.id ?? -1);
+    }
+    async completeSyncSession(sessionId, messagesCount, mediaCount) {
+        if (sessionId < 0)
+            return;
+        await this.pool.query(`UPDATE ${this.table("sync_history")}
+          SET sync_completed_at = $1, messages_count = $2, media_count = $3, status = 'completed'
+        WHERE id = $4`, [new Date().toISOString(), messagesCount, mediaCount, sessionId]);
+    }
+    async getLastMessageInfo(chatId) {
+        const result = await this.pool.query(`SELECT message_id, date FROM ${this.table("messages")}
+        WHERE chat_id = $1
+        ORDER BY date DESC, message_id DESC
+        LIMIT 1`, [chatId]);
+        const row = result.rows[0];
+        return row ? { messageId: Number(row.message_id), date: toIsoDate(row.date) } : undefined;
+    }
+    async getChatUsername(chatId) {
+        const result = await this.pool.query(`SELECT username FROM ${this.table("chats")} WHERE chat_id = $1`, [chatId]);
+        const username = result.rows[0]?.username?.trim();
+        return username || undefined;
+    }
+    async messageIdsExist(chatId, messageIds) {
+        const ids = [...new Set(messageIds.map(Number))];
+        if (!ids.length)
+            return new Set();
+        const placeholders = ids.map((_, index) => `$${index + 2}`).join(", ");
+        const result = await this.pool.query(`SELECT message_id FROM ${this.table("messages")}
+        WHERE chat_id = $1 AND message_id IN (${placeholders})`, [chatId, ...ids]);
+        return new Set(result.rows.map((row) => Number(row.message_id)));
+    }
+    async saveBatch(messages, media) {
+        if (!messages.length)
+            return { messages: 0, media: 0 };
+        const client = await this.pool.connect();
+        try {
+            await client.query("BEGIN");
+            const messagePlaceholders = messages
+                .map((_, index) => `(${paramPlaceholders(index * 20 + 1, 20)})`)
+                .join(", ");
+            const messageValues = [];
+            for (const row of messages) {
+                messageValues.push(row.messageId, row.chatId, row.groupedId ?? null, row.chatTitle ?? null, row.senderId ?? null, row.senderUsername ?? null, row.senderFirstName ?? null, row.senderLastName ?? null, toIsoDate(row.date), row.text, entitiesJson(row.entities), row.messageType ?? "text", row.replyToMessageId ?? null, row.forwardFromId ?? null, row.forwardFromName ?? null, row.hasMedia, row.mediaType ?? null, row.mediaFilePath ?? null, new Date().toISOString(), blockedOf(row, this.blockwords, this.blockedUsers));
+            }
+            const inserted = await client.query(`INSERT INTO ${this.table("messages")}
+          (message_id, chat_id, grouped_id, chat_title, sender_id, sender_username,
+           sender_first_name, sender_last_name, date, text, entities, message_type,
+           reply_to_msg_id, forward_from_id, forward_from_name,
+           has_media, media_type, media_path, created_at, blocked)
+         VALUES ${messagePlaceholders}
+         ON CONFLICT (message_id, chat_id) DO NOTHING`, messageValues);
+            let mediaInserted = 0;
+            if (media.length) {
+                const mediaPlaceholders = media
+                    .map((_, index) => `(${paramPlaceholders(index * 7 + 1, 7)})`)
+                    .join(", ");
+                const mediaValues = [];
+                for (const row of media) {
+                    mediaValues.push(row.messageId, row.chatId, row.mediaType, row.fileName, row.filePath, row.fileSize ?? null, row.mimeType ?? null);
+                }
+                const mediaResult = await client.query(`INSERT INTO ${this.table("media_files")}
+            (message_id, chat_id, media_type, file_name, file_path, file_size, mime_type)
+           VALUES ${mediaPlaceholders}`, mediaValues);
+                mediaInserted = mediaResult.rowCount ?? 0;
+            }
+            await client.query("COMMIT");
+            return { messages: inserted.rowCount ?? 0, media: mediaInserted };
+        }
+        catch (error) {
+            await client.query("ROLLBACK");
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    async listBlockwords() {
+        return { words: [...this.blockwords], version: this.blockwordsVersion };
+    }
+    async addBlockword(word) {
+        const normalized = normalizeBlockword(word);
+        if (normalized === undefined)
+            return "invalid";
+        const client = await this.pool.connect();
+        try {
+            await client.query("BEGIN");
+            const result = await client.query(`INSERT INTO ${this.table("blockwords")} (word) VALUES ($1)
+         ON CONFLICT (word) DO NOTHING`, [normalized]);
+            if (result.rowCount === 0) {
+                await client.query("ROLLBACK");
+                return "exists";
+            }
+            await this.markBlockwordMatches(client, normalized);
+            this.blockwords = [...this.blockwords, normalized];
+            this.blockwordsVersion += 1;
+            await client.query("COMMIT");
+            return "added";
+        }
+        catch (error) {
+            await client.query("ROLLBACK");
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    async removeBlockword(word) {
+        const normalized = normalizeBlockword(word);
+        if (normalized === undefined)
+            return false;
+        const client = await this.pool.connect();
+        try {
+            await client.query("BEGIN");
+            const result = await client.query(`DELETE FROM ${this.table("blockwords")} WHERE word = $1`, [normalized]);
+            if (result.rowCount === 0) {
+                await client.query("ROLLBACK");
+                return false;
+            }
+            const next = this.blockwords.filter((item) => item !== normalized);
+            await this.unmarkBlockwordMatches(client, normalized, next);
+            this.blockwords = next;
+            this.blockwordsVersion += 1;
+            await client.query("COMMIT");
+            return true;
+        }
+        catch (error) {
+            await client.query("ROLLBACK");
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    async listBlockedUsers() {
+        return { users: [...this.blockedUsers], version: this.blockedUsersVersion };
+    }
+    async addBlockedUser(input) {
+        const userId = normalizeUserId(input.userId);
+        if (userId === undefined)
+            return "invalid";
+        const client = await this.pool.connect();
+        try {
+            await client.query("BEGIN");
+            const result = await client.query(`INSERT INTO ${this.table("blocked_users")} (user_id, name, username) VALUES ($1, $2, $3)
+         ON CONFLICT (user_id) DO NOTHING`, [userId, input.name?.trim() || null, input.username?.trim() || null]);
+            if (result.rowCount === 0) {
+                await client.query("ROLLBACK");
+                return "exists";
+            }
+            await this.markBlockedUserMatches(client, userId);
+            this.blockedUsers = [...this.blockedUsers, toBlockedUserInfo({
+                    user_id: userId,
+                    name: input.name?.trim() || null,
+                    username: input.username?.trim() || null
+                })];
+            this.blockedUsersVersion += 1;
+            await client.query("COMMIT");
+            return "added";
+        }
+        catch (error) {
+            await client.query("ROLLBACK");
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    async removeBlockedUser(userId) {
+        const normalized = normalizeUserId(userId);
+        if (normalized === undefined)
+            return false;
+        const client = await this.pool.connect();
+        try {
+            await client.query("BEGIN");
+            const result = await client.query(`DELETE FROM ${this.table("blocked_users")} WHERE user_id = $1`, [normalized]);
+            if (result.rowCount === 0) {
+                await client.query("ROLLBACK");
+                return false;
+            }
+            const next = this.blockedUsers.filter((user) => user.userId !== normalized);
+            await this.unmarkBlockedUserMatches(client, normalized, next);
+            this.blockedUsers = next;
+            this.blockedUsersVersion += 1;
+            await client.query("COMMIT");
+            return true;
+        }
+        catch (error) {
+            await client.query("ROLLBACK");
+            throw error;
+        }
+        finally {
+            client.release();
+        }
+    }
+    /** 增量置位：只置未屏蔽行中命中新词的行的标志。 */
+    async markBlockwordMatches(client, word) {
+        await client.query(`UPDATE ${this.table("messages")} SET blocked = TRUE
+        WHERE id IN (
+          SELECT id FROM ${this.table("messages")}
+           WHERE blocked = FALSE AND strpos(lower(COALESCE(text, '')), $1) > 0
+        )`, [word]);
+    }
+    /** 增量解锁：只重算已屏蔽且命中被删词的行，其余行不受影响。 */
+    async unmarkBlockwordMatches(client, word, remaining) {
+        const holds = [
+            ...remaining.map((_, index) => `strpos(lower(COALESCE(text, '')), $${index + 2}) > 0`),
+            ...this.blockedUsers.map((_, index) => `COALESCE(sender_id::text, '') = $${remaining.length + index + 2}`)
+        ];
+        const values = [...remaining, ...this.blockedUsers.map((user) => user.userId)];
+        await this.runUnblock(client, `strpos(lower(COALESCE(text, '')), $1) > 0`, word, holds, values);
+    }
+    /** 增量置位：只置未屏蔽行中发送者为新用户的行的标志。 */
+    async markBlockedUserMatches(client, userId) {
+        await client.query(`UPDATE ${this.table("messages")} SET blocked = TRUE
+        WHERE id IN (
+          SELECT id FROM ${this.table("messages")}
+           WHERE blocked = FALSE AND COALESCE(sender_id::text, '') = $1
+        )`, [userId]);
+    }
+    /** 增量解锁：只重算已屏蔽且发送者为被删用户的行。 */
+    async unmarkBlockedUserMatches(client, userId, remaining) {
+        const holds = [
+            ...remaining.map((_, index) => `COALESCE(sender_id::text, '') = $${index + 2}`),
+            ...this.blockwords.map((_, index) => `strpos(lower(COALESCE(text, '')), $${remaining.length + index + 2}) > 0`)
+        ];
+        const values = [...remaining.map((user) => user.userId), ...this.blockwords];
+        await this.runUnblock(client, "COALESCE(sender_id::text, '') = $1", userId, holds, values);
+    }
+    /**
+     * 增量解锁执行：match 命中已屏蔽行，剩余规则（holds）全不命中时解锁。
+     * 命中行先经子查询物化，避免 UPDATE 边扫索引边改 blocked 时漏行。
+     */
+    async runUnblock(client, match, matchValue, holds, holdValues) {
+        const inner = holds.length
+            ? `SELECT id FROM ${this.table("messages")} WHERE blocked = TRUE AND ${match} AND NOT (${holds.join(" OR ")})`
+            : `SELECT id FROM ${this.table("messages")} WHERE blocked = TRUE AND ${match}`;
+        await client.query(`UPDATE ${this.table("messages")} SET blocked = FALSE WHERE id IN (${inner})`, [matchValue, ...holdValues]);
+    }
+    async searchStructured(query) {
+        const { where: rowWhere, values: rowValues } = buildWhere(query);
+        const { where: memberWhere, values: memberValues } = buildWhere(query, "g");
+        const entry = buildSearchWhere(this.table("messages"), rowWhere, rowValues, memberWhere, memberValues);
+        const limit = normalizeLimit(query.limit);
+        const offset = normalizeOffset(query.offset);
+        const row = rowWhere.replace(/^WHERE\s+/, "");
+        const shifted = row.replace(/\$(\d+)/g, (_match, index) => `$${rowValues.length + Number(index)}`);
+        const [entryCount, messageCount] = await Promise.all([
+            this.pool.query(`SELECT (
+           (SELECT COUNT(*)::int FROM ${this.table("messages")} m
+             WHERE m.grouped_id IS NULL AND ${row})
+           + (SELECT COUNT(DISTINCT m.grouped_id)::int FROM ${this.table("messages")} m
+             WHERE m.grouped_id IS NOT NULL AND ${shifted})
+         )::int AS count`, [...rowValues, ...rowValues]),
+            this.pool.query(`SELECT COUNT(*)::int AS count FROM ${this.table("messages")} m ${rowWhere}`, rowValues)
+        ]);
+        const rows = await this.pool.query(`${messageSelect(this.table("messages"), this.table("media_files"))}
+       ${entry.where}
+       ORDER BY m.date DESC, m.id DESC
+       LIMIT $${entry.values.length + 1} OFFSET $${entry.values.length + 2}`, [...entry.values, limit, offset]);
+        const attached = await this.attachMedia(rows.rows);
+        const groupMap = await this.fetchGroupsAttached(uniqueGroupKeys(rows.rows));
+        return {
+            items: buildContextEntries(attached, groupMap),
+            total: Number(entryCount.rows[0]?.count ?? 0),
+            totalMessages: Number(messageCount.rows[0]?.count ?? 0),
+            limit,
+            offset
+        };
+    }
+    async listChatLedger(limit) {
+        const cap = Math.min(500, Math.max(1, Number.isInteger(limit) ? limit : 300));
+        const result = await this.pool.query(`SELECT
+         c.chat_id::text AS chat_id,
+         COALESCE(c.title, latest.chat_title) AS title,
+         c.username,
+         c.type,
+         c.description,
+         c.members_count,
+         agg.count,
+         latest.date AS last_date,
+         LEFT(latest.text, 120) AS last_text
+       FROM ${this.table("chats")} c
+       LEFT JOIN LATERAL (
+         SELECT COUNT(*)::int AS count
+           FROM ${this.table("messages")} m
+          WHERE m.chat_id = c.chat_id AND m.blocked = FALSE
+       ) agg ON TRUE
+       LEFT JOIN LATERAL (
+         SELECT chat_title, date, text
+           FROM ${this.table("messages")} m
+          WHERE m.chat_id = c.chat_id AND m.blocked = FALSE
+          ORDER BY m.date DESC, m.id DESC
+          LIMIT 1
+       ) latest ON TRUE
+       ORDER BY latest.date DESC NULLS LAST, c.chat_id
+       LIMIT $1`, [cap]);
+        return result.rows.map((row) => ({
+            chatId: String(row.chat_id),
+            title: String(row.title ?? row.chat_id),
+            username: optionalString(row.username),
+            type: optionalString(row.type),
+            description: optionalString(row.description),
+            membersCount: optionalNumber(row.members_count),
+            count: Number(row.count ?? 0),
+            lastDate: row.last_date === null || row.last_date === undefined ? undefined : toIsoDate(row.last_date),
+            lastText: optionalString(row.last_text)
+        }));
+    }
+    async getMessageByRecordId(recordId) {
+        const result = await this.pool.query(`${messageSelect(this.table("messages"), this.table("media_files"))}
+       WHERE m.id = $1::bigint AND m.blocked = FALSE
+       LIMIT 1`, [normalizeRecordId(recordId)]);
+        return result.rows.length ? (await this.attachMedia(result.rows))[0] : undefined;
+    }
+    async searchSenders(query) {
+        const limit = normalizeLimit(query.limit);
+        const conditions = ["m.blocked = FALSE"];
+        const values = [];
+        const q = (query.q ?? "").trim().replace(/^@/, "");
+        if (q) {
+            conditions.push(`(
+        m.sender_id = $${values.length + 1}
+        OR m.sender_id ILIKE $${values.length + 2}
+        OR m.sender_username ILIKE $${values.length + 3}
+        OR m.sender_first_name ILIKE $${values.length + 4}
+        OR m.sender_last_name ILIKE $${values.length + 5}
+      )`);
+            const like = `%${escapeLike(q)}%`;
+            values.push(q, like, like, like, like);
+        }
+        if (query.chatId?.trim()) {
+            conditions.push(`m.chat_id = $${values.length + 1}`);
+            values.push(query.chatId.trim());
+        }
+        const where = `WHERE ${conditions.join(" AND ")}`;
+        const total = await this.pool.query(`SELECT COUNT(DISTINCT m.sender_id)::int AS count FROM ${this.table("messages")} m ${where}`, values);
+        const rows = await this.pool.query(`SELECT m.sender_id,
+              MAX(COALESCE(m.sender_username, ''))::text AS sender_username,
+              MAX(COALESCE(m.sender_first_name, ''))::text AS sender_first_name,
+              MAX(COALESCE(m.sender_last_name, ''))::text AS sender_last_name,
+              COUNT(*)::int AS count
+         FROM ${this.table("messages")} m
+         ${where}
+        GROUP BY m.sender_id
+        ORDER BY count DESC, m.sender_id
+        LIMIT $${values.length + 1}`, [...values, limit]);
+        return {
+            items: rows.rows.filter((row) => row.sender_id !== null && row.sender_id !== "").map(toSenderInfo),
+            total: Number(total.rows[0]?.count ?? 0)
+        };
+    }
+    async getSenderSummary(senderId, chatId) {
+        const id = senderId.trim();
+        const exists = await this.pool.query(`SELECT 1 FROM ${this.table("messages")} WHERE sender_id = $1 LIMIT 1`, [id]);
+        if (exists.rows.length === 0)
+            return undefined;
+        const chatFilter = chatId?.trim() ? " AND m.chat_id = $2" : "";
+        const values = chatId?.trim() ? [id, chatId.trim()] : [id];
+        const stats = await this.pool.query(`SELECT COUNT(*)::int AS total, MIN(m.date) AS first_date, MAX(m.date) AS last_date
+         FROM ${this.table("messages")} m
+        WHERE m.sender_id = $1 AND m.blocked = FALSE${chatFilter}`, values);
+        const chatRows = await this.pool.query(`SELECT m.chat_id, c.title AS chat_title, COUNT(*)::int AS count,
+              MAX(m.date) AS last_date,
+              (
+                SELECT LEFT(m2.text, 120)
+                  FROM ${this.table("messages")} m2
+                 WHERE m2.chat_id = m.chat_id AND m2.sender_id = m.sender_id AND m2.blocked = FALSE
+                 ORDER BY m2.date DESC, m2.id DESC LIMIT 1
+              ) AS last_text
+         FROM ${this.table("messages")} m
+         LEFT JOIN ${this.table("chats")} c ON c.chat_id = m.chat_id
+        WHERE m.sender_id = $1 AND m.blocked = FALSE${chatFilter}
+        GROUP BY m.chat_id, c.title
+        ORDER BY last_date DESC, m.chat_id`, values);
+        const identity = await this.pool.query(`SELECT m.sender_id,
+              MAX(COALESCE(m.sender_username, ''))::text AS sender_username,
+              MAX(COALESCE(m.sender_first_name, ''))::text AS sender_first_name,
+              MAX(COALESCE(m.sender_last_name, ''))::text AS sender_last_name
+         FROM ${this.table("messages")} m
+        WHERE m.sender_id = $1
+        GROUP BY m.sender_id`, [id]);
+        const statRow = stats.rows[0];
+        return {
+            sender: toSenderInfo(identity.rows[0] ?? { sender_id: id }),
+            total: Number(statRow?.total ?? 0),
+            firstDate: statRow?.first_date === null || statRow?.first_date === undefined ? undefined : toIsoDate(statRow.first_date),
+            lastDate: statRow?.last_date === null || statRow?.last_date === undefined ? undefined : toIsoDate(statRow.last_date),
+            chats: chatRows.rows.map((row) => ({
+                chatId: String(row.chat_id),
+                chatTitle: optionalString(row.chat_title),
+                count: Number(row.count),
+                lastDate: row.last_date === null || row.last_date === undefined ? undefined : toIsoDate(row.last_date),
+                lastText: optionalString(row.last_text)
+            }))
+        };
+    }
+    async getReplyChain(recordId) {
+        const anchor = await this.getMessageByRecordId(recordId);
+        if (!anchor)
+            return undefined;
+        let parent;
+        if (anchor.replyToMessageId !== undefined) {
+            const result = await this.pool.query(`${messageSelect(this.table("messages"), this.table("media_files"))}
+         WHERE m.chat_id = $1 AND m.message_id = $2::bigint AND m.blocked = FALSE
+         LIMIT 1`, [anchor.chatId, anchor.replyToMessageId]);
+            if (result.rows.length > 0) {
+                const records = await this.attachMedia(result.rows);
+                const parentRecord = records[0];
+                if (parentRecord?.groupedId !== undefined) {
+                    const group = await this.fetchGroupAttached(parentRecord.chatId, parentRecord.groupedId);
+                    parent = group.length > 1
+                        ? albumEntryOf(group, parentRecord.recordId)
+                        : { kind: "message", record: parentRecord };
+                }
+                else if (parentRecord !== undefined) {
+                    parent = { kind: "message", record: parentRecord };
+                }
+            }
+        }
+        const childRows = await this.pool.query(`${messageSelect(this.table("messages"), this.table("media_files"))}
+       WHERE m.chat_id = $1 AND m.reply_to_msg_id = $2::bigint AND m.blocked = FALSE
+       ORDER BY m.date ASC, m.id ASC
+       LIMIT 100`, [anchor.chatId, anchor.messageId]);
+        const children = childRows.rows.length
+            ? buildContextEntries(await this.attachMedia(childRows.rows), await this.fetchGroupsAttached(uniqueGroupKeys(childRows.rows)))
+            : [];
+        return {
+            parent,
+            children,
+            replyToMessageId: anchor.replyToMessageId
+        };
+    }
+    async resolveMessageRef(ref) {
+        let chatId = ref.chatId?.trim() || "";
+        if (!chatId && ref.username?.trim()) {
+            const row = await this.pool.query(`SELECT chat_id FROM ${this.table("chats")} WHERE username = $1 LIMIT 1`, [ref.username.trim()]);
+            chatId = row.rows.length ? String(row.rows[0].chat_id) : "";
+        }
+        if (!chatId && ref.code?.trim()) {
+            const row = await this.pool.query(`SELECT chat_id FROM ${this.table("chats")} WHERE chat_id IN (('-100' || $1)::bigint, ('-' || $1)::bigint) LIMIT 1`, [ref.code.trim()]);
+            chatId = row.rows.length ? String(row.rows[0].chat_id) : "";
+        }
+        if (!chatId)
+            return undefined;
+        const row = await this.pool.query(`SELECT id FROM ${this.table("messages")} WHERE chat_id = $1 AND message_id = $2::bigint LIMIT 1`, [chatId, ref.messageId]);
+        return row.rows.length ? String(row.rows[0].id) : undefined;
+    }
+    async getMessageContext(recordId, beforeN, afterN, beforeOffset = 0, afterOffset = 0) {
+        const beforeLimit = normalizeContextLimit(beforeN);
+        const afterLimit = normalizeContextLimit(afterN);
+        const beforeOff = normalizeOffset(beforeOffset);
+        const afterOff = normalizeOffset(afterOffset);
+        const anchorRecord = await this.getMessageByRecordId(recordId);
+        if (!anchorRecord) {
+            return { anchor: undefined, before: [], after: [], beforeN: 0, afterN: 0 };
+        }
+        let lower = { date: anchorRecord.date, id: Number(anchorRecord.recordId) };
+        let upper = lower;
+        let anchorGroup;
+        if (anchorRecord.groupedId !== undefined) {
+            const group = await this.fetchGroupAttached(anchorRecord.chatId, anchorRecord.groupedId);
+            if (group.length > 1) {
+                anchorGroup = group;
+                lower = { date: group[0].date, id: Number(group[0].recordId) };
+                const last = group[group.length - 1];
+                upper = { date: last.date, id: Number(last.recordId) };
+            }
+        }
+        const tables = this.table("messages");
+        const first = groupFirstCondition(tables);
+        const beforeResult = beforeLimit
+            ? await this.pool.query(`${messageSelect(this.table("messages"), this.table("media_files"))}
+           WHERE m.chat_id = $1
+             AND (m.date < $2::timestamptz OR (m.date = $2::timestamptz AND m.id < $3::bigint))
+             AND ${first}
+             AND m.blocked = FALSE
+           ORDER BY m.date DESC, m.id DESC
+           LIMIT $4 OFFSET $5`, [anchorRecord.chatId, lower.date, lower.id, beforeLimit, beforeOff])
+            : { rows: [] };
+        const afterResult = afterLimit
+            ? await this.pool.query(`${messageSelect(this.table("messages"), this.table("media_files"))}
+           WHERE m.chat_id = $1
+             AND (m.date > $2::timestamptz OR (m.date = $2::timestamptz AND m.id > $3::bigint))
+             AND ${first}
+             AND m.blocked = FALSE
+           ORDER BY m.date ASC, m.id ASC
+           LIMIT $4 OFFSET $5`, [anchorRecord.chatId, upper.date, upper.id, afterLimit, afterOff])
+            : { rows: [] };
+        const beforeCount = await this.pool.query(`SELECT COUNT(*)::int AS count FROM ${tables} m
+        WHERE m.chat_id = $1
+          AND (m.date < $2::timestamptz OR (m.date = $2::timestamptz AND m.id < $3::bigint))
+          AND ${first}
+          AND m.blocked = FALSE`, [anchorRecord.chatId, lower.date, lower.id]);
+        const afterCount = await this.pool.query(`SELECT COUNT(*)::int AS count FROM ${tables} m
+        WHERE m.chat_id = $1
+          AND (m.date > $2::timestamptz OR (m.date = $2::timestamptz AND m.id > $3::bigint))
+          AND ${first}
+          AND m.blocked = FALSE`, [anchorRecord.chatId, upper.date, upper.id]);
+        const groupMap = await this.fetchGroupsAttached(uniqueGroupKeys([...beforeResult.rows, ...afterResult.rows]));
+        const before = buildContextEntries(await this.attachMedia([...beforeResult.rows].reverse()), groupMap);
+        const after = buildContextEntries(await this.attachMedia(afterResult.rows), groupMap);
+        const anchor = anchorGroup !== undefined
+            ? albumEntryOf(anchorGroup, anchorRecord.recordId)
+            : { kind: "message", record: anchorRecord };
+        return {
+            anchor,
+            before,
+            after,
+            beforeN: Number(beforeCount.rows[0]?.count ?? 0),
+            afterN: Number(afterCount.rows[0]?.count ?? 0)
+        };
+    }
+    async fetchGroupAttached(chatId, groupedId) {
+        return this.attachMedia(await this.groupRows([[chatId, groupedId]]));
+    }
+    async fetchGroupsAttached(keys) {
+        const result = new Map();
+        if (!keys.length)
+            return result;
+        const rows = await this.groupRows(keys);
+        const byKey = new Map();
+        for (const row of rows) {
+            pushGrouped(byKey, groupKey(String(row.chat_id), String(row.grouped_id)), row);
+        }
+        for (const [key, groupRows] of byKey) {
+            result.set(key, await this.attachMedia(groupRows));
+        }
+        return result;
+    }
+    async groupRows(keys) {
+        const { clause, params } = pairSql("grouped_id", keys);
+        const result = await this.pool.query(`${messageSelect(this.table("messages"), this.table("media_files"))}
+       WHERE m.has_media = TRUE AND m.blocked = FALSE AND ${clause}
+       ORDER BY m.date ASC, m.id ASC`, params);
+        return result.rows;
+    }
+    async getMediaFileById(id) {
+        const result = await this.pool.query(`SELECT mf.id::text AS id, mf.message_id, mf.chat_id, mf.media_type, mf.file_name,
+              mf.file_path, mf.file_size, mf.mime_type
+         FROM ${this.table("media_files")} mf
+        WHERE mf.id = $1::bigint
+          AND NOT EXISTS (
+            SELECT 1 FROM ${this.table("messages")} m
+             WHERE m.chat_id = mf.chat_id AND m.message_id = mf.message_id
+               AND m.blocked = TRUE
+          )
+        LIMIT 1`, [normalizeRecordId(id)]);
+        return result.rows.length ? toStoredMediaFile(result.rows[0]) : undefined;
+    }
+    async attachMedia(rows) {
+        if (!rows.length)
+            return [];
+        const pairKeys = rows.map((row) => [String(row.chat_id), Number(row.message_id)]);
+        const groupKeys = rows
+            .filter((row) => row.grouped_id !== null && row.grouped_id !== undefined)
+            .map((row) => [String(row.chat_id), String(row.grouped_id)]);
+        const mediaMap = await this.fetchMediaFiles(pairKeys);
+        const albumMap = groupKeys.length
+            ? await this.fetchAlbumRows(groupKeys)
+            : new Map();
+        return rows.map((row) => toMessageRecord(row, mediaMap, albumMap));
+    }
+    async fetchMediaFiles(pairs) {
+        const { clause, params } = pairSql("message_id", pairs);
+        const result = await this.pool.query(`SELECT id::text AS id, message_id, chat_id, media_type, mime_type
+         FROM ${this.table("media_files")}
+        WHERE ${clause}
+        ORDER BY id ASC`, params);
+        const map = new Map();
+        for (const row of result.rows) {
+            pushGrouped(map, mediaKey(String(row.chat_id), Number(row.message_id)), toStoredMediaFile(row));
+        }
+        return map;
+    }
+    async fetchAlbumRows(pairs) {
+        const { clause, params } = pairSql("grouped_id", pairs);
+        const result = await this.pool.query(`${albumSelect(this.table("messages"), this.table("media_files"))}
+       WHERE m.has_media = TRUE AND m.blocked = FALSE AND ${clause}
+       ORDER BY m.date ASC, m.id ASC`, params);
+        const map = new Map();
+        for (const row of result.rows) {
+            pushGrouped(map, groupKey(String(row.chat_id), String(row.grouped_id)), toAlbumRow(row));
+        }
+        return map;
+    }
+    table(name) {
+        return `${this.schema}.${quoteIdentifier(name)}`;
+    }
+}
+/** 相册以组内第一条在时间线中定位，窗口按条目推进。 */
+function groupFirstCondition(messages) {
+    return `(
+    m.grouped_id IS NULL
+    OR NOT EXISTS (
+      SELECT 1 FROM ${messages} g
+       WHERE g.chat_id = m.chat_id AND g.grouped_id = m.grouped_id
+         AND g.blocked = FALSE
+         AND (g.date < m.date OR (g.date = m.date AND g.id < m.id))
+    )
+  )`;
+}
+function messageSelect(messages, mediaFiles) {
+    return `SELECT
+      m.id::text AS row_id,
+      m.message_id,
+      m.chat_id,
+      m.grouped_id,
+      m.chat_title,
+      m.date,
+      m.sender_id,
+      m.sender_username,
+      m.sender_first_name,
+      m.sender_last_name,
+      m.has_media,
+      m.media_type,
+      m.message_type,
+      m.text,
+      m.entities,
+      m.reply_to_msg_id,
+      m.forward_from_id,
+      m.forward_from_name,
+      (
+        SELECT substr(p.text, 1, 120) FROM ${messages} p
+         WHERE p.chat_id = m.chat_id AND p.message_id = m.reply_to_msg_id
+           AND p.blocked = FALSE
+         LIMIT 1
+      ) AS reply_to_text,
+      media_meta.mime_type
+    FROM ${messages} m
+    LEFT JOIN LATERAL (
+      SELECT mf.mime_type
+        FROM ${mediaFiles} mf
+       WHERE mf.message_id = m.message_id AND mf.chat_id = m.chat_id
+       ORDER BY mf.id DESC
+       LIMIT 1
+    ) media_meta ON TRUE`;
+}
+function albumSelect(messages, mediaFiles) {
+    return `SELECT
+      m.id::text AS row_id,
+      m.message_id,
+      m.chat_id,
+      m.grouped_id,
+      m.date,
+      m.has_media,
+      m.media_type,
+      m.message_type,
+      media_meta.mime_type
+    FROM ${messages} m
+    LEFT JOIN LATERAL (
+      SELECT mf.mime_type
+        FROM ${mediaFiles} mf
+       WHERE mf.message_id = m.message_id AND mf.chat_id = m.chat_id
+       ORDER BY mf.id DESC
+       LIMIT 1
+    ) media_meta ON TRUE`;
+}
+/** 检索基线：普通消息取自身；相册取组内第一条作代表，组内任一条命中即整体命中。 */
+function buildSearchWhere(messages, rowWhere, rowValues, memberWhere, memberValues) {
+    const row = rowWhere?.replace(/^WHERE\s+/, "");
+    const member = memberWhere
+        ? ` OR EXISTS (
+        SELECT 1 FROM ${messages} g
+         WHERE g.chat_id = m.chat_id AND g.grouped_id = m.grouped_id
+           AND ${memberWhere
+            .replace(/^WHERE\s+/, "")
+            .replace(/\$(\d+)/g, (_match, index) => `$${rowValues.length + Number(index)}`)}
+      )`
+        : "";
+    const matches = row ? ` AND (${row}${member})` : "";
+    return {
+        where: `WHERE ${groupFirstCondition(messages)} AND m.blocked = FALSE${matches}`,
+        values: [...rowValues, ...memberValues]
+    };
+}
+function buildWhere(query, alias = "m") {
+    const conditions = [];
+    const values = [];
+    const add = (condition, value) => {
+        values.push(value);
+        conditions.push(condition.replace("$N", `$${values.length}`));
+    };
+    for (const term of splitTerms(query.keyword))
+        add(`lower(${alias}.text) LIKE $N`, `%${term.toLowerCase()}%`);
+    for (const term of splitTerms(query.excludeKeyword))
+        add(`lower(${alias}.text) NOT LIKE $N`, `%${term.toLowerCase()}%`);
+    const chatIds = (query.chatIds ?? []).map((id) => id.trim()).filter(Boolean);
+    if (chatIds.length > 0) {
+        const start = values.length + 1;
+        values.push(...chatIds);
+        conditions.push(`${alias}.chat_id IN (${paramPlaceholders(start, chatIds.length)})`);
+    }
+    if (query.chatTitle?.trim())
+        add(`${alias}.chat_title ILIKE $N`, `%${query.chatTitle.trim()}%`);
+    const senderIds = (query.senderIds ?? []).map((id) => id.trim()).filter(Boolean);
+    if (senderIds.length > 0) {
+        const start = values.length + 1;
+        values.push(...senderIds);
+        conditions.push(`${alias}.sender_id IN (${paramPlaceholders(start, senderIds.length)})`);
+    }
+    if (query.forwardFromId?.trim())
+        add(`${alias}.forward_from_id = $N`, query.forwardFromId.trim());
+    if (query.forwardFromName?.trim())
+        add(`${alias}.forward_from_name ILIKE $N`, `%${escapeLike(query.forwardFromName.trim())}%`);
+    appendTimeWhere(conditions, values, query, add, alias);
+    conditions.push(`${alias}.blocked = FALSE`);
+    return { where: conditions.length ? `WHERE ${conditions.join(" AND ")}` : "", values };
+}
+function appendTimeWhere(conditions, values, query, add, alias = "m") {
+    const mode = normalizeTimeMode(query.timeMode);
+    if (mode === "off" || (!query.dateFrom && !query.dateTo))
+        return;
+    const column = `${alias}.date`;
+    if (mode === "include") {
+        if (query.dateFrom)
+            add(`${column} >= $N::timestamptz`, query.dateFrom);
+        if (query.dateTo)
+            add(`${column} <= $N::timestamptz`, query.dateTo);
+        return;
+    }
+    if (query.dateFrom && query.dateTo) {
+        const fromIndex = values.length + 1;
+        values.push(query.dateFrom, query.dateTo);
+        conditions.push(`NOT (${column} >= $${fromIndex}::timestamptz AND ${column} <= $${fromIndex + 1}::timestamptz)`);
+    }
+    else if (query.dateFrom) {
+        add(`${column} < $N::timestamptz`, query.dateFrom);
+    }
+    else if (query.dateTo) {
+        add(`${column} > $N::timestamptz`, query.dateTo);
+    }
+}
+function pairSql(column, pairs) {
+    const params = [];
+    const parts = [];
+    for (const [chatId, second] of pairs) {
+        const start = params.length + 1;
+        parts.push(`(chat_id = $${start} AND ${column} = $${start + 1})`);
+        params.push(chatId, second);
+    }
+    return { clause: parts.join(" OR "), params };
+}
+function toStoredMediaFile(row) {
+    return {
+        id: String(row.id),
+        messageId: Number(row.message_id),
+        chatId: String(row.chat_id),
+        mediaType: String(row.media_type ?? "document"),
+        fileName: optionalString(row.file_name),
+        filePath: optionalString(row.file_path),
+        fileSize: optionalNumber(row.file_size),
+        mimeType: optionalString(row.mime_type)
+    };
+}
+function toAlbumRow(row) {
+    return {
+        recordId: String(row.row_id),
+        messageId: Number(row.message_id),
+        chatId: String(row.chat_id),
+        groupedId: String(row.grouped_id),
+        date: toIsoDate(row.date),
+        hasMedia: Boolean(row.has_media),
+        mediaType: optionalString(row.media_type),
+        messageType: optionalString(row.message_type),
+        mimeType: optionalString(row.mime_type)
+    };
+}
+function toMessageRecord(row, mediaMap, albumMap) {
+    const chatId = String(row.chat_id);
+    const messageId = Number(row.message_id);
+    const groupedId = optionalString(row.grouped_id);
+    const mediaFiles = mediaMap.get(mediaKey(chatId, messageId)) ?? [];
+    const albumRows = groupedId !== undefined ? albumMap.get(groupKey(chatId, groupedId)) ?? [] : [];
+    return {
+        recordId: String(row.row_id),
+        messageId,
+        chatId,
+        groupedId,
+        chatTitle: optionalString(row.chat_title),
+        date: toIsoDate(row.date),
+        senderId: optionalString(row.sender_id),
+        senderUsername: optionalString(row.sender_username),
+        senderFirstName: optionalString(row.sender_first_name),
+        senderLastName: optionalString(row.sender_last_name),
+        hasMedia: Boolean(row.has_media),
+        mediaType: optionalString(row.media_type),
+        messageType: optionalString(row.message_type) ?? "text",
+        mimeType: optionalString(row.mime_type),
+        replyToMessageId: row.reply_to_msg_id === null || row.reply_to_msg_id === undefined ? undefined : Number(row.reply_to_msg_id),
+        replyToText: optionalString(row.reply_to_text),
+        forwardFromId: optionalString(row.forward_from_id),
+        forwardFromName: optionalString(row.forward_from_name),
+        entities: parseEntities(row.entities),
+        text: String(row.text ?? ""),
+        mediaFiles,
+        albumRows
+    };
+}
+function pushGrouped(map, key, item) {
+    let items = map.get(key);
+    if (!items) {
+        items = [];
+        map.set(key, items);
+    }
+    items.push(item);
+}
+function quoteIdentifier(value) {
+    return `"${value.replaceAll('"', '""')}"`;
+}
+/** 屏蔽词命中判定（子串、大小写不敏感）；空文本永不命中。 */
+function matchesBlockword(text, words) {
+    if (text === undefined || text === "")
+        return false;
+    const lower = text.toLowerCase();
+    return words.some((word) => lower.includes(word));
+}
+/** 新落库消息是否屏蔽：命中任一屏蔽词，或发送者在屏蔽用户名单内。 */
+function blockedOf(row, words, users) {
+    if (matchesBlockword(row.text, words))
+        return true;
+    return row.senderId !== undefined && users.some((user) => user.userId === row.senderId);
+}
+function toBlockedUserInfo(row) {
+    return {
+        userId: String(row.user_id),
+        name: optionalString(row.name),
+        username: optionalString(row.username)
+    };
+}
+function toSenderInfo(row) {
+    return {
+        senderId: String(row.sender_id),
+        username: optionalString(row.sender_username),
+        firstName: optionalString(row.sender_first_name),
+        lastName: optionalString(row.sender_last_name)
+    };
+}
+/** LIKE/ILIKE 通配符转义：pg 默认转义符为反斜杠。 */
+function escapeLike(value) {
+    return value.replaceAll("\\", "\\\\").replaceAll("%", "\\%").replaceAll("_", "\\_");
+}
+function optionalString(value) {
+    return value === undefined || value === null || value === "" ? undefined : String(value);
+}
+function optionalNumber(value) {
+    return value === undefined || value === null || value === "" ? undefined : Number(value);
+}
+//# sourceMappingURL=postgres.js.map