wowok 2.1.9 → 2.1.12

package/dist/cjs/w/messenger/storage.js

@@ -1,85 +1,114 @@
 "use strict";
-var __create = Object.create;
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __getProtoOf = Object.getPrototypeOf;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
-  // If the importer is in node compatibility mode or this is not an ESM
-  // file that has been converted to a CommonJS file using a Babel-
-  // compatible transform (i.e. "__esModule" has not been set), then set
-  // "default" to the CommonJS "module.exports" for node compatibility.
-  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
-  mod
-));
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
-var storage_exports = {};
-__export(storage_exports, {
-  MessageStorage: () => MessageStorage,
-  MessengerPersistStorage: () => MessengerPersistStorage,
-  SessionStateStorage: () => SessionStateStorage,
-  SignalProtocolStorage: () => SignalProtocolStorage,
-  defaultPersistStorage: () => defaultPersistStorage,
-  resetDbConnection: () => resetDbConnection
+// Copyright (c) Wowok.
+// SPDX-License-Identifier: Apache-2.0
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
 });
-module.exports = __toCommonJS(storage_exports);
-var import_better_sqlite3 = __toESM(require("better-sqlite3"));
-var path = __toESM(require("path"));
-var fs = __toESM(require("fs"));
-var import_util = require("../local/util.js");
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.defaultPersistStorage = exports.MessengerPersistStorage = exports.SessionStateStorage = exports.MessageStorage = exports.SignalProtocolStorage = void 0;
+exports.resetDbConnection = resetDbConnection;
+/**
+ * Messenger 存储模块
+ *
+ * 使用与 wowok SDK 相同的 SQLite 存储方案
+ * 数据库文件：messenger.db（位于 ~/.wow/ 目录）
+ *
+ * 存储内容：
+ * 1. Signal Protocol 状态（身份密钥、预密钥、会话）
+ * 2. 消息记录
+ * 3. 会话状态（Merkle Tree 根哈希等）
+ */
+const better_sqlite3_1 = __importDefault(require("better-sqlite3"));
+const path = __importStar(require("path"));
+const fs = __importStar(require("fs"));
+const util_js_1 = require("../local/util.js");
+// 动态获取数据目录（支持测试时通过环境变量覆盖）
 function getWowDirDynamic() {
-  return (0, import_util.getWowDir)();
+    return (0, util_js_1.getWowDir)();
 }
+// 按用户分库的数据库目录（动态计算）
 function getMessengerDbDir() {
-  return path.join(getWowDirDynamic(), "messenger");
+    return path.join(getWowDirDynamic(), "messenger");
 }
 function ensureDirExists(dir) {
-  if (!fs.existsSync(dir)) {
-    fs.mkdirSync(dir, { recursive: true });
-  }
+    if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+    }
 }
+// 获取用户特定的数据库路径
 function getUserDbPath(userAddress) {
-  const shortAddress = userAddress.toLowerCase().replace(/^0x/, "").slice(0, 16);
-  const messengerDbDir = getMessengerDbDir();
-  ensureDirExists(messengerDbDir);
-  return path.join(messengerDbDir, `${shortAddress}.db`);
+    // 使用用户地址的前16位作为数据库文件名
+    const shortAddress = userAddress
+        .toLowerCase()
+        .replace(/^0x/, "")
+        .slice(0, 16);
+    const messengerDbDir = getMessengerDbDir();
+    ensureDirExists(messengerDbDir);
+    return path.join(messengerDbDir, `${shortAddress}.db`);
 }
+// 获取全局数据库路径（用于 MessengerManager 状态和 SSE 锁）
 function getGlobalDbPath() {
-  const messengerDbDir = getMessengerDbDir();
-  ensureDirExists(messengerDbDir);
-  return path.join(messengerDbDir, "global.db");
+    const messengerDbDir = getMessengerDbDir();
+    ensureDirExists(messengerDbDir);
+    return path.join(messengerDbDir, "global.db");
 }
-const userDbInstances = /* @__PURE__ */ new Map();
+// 用户数据库实例缓存
+const userDbInstances = new Map();
+// 全局数据库实例缓存
 let globalDbInstance = null;
 function getDatabase(userAddress) {
-  if (!userAddress) {
-    throw new Error("userAddress is required to get database");
-  }
-  const cached = userDbInstances.get(userAddress);
-  if (cached) {
-    return cached;
-  }
-  const dbPath = getUserDbPath(userAddress);
-  const db = new import_better_sqlite3.default(dbPath);
-  db.pragma("journal_mode = WAL");
-  db.pragma("synchronous = NORMAL");
-  db.pragma("cache_size = -64000");
-  db.exec(`
+    if (!userAddress) {
+        throw new Error("userAddress is required to get database");
+    }
+    // 检查缓存
+    const cached = userDbInstances.get(userAddress);
+    if (cached) {
+        return cached;
+    }
+    // 创建新连接
+    const dbPath = getUserDbPath(userAddress);
+    const db = new better_sqlite3_1.default(dbPath);
+    // 启用 WAL 模式以支持多进程访问
+    db.pragma("journal_mode = WAL");
+    // 设置同步模式为 NORMAL 以平衡性能和安全性
+    db.pragma("synchronous = NORMAL");
+    // 启用内存共享缓存
+    db.pragma("cache_size = -64000");
+    // Signal Protocol 存储表
+    db.exec(`
     CREATE TABLE IF NOT EXISTS signal_store (
       store_type TEXT NOT NULL,
       key TEXT NOT NULL,
@@ -89,8 +118,9 @@ function getDatabase(userAddress) {
     );
     CREATE INDEX IF NOT EXISTS idx_signal_store_type ON signal_store(store_type);
   `);
-  try {
-    db.exec(`
+    // 消息表 - 分开创建，先不创建索引
+    try {
+        db.exec(`
       CREATE TABLE IF NOT EXISTS messages (
         message_id TEXT PRIMARY KEY,
         from_address TEXT NOT NULL,
@@ -117,59 +147,68 @@ function getDatabase(userAddress) {
         received_at INTEGER
       );
     `);
-  } catch (e) {
-  }
-  try {
-    db.exec("ALTER TABLE messages ADD COLUMN guard_address TEXT");
-  } catch (e) {
-  }
-  try {
-    db.exec("ALTER TABLE messages ADD COLUMN passport_address TEXT");
-  } catch (e) {
-  }
-  try {
-    db.exec(
-      "ALTER TABLE messages ADD COLUMN last_received_leaf_index INTEGER"
-    );
-  } catch (e) {
-  }
-  try {
-    db.exec(
-      "CREATE INDEX IF NOT EXISTS idx_messages_from_to ON messages(from_address, to_address)"
-    );
-  } catch (e) {
-  }
-  try {
-    db.exec(
-      "CREATE INDEX IF NOT EXISTS idx_messages_status ON messages(status)"
-    );
-  } catch (e) {
-  }
-  try {
-    db.exec(
-      "CREATE INDEX IF NOT EXISTS idx_messages_leaf ON messages(leaf_index)"
-    );
-  } catch (e) {
-  }
-  try {
-    db.exec(
-      "CREATE INDEX IF NOT EXISTS idx_messages_created_at ON messages(created_at)"
-    );
-  } catch (e) {
-  }
-  try {
-    db.exec(
-      "CREATE INDEX IF NOT EXISTS idx_messages_server_ts ON messages(server_timestamp)"
-    );
-  } catch (e) {
-  }
-  try {
-    db.exec(
-      "CREATE INDEX IF NOT EXISTS idx_messages_received_at ON messages(received_at)"
-    );
-  } catch (e) {
-  }
-  db.exec(`
+    }
+    catch (e) {
+        // 表可能已存在，忽略错误
+    }
+    // 安全地添加新列（如果还不存在）
+    try {
+        db.exec("ALTER TABLE messages ADD COLUMN guard_address TEXT");
+    }
+    catch (e) {
+        // 列可能已存在，忽略错误
+    }
+    try {
+        db.exec("ALTER TABLE messages ADD COLUMN passport_address TEXT");
+    }
+    catch (e) {
+        // 列可能已存在，忽略错误
+    }
+    try {
+        db.exec("ALTER TABLE messages ADD COLUMN last_received_leaf_index INTEGER");
+    }
+    catch (e) {
+        // 列可能已存在，忽略错误
+    }
+    // 安全地创建索引
+    try {
+        db.exec("CREATE INDEX IF NOT EXISTS idx_messages_from_to ON messages(from_address, to_address)");
+    }
+    catch (e) {
+        // 索引可能已存在，忽略错误
+    }
+    try {
+        db.exec("CREATE INDEX IF NOT EXISTS idx_messages_status ON messages(status)");
+    }
+    catch (e) {
+        // 索引可能已存在，忽略错误
+    }
+    try {
+        db.exec("CREATE INDEX IF NOT EXISTS idx_messages_leaf ON messages(leaf_index)");
+    }
+    catch (e) {
+        // 索引可能已存在，忽略错误
+    }
+    try {
+        db.exec("CREATE INDEX IF NOT EXISTS idx_messages_created_at ON messages(created_at)");
+    }
+    catch (e) {
+        // 索引可能已存在，忽略错误
+    }
+    try {
+        db.exec("CREATE INDEX IF NOT EXISTS idx_messages_server_ts ON messages(server_timestamp)");
+    }
+    catch (e) {
+        // 索引可能已存在，忽略错误
+    }
+    try {
+        db.exec("CREATE INDEX IF NOT EXISTS idx_messages_received_at ON messages(received_at)");
+    }
+    catch (e) {
+        // 索引可能已存在，忽略错误
+    }
+    // 会话状态表
+    db.exec(`
     CREATE TABLE IF NOT EXISTS session_states (
       session_id TEXT PRIMARY KEY,
       my_address TEXT NOT NULL,
@@ -185,249 +224,294 @@ function getDatabase(userAddress) {
     );
     CREATE INDEX IF NOT EXISTS idx_session_my_peer ON session_states(my_address, peer_address);
   `);
-  userDbInstances.set(userAddress, db);
-  return db;
+    // 缓存实例
+    userDbInstances.set(userAddress, db);
+    return db;
 }
+// 获取全局数据库实例（用于 MessengerManager 状态和 SSE 锁）
 function getGlobalDatabase() {
-  if (globalDbInstance) {
-    return globalDbInstance;
-  }
-  const dbPath = getGlobalDbPath();
-  const db = new import_better_sqlite3.default(dbPath);
-  db.pragma("journal_mode = WAL");
-  db.pragma("synchronous = NORMAL");
-  db.pragma("cache_size = -10000");
-  db.pragma("busy_timeout = 5000");
-  globalDbInstance = db;
-  return db;
+    if (globalDbInstance) {
+        return globalDbInstance;
+    }
+    // 创建新连接
+    const dbPath = getGlobalDbPath();
+    const db = new better_sqlite3_1.default(dbPath);
+    // 启用 WAL 模式以支持多进程访问
+    db.pragma("journal_mode = WAL");
+    // 设置同步模式为 NORMAL 以平衡性能和安全性
+    db.pragma("synchronous = NORMAL");
+    // 设置缓存大小
+    db.pragma("cache_size = -10000");
+    // 设置忙等待超时（5秒）
+    db.pragma("busy_timeout = 5000");
+    // 缓存实例
+    globalDbInstance = db;
+    return db;
 }
+/**
+ * 重置数据库连接
+ * 用于测试时清理数据后重新创建连接
+ */
 function resetDbConnection() {
-  for (const [_, db] of userDbInstances) {
-    try {
-      db.pragma("journal_mode = DELETE");
-    } catch (e) {
-    }
-  }
-  for (const [_, db] of userDbInstances) {
-    try {
-      db.close();
-    } catch (e) {
+    // 先切换到 DELETE 模式，这样就不会有 WAL 文件
+    for (const [_, db] of userDbInstances) {
+        try {
+            db.pragma("journal_mode = DELETE");
+        }
+        catch (e) {
+            // 忽略模式切换错误
+        }
     }
-  }
-  userDbInstances.clear();
-  if (globalDbInstance) {
-    try {
-      globalDbInstance.pragma("journal_mode = DELETE");
-    } catch (e) {
+    // 关闭所有用户的数据库连接
+    for (const [_, db] of userDbInstances) {
+        try {
+            db.close();
+        }
+        catch (e) {
+            // 忽略关闭错误
+        }
     }
-    try {
-      globalDbInstance.close();
-    } catch (e) {
+    userDbInstances.clear();
+    // 关闭全局数据库连接
+    if (globalDbInstance) {
+        try {
+            globalDbInstance.pragma("journal_mode = DELETE");
+        }
+        catch (e) {
+            // 忽略模式切换错误
+        }
+        try {
+            globalDbInstance.close();
+        }
+        catch (e) {
+            // 忽略关闭错误
+        }
+        globalDbInstance = null;
     }
-    globalDbInstance = null;
-  }
 }
+// 注册进程退出时的清理函数
 process.on("exit", () => {
-  resetDbConnection();
+    resetDbConnection();
 });
+// 处理 SIGINT (Ctrl+C)
 process.on("SIGINT", () => {
-  resetDbConnection();
-  process.exit(0);
+    resetDbConnection();
+    process.exit(0);
 });
+// 处理 SIGTERM
 process.on("SIGTERM", () => {
-  resetDbConnection();
-  process.exit(0);
+    resetDbConnection();
+    process.exit(0);
 });
+// ========== Signal Protocol 存储 ==========
 const STORE_TYPES = {
-  IDENTITY: "identity",
-  REGISTRATION_ID: "registration_id",
-  IDENTITIES: "identities",
-  PREKEYS: "prekeys",
-  SIGNED_PREKEYS: "signed_prekeys",
-  SESSIONS: "sessions",
-  META: "meta"
+    IDENTITY: "identity",
+    REGISTRATION_ID: "registration_id",
+    IDENTITIES: "identities",
+    PREKEYS: "prekeys",
+    SIGNED_PREKEYS: "signed_prekeys",
+    SESSIONS: "sessions",
+    META: "meta",
 };
 function arrayBufferToBase64(buffer) {
-  return Buffer.from(new Uint8Array(buffer)).toString("base64");
+    return Buffer.from(new Uint8Array(buffer)).toString("base64");
 }
 function base64ToArrayBuffer(base64) {
-  return Uint8Array.from(Buffer.from(base64, "base64")).buffer;
+    return Uint8Array.from(Buffer.from(base64, "base64")).buffer;
 }
 function bufferEqual(a, b) {
-  if (a.byteLength !== b.byteLength) return false;
-  const va = new Uint8Array(a);
-  const vb = new Uint8Array(b);
-  for (let i = 0; i < va.length; i++) {
-    if (va[i] !== vb[i]) return false;
-  }
-  return true;
+    if (a.byteLength !== b.byteLength)
+        return false;
+    const va = new Uint8Array(a);
+    const vb = new Uint8Array(b);
+    for (let i = 0; i < va.length; i++) {
+        if (va[i] !== vb[i])
+            return false;
+    }
+    return true;
 }
+/**
+ * Signal Protocol SQLite 存储
+ * 兼容 libsignal-protocol-typescript 的 StorageType 接口
+ */
 class SignalProtocolStorage {
-  constructor(userAddress, storeType = "default") {
-    __publicField(this, "storePrefix");
-    __publicField(this, "userAddress");
-    this.userAddress = userAddress;
-    this.storePrefix = storeType;
-  }
-  get db() {
-    return getDatabase(this.userAddress);
-  }
-  // Identity
-  async setIdentity(identity, registrationId) {
-    const persisted = {
-      pubKey: arrayBufferToBase64(identity.pubKey),
-      privKey: arrayBufferToBase64(identity.privKey)
-    };
-    this.putValue(STORE_TYPES.IDENTITY, JSON.stringify(persisted));
-    this.putValue(STORE_TYPES.REGISTRATION_ID, registrationId.toString());
-  }
-  async getIdentityKeyPair() {
-    const value = this.getValue(STORE_TYPES.IDENTITY);
-    if (!value) return void 0;
-    const persisted = JSON.parse(value);
-    return {
-      pubKey: base64ToArrayBuffer(persisted.pubKey),
-      privKey: base64ToArrayBuffer(persisted.privKey)
-    };
-  }
-  async getLocalRegistrationId() {
-    const value = this.getValue(STORE_TYPES.REGISTRATION_ID);
-    return value ? parseInt(value, 10) : void 0;
-  }
-  // Identity verification
-  async isTrustedIdentity(identifier, identityKey) {
-    const key = `${STORE_TYPES.IDENTITIES}:${identifier}`;
-    const existing = this.getValue(key);
-    if (!existing) return true;
-    return bufferEqual(base64ToArrayBuffer(existing), identityKey);
-  }
-  async saveIdentity(encodedAddress, publicKey) {
-    const key = `${STORE_TYPES.IDENTITIES}:${encodedAddress}`;
-    const existing = this.getValue(key);
-    const newValue = arrayBufferToBase64(publicKey);
-    this.putValue(key, newValue);
-    return existing ? bufferEqual(base64ToArrayBuffer(existing), publicKey) : true;
-  }
-  async loadIdentity(address) {
-    const key = `${STORE_TYPES.IDENTITIES}:${address}`;
-    const value = this.getValue(key);
-    return value ? base64ToArrayBuffer(value) : void 0;
-  }
-  // PreKeys
-  async loadPreKey(keyId) {
-    const key = `${STORE_TYPES.PREKEYS}:${keyId}`;
-    const value = this.getValue(key);
-    if (!value) return void 0;
-    const persisted = JSON.parse(value);
-    return {
-      pubKey: base64ToArrayBuffer(persisted.pubKey),
-      privKey: base64ToArrayBuffer(persisted.privKey)
-    };
-  }
-  async storePreKey(keyId, keyPair) {
-    const key = `${STORE_TYPES.PREKEYS}:${keyId}`;
-    const persisted = {
-      pubKey: arrayBufferToBase64(keyPair.pubKey),
-      privKey: arrayBufferToBase64(keyPair.privKey)
-    };
-    this.putValue(key, JSON.stringify(persisted));
-  }
-  async removePreKey(keyId) {
-    const key = `${STORE_TYPES.PREKEYS}:${keyId}`;
-    this.delValue(key);
-  }
-  // Signed PreKeys
-  async loadSignedPreKey(keyId) {
-    const key = `${STORE_TYPES.SIGNED_PREKEYS}:${keyId}`;
-    const value = this.getValue(key);
-    if (!value) return void 0;
-    const persisted = JSON.parse(value);
-    return {
-      pubKey: base64ToArrayBuffer(persisted.pubKey),
-      privKey: base64ToArrayBuffer(persisted.privKey)
-    };
-  }
-  async storeSignedPreKey(keyId, keyPair) {
-    const key = `${STORE_TYPES.SIGNED_PREKEYS}:${keyId}`;
-    const persisted = {
-      pubKey: arrayBufferToBase64(keyPair.pubKey),
-      privKey: arrayBufferToBase64(keyPair.privKey)
-    };
-    this.putValue(key, JSON.stringify(persisted));
-  }
-  async removeSignedPreKey(keyId) {
-    const key = `${STORE_TYPES.SIGNED_PREKEYS}:${keyId}`;
-    this.delValue(key);
-  }
-  // Sessions
-  async loadSession(encodedAddress) {
-    const key = `${STORE_TYPES.SESSIONS}:${encodedAddress}`;
-    return this.getValue(key);
-  }
-  async storeSession(encodedAddress, record) {
-    const key = `${STORE_TYPES.SESSIONS}:${encodedAddress}`;
-    this.putValue(key, record);
-  }
-  async removeSession(encodedAddress) {
-    const key = `${STORE_TYPES.SESSIONS}:${encodedAddress}`;
-    this.delValue(key);
-  }
-  // Meta
-  async getMeta() {
-    const value = this.getValue(STORE_TYPES.META);
-    if (!value) return { nextPreKeyId: 1 };
-    return JSON.parse(value);
-  }
-  async setMeta(meta) {
-    const current = await this.getMeta();
-    const updated = { ...current, ...meta };
-    this.putValue(STORE_TYPES.META, JSON.stringify(updated));
-  }
-  listPreKeys() {
-    const rows = this.db.prepare(
-      "SELECT key FROM signal_store WHERE store_type = ? AND key LIKE ?"
-    ).all(this.storePrefix, `${STORE_TYPES.PREKEYS}:%`);
-    return rows.map((r) => {
-      const id = r.key.replace(`${STORE_TYPES.PREKEYS}:`, "");
-      return Number(id);
-    });
-  }
-  // Private helpers
-  getValue(key) {
-    const row = this.db.prepare(
-      "SELECT value FROM signal_store WHERE store_type = ? AND key = ?"
-    ).get(this.storePrefix, key);
-    return row?.value;
-  }
-  putValue(key, value) {
-    this.db.prepare(
-      `
+    storePrefix;
+    userAddress;
+    constructor(userAddress, storeType = "default") {
+        this.userAddress = userAddress;
+        this.storePrefix = storeType;
+    }
+    get db() {
+        return getDatabase(this.userAddress);
+    }
+    // Identity
+    async setIdentity(identity, registrationId) {
+        const persisted = {
+            pubKey: arrayBufferToBase64(identity.pubKey),
+            privKey: arrayBufferToBase64(identity.privKey),
+        };
+        this.putValue(STORE_TYPES.IDENTITY, JSON.stringify(persisted));
+        this.putValue(STORE_TYPES.REGISTRATION_ID, registrationId.toString());
+    }
+    async getIdentityKeyPair() {
+        const value = this.getValue(STORE_TYPES.IDENTITY);
+        if (!value)
+            return undefined;
+        const persisted = JSON.parse(value);
+        return {
+            pubKey: base64ToArrayBuffer(persisted.pubKey),
+            privKey: base64ToArrayBuffer(persisted.privKey),
+        };
+    }
+    async getLocalRegistrationId() {
+        const value = this.getValue(STORE_TYPES.REGISTRATION_ID);
+        return value ? parseInt(value, 10) : undefined;
+    }
+    // Identity verification
+    async isTrustedIdentity(identifier, identityKey) {
+        const key = `${STORE_TYPES.IDENTITIES}:${identifier}`;
+        const existing = this.getValue(key);
+        if (!existing)
+            return true;
+        return bufferEqual(base64ToArrayBuffer(existing), identityKey);
+    }
+    async saveIdentity(encodedAddress, publicKey) {
+        const key = `${STORE_TYPES.IDENTITIES}:${encodedAddress}`;
+        const existing = this.getValue(key);
+        const newValue = arrayBufferToBase64(publicKey);
+        this.putValue(key, newValue);
+        return existing
+            ? bufferEqual(base64ToArrayBuffer(existing), publicKey)
+            : true;
+    }
+    async loadIdentity(address) {
+        const key = `${STORE_TYPES.IDENTITIES}:${address}`;
+        const value = this.getValue(key);
+        return value ? base64ToArrayBuffer(value) : undefined;
+    }
+    // PreKeys
+    async loadPreKey(keyId) {
+        const key = `${STORE_TYPES.PREKEYS}:${keyId}`;
+        const value = this.getValue(key);
+        if (!value)
+            return undefined;
+        const persisted = JSON.parse(value);
+        return {
+            pubKey: base64ToArrayBuffer(persisted.pubKey),
+            privKey: base64ToArrayBuffer(persisted.privKey),
+        };
+    }
+    async storePreKey(keyId, keyPair) {
+        const key = `${STORE_TYPES.PREKEYS}:${keyId}`;
+        const persisted = {
+            pubKey: arrayBufferToBase64(keyPair.pubKey),
+            privKey: arrayBufferToBase64(keyPair.privKey),
+        };
+        this.putValue(key, JSON.stringify(persisted));
+    }
+    async removePreKey(keyId) {
+        const key = `${STORE_TYPES.PREKEYS}:${keyId}`;
+        this.delValue(key);
+    }
+    // Signed PreKeys
+    async loadSignedPreKey(keyId) {
+        const key = `${STORE_TYPES.SIGNED_PREKEYS}:${keyId}`;
+        const value = this.getValue(key);
+        if (!value)
+            return undefined;
+        const persisted = JSON.parse(value);
+        return {
+            pubKey: base64ToArrayBuffer(persisted.pubKey),
+            privKey: base64ToArrayBuffer(persisted.privKey),
+        };
+    }
+    async storeSignedPreKey(keyId, keyPair) {
+        const key = `${STORE_TYPES.SIGNED_PREKEYS}:${keyId}`;
+        const persisted = {
+            pubKey: arrayBufferToBase64(keyPair.pubKey),
+            privKey: arrayBufferToBase64(keyPair.privKey),
+        };
+        this.putValue(key, JSON.stringify(persisted));
+    }
+    async removeSignedPreKey(keyId) {
+        const key = `${STORE_TYPES.SIGNED_PREKEYS}:${keyId}`;
+        this.delValue(key);
+    }
+    // Sessions
+    async loadSession(encodedAddress) {
+        const key = `${STORE_TYPES.SESSIONS}:${encodedAddress}`;
+        return this.getValue(key);
+    }
+    async storeSession(encodedAddress, record) {
+        const key = `${STORE_TYPES.SESSIONS}:${encodedAddress}`;
+        this.putValue(key, record);
+    }
+    async removeSession(encodedAddress) {
+        const key = `${STORE_TYPES.SESSIONS}:${encodedAddress}`;
+        this.delValue(key);
+    }
+    // Meta
+    async getMeta() {
+        const value = this.getValue(STORE_TYPES.META);
+        if (!value)
+            return { nextPreKeyId: 1 };
+        return JSON.parse(value);
+    }
+    async setMeta(meta) {
+        const current = await this.getMeta();
+        const updated = { ...current, ...meta };
+        this.putValue(STORE_TYPES.META, JSON.stringify(updated));
+    }
+    listPreKeys() {
+        const rows = this.db
+            .prepare("SELECT key FROM signal_store WHERE store_type = ? AND key LIKE ?")
+            .all(this.storePrefix, `${STORE_TYPES.PREKEYS}:%`);
+        return rows.map((r) => {
+            const id = r.key.replace(`${STORE_TYPES.PREKEYS}:`, "");
+            return Number(id);
+        });
+    }
+    // Private helpers
+    getValue(key) {
+        const row = this.db
+            .prepare("SELECT value FROM signal_store WHERE store_type = ? AND key = ?")
+            .get(this.storePrefix, key);
+        return row?.value;
+    }
+    putValue(key, value) {
+        this.db
+            .prepare(`
       INSERT INTO signal_store (store_type, key, value, updated_at) 
       VALUES (?, ?, ?, strftime('%s', 'now'))
       ON CONFLICT(store_type, key) DO UPDATE SET 
         value = excluded.value, 
         updated_at = excluded.updated_at
-    `
-    ).run(this.storePrefix, key, value);
-  }
-  delValue(key) {
-    this.db.prepare(
-      "DELETE FROM signal_store WHERE store_type = ? AND key = ?"
-    ).run(this.storePrefix, key);
-  }
+    `)
+            .run(this.storePrefix, key, value);
+    }
+    delValue(key) {
+        this.db
+            .prepare("DELETE FROM signal_store WHERE store_type = ? AND key = ?")
+            .run(this.storePrefix, key);
+    }
 }
+exports.SignalProtocolStorage = SignalProtocolStorage;
+// ========== 消息存储 ==========
 class MessageStorage {
-  constructor(userAddress) {
-    __publicField(this, "userAddress");
-    this.userAddress = userAddress;
-  }
-  get db() {
-    return getDatabase(this.userAddress);
-  }
-  saveMessage(message) {
-    const receivedAt = message.direction === "received" && !message.receivedAt ? Date.now() : message.receivedAt;
-    this.db.prepare(
-      `
+    userAddress;
+    constructor(userAddress) {
+        this.userAddress = userAddress;
+    }
+    get db() {
+        return getDatabase(this.userAddress);
+    }
+    saveMessage(message) {
+        // 如果是接收消息且没有 receivedAt，设置当前时间
+        const receivedAt = message.direction === "received" && !message.receivedAt
+            ? Date.now()
+            : message.receivedAt;
+        this.db
+            .prepare(`
       INSERT INTO messages (
         message_id, from_address, to_address, plaintext_hash,
         plaintext, guard_address, passport_address, last_received_leaf_index, direction, status, msg_type,
@@ -453,279 +537,312 @@ class MessageStorage {
         ark_timestamp = COALESCE(excluded.ark_timestamp, ark_timestamp),
         zip_metadata = COALESCE(excluded.zip_metadata, zip_metadata),
         received_at = COALESCE(excluded.received_at, received_at)
-    `
-    ).run(
-      message.messageId,
-      message.fromAddress,
-      message.toAddress,
-      message.plaintextHash,
-      message.plaintext === void 0 ? null : message.plaintext,
-      message.guardAddress === void 0 ? null : message.guardAddress,
-      message.passportAddress === void 0 ? null : message.passportAddress,
-      message.lastReceivedLeafIndex === void 0 ? null : message.lastReceivedLeafIndex,
-      message.direction,
-      message.status,
-      message.msgType,
-      message.leafIndex === void 0 ? null : message.leafIndex,
-      message.prevRoot === void 0 ? null : message.prevRoot,
-      message.newRoot === void 0 ? null : message.newRoot,
-      message.serverSignature === void 0 ? null : message.serverSignature,
-      message.serverPublicKey === void 0 ? null : message.serverPublicKey,
-      message.serverTimestamp === void 0 ? null : message.serverTimestamp,
-      message.arkConfirmed?.recipient === void 0 ? null : message.arkConfirmed?.recipient,
-      message.arkConfirmed?.recipientPublicKey === void 0 ? null : message.arkConfirmed?.recipientPublicKey,
-      message.arkConfirmed?.signature === void 0 ? null : message.arkConfirmed?.signature,
-      message.arkConfirmed?.timestamp === void 0 ? null : message.arkConfirmed?.timestamp,
-      message.zipMetadata ? JSON.stringify(message.zipMetadata) : null,
-      message.createdAt,
-      receivedAt === void 0 ? null : receivedAt
-    );
-  }
-  getMessage(messageId) {
-    const row = this.db.prepare("SELECT * FROM messages WHERE message_id = ?").get(messageId);
-    if (!row) return void 0;
-    return this.rowToMessage(row);
-  }
-  getMessagesBySession(myAddress, peerAddress) {
-    const rows = this.db.prepare(
-      `
+    `)
+            .run(message.messageId, message.fromAddress, message.toAddress, message.plaintextHash, message.plaintext === undefined ? null : message.plaintext, message.guardAddress === undefined
+            ? null
+            : message.guardAddress, message.passportAddress === undefined
+            ? null
+            : message.passportAddress, message.lastReceivedLeafIndex === undefined
+            ? null
+            : message.lastReceivedLeafIndex, message.direction, message.status, message.msgType, message.leafIndex === undefined ? null : message.leafIndex, message.prevRoot === undefined ? null : message.prevRoot, message.newRoot === undefined ? null : message.newRoot, message.serverSignature === undefined
+            ? null
+            : message.serverSignature, message.serverPublicKey === undefined
+            ? null
+            : message.serverPublicKey, message.serverTimestamp === undefined
+            ? null
+            : message.serverTimestamp, message.arkConfirmed?.recipient === undefined
+            ? null
+            : message.arkConfirmed?.recipient, message.arkConfirmed?.recipientPublicKey === undefined
+            ? null
+            : message.arkConfirmed?.recipientPublicKey, message.arkConfirmed?.signature === undefined
+            ? null
+            : message.arkConfirmed?.signature, message.arkConfirmed?.timestamp === undefined
+            ? null
+            : message.arkConfirmed?.timestamp, message.zipMetadata
+            ? JSON.stringify(message.zipMetadata)
+            : null, message.createdAt, receivedAt === undefined ? null : receivedAt);
+    }
+    getMessage(messageId) {
+        const row = this.db
+            .prepare("SELECT * FROM messages WHERE message_id = ?")
+            .get(messageId);
+        if (!row)
+            return undefined;
+        return this.rowToMessage(row);
+    }
+    getMessagesBySession(myAddress, peerAddress) {
+        const rows = this.db
+            .prepare(`
         SELECT * FROM messages 
         WHERE (from_address = ? AND to_address = ?) 
            OR (from_address = ? AND to_address = ?)
         ORDER BY created_at ASC
-      `
-    ).all(myAddress, peerAddress, peerAddress, myAddress);
-    return rows.map((r) => this.rowToMessage(r));
-  }
-  getPendingMessages(myAddress) {
-    const rows = this.db.prepare(
-      `
+      `)
+            .all(myAddress, peerAddress, peerAddress, myAddress);
+        return rows.map((r) => this.rowToMessage(r));
+    }
+    getPendingMessages(myAddress) {
+        const rows = this.db
+            .prepare(`
         SELECT * FROM messages 
         WHERE (from_address = ? OR to_address = ?) AND status = ?
         ORDER BY created_at ASC
-      `
-    ).all(myAddress, myAddress, "pending");
-    return rows.map((r) => this.rowToMessage(r));
-  }
-  updateMessageStatus(messageId, status, updates) {
-    const fields = ["status = ?"];
-    const values = [status];
-    if (updates?.leafIndex !== void 0) {
-      fields.push("leaf_index = ?");
-      values.push(updates.leafIndex);
-    }
-    if (updates?.newRoot !== void 0) {
-      fields.push("new_root = ?");
-      values.push(updates.newRoot);
-    }
-    if (updates?.serverSignature !== void 0) {
-      fields.push("server_signature = ?");
-      values.push(updates.serverSignature);
-    }
-    if (updates?.serverTimestamp !== void 0) {
-      fields.push("server_timestamp = ?");
-      values.push(updates.serverTimestamp);
-    }
-    if (updates?.serverPublicKey !== void 0) {
-      fields.push("server_public_key = ?");
-      values.push(updates.serverPublicKey);
-    }
-    values.push(messageId);
-    this.db.prepare(
-      `
+      `)
+            .all(myAddress, myAddress, "pending");
+        return rows.map((r) => this.rowToMessage(r));
+    }
+    updateMessageStatus(messageId, status, updates) {
+        const fields = ["status = ?"];
+        const values = [status];
+        if (updates?.leafIndex !== undefined) {
+            fields.push("leaf_index = ?");
+            values.push(updates.leafIndex);
+        }
+        if (updates?.newRoot !== undefined) {
+            fields.push("new_root = ?");
+            values.push(updates.newRoot);
+        }
+        if (updates?.serverSignature !== undefined) {
+            fields.push("server_signature = ?");
+            values.push(updates.serverSignature);
+        }
+        if (updates?.serverTimestamp !== undefined) {
+            fields.push("server_timestamp = ?");
+            values.push(updates.serverTimestamp);
+        }
+        if (updates?.serverPublicKey !== undefined) {
+            fields.push("server_public_key = ?");
+            values.push(updates.serverPublicKey);
+        }
+        values.push(messageId);
+        this.db
+            .prepare(`
       UPDATE messages SET ${fields.join(", ")} WHERE message_id = ?
-    `
-    ).run(...values);
-  }
-  /**
-   * 获取用户的所有消息
-   *
-   * @param userAddress 用户地址
-   * @returns 所有相关消息（发送和接收的）
-   */
-  getAllMessages(userAddress) {
-    const rows = this.db.prepare(
-      `
+    `)
+            .run(...values);
+    }
+    /**
+     * 获取用户的所有消息
+     *
+     * @param userAddress 用户地址
+     * @returns 所有相关消息（发送和接收的）
+     */
+    getAllMessages(userAddress) {
+        const rows = this.db
+            .prepare(`
         SELECT * FROM messages 
         WHERE from_address = ? OR to_address = ?
         ORDER BY created_at DESC
-      `
-    ).all(userAddress, userAddress);
-    return rows.map((r) => this.rowToMessage(r));
-  }
-  /**
-   * 通过消息 ID 获取消息
-   *
-   * @param messageId 消息 ID
-   * @returns 消息对象或 undefined
-   */
-  getMessageById(messageId) {
-    const row = this.db.prepare(
-      `
+      `)
+            .all(userAddress, userAddress);
+        return rows.map((r) => this.rowToMessage(r));
+    }
+    /**
+     * 通过消息 ID 获取消息
+     *
+     * @param messageId 消息 ID
+     * @returns 消息对象或 undefined
+     */
+    getMessageById(messageId) {
+        const row = this.db
+            .prepare(`
         SELECT * FROM messages
         WHERE message_id = ?
-      `
-    ).get(messageId);
-    if (!row) return void 0;
-    return this.rowToMessage(row);
-  }
-  /**
-   * 更新消息 ID（用于 Guard 消息确认后更新 messageId）
-   *
-   * 由于 message_id 是主键，需要通过删除旧记录并插入新记录来实现
-   *
-   * @param oldMessageId 旧消息 ID（internal_id）
-   * @param newMessageId 新消息 ID（正式的 message_id）
-   * @param updates 可选的更新字段
-   * @returns 是否成功更新
-   */
-  updateMessageId(oldMessageId, newMessageId, updates) {
-    const row = this.db.prepare("SELECT * FROM messages WHERE message_id = ?").get(oldMessageId);
-    if (!row) return false;
-    const getUpdatedValue = (updateValue, originalValue) => {
-      if (updateValue !== void 0) return updateValue;
-      return originalValue === null ? void 0 : originalValue;
-    };
-    const message = {
-      messageId: newMessageId,
-      fromAddress: row.from_address,
-      toAddress: row.to_address,
-      plaintextHash: row.plaintext_hash,
-      plaintext: row.plaintext === null ? void 0 : row.plaintext,
-      guardAddress: row.guard_address === null ? void 0 : row.guard_address,
-      passportAddress: row.passport_address === null ? void 0 : row.passport_address,
-      direction: row.direction,
-      status: updates?.status !== void 0 ? updates.status : row.status,
-      msgType: row.msg_type,
-      leafIndex: getUpdatedValue(updates?.leafIndex, row.leaf_index),
-      prevRoot: getUpdatedValue(updates?.prevRoot, row.prev_root),
-      newRoot: getUpdatedValue(updates?.newRoot, row.new_root),
-      serverSignature: getUpdatedValue(
-        updates?.serverSignature,
-        row.server_signature
-      ),
-      serverPublicKey: getUpdatedValue(
-        updates?.serverPublicKey,
-        row.server_public_key
-      ),
-      serverTimestamp: getUpdatedValue(
-        updates?.serverTimestamp,
-        row.server_timestamp
-      ),
-      arkConfirmed: row.ark_recipient ? {
-        recipient: row.ark_recipient,
-        recipientPublicKey: row.ark_recipient_pubkey,
-        signature: row.ark_signature,
-        timestamp: row.ark_timestamp
-      } : void 0,
-      zipMetadata: row.zip_metadata ? JSON.parse(row.zip_metadata) : void 0,
-      createdAt: row.created_at,
-      receivedAt: row.received_at === null ? void 0 : row.received_at
-    };
-    const transaction = this.db.transaction(() => {
-      this.db.prepare("DELETE FROM messages WHERE message_id = ?").run(oldMessageId);
-      this.saveMessage(message);
-    });
-    try {
-      transaction();
-      return true;
-    } catch (error) {
-      return false;
-    }
-  }
-  rowToMessage(row) {
-    if (row.to_address === null) {
-      throw new Error(`Message ${row.message_id} has null toAddress`);
-    }
-    return {
-      messageId: row.message_id,
-      fromAddress: row.from_address,
-      toAddress: row.to_address,
-      plaintextHash: row.plaintext_hash,
-      plaintext: row.plaintext === null ? void 0 : row.plaintext,
-      guardAddress: row.guard_address === null ? void 0 : row.guard_address,
-      passportAddress: row.passport_address === null ? void 0 : row.passport_address,
-      lastReceivedLeafIndex: row.last_received_leaf_index === null ? void 0 : row.last_received_leaf_index,
-      direction: row.direction,
-      status: row.status,
-      msgType: row.msg_type,
-      leafIndex: row.leaf_index === null ? void 0 : row.leaf_index,
-      prevRoot: row.prev_root === null ? void 0 : row.prev_root,
-      newRoot: row.new_root === null ? void 0 : row.new_root,
-      serverSignature: row.server_signature === null ? void 0 : row.server_signature,
-      serverPublicKey: row.server_public_key === null ? void 0 : row.server_public_key,
-      serverTimestamp: row.server_timestamp === null ? void 0 : row.server_timestamp,
-      arkConfirmed: row.ark_recipient ? {
-        recipient: row.ark_recipient,
-        recipientPublicKey: row.ark_recipient_pubkey,
-        signature: row.ark_signature,
-        timestamp: row.ark_timestamp
-      } : void 0,
-      zipMetadata: row.zip_metadata ? JSON.parse(row.zip_metadata) : void 0,
-      createdAt: row.created_at,
-      receivedAt: row.received_at === null ? void 0 : row.received_at
-    };
-  }
-  /**
-   * 获取最后收到的消息序号（用于消息同步确认）
-   *
-   * 查询指定会话中，用户作为接收者时，发送者最后收到的消息序号
-   * 用于在发送新消息时，告诉对方我已经收到了哪些消息
-   *
-   * @param myAddress 当前用户地址
-   * @param peerAddress 对方地址
-   * @returns 最后收到的消息序号（leafIndex），如果没有则返回 -1
-   */
-  getLastReceivedLeafIndex(myAddress, peerAddress) {
-    const row = this.db.prepare(
-      `
+      `)
+            .get(messageId);
+        if (!row)
+            return undefined;
+        return this.rowToMessage(row);
+    }
+    /**
+     * 更新消息 ID（用于 Guard 消息确认后更新 messageId）
+     *
+     * 由于 message_id 是主键，需要通过删除旧记录并插入新记录来实现
+     *
+     * @param oldMessageId 旧消息 ID（internal_id）
+     * @param newMessageId 新消息 ID（正式的 message_id）
+     * @param updates 可选的更新字段
+     * @returns 是否成功更新
+     */
+    updateMessageId(oldMessageId, newMessageId, updates) {
+        const row = this.db
+            .prepare("SELECT * FROM messages WHERE message_id = ?")
+            .get(oldMessageId);
+        if (!row)
+            return false;
+        // 辅助函数：获取更新值，优先使用 updates 中的值，但不使用 ?? 避免改写数据
+        // 如果 updates 中有定义该字段（包括 null），使用 updates 的值；否则使用原值
+        const getUpdatedValue = (updateValue, originalValue) => {
+            if (updateValue !== undefined)
+                return updateValue;
+            return originalValue === null ? undefined : originalValue;
+        };
+        // 构建更新后的消息对象
+        const message = {
+            messageId: newMessageId,
+            fromAddress: row.from_address,
+            toAddress: row.to_address,
+            plaintextHash: row.plaintext_hash,
+            plaintext: row.plaintext === null ? undefined : row.plaintext,
+            guardAddress: row.guard_address === null ? undefined : row.guard_address,
+            passportAddress: row.passport_address === null
+                ? undefined
+                : row.passport_address,
+            direction: row.direction,
+            status: updates?.status !== undefined ? updates.status : row.status,
+            msgType: row.msg_type,
+            leafIndex: getUpdatedValue(updates?.leafIndex, row.leaf_index),
+            prevRoot: getUpdatedValue(updates?.prevRoot, row.prev_root),
+            newRoot: getUpdatedValue(updates?.newRoot, row.new_root),
+            serverSignature: getUpdatedValue(updates?.serverSignature, row.server_signature),
+            serverPublicKey: getUpdatedValue(updates?.serverPublicKey, row.server_public_key),
+            serverTimestamp: getUpdatedValue(updates?.serverTimestamp, row.server_timestamp),
+            arkConfirmed: row.ark_recipient
+                ? {
+                    recipient: row.ark_recipient,
+                    recipientPublicKey: row.ark_recipient_pubkey,
+                    signature: row.ark_signature,
+                    timestamp: row.ark_timestamp,
+                }
+                : undefined,
+            zipMetadata: row.zip_metadata
+                ? JSON.parse(row.zip_metadata)
+                : undefined,
+            createdAt: row.created_at,
+            receivedAt: row.received_at === null ? undefined : row.received_at,
+        };
+        // 使用事务确保原子性
+        const transaction = this.db.transaction(() => {
+            // 删除旧记录
+            this.db
+                .prepare("DELETE FROM messages WHERE message_id = ?")
+                .run(oldMessageId);
+            // 插入新记录
+            this.saveMessage(message);
+        });
+        try {
+            transaction();
+            return true;
+        }
+        catch (error) {
+            //console.error("Failed to update message ID:", error);
+            return false;
+        }
+    }
+    rowToMessage(row) {
+        if (row.to_address === null) {
+            throw new Error(`Message ${row.message_id} has null toAddress`);
+        }
+        return {
+            messageId: row.message_id,
+            fromAddress: row.from_address,
+            toAddress: row.to_address,
+            plaintextHash: row.plaintext_hash,
+            plaintext: row.plaintext === null ? undefined : row.plaintext,
+            guardAddress: row.guard_address === null ? undefined : row.guard_address,
+            passportAddress: row.passport_address === null
+                ? undefined
+                : row.passport_address,
+            lastReceivedLeafIndex: row.last_received_leaf_index === null
+                ? undefined
+                : row.last_received_leaf_index,
+            direction: row.direction,
+            status: row.status,
+            msgType: row.msg_type,
+            leafIndex: row.leaf_index === null ? undefined : row.leaf_index,
+            prevRoot: row.prev_root === null ? undefined : row.prev_root,
+            newRoot: row.new_root === null ? undefined : row.new_root,
+            serverSignature: row.server_signature === null
+                ? undefined
+                : row.server_signature,
+            serverPublicKey: row.server_public_key === null
+                ? undefined
+                : row.server_public_key,
+            serverTimestamp: row.server_timestamp === null
+                ? undefined
+                : row.server_timestamp,
+            arkConfirmed: row.ark_recipient
+                ? {
+                    recipient: row.ark_recipient,
+                    recipientPublicKey: row.ark_recipient_pubkey,
+                    signature: row.ark_signature,
+                    timestamp: row.ark_timestamp,
+                }
+                : undefined,
+            zipMetadata: row.zip_metadata
+                ? JSON.parse(row.zip_metadata)
+                : undefined,
+            createdAt: row.created_at,
+            receivedAt: row.received_at === null ? undefined : row.received_at,
+        };
+    }
+    /**
+     * 获取最后收到的消息序号（用于消息同步确认）
+     *
+     * 查询指定会话中，用户作为接收者时，发送者最后收到的消息序号
+     * 用于在发送新消息时，告诉对方我已经收到了哪些消息
+     *
+     * @param myAddress 当前用户地址
+     * @param peerAddress 对方地址
+     * @returns 最后收到的消息序号（leafIndex），如果没有则返回 -1
+     */
+    getLastReceivedLeafIndex(myAddress, peerAddress) {
+        // 查询从 peerAddress 发送给 myAddress 的消息中，
+        // 最大的 last_received_leaf_index
+        const row = this.db
+            .prepare(`
 				SELECT MAX(last_received_leaf_index) as max_index
 				FROM messages
 				WHERE from_address = ? AND to_address = ?
 					AND last_received_leaf_index IS NOT NULL
-			`
-    ).get(peerAddress, myAddress);
-    return row?.max_index ?? -1;
-  }
+			`)
+            .get(peerAddress, myAddress);
+        return row?.max_index ?? -1;
+    }
 }
+exports.MessageStorage = MessageStorage;
+// ========== 会话状态存储 ==========
 class SessionStateStorage {
-  constructor(userAddress) {
-    __publicField(this, "userAddress");
-    this.userAddress = userAddress;
-  }
-  get db() {
-    return getDatabase(this.userAddress);
-  }
-  getSessionId(myAddress, peerAddress) {
-    const addresses = [myAddress, peerAddress].sort();
-    return `${addresses[0]}:${addresses[1]}`;
-  }
-  getSessionState(myAddress, peerAddress) {
-    const sessionId = this.getSessionId(myAddress, peerAddress);
-    const row = this.db.prepare("SELECT * FROM session_states WHERE session_id = ?").get(sessionId);
-    if (!row) return void 0;
-    return {
-      sessionId: row.session_id,
-      myAddress: row.my_address,
-      peerAddress: row.peer_address,
-      currentRoot: row.current_root,
-      prevRoot: row.prev_root,
-      messageCount: row.message_count,
-      lastLeafIndex: row.last_leaf_index,
-      lastConfirmedHash: row.last_confirmed_hash === null ? "" : row.last_confirmed_hash,
-      lastSyncAt: row.last_sync_at === null ? 0 : row.last_sync_at,
-      createdAt: row.created_at,
-      updatedAt: row.updated_at
-    };
-  }
-  updateSessionState(myAddress, peerAddress, updates) {
-    const sessionId = this.getSessionId(myAddress, peerAddress);
-    const existing = this.getSessionState(myAddress, peerAddress);
-    const now = Date.now();
-    if (existing) {
-      const updated = { ...existing, ...updates, updatedAt: now };
-      this.db.prepare(
-        `
+    userAddress;
+    constructor(userAddress) {
+        this.userAddress = userAddress;
+    }
+    get db() {
+        return getDatabase(this.userAddress);
+    }
+    getSessionId(myAddress, peerAddress) {
+        const addresses = [myAddress, peerAddress].sort();
+        return `${addresses[0]}:${addresses[1]}`;
+    }
+    getSessionState(myAddress, peerAddress) {
+        const sessionId = this.getSessionId(myAddress, peerAddress);
+        const row = this.db
+            .prepare("SELECT * FROM session_states WHERE session_id = ?")
+            .get(sessionId);
+        if (!row)
+            return undefined;
+        return {
+            sessionId: row.session_id,
+            myAddress: row.my_address,
+            peerAddress: row.peer_address,
+            currentRoot: row.current_root,
+            prevRoot: row.prev_root,
+            messageCount: row.message_count,
+            lastLeafIndex: row.last_leaf_index,
+            lastConfirmedHash: row.last_confirmed_hash === null ? "" : row.last_confirmed_hash,
+            lastSyncAt: row.last_sync_at === null ? 0 : row.last_sync_at,
+            createdAt: row.created_at,
+            updatedAt: row.updated_at,
+        };
+    }
+    updateSessionState(myAddress, peerAddress, updates) {
+        const sessionId = this.getSessionId(myAddress, peerAddress);
+        const existing = this.getSessionState(myAddress, peerAddress);
+        const now = Date.now();
+        if (existing) {
+            const updated = { ...existing, ...updates, updatedAt: now };
+            this.db
+                .prepare(`
         UPDATE session_states SET
           current_root = ?,
           prev_root = ?,
@@ -735,64 +852,48 @@ class SessionStateStorage {
           last_sync_at = ?,
           updated_at = ?
         WHERE session_id = ?
-      `
-      ).run(
-        updated.currentRoot,
-        updated.prevRoot,
-        updated.messageCount,
-        updated.lastLeafIndex,
-        updated.lastConfirmedHash,
-        updated.lastSyncAt,
-        updated.updatedAt,
-        sessionId
-      );
-    } else {
-      const initialRoot = "0".repeat(64);
-      this.db.prepare(
-        `
+      `)
+                .run(updated.currentRoot, updated.prevRoot, updated.messageCount, updated.lastLeafIndex, updated.lastConfirmedHash, updated.lastSyncAt, updated.updatedAt, sessionId);
+        }
+        else {
+            const initialRoot = "0".repeat(64);
+            this.db
+                .prepare(`
         INSERT INTO session_states (
           session_id, my_address, peer_address, current_root, prev_root,
           message_count, last_leaf_index, last_confirmed_hash, last_sync_at,
           created_at, updated_at
         ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-      `
-      ).run(
-        sessionId,
-        myAddress,
-        peerAddress,
-        updates.currentRoot ?? initialRoot,
-        updates.prevRoot ?? initialRoot,
-        updates.messageCount ?? 0,
-        updates.lastLeafIndex ?? 0,
-        updates.lastConfirmedHash ?? "",
-        updates.lastSyncAt ?? 0,
-        now,
-        now
-      );
+      `)
+                .run(sessionId, myAddress, peerAddress, updates.currentRoot ?? initialRoot, updates.prevRoot ?? initialRoot, updates.messageCount ?? 0, updates.lastLeafIndex ?? 0, updates.lastConfirmedHash ?? "", updates.lastSyncAt ?? 0, now, now);
+        }
+    }
+    getCurrentRoot(myAddress, peerAddress) {
+        const state = this.getSessionState(myAddress, peerAddress);
+        return state?.currentRoot ?? "0".repeat(64);
     }
-  }
-  getCurrentRoot(myAddress, peerAddress) {
-    const state = this.getSessionState(myAddress, peerAddress);
-    return state?.currentRoot ?? "0".repeat(64);
-  }
 }
+exports.SessionStateStorage = SessionStateStorage;
+/**
+ * MessengerManager 持久化存储类
+ *
+ * 使用 SQLite 存储 MessengerManager 的配置和状态
+ */
 class MessengerPersistStorage {
-  constructor() {
-    __publicField(this, "initialized", false);
-  }
-  get db() {
-    const db = getGlobalDatabase();
-    if (!this.initialized) {
-      this.initTable(db);
-      this.initialized = true;
+    initialized = false;
+    get db() {
+        const db = getGlobalDatabase();
+        if (!this.initialized) {
+            this.initTable(db);
+            this.initialized = true;
+        }
+        return db;
     }
-    return db;
-  }
-  /**
-   * 初始化存储表
-   */
-  initTable(db) {
-    db.exec(`
+    /**
+     * 初始化存储表
+     */
+    initTable(db) {
+        db.exec(`
 			CREATE TABLE IF NOT EXISTS messenger_manager_state (
 				instance_id TEXT PRIMARY KEY,
 				state_data TEXT NOT NULL,
@@ -800,7 +901,7 @@ class MessengerPersistStorage {
 			);
 			CREATE INDEX IF NOT EXISTS idx_messenger_state_updated ON messenger_manager_state(updated_at);
 		`);
-    db.exec(`
+        db.exec(`
 			CREATE TABLE IF NOT EXISTS sse_connection_locks (
 				account_address TEXT PRIMARY KEY,
 				process_id TEXT NOT NULL,
@@ -810,91 +911,94 @@ class MessengerPersistStorage {
 			);
 			CREATE INDEX IF NOT EXISTS idx_sse_lock_heartbeat ON sse_connection_locks(last_heartbeat);
 		`);
-  }
-  /**
-   * 保存 Messenger 配置和账号信息
-   *
-   * @param instanceId 实例标识（用于区分不同的 MessengerManager 实例）
-   * @param state 要保存的状态
-   */
-  saveState(instanceId, state) {
-    const data = JSON.stringify(state);
-    const now = Date.now();
-    this.db.prepare(
-      `
+    }
+    /**
+     * 保存 Messenger 配置和账号信息
+     *
+     * @param instanceId 实例标识（用于区分不同的 MessengerManager 实例）
+     * @param state 要保存的状态
+     */
+    saveState(instanceId, state) {
+        const data = JSON.stringify(state);
+        const now = Date.now();
+        this.db
+            .prepare(`
 				INSERT INTO messenger_manager_state (instance_id, state_data, updated_at)
 				VALUES (?, ?, ?)
 				ON CONFLICT(instance_id) DO UPDATE SET
 					state_data = excluded.state_data,
 					updated_at = excluded.updated_at
-			`
-    ).run(instanceId, data, now);
-  }
-  /**
-   * 加载 Messenger 配置和账号信息
-   *
-   * @param instanceId 实例标识
-   * @returns 保存的状态，如果不存在返回 null
-   */
-  loadState(instanceId) {
-    const row = this.db.prepare(
-      "SELECT state_data FROM messenger_manager_state WHERE instance_id = ?"
-    ).get(instanceId);
-    if (!row?.state_data) return null;
-    try {
-      return JSON.parse(row.state_data);
-    } catch (error) {
-      return null;
-    }
-  }
-  /**
-   * 删除保存的状态
-   *
-   * @param instanceId 实例标识
-   */
-  deleteState(instanceId) {
-    this.db.prepare(
-      "DELETE FROM messenger_manager_state WHERE instance_id = ?"
-    ).run(instanceId);
-  }
-  /**
-   * 获取所有实例ID列表
-   */
-  getAllInstanceIds() {
-    const rows = this.db.prepare(
-      "SELECT instance_id FROM messenger_manager_state ORDER BY updated_at DESC"
-    ).all();
-    return rows.map((r) => r.instance_id);
-  }
-  get processId() {
-    return `pid_${process.pid}_${Date.now()}`;
-  }
-  /**
-   * 尝试获取指定账号的 SSE 连接锁
-   * @param accountAddress 账号地址
-   * @param instanceId 当前实例ID
-   * @returns 是否成功获取锁
-   */
-  tryAcquireSSELock(accountAddress, instanceId) {
-    const now = Date.now();
-    const heartbeatTimeout = 15e3;
-    const transaction = this.db.transaction(() => {
-      const row = this.db.prepare(
-        "SELECT * FROM sse_connection_locks WHERE account_address = ?"
-      ).get(accountAddress.toLowerCase());
-      if (row) {
-        if (row.instance_id === instanceId) {
-          this.db.prepare(
-            "UPDATE sse_connection_locks SET last_heartbeat = ? WHERE account_address = ?"
-          ).run(now, accountAddress.toLowerCase());
-          return true;
+			`)
+            .run(instanceId, data, now);
+    }
+    /**
+     * 加载 Messenger 配置和账号信息
+     *
+     * @param instanceId 实例标识
+     * @returns 保存的状态，如果不存在返回 null
+     */
+    loadState(instanceId) {
+        const row = this.db
+            .prepare("SELECT state_data FROM messenger_manager_state WHERE instance_id = ?")
+            .get(instanceId);
+        if (!row?.state_data)
+            return null;
+        try {
+            return JSON.parse(row.state_data);
         }
-        if (now - row.last_heartbeat < heartbeatTimeout) {
-          return false;
+        catch (error) {
+            //console.error("Failed to parse persisted state:", error);
+            return null;
         }
-      }
-      this.db.prepare(
-        `
+    }
+    /**
+     * 删除保存的状态
+     *
+     * @param instanceId 实例标识
+     */
+    deleteState(instanceId) {
+        this.db
+            .prepare("DELETE FROM messenger_manager_state WHERE instance_id = ?")
+            .run(instanceId);
+    }
+    /**
+     * 获取所有实例ID列表
+     */
+    getAllInstanceIds() {
+        const rows = this.db
+            .prepare("SELECT instance_id FROM messenger_manager_state ORDER BY updated_at DESC")
+            .all();
+        return rows.map((r) => r.instance_id);
+    }
+    get processId() {
+        return `pid_${process.pid}_${Date.now()}`;
+    }
+    /**
+     * 尝试获取指定账号的 SSE 连接锁
+     * @param accountAddress 账号地址
+     * @param instanceId 当前实例ID
+     * @returns 是否成功获取锁
+     */
+    tryAcquireSSELock(accountAddress, instanceId) {
+        const now = Date.now();
+        const heartbeatTimeout = 15000;
+        const transaction = this.db.transaction(() => {
+            const row = this.db
+                .prepare("SELECT * FROM sse_connection_locks WHERE account_address = ?")
+                .get(accountAddress.toLowerCase());
+            if (row) {
+                if (row.instance_id === instanceId) {
+                    this.db
+                        .prepare("UPDATE sse_connection_locks SET last_heartbeat = ? WHERE account_address = ?")
+                        .run(now, accountAddress.toLowerCase());
+                    return true;
+                }
+                if (now - row.last_heartbeat < heartbeatTimeout) {
+                    return false;
+                }
+            }
+            this.db
+                .prepare(`
 					INSERT INTO sse_connection_locks (
 						account_address, process_id, instance_id, last_heartbeat, created_at
 					) VALUES (?, ?, ?, ?, ?)
@@ -902,63 +1006,63 @@ class MessengerPersistStorage {
 						process_id = excluded.process_id,
 						instance_id = excluded.instance_id,
 						last_heartbeat = excluded.last_heartbeat
-				`
-      ).run(
-        accountAddress.toLowerCase(),
-        this.processId,
-        instanceId,
-        now,
-        now
-      );
-      return true;
-    });
-    try {
-      return transaction();
-    } catch (error) {
-      console.warn("Failed to acquire SSE lock:", error);
-      return false;
-    }
-  }
-  /**
-   * 更新 SSE 连接锁的心跳
-   * @param accountAddress 账号地址
-   * @param instanceId 当前实例ID
-   */
-  updateSSELockHeartbeat(accountAddress, instanceId) {
-    const now = Date.now();
-    try {
-      this.db.prepare(
-        "UPDATE sse_connection_locks SET last_heartbeat = ? WHERE account_address = ? AND instance_id = ?"
-      ).run(now, accountAddress.toLowerCase(), instanceId);
-    } catch (error) {
-      console.warn("Failed to update SSE lock heartbeat:", error);
-    }
-  }
-  /**
-   * 释放指定账号的 SSE 连接锁
-   * @param accountAddress 账号地址
-   * @param instanceId 当前实例ID
-   */
-  releaseSSELock(accountAddress, instanceId) {
-    try {
-      this.db.prepare(
-        "DELETE FROM sse_connection_locks WHERE account_address = ? AND instance_id = ?"
-      ).run(accountAddress.toLowerCase(), instanceId);
-    } catch (error) {
-      console.warn("Failed to release SSE lock:", error);
-    }
-  }
-  /**
-   * 检查是否持有指定账号的 SSE 连接锁
-   * @param accountAddress 账号地址
-   * @param instanceId 当前实例ID
-   * @returns 是否持有锁
-   */
-  holdsSSELock(accountAddress, instanceId) {
-    const row = this.db.prepare(
-      "SELECT * FROM sse_connection_locks WHERE account_address = ? AND instance_id = ?"
-    ).get(accountAddress.toLowerCase(), instanceId);
-    return !!row;
-  }
+				`)
+                .run(accountAddress.toLowerCase(), this.processId, instanceId, now, now);
+            return true;
+        });
+        try {
+            return transaction();
+        }
+        catch (error) {
+            console.warn("Failed to acquire SSE lock:", error);
+            return false;
+        }
+    }
+    /**
+     * 更新 SSE 连接锁的心跳
+     * @param accountAddress 账号地址
+     * @param instanceId 当前实例ID
+     */
+    updateSSELockHeartbeat(accountAddress, instanceId) {
+        const now = Date.now();
+        try {
+            this.db
+                .prepare("UPDATE sse_connection_locks SET last_heartbeat = ? WHERE account_address = ? AND instance_id = ?")
+                .run(now, accountAddress.toLowerCase(), instanceId);
+        }
+        catch (error) {
+            console.warn("Failed to update SSE lock heartbeat:", error);
+        }
+    }
+    /**
+     * 释放指定账号的 SSE 连接锁
+     * @param accountAddress 账号地址
+     * @param instanceId 当前实例ID
+     */
+    releaseSSELock(accountAddress, instanceId) {
+        try {
+            this.db
+                .prepare("DELETE FROM sse_connection_locks WHERE account_address = ? AND instance_id = ?")
+                .run(accountAddress.toLowerCase(), instanceId);
+        }
+        catch (error) {
+            console.warn("Failed to release SSE lock:", error);
+        }
+    }
+    /**
+     * 检查是否持有指定账号的 SSE 连接锁
+     * @param accountAddress 账号地址
+     * @param instanceId 当前实例ID
+     * @returns 是否持有锁
+     */
+    holdsSSELock(accountAddress, instanceId) {
+        const row = this.db
+            .prepare("SELECT * FROM sse_connection_locks WHERE account_address = ? AND instance_id = ?")
+            .get(accountAddress.toLowerCase(), instanceId);
+        return !!row;
+    }
 }
-const defaultPersistStorage = new MessengerPersistStorage();
+exports.MessengerPersistStorage = MessengerPersistStorage;
+// 默认持久化存储实例
+exports.defaultPersistStorage = new MessengerPersistStorage();
+//# sourceMappingURL=storage.js.map