agent-remnote 0.1.0 → 0.3.0

package/dist/packages/mcp/src/tools/shared.js

@@ -1,393 +0,0 @@
-import BetterSqlite3 from "better-sqlite3";
-import { promises as fs } from "fs";
-import os from "os";
-import path from "path";
-import { format as formatDate } from "date-fns";
-export const REMNOTE_RELATIVE_DIR = "remnote";
-export const REMNOTE_DB_NAME = "remnote.db";
-export const REMNOTE_DIR_PREFIX = "remnote-";
-const SECONDARY_DIRS = new Set(["remnote-browser", "lnotes"]);
-export const SYSTEM_REM_IDS = new Set([
-    "u71eGVAt1upM7uwC6", // Date String
-    "DikxdKKfeGh52AP7p", // timestamp
-    "SLg5kZsuzw3GyN6Kh", // Status
-]);
-export const SYSTEM_REM_KEYS = new Set(["Date String", "timestamp", "Status"]);
-export async function withResolvedDatabase(dbPath, fn) {
-    const info = await resolveDatabasePath(dbPath);
-    let db;
-    try {
-        db = new BetterSqlite3(info.dbPath, { readonly: true });
-    }
-    catch (error) {
-        const normalized = String(error ?? "");
-        if (/\b(SQLITE_(BUSY|LOCKED|CANTOPEN))\b/i.test(normalized) || /busy/i.test(normalized)) {
-            throw new Error(`无法打开数据库 ${info.dbPath}（可能正在被 RemNote 客户端占用或尚未写入完成）。` +
-                " 请确认 RemNote 已完成同步，或稍候重试。如需立即使用，可显式指定备份路径 dbPath=...。");
-        }
-        throw error;
-    }
-    try {
-        try {
-            const result = await fn(db);
-            return { result, info };
-        }
-        catch (inner) {
-            const message = String(inner ?? "");
-            if (/no such table: (remsSearchInfos|remsContents)/i.test(message)) {
-                throw new Error("数据库缺少搜索索引表 remsSearchInfos/remsContents。请在 RemNote 客户端完成索引生成或使用较新的备份后重试。");
-            }
-            if (/no such table: quanta/i.test(message)) {
-                throw new Error("数据库结构不匹配（缺少 quanta 表）。请确认 dbPath 是否指向 RemNote 的 remnote.db 文件，或选择正确账户目录下的数据库/备份。");
-            }
-            if (/malformed MATCH/i.test(message) || /no such tokenizer/i.test(message)) {
-                throw new Error(`全文检索 FTS 当前不可用（分词器未加载或语法不兼容）。建议改用 mode=\"like\"，或在 RemNote 客户端完成索引后重试。原始错误：${message}`);
-            }
-            throw inner;
-        }
-    }
-    finally {
-        db?.close();
-    }
-}
-export async function resolveDatabasePath(dbPath) {
-    if (dbPath) {
-        const expanded = expandHome(dbPath.trim());
-        const stat = await fs.stat(expanded);
-        if (!stat.isFile()) {
-            throw new Error(`${expanded} is not a file`);
-        }
-        return {
-            dbPath: expanded,
-            source: "explicit",
-            baseDir: path.dirname(expanded),
-        };
-    }
-    const baseDir = path.join(os.homedir(), REMNOTE_RELATIVE_DIR);
-    const entries = await fs
-        .readdir(baseDir, { withFileTypes: true })
-        .catch(() => {
-        throw new Error(`RemNote directory ${baseDir} not found – please specify dbPath manually`);
-    });
-    const candidates = [];
-    for (const entry of entries) {
-        if (!entry.isDirectory())
-            continue;
-        if (entry.name === "backups")
-            continue;
-        const isPrimary = entry.name.startsWith(REMNOTE_DIR_PREFIX);
-        const isSecondary = SECONDARY_DIRS.has(entry.name);
-        if (!isPrimary && !isSecondary)
-            continue;
-        const candidatePath = path.join(baseDir, entry.name, REMNOTE_DB_NAME);
-        try {
-            const stat = await fs.stat(candidatePath);
-            candidates.push({
-                dbPath: candidatePath,
-                dirName: entry.name,
-                mtimeMs: stat.mtimeMs,
-                priority: isPrimary ? 0 : 1,
-            });
-        }
-        catch (_) {
-            continue;
-        }
-    }
-    if (candidates.length === 0) {
-        throw new Error(`No ${REMNOTE_DB_NAME} found under ${baseDir}. Specify dbPath manually or ensure RemNote has run.`);
-    }
-    candidates.sort((a, b) => {
-        if (a.priority !== b.priority)
-            return a.priority - b.priority;
-        return b.mtimeMs - a.mtimeMs;
-    });
-    const best = candidates[0];
-    return {
-        dbPath: best.dbPath,
-        source: "auto",
-        dirName: best.dirName,
-        baseDir,
-    };
-}
-export async function discoverBackups(basePath) {
-    const backups = [];
-    let entries;
-    try {
-        entries = await fs.readdir(basePath, { withFileTypes: true });
-    }
-    catch (error) {
-        throw new Error(`Unable to read ${basePath}: ${String(error)}`);
-    }
-    for (const entry of entries) {
-        if (!entry.isDirectory())
-            continue;
-        const accountDir = path.join(basePath, entry.name);
-        const backupDir = path.join(accountDir, "backups");
-        let backupFiles;
-        try {
-            backupFiles = await fs.readdir(backupDir, { withFileTypes: true });
-        }
-        catch (_) {
-            continue;
-        }
-        for (const file of backupFiles) {
-            if (!file.isFile())
-                continue;
-            if (!file.name.endsWith(".db") && !file.name.endsWith(".db.zip"))
-                continue;
-            const fullPath = path.join(backupDir, file.name);
-            const stat = await fs.stat(fullPath);
-            backups.push({
-                accountDir: entry.name,
-                file: file.name,
-                fullPath,
-                size: stat.size,
-                mtime: new Date(stat.mtimeMs).toISOString(),
-                type: file.name.endsWith(".zip") ? "zip" : "db",
-            });
-        }
-    }
-    backups.sort((a, b) => new Date(b.mtime).getTime() - new Date(a.mtime).getTime());
-    return backups;
-}
-export function expandHome(targetPath) {
-    if (!targetPath.startsWith("~"))
-        return targetPath;
-    const home = os.homedir();
-    if (targetPath === "~")
-        return home;
-    if (targetPath.startsWith("~/"))
-        return path.join(home, targetPath.slice(2));
-    return targetPath.replace(/^~(?=$|\/)/, home);
-}
-export function safeJsonParse(input) {
-    if (typeof input !== "string")
-        return null;
-    try {
-        return JSON.parse(input);
-    }
-    catch (_) {
-        return null;
-    }
-}
-export function summarizeKey(rawKey, db, options) {
-    const fragments = [];
-    const references = new Set();
-    const visited = new Set();
-    const stmt = db ? db.prepare("SELECT doc FROM quanta WHERE _id = ?") : undefined;
-    const helper = (key, depth) => {
-        if (!Array.isArray(key))
-            return;
-        for (const item of key) {
-            if (typeof item === "string") {
-                fragments.push(item);
-                continue;
-            }
-            if (item && typeof item === "object") {
-                const maybeObj = item;
-                if (maybeObj.i === "q" && typeof maybeObj._id === "string") {
-                    const refId = maybeObj._id;
-                    references.add(refId);
-                    let expanded = false;
-                    if (options.expand &&
-                        depth < options.maxDepth &&
-                        stmt &&
-                        !visited.has(refId)) {
-                        visited.add(refId);
-                        try {
-                            const row = stmt.get(refId);
-                            if (row) {
-                                const refDoc = safeJsonParse(row.doc);
-                                helper(refDoc?.key, depth + 1);
-                                expanded = true;
-                            }
-                        }
-                        finally {
-                            // keep visited entry to prevent cycles
-                        }
-                    }
-                    if (!expanded) {
-                        fragments.push(`{ref:${refId}}`);
-                    }
-                    continue;
-                }
-                const richText = extractRichText(maybeObj);
-                if (richText) {
-                    fragments.push(richText);
-                    continue;
-                }
-                if (typeof maybeObj.t === "string") {
-                    fragments.push(maybeObj.t);
-                    continue;
-                }
-                fragments.push(JSON.stringify(item));
-                continue;
-            }
-            if (item != null) {
-                fragments.push(String(item));
-            }
-        }
-    };
-    helper(rawKey, 0);
-    const text = fragments
-        .join("")
-        .replace(/\s+/g, " ")
-        .trim();
-    return {
-        text,
-        references: Array.from(references),
-    };
-}
-function extractRichText(obj) {
-    if (typeof obj.text === "string" && obj.text.trim()) {
-        return obj.text;
-    }
-    if (typeof obj.title === "string" && obj.title.trim()) {
-        const suffix = typeof obj.text === "string" && obj.text.trim() ? ` ${obj.text}` : "";
-        return `${obj.title}${suffix}`.trim();
-    }
-    if (Array.isArray(obj.children)) {
-        const nested = obj.children
-            .map((child) => {
-            if (typeof child === "string")
-                return child;
-            if (child && typeof child === "object") {
-                return extractRichText(child) ?? "";
-            }
-            return "";
-        })
-            .filter(Boolean)
-            .join("");
-        if (nested) {
-            return nested;
-        }
-    }
-    return undefined;
-}
-export function makeToolResponse(structuredContent) {
-    return {
-        content: [
-            {
-                type: "text",
-                text: JSON.stringify(structuredContent, null, 2),
-            },
-        ],
-        structuredContent,
-    };
-}
-export function parseOrThrow(schema, input, options) {
-    const res = schema.safeParse(input);
-    if (res.success)
-        return res.data;
-    const issues = res.error.issues;
-    const lines = issues.map((i) => formatZodIssueCN(i));
-    const prefix = options?.label ? `参数错误（${options.label}）：` : "参数错误：";
-    throw new Error(prefix + lines.join("； "));
-}
-function formatZodIssueCN(issue) {
-    const path = issue.path && issue.path.length > 0 ? issue.path.join(".") : undefined;
-    const where = path ? `字段 ${path}：` : "";
-    switch (issue.code) {
-        case "invalid_type": {
-            const expected = toCNType(issue.expected);
-            const received = toCNType(issue.received);
-            return `${where}类型应为 ${expected}，实际为 ${received}`;
-        }
-        case "too_small": {
-            if (issue.type === "string") {
-                return `${where}长度至少 ${issue.minimum}`;
-            }
-            if (issue.type === "number") {
-                return `${where}数值应 ≥ ${issue.minimum}`;
-            }
-            if (issue.type === "array") {
-                return `${where}至少包含 ${issue.minimum} 个元素`;
-            }
-            return `${where}值过小`;
-        }
-        case "too_big": {
-            if (issue.type === "string") {
-                return `${where}长度至多 ${issue.maximum}`;
-            }
-            if (issue.type === "number") {
-                return `${where}数值应 ≤ ${issue.maximum}`;
-            }
-            if (issue.type === "array") {
-                return `${where}至多包含 ${issue.maximum} 个元素`;
-            }
-            return `${where}值过大`;
-        }
-        case "invalid_string": {
-            return `${where}字符串不符合要求`;
-        }
-        case "invalid_enum_value": {
-            const opts = Array.isArray(issue.options) ? issue.options.join("/") : String(issue.options);
-            return `${where}不在允许取值范围（${opts}）内`;
-        }
-        case "invalid_union":
-        case "invalid_union_discriminator": {
-            return `${where}不符合联合类型约束`;
-        }
-        case "custom": {
-            return `${where}${issue.message || "不符合约束"}`;
-        }
-        default: {
-            return `${where}${issue.message || "参数不合法"}`;
-        }
-    }
-}
-function toCNType(t) {
-    const map = {
-        string: "字符串",
-        number: "数字",
-        boolean: "布尔",
-        bigint: "大整数",
-        date: "日期",
-        undefined: "未定义",
-        null: "空",
-        object: "对象",
-        array: "数组",
-    };
-    return map[String(t)] ?? String(t);
-}
-export function buildGuidedResponse(structuredContent, suggestions) {
-    const payload = {
-        ...structuredContent,
-    };
-    if (suggestions && suggestions.length > 0) {
-        payload.next = suggestions;
-    }
-    return makeToolResponse(payload);
-}
-export async function getUserSetting(db, key) {
-    const row = db
-        .prepare("SELECT json_extract(doc, '$.value') AS value FROM user_data WHERE json_extract(doc, '$.key') = ?")
-        .get(key);
-    if (!row)
-        return undefined;
-    if (typeof row.value === "string") {
-        return (safeJsonParse(row.value) ?? row.value);
-    }
-    return row.value;
-}
-export async function getDateFormatting(db) {
-    const format = await getUserSetting(db, "dateFormatting");
-    if (typeof format === "string" && format.trim()) {
-        return normalizeDateFormat(format.trim());
-    }
-    return undefined;
-}
-export function formatDateWithPattern(date, pattern) {
-    try {
-        return formatDate(date, pattern);
-    }
-    catch (_) {
-        return formatDate(date, "yyyy/MM/dd");
-    }
-}
-function normalizeDateFormat(format) {
-    return format
-        .replace(/YYYY/g, "yyyy")
-        .replace(/YY/g, "yy")
-        .replace(/DD/g, "dd");
-}

package/dist/packages/mcp/src/tools/summarizeDailyNotes.js

@@ -1,221 +0,0 @@
-import { z } from "zod";
-import { buildGuidedResponse, withResolvedDatabase, getDateFormatting, formatDateWithPattern, parseOrThrow, } from "./shared.js";
-import { executeSearchRemOverview } from "./searchRemOverview.js";
-import { executeOutlineRemSubtree } from "./outlineRemSubtree.js";
-const inputShape = {
-    days: z
-        .number()
-        .int()
-        .min(1)
-        .max(30)
-        .default(7)
-        .describe("统计最近 N 天（默认 7）"),
-    includeEmpty: z.boolean().optional().describe("是否包含空行/空文档（默认 false）"),
-    expandReferences: z.boolean().optional().describe("是否展开 [[引用]] 文本（默认 true）"),
-    maxReferenceDepth: z
-        .number()
-        .int()
-        .min(0)
-        .max(5)
-        .optional()
-        .describe("引用展开的最大深度（默认 1）"),
-    maxLines: z
-        .number()
-        .int()
-        .min(1)
-        .max(200)
-        .optional()
-        .describe("单篇大纲的最大行数（默认 40）"),
-    dbPath: z.string().optional().describe("数据库文件路径（默认自动发现）"),
-    detail: z.boolean().optional().describe("是否返回逐日的节点详情（默认 false）"),
-};
-export const summarizeDailyNotesSchema = z.object(inputShape);
-export async function executeSummarizeDailyNotes(params) {
-    const parsed = parseOrThrow(summarizeDailyNotesSchema, params, { label: "summarize_daily_notes" });
-    const days = parsed.days;
-    const includeEmpty = parsed.includeEmpty ?? false;
-    const expandReferences = parsed.expandReferences ?? true;
-    const maxReferenceDepth = parsed.maxReferenceDepth ?? 1;
-    const maxLines = parsed.maxLines ?? 40;
-    const detail = parsed.detail ?? false;
-    const detailResults = [];
-    const summaries = [];
-    const sections = [];
-    const referenceIndex = new Map();
-    const { info } = await withResolvedDatabase(parsed.dbPath, async (db) => {
-        const format = (await getDateFormatting(db)) ?? "yyyy/MM/dd";
-        for (let offset = 0; offset > -days; offset--) {
-            const targetDate = new Date();
-            targetDate.setDate(targetDate.getDate() + offset);
-            const dateString = formatDateWithPattern(targetDate, format);
-            const sectionLines = [`## ${dateString}`];
-            try {
-                const searchResult = await executeSearchRemOverview({
-                    query: dateString,
-                    dbPath: parsed.dbPath,
-                });
-                if (searchResult.count === 0 || searchResult.matches.length === 0) {
-                    summaries.push({ date: dateString, status: "not_found" });
-                    sectionLines.push("- 未找到对应笔记");
-                    detailResults.push({ date: dateString });
-                    sections.push(sectionLines.join("\n"));
-                    continue;
-                }
-                const remId = searchResult.matches[0]?.id;
-                if (!remId) {
-                    summaries.push({ date: dateString, status: "error" });
-                    sectionLines.push("- 未找到有效 Rem ID");
-                    detailResults.push({
-                        date: dateString,
-                        error: "未找到有效 Rem ID",
-                    });
-                    sections.push(sectionLines.join("\n"));
-                    continue;
-                }
-                const outlineResult = await executeOutlineRemSubtree({
-                    id: remId,
-                    dbPath: parsed.dbPath,
-                    includeEmpty,
-                    expandReferences,
-                    maxReferenceDepth,
-                    startOffset: 0,
-                    maxNodes: maxLines,
-                    format: "markdown",
-                    detail: true,
-                });
-                const markdown = outlineResult.markdown ?? `- ${dateString}`;
-                const lines = markdown.split("\n");
-                const trimmed = lines.slice(0, maxLines).join("\n");
-                const truncated = lines.length > maxLines || outlineResult.hasMore;
-                const nodes = Array.isArray(outlineResult.tree)
-                    ? (outlineResult.tree ?? [])
-                    : [];
-                collectReferenceSummaries(referenceIndex, nodes, remId, outlineResult.title ?? remId);
-                summaries.push({
-                    date: dateString,
-                    status: truncated ? "truncated" : "ok",
-                    remId,
-                    title: outlineResult.title ?? remId,
-                    lineCount: lines.length,
-                });
-                sectionLines.push(trimmed);
-                if (truncated) {
-                    sectionLines.push(`> 内容已截断，请调用 outline_rem_subtree id=${remId} startOffset=${maxLines} 继续阅读`);
-                }
-                const detailEntry = {
-                    date: dateString,
-                    remId,
-                    markdown: trimmed,
-                    truncated,
-                    lineCount: lines.length,
-                };
-                if (detail) {
-                    detailEntry.nodes = nodes;
-                }
-                detailResults.push(detailEntry);
-            }
-            catch (error) {
-                summaries.push({ date: dateString, status: "error" });
-                sectionLines.push(`- 读取失败：${String(error)}`);
-                detailResults.push({
-                    date: dateString,
-                    error: String(error),
-                });
-            }
-            sections.push(sectionLines.join("\n"));
-        }
-    });
-    const aggregatedMarkdown = buildAggregateMarkdown(days, sections);
-    const response = {
-        dbPath: info.dbPath,
-        resolution: info.source,
-        dirName: info.dirName,
-        days,
-        maxLines,
-        count: summaries.length,
-        markdown: aggregatedMarkdown,
-        daysSummary: summaries,
-        referenceIndex: convertReferenceIndex(referenceIndex),
-    };
-    if (detail) {
-        Object.assign(response, { results: detailResults });
-    }
-    return response;
-}
-export function registerSummarizeDailyNotes(server) {
-    server.tool("summarize_daily_notes", `<usecase>汇总最近 N 天的每日笔记，自动匹配用户设定的日期格式。</usecase>
-<instructions>
-- 默认统计最近 7 天，可通过 days 调整范围。
-- maxLines 控制单篇输出的最大行数；若被截断，结果会返回 truncated=true。
-- truncated=true 时，可进一步调用 outline_rem_subtree id=<remId> startOffset=<maxLines> 继续阅读。
-- 输出为 Markdown 大纲。若需要原始字段，可结合 inspect_rem_doc。
-</instructions>`, inputShape, async (input) => {
-        const parsed = parseOrThrow(summarizeDailyNotesSchema, input, { label: "summarize_daily_notes" });
-        const result = await executeSummarizeDailyNotes(parsed);
-        const suggestions = [
-            "如需深入某一天的笔记，可继续调用 outline_rem_subtree 或 inspect_rem_doc",
-        ];
-        const truncatedExists = Array.isArray(result.results)
-            ? result.results.some((item) => item.truncated && item.remId)
-            : Array.isArray(result.daysSummary)
-                ? result.daysSummary.some((item) => item.status === "truncated")
-                : false;
-        if (truncatedExists) {
-            suggestions.unshift("若看到 truncated=true，可使用 outline_rem_subtree id=<remId> startOffset=<maxLines> 继续阅读");
-        }
-        const guidance = `已汇总最近 ${result.days} 天的每日笔记（按用户日期格式，单条最多 ${result.maxLines} 行）。`;
-        return buildGuidedResponse({ guidance, ...result }, suggestions);
-    });
-}
-function buildAggregateMarkdown(days, sections) {
-    const header = `# 最近 ${days} 天每日笔记`;
-    if (sections.length === 0) {
-        return `${header}\n\n- 未生成任何结果`;
-    }
-    return [header, ...sections].join("\n\n");
-}
-function collectReferenceSummaries(index, nodes, remId, remTitle) {
-    if (!nodes || nodes.length === 0)
-        return;
-    for (const node of nodes) {
-        if (!node.references || node.references.length === 0)
-            continue;
-        for (const refId of node.references) {
-            if (!refId)
-                continue;
-            const entry = ensureReferenceEntry(index, refId);
-            entry.totalOccurrences += 1;
-            if (entry.samples.length < 5) {
-                entry.samples.push({
-                    remId,
-                    remTitle,
-                    nodeId: node.id,
-                    text: node.text,
-                    depth: node.depth,
-                });
-            }
-        }
-    }
-}
-function ensureReferenceEntry(index, id) {
-    let entry = index.get(id);
-    if (!entry) {
-        entry = {
-            id,
-            totalOccurrences: 0,
-            samples: [],
-        };
-        index.set(id, entry);
-    }
-    return entry;
-}
-function convertReferenceIndex(index) {
-    const result = {};
-    for (const [id, entry] of index.entries()) {
-        result[id] = {
-            totalOccurrences: entry.totalOccurrences,
-            samples: entry.samples,
-        };
-    }
-    return result;
-}