agent-remnote 0.1.0 → 0.3.0

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

@@ -1,130 +0,0 @@
-import { z } from "zod";
-export const TIME_RANGE_PATTERN = /^(\d+)\s*([hdwmy])$/i;
-// 允许将 "all" 或 "*" 视为无时间限制（即不应用任何时间过滤）
-const ALL_TIME_RANGE_TOKENS = new Set(["all", "*"]);
-export const timeValueSchema = z.union([z.number().finite(), z.string().min(1)]);
-export function resolveTimeFilters(input, options) {
-    const filters = {};
-    const summary = {};
-    // 确保在所有分支中都有定义，避免引用错误
-    let effectiveTimeRange;
-    const explicitProvided = input.timeRange !== undefined ||
-        input.createdAfter !== undefined ||
-        input.createdBefore !== undefined ||
-        input.updatedAfter !== undefined ||
-        input.updatedBefore !== undefined;
-    // 规范化 timeRange 字符串
-    const normalizedTimeRange = typeof input.timeRange === "string" ? input.timeRange.trim().toLowerCase() : undefined;
-    // 如果显式传入 all/*，表示无时间限制：不应用任何时间过滤条件
-    if (normalizedTimeRange && ALL_TIME_RANGE_TOKENS.has(normalizedTimeRange)) {
-        summary.timeRange = input.timeRange ?? "all";
-        // 直接跳过默认 timeRange 与阈值计算
-    }
-    else {
-        effectiveTimeRange =
-            input.timeRange ?? (!explicitProvided ? options?.defaultTimeRange : undefined);
-        if (effectiveTimeRange) {
-            const durationMs = parseTimeRange(effectiveTimeRange);
-            if (durationMs == null) {
-                throw new Error("timeRange 解析失败，需形如 '30d'、'2w'、'12h'");
-            }
-            const threshold = Date.now() - durationMs;
-            if (input.updatedAfter === undefined) {
-                filters.updatedAfter = threshold;
-                summary.updatedAfter = describeTimestamp(threshold);
-            }
-            if (input.createdAfter === undefined) {
-                filters.createdAfter = threshold;
-                summary.createdAfter = describeTimestamp(threshold);
-            }
-            summary.timeRange = effectiveTimeRange;
-        }
-    }
-    const explicitMappings = [
-        { key: "createdAfter", value: input.createdAfter, summaryKey: "createdAfter" },
-        { key: "createdBefore", value: input.createdBefore, summaryKey: "createdBefore" },
-        { key: "updatedAfter", value: input.updatedAfter, summaryKey: "updatedAfter" },
-        { key: "updatedBefore", value: input.updatedBefore, summaryKey: "updatedBefore" },
-    ];
-    for (const entry of explicitMappings) {
-        if (entry.value === undefined || entry.value === null)
-            continue;
-        const parsed = parseTemporalInput(entry.value);
-        if (parsed == null) {
-            throw new Error(`${entry.summaryKey} 解析失败，需要传入毫秒时间戳或 ISO 日期字符串`);
-        }
-        ;
-        filters[entry.key] = parsed;
-        summary[entry.summaryKey] = describeTimestamp(parsed);
-    }
-    return { filters, summary, effectiveTimeRange };
-}
-export function parseTemporalInput(value) {
-    if (typeof value === "number" && Number.isFinite(value)) {
-        return Math.trunc(value);
-    }
-    if (typeof value === "string") {
-        const trimmed = value.trim();
-        if (!trimmed)
-            return null;
-        if (/^-?\d+$/.test(trimmed)) {
-            const num = Number(trimmed);
-            return Number.isFinite(num) ? Math.trunc(num) : null;
-        }
-        const parsed = Date.parse(trimmed);
-        if (Number.isNaN(parsed)) {
-            return null;
-        }
-        return parsed;
-    }
-    return null;
-}
-export function parseTimeRange(input) {
-    const match = input.match(TIME_RANGE_PATTERN);
-    if (!match)
-        return null;
-    const value = Number(match[1]);
-    if (!Number.isFinite(value))
-        return null;
-    const unit = match[2].toLowerCase();
-    const base = {
-        h: 60 * 60 * 1000,
-        d: 24 * 60 * 60 * 1000,
-        w: 7 * 24 * 60 * 60 * 1000,
-        m: 30 * 24 * 60 * 60 * 1000,
-        y: 365 * 24 * 60 * 60 * 1000,
-    };
-    const multiplier = base[unit];
-    if (!multiplier)
-        return null;
-    return value * multiplier;
-}
-export function describeTimestamp(ms) {
-    return {
-        ms,
-        iso: new Date(ms).toISOString(),
-    };
-}
-export function describeFilterSummary(summary) {
-    if (!summary)
-        return null;
-    const parts = [];
-    if (summary.timeRange) {
-        parts.push(`timeRange=${summary.timeRange}`);
-    }
-    if (summary.updatedAfter) {
-        parts.push(`updated ≥ ${summary.updatedAfter.iso}`);
-    }
-    if (summary.updatedBefore) {
-        parts.push(`updated ≤ ${summary.updatedBefore.iso}`);
-    }
-    if (summary.createdAfter) {
-        parts.push(`created ≥ ${summary.createdAfter.iso}`);
-    }
-    if (summary.createdBefore) {
-        parts.push(`created ≤ ${summary.createdBefore.iso}`);
-    }
-    if (parts.length === 0)
-        return null;
-    return parts.join("，");
-}

package/dist/packages/mcp/src/ws/bridge.js

@@ -1,377 +0,0 @@
-import { WebSocketServer } from "ws";
-import fs from "node:fs";
-import path from "node:path";
-import { openQueueDb } from "../queue/db.js";
-import { ackDead, ackRetry, ackSuccess, claimNextOp, getTxnIdByOpId, queueStats, recoverExpiredLeases, upsertIdMap } from "../queue/dao.js";
-const clientMeta = new Map();
-const GLOBAL_BRIDGE_KEY = Symbol.for("__REMNOTE_WS_BRIDGE__");
-const globalAny = globalThis;
-function setStoredBridge(bridge) {
-    if (bridge) {
-        globalAny[GLOBAL_BRIDGE_KEY] = bridge;
-    }
-    else {
-        delete globalAny[GLOBAL_BRIDGE_KEY];
-    }
-}
-function isWssListening(wss) {
-    if (!wss)
-        return false;
-    try {
-        const addr = wss.address();
-        if (!addr)
-            return false;
-        if (typeof addr === "string")
-            return addr.length > 0;
-        if (typeof addr.port === "number")
-            return true;
-        return false;
-    }
-    catch (_) {
-        return false;
-    }
-}
-function getStoredBridge() {
-    const existing = globalAny[GLOBAL_BRIDGE_KEY];
-    if (!existing)
-        return undefined;
-    if (!isWssListening(existing.wss)) {
-        setStoredBridge(undefined);
-        return undefined;
-    }
-    return existing;
-}
-export function getWsStatus() {
-    const rows = [];
-    for (const ws of clientMeta.keys()) {
-        const meta = clientMeta.get(ws);
-        rows.push({
-            consumerId: meta.consumerId,
-            connectedAt: meta.connectedAt,
-            lastSeenAt: meta.lastSeenAt,
-            remoteAddr: meta.remoteAddr,
-            readyState: ws.readyState,
-        });
-    }
-    return { clients: rows };
-}
-export function notifyStartSync(targetConsumerId) {
-    let sent = 0;
-    for (const [ws, meta] of clientMeta.entries()) {
-        if (targetConsumerId && meta.consumerId !== targetConsumerId)
-            continue;
-        try {
-            ws.send(JSON.stringify({ type: "StartSync" }));
-            if (DEBUG)
-                log("StartSync sent", { to: meta.consumerId, remote: meta.remoteAddr });
-            sent += 1;
-        }
-        catch (_) { }
-    }
-    return { sent };
-}
-/**
- * 启动最小可用的 WebSocket 桥接服务。
- * - 默认端口：从环境变量 `REMNOTE_WS_PORT`、`WS_PORT`，否则 3010
- * - 默认路径：`/ws`
- * - 支持心跳保活，客户端可按需发送/响应 `ping/pong`
- */
-export function startWebSocketBridge(opts = {}) {
-    const enabled = opts.enable ?? envEnabled();
-    if (!enabled)
-        return undefined;
-    const port = opts.port ?? envPort();
-    const path = opts.path ?? process.env.REMNOTE_WS_PATH ?? "/ws";
-    const heartbeatIntervalMs = opts.heartbeatIntervalMs ?? 30_000;
-    // 单进程热更新守卫：若已启动则直接复用，避免重复绑定端口
-    const existing = getStoredBridge();
-    if (existing)
-        return existing;
-    let wss;
-    try {
-        wss = new WebSocketServer({ port, path });
-    }
-    catch (e) {
-        const msg = String(e?.message || e || "");
-        if (msg.includes("EADDRINUSE")) {
-            // 端口占用常见于 watch 重启时旧进程尚未完全释放；此处仅提示并返回。
-            // 建议配合进程退出时的 close() 钩子与单进程守卫，减少该情况出现概率。
-            console.error(`ws port already in use (${port}${path}); likely due to hot reload overlap.`);
-            return undefined;
-        }
-        console.error(`failed to start websocket bridge on port ${port}${path}:`, msg);
-        return undefined;
-    }
-    const db = openQueueDb();
-    if (DEBUG)
-        log("WS bridge started", { port, path });
-    wss.on("listening", () => {
-        console.log(`websocket bridge ready at ws://localhost:${port}${path}`);
-    });
-    wss.on("connection", (ws, req) => {
-        ;
-        ws.isAlive = true;
-        const now = Date.now();
-        const remoteAddr = req.socket.remoteAddress;
-        const userAgent = req.headers["user-agent"];
-        clientMeta.set(ws, { connectedAt: now, lastSeenAt: now, remoteAddr, userAgent });
-        if (DEBUG)
-            log("connection", { remoteAddr, userAgent });
-        ws.on("pong", () => {
-            ;
-            ws.isAlive = true;
-            const meta = clientMeta.get(ws);
-            if (meta)
-                meta.lastSeenAt = Date.now();
-        });
-        ws.on("message", (raw) => {
-            try {
-                const txt = raw.toString();
-                if (txt === "ping") {
-                    ws.send("pong");
-                    return;
-                }
-                const msg = JSON.parse(txt);
-                const meta = clientMeta.get(ws);
-                if (meta)
-                    meta.lastSeenAt = Date.now();
-                if (DEBUG)
-                    log("message", { from: meta?.consumerId, type: msg?.type });
-                handleMessage(ws, db, msg, ws);
-            }
-            catch (e) {
-                ws.send(JSON.stringify({ type: "error", message: e.message }));
-            }
-        });
-        ws.on("close", () => {
-            // 连接关闭时清理
-            clientMeta.delete(ws);
-            if (DEBUG)
-                log("close");
-        });
-    });
-    // 心跳保活，剔除断开连接的客户端
-    const interval = setInterval(() => {
-        wss.clients.forEach((ws) => {
-            const sock = ws;
-            if (sock.isAlive === false)
-                return ws.terminate();
-            sock.isAlive = false;
-            ws.ping();
-        });
-        // 回收过期租约，避免任务卡死
-        try {
-            const n = recoverExpiredLeases(db);
-            if (DEBUG && n > 0)
-                log("lease_recovered", { count: n });
-        }
-        catch (_) { }
-    }, heartbeatIntervalMs);
-    const started = {
-        wss,
-        close: async () => {
-            clearInterval(interval);
-            await new Promise((resolve) => wss.close(() => resolve()));
-            if (globalAny[GLOBAL_BRIDGE_KEY] === started) {
-                setStoredBridge(undefined);
-            }
-        },
-    };
-    // 记录到 global，避免单进程内的二次初始化
-    setStoredBridge(started);
-    // 监听退出信号，尽快释放端口，减少 watch 重启时的占用窗口
-    const onExit = async () => {
-        try {
-            await started.close();
-        }
-        catch (_) { }
-    };
-    try {
-        process.once("SIGINT", onExit);
-        process.once("SIGTERM", onExit);
-        process.once("exit", onExit);
-    }
-    catch (_) { }
-    return started;
-}
-export function ensureWebSocketBridge(opts = {}) {
-    const existing = getStoredBridge();
-    if (existing)
-        return { bridge: existing, restarted: false };
-    const bridge = startWebSocketBridge({ ...opts, enable: opts.enable ?? true });
-    return { bridge, restarted: !!bridge };
-}
-function envPort() {
-    const raw = process.env.REMNOTE_WS_PORT || process.env.WS_PORT;
-    const n = Number(raw);
-    return Number.isFinite(n) && n > 0 ? n : 3010;
-}
-function envEnabled() {
-    const flag = (process.env.REMNOTE_WS_ENABLED || process.env.WS_ENABLED || "").toLowerCase();
-    const disabled = (process.env.REMNOTE_WS_DISABLED || process.env.NO_WS || "").toLowerCase();
-    if (disabled === "1" || disabled === "true")
-        return false;
-    if (flag === "0" || flag === "false")
-        return false;
-    // 默认开启
-    return true;
-}
-function handleMessage(ws, db, msg, sock) {
-    const send = (obj) => ws.send(JSON.stringify(obj));
-    switch (msg?.type) {
-        case "Hello": {
-            // 暂不强制鉴权，保留占位
-            send({ type: "HelloAck", ok: true });
-            break;
-        }
-        case "Register": {
-            const consumerId = String(msg.consumerId || "") || undefined;
-            const meta = clientMeta.get(sock);
-            if (meta)
-                meta.consumerId = consumerId;
-            if (DEBUG)
-                log("registered", { consumerId });
-            return send({ type: "Registered", consumerId });
-        }
-        case "RequestOp": {
-            const consumer = msg.consumerId || "plugin";
-            const meta = clientMeta.get(sock);
-            if (meta)
-                meta.consumerId = consumer;
-            const leaseMs = typeof msg.leaseMs === "number" ? msg.leaseMs : 30000;
-            const op = claimNextOp(db, consumer, leaseMs);
-            if (!op) {
-                if (DEBUG)
-                    log("no_work", { consumer });
-                return send({ type: "NoWork" });
-            }
-            if (DEBUG)
-                log("dispatch", { op_id: op.op_id, txn_id: op.txn_id, type: op.type, consumer });
-            return send({
-                type: "OpDispatch",
-                op_id: op.op_id,
-                txn_id: op.txn_id,
-                op_seq: op.op_seq,
-                op_type: op.type,
-                payload: safeParseJson(op.payload_json),
-                idempotency_key: op.idempotency_key,
-            });
-        }
-        case "TriggerStartSync": {
-            // 开发用途：收到管理指令后向所有或指定 consumerId 广播 StartSync
-            const target = typeof msg?.consumerId === 'string' && msg.consumerId.trim() ? String(msg.consumerId) : undefined;
-            const res = notifyStartSync(target);
-            if (DEBUG)
-                log("trigger_start_sync", { target, sent: res.sent });
-            return send({ type: "StartSyncTriggered", sent: res.sent });
-        }
-        case "OpAck": {
-            const { op_id, status } = msg;
-            if (!op_id || !status)
-                return send({ type: "Error", message: "invalid OpAck" });
-            if (status === "success") {
-                ackSuccess(db, op_id, msg.result || null);
-                // 可选的 ID 映射回填
-                try {
-                    const mappings = [];
-                    const created = msg?.result?.created;
-                    if (created?.client_temp_id && created?.remote_id)
-                        mappings.push({ client_temp_id: created.client_temp_id, remote_id: created.remote_id, remote_type: created.remote_type || 'rem' });
-                    if (Array.isArray(msg?.result?.id_map)) {
-                        for (const m of msg.result.id_map)
-                            if (m?.client_temp_id && m?.remote_id)
-                                mappings.push({ client_temp_id: m.client_temp_id, remote_id: m.remote_id, remote_type: m.remote_type || 'rem' });
-                    }
-                    if (Array.isArray(msg?.remote_id_map)) {
-                        for (const m of msg.remote_id_map)
-                            if (m?.client_temp_id && m?.remote_id)
-                                mappings.push({ client_temp_id: m.client_temp_id, remote_id: m.remote_id, remote_type: m.remote_type || 'rem' });
-                    }
-                    if (mappings.length > 0) {
-                        const txn_id = getTxnIdByOpId(db, op_id);
-                        upsertIdMap(db, mappings.map((m) => ({ ...m, source_txn: txn_id })));
-                    }
-                    if (DEBUG)
-                        log("ack_success", { op_id, mappings: msg?.result?.created ? 1 : (Array.isArray(msg?.result?.id_map) ? msg.result.id_map.length : 0) });
-                }
-                catch (_) { }
-                return send({ type: "AckOk", op_id });
-            }
-            else if (status === "retry") {
-                ackRetry(db, op_id, { code: msg.error_code, message: msg.error_message, retryAfterMs: msg.retry_after_ms });
-                if (DEBUG)
-                    log("ack_retry", { op_id, code: msg.error_code });
-                return send({ type: "AckOk", op_id });
-            }
-            else if (status === "failed" || status === "dead") {
-                ackDead(db, op_id, { code: msg.error_code, message: msg.error_message });
-                if (DEBUG)
-                    log("ack_dead", { op_id, code: msg.error_code });
-                return send({ type: "AckOk", op_id });
-            }
-            return send({ type: "Error", message: `unknown status ${status}` });
-        }
-        case "QueryStats": {
-            const st = queueStats(db);
-            if (DEBUG)
-                log("stats", st);
-            return send({ type: "Stats", ...st });
-        }
-        case "QueryClients": {
-            const status = getWsStatus();
-            return send({ type: "Clients", clients: status.clients });
-        }
-        case "WhoAmI": {
-            const meta = clientMeta.get(sock);
-            return send({ type: "YouAre", consumerId: meta?.consumerId, lastSeenAt: meta?.lastSeenAt });
-        }
-        default: {
-            return send({ type: "Error", message: "unknown message type" });
-        }
-    }
-}
-function safeParseJson(s) {
-    try {
-        return JSON.parse(s);
-    }
-    catch {
-        return null;
-    }
-}
-const DEBUG = envDebug();
-const LOG_FILE = envLogFilePath();
-function envDebug() {
-    const v = (process.env.REMNOTE_WS_DEBUG || process.env.WS_DEBUG || '').toLowerCase();
-    return v === '1' || v === 'true';
-}
-function log(msg, ctx) {
-    try {
-        const line = `[ws] ${msg}`;
-        const ctxStr = ctx ? JSON.stringify(ctx) : '';
-        console.log(line, ctxStr);
-        if (LOG_FILE) {
-            try {
-                const record = `${new Date().toISOString()} ${line} ${ctxStr}\n`;
-                fs.appendFileSync(LOG_FILE, record);
-            }
-            catch (_) { }
-        }
-    }
-    catch {
-        console.log(`[ws] ${msg}`);
-    }
-}
-function envLogFilePath() {
-    const p = process.env.REMNOTE_WS_LOGFILE || process.env.WS_LOGFILE || '';
-    const home = process.env.HOME || process.env.USERPROFILE || '.';
-    const def = DEBUG ? path.join(home, '.remnote-mcp', 'ws-debug.log') : '';
-    const out = p || def;
-    if (!out)
-        return undefined;
-    try {
-        const dir = path.dirname(out);
-        fs.mkdirSync(dir, { recursive: true });
-    }
-    catch (_) { }
-    return out;
-}