@memtensor/memos-local-openclaw-plugin 1.0.4-beta.10 → 1.0.4-beta.12

package/dist/storage/sqlite.js

@@ -148,6 +148,8 @@ class SqliteStore {
         this.migrateHubTables();
         this.migrateHubFtsToTrigram();
         this.migrateLocalSharedTasksOwner();
+        this.migrateHubUserIdentityFields();
+        this.migrateClientHubConnectionIdentityFields();
         this.log.debug("Database schema initialized");
     }
     migrateChunksIndexesForRecall() {
@@ -163,6 +165,51 @@ class SqliteStore {
         }
         catch { /* table may not exist yet */ }
     }
+    migrateHubUserIdentityFields() {
+        try {
+            const cols = this.db.prepare("PRAGMA table_info(hub_users)").all();
+            if (cols.length === 0)
+                return;
+            if (!cols.some(c => c.name === "identity_key")) {
+                this.db.exec("ALTER TABLE hub_users ADD COLUMN identity_key TEXT NOT NULL DEFAULT ''");
+                this.db.exec("CREATE INDEX IF NOT EXISTS idx_hub_users_identity_key ON hub_users(identity_key)");
+                this.log.info("Migrated: added identity_key to hub_users");
+            }
+            if (!cols.some(c => c.name === "left_at")) {
+                this.db.exec("ALTER TABLE hub_users ADD COLUMN left_at INTEGER");
+                this.log.info("Migrated: added left_at to hub_users");
+            }
+            if (!cols.some(c => c.name === "removed_at")) {
+                this.db.exec("ALTER TABLE hub_users ADD COLUMN removed_at INTEGER");
+                this.log.info("Migrated: added removed_at to hub_users");
+            }
+            if (!cols.some(c => c.name === "rejected_at")) {
+                this.db.exec("ALTER TABLE hub_users ADD COLUMN rejected_at INTEGER");
+                this.log.info("Migrated: added rejected_at to hub_users");
+            }
+            if (!cols.some(c => c.name === "rejoin_requested_at")) {
+                this.db.exec("ALTER TABLE hub_users ADD COLUMN rejoin_requested_at INTEGER");
+                this.log.info("Migrated: added rejoin_requested_at to hub_users");
+            }
+        }
+        catch { /* table may not exist yet */ }
+    }
+    migrateClientHubConnectionIdentityFields() {
+        try {
+            const cols = this.db.prepare("PRAGMA table_info(client_hub_connection)").all();
+            if (cols.length === 0)
+                return;
+            if (!cols.some(c => c.name === "identity_key")) {
+                this.db.exec("ALTER TABLE client_hub_connection ADD COLUMN identity_key TEXT NOT NULL DEFAULT ''");
+                this.log.info("Migrated: added identity_key to client_hub_connection");
+            }
+            if (!cols.some(c => c.name === "last_known_status")) {
+                this.db.exec("ALTER TABLE client_hub_connection ADD COLUMN last_known_status TEXT NOT NULL DEFAULT ''");
+                this.log.info("Migrated: added last_known_status to client_hub_connection");
+            }
+        }
+        catch { /* table may not exist yet */ }
+    }
     migrateOwnerFields() {
         const chunkCols = this.db.prepare("PRAGMA table_info(chunks)").all();
         if (!chunkCols.some((c) => c.name === "owner")) {
@@ -726,6 +773,20 @@ class SqliteStore {
       CREATE INDEX IF NOT EXISTS idx_hub_users_status ON hub_users(status);
       CREATE INDEX IF NOT EXISTS idx_hub_users_role ON hub_users(role);
 
+      CREATE TABLE IF NOT EXISTS hub_groups (
+        id          TEXT PRIMARY KEY,
+        name        TEXT NOT NULL,
+        description TEXT NOT NULL DEFAULT '',
+        created_at  INTEGER NOT NULL
+      );
+
+      CREATE TABLE IF NOT EXISTS hub_group_members (
+        group_id   TEXT NOT NULL REFERENCES hub_groups(id) ON DELETE CASCADE,
+        user_id    TEXT NOT NULL REFERENCES hub_users(id) ON DELETE CASCADE,
+        joined_at  INTEGER NOT NULL,
+        PRIMARY KEY (group_id, user_id)
+      );
+
       CREATE TABLE IF NOT EXISTS hub_tasks (
         id             TEXT PRIMARY KEY,
         source_task_id TEXT NOT NULL,
@@ -1580,16 +1641,18 @@ class SqliteStore {
     // ─── Hub / Client connection ───
     setClientHubConnection(conn) {
         this.db.prepare(`
-      INSERT INTO client_hub_connection (id, hub_url, user_id, username, user_token, role, connected_at)
-      VALUES (1, ?, ?, ?, ?, ?, ?)
+      INSERT INTO client_hub_connection (id, hub_url, user_id, username, user_token, role, connected_at, identity_key, last_known_status)
+      VALUES (1, ?, ?, ?, ?, ?, ?, ?, ?)
       ON CONFLICT(id) DO UPDATE SET
         hub_url = excluded.hub_url,
         user_id = excluded.user_id,
         username = excluded.username,
         user_token = excluded.user_token,
         role = excluded.role,
-        connected_at = excluded.connected_at
-    `).run(conn.hubUrl, conn.userId, conn.username, conn.userToken, conn.role, conn.connectedAt);
+        connected_at = excluded.connected_at,
+        identity_key = excluded.identity_key,
+        last_known_status = excluded.last_known_status
+    `).run(conn.hubUrl, conn.userId, conn.username, conn.userToken, conn.role, conn.connectedAt, conn.identityKey ?? "", conn.lastKnownStatus ?? "");
     }
     getClientHubConnection() {
         const row = this.db.prepare('SELECT * FROM client_hub_connection WHERE id = 1').get();
@@ -1682,8 +1745,8 @@ class SqliteStore {
     // ─── Hub Users / Groups ───
     upsertHubUser(user) {
         this.db.prepare(`
-      INSERT INTO hub_users (id, username, device_name, role, status, token_hash, created_at, approved_at)
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+      INSERT INTO hub_users (id, username, device_name, role, status, token_hash, created_at, approved_at, identity_key, left_at, removed_at, rejected_at, rejoin_requested_at)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
       ON CONFLICT(id) DO UPDATE SET
         username = excluded.username,
         device_name = excluded.device_name,
@@ -1691,20 +1754,31 @@ class SqliteStore {
         status = excluded.status,
         token_hash = excluded.token_hash,
         created_at = excluded.created_at,
-        approved_at = excluded.approved_at
-    `).run(user.id, user.username, user.deviceName ?? "", user.role, user.status, user.tokenHash, user.createdAt, user.approvedAt);
+        approved_at = excluded.approved_at,
+        identity_key = excluded.identity_key,
+        left_at = excluded.left_at,
+        removed_at = excluded.removed_at,
+        rejected_at = excluded.rejected_at,
+        rejoin_requested_at = excluded.rejoin_requested_at
+    `).run(user.id, user.username, user.deviceName ?? "", user.role, user.status, user.tokenHash, user.createdAt, user.approvedAt, user.identityKey ?? "", user.leftAt ?? null, user.removedAt ?? null, user.rejectedAt ?? null, user.rejoinRequestedAt ?? null);
     }
     getHubUser(userId) {
         const row = this.db.prepare('SELECT * FROM hub_users WHERE id = ?').get(userId);
         if (!row)
             return null;
-        return rowToHubUser(row);
+        const user = rowToHubUser(row);
+        user.groups = this.getGroupsForHubUser(userId);
+        return user;
     }
     listHubUsers(status) {
         const rows = status
             ? this.db.prepare('SELECT * FROM hub_users WHERE status = ? ORDER BY created_at').all(status)
             : this.db.prepare('SELECT * FROM hub_users ORDER BY created_at').all();
-        return rows.map(rowToHubUser);
+        return rows.map(r => {
+            const user = rowToHubUser(r);
+            user.groups = this.getGroupsForHubUser(r.id);
+            return user;
+        });
     }
     deleteHubUser(userId, cleanResources = false) {
         if (cleanResources) {
@@ -1714,12 +1788,46 @@ class SqliteStore {
             const result = this.db.prepare('DELETE FROM hub_users WHERE id = ?').run(userId);
             return result.changes > 0;
         }
-        const result = this.db.prepare("UPDATE hub_users SET status = 'removed', token_hash = '' WHERE id = ?").run(userId);
+        const result = this.db.prepare("UPDATE hub_users SET status = 'removed', token_hash = '', removed_at = ? WHERE id = ?").run(Date.now(), userId);
+        return result.changes > 0;
+    }
+    findHubUserByIdentityKey(identityKey) {
+        if (!identityKey)
+            return null;
+        const row = this.db.prepare('SELECT * FROM hub_users WHERE identity_key = ?').get(identityKey);
+        return row ? rowToHubUser(row) : null;
+    }
+    markHubUserLeft(userId) {
+        const result = this.db.prepare("UPDATE hub_users SET status = 'left', token_hash = '', left_at = ? WHERE id = ?").run(Date.now(), userId);
         return result.changes > 0;
     }
     updateHubUserActivity(userId, ip, timestamp) {
         this.db.prepare('UPDATE hub_users SET last_ip = ?, last_active_at = ? WHERE id = ?').run(ip, timestamp ?? Date.now(), userId);
     }
+    // ─── Hub Groups ───
+    upsertHubGroup(group) {
+        this.db.prepare(`
+      INSERT INTO hub_groups (id, name, description, created_at)
+      VALUES (?, ?, ?, ?)
+      ON CONFLICT(id) DO UPDATE SET name = excluded.name, description = excluded.description
+    `).run(group.id, group.name, group.description ?? "", group.createdAt);
+    }
+    addHubGroupMember(groupId, userId, joinedAt) {
+        this.db.prepare(`
+      INSERT OR IGNORE INTO hub_group_members (group_id, user_id, joined_at)
+      VALUES (?, ?, ?)
+    `).run(groupId, userId, joinedAt);
+    }
+    removeHubGroupMember(groupId, userId) {
+        this.db.prepare('DELETE FROM hub_group_members WHERE group_id = ? AND user_id = ?').run(groupId, userId);
+    }
+    getGroupsForHubUser(userId) {
+        return this.db.prepare(`
+      SELECT g.id, g.name, g.description FROM hub_groups g
+      JOIN hub_group_members m ON m.group_id = g.id
+      WHERE m.user_id = ?
+    `).all(userId);
+    }
     getHubUserContributions() {
         const result = {};
         const memRows = this.db.prepare('SELECT source_user_id, COUNT(*) as cnt FROM hub_memories GROUP BY source_user_id').all();
@@ -1836,20 +1944,36 @@ class SqliteStore {
             out.push(row.vector.readFloatLE(i * 4));
         return out;
     }
+    getVisibleHubSkillEmbeddings() {
+        const rows = this.db.prepare(`
+      SELECT hse.skill_id, hse.vector, hse.dimensions
+      FROM hub_skill_embeddings hse
+      JOIN hub_skills hs ON hs.id = hse.skill_id
+    `).all();
+        return rows.map(r => ({
+            skillId: r.skill_id,
+            vector: new Float32Array(r.vector.buffer, r.vector.byteOffset, r.dimensions),
+        }));
+    }
     searchHubChunks(query, options) {
         const limit = options?.maxResults ?? 10;
         const userId = options?.userId ?? "";
         const rows = this.db.prepare(`
-      SELECT hc.id, hc.content, hc.summary, hc.role, hc.created_at, ht.title as task_title, ht.visibility, '' as group_name, hu.username as owner_name,
+      SELECT hc.id, hc.content, hc.summary, hc.role, hc.created_at, ht.title as task_title, ht.visibility,
+             COALESCE(hg.name, '') as group_name, hu.username as owner_name,
              bm25(hub_chunks_fts) as rank
       FROM hub_chunks_fts f
       JOIN hub_chunks hc ON hc.rowid = f.rowid
       JOIN hub_tasks ht ON ht.id = hc.hub_task_id
       LEFT JOIN hub_users hu ON hu.id = ht.source_user_id
+      LEFT JOIN hub_groups hg ON hg.id = ht.group_id
       WHERE hub_chunks_fts MATCH ?
+        AND (ht.visibility = 'public'
+             OR ht.source_user_id = ?
+             OR EXISTS (SELECT 1 FROM hub_group_members gm WHERE gm.group_id = ht.group_id AND gm.user_id = ?))
       ORDER BY rank
       LIMIT ?
-    `).all(sanitizeFtsQuery(query), limit);
+    `).all(sanitizeFtsQuery(query), userId, userId, limit);
         return rows.map((row, idx) => ({ hit: row, rank: idx + 1 }));
     }
     upsertHubEmbedding(chunkId, vector) {
@@ -1872,7 +1996,10 @@ class SqliteStore {
       FROM hub_embeddings he
       JOIN hub_chunks hc ON hc.id = he.chunk_id
       JOIN hub_tasks ht ON ht.id = hc.hub_task_id
-    `).all();
+      WHERE ht.visibility = 'public'
+        OR ht.source_user_id = ?
+        OR EXISTS (SELECT 1 FROM hub_group_members gm WHERE gm.group_id = ht.group_id AND gm.user_id = ?)
+    `).all(userId, userId);
         return rows.map(r => ({
             chunkId: r.chunk_id,
             vector: new Float32Array(r.vector.buffer, r.vector.byteOffset, r.dimensions),
@@ -1880,14 +2007,19 @@ class SqliteStore {
     }
     getVisibleHubSearchHitByChunkId(chunkId, userId) {
         const row = this.db.prepare(`
-      SELECT hc.id, hc.content, hc.summary, hc.role, hc.created_at, ht.title as task_title, ht.visibility, '' as group_name, hu.username as owner_name,
+      SELECT hc.id, hc.content, hc.summary, hc.role, hc.created_at, ht.title as task_title, ht.visibility,
+             COALESCE(hg.name, '') as group_name, hu.username as owner_name,
              0 as rank
       FROM hub_chunks hc
       JOIN hub_tasks ht ON ht.id = hc.hub_task_id
       LEFT JOIN hub_users hu ON hu.id = ht.source_user_id
+      LEFT JOIN hub_groups hg ON hg.id = ht.group_id
       WHERE hc.id = ?
+        AND (ht.visibility = 'public'
+             OR ht.source_user_id = ?
+             OR EXISTS (SELECT 1 FROM hub_group_members gm WHERE gm.group_id = ht.group_id AND gm.user_id = ?))
       LIMIT 1
-    `).get(chunkId);
+    `).get(chunkId, userId, userId);
         return row ?? null;
     }
     getHubChunkById(chunkId) {
@@ -2276,6 +2408,8 @@ function rowToClientHubConnection(row) {
         userToken: row.user_token,
         role: row.role,
         connectedAt: row.connected_at,
+        identityKey: row.identity_key || "",
+        lastKnownStatus: row.last_known_status || "",
     };
 }
 function rowToHubUser(row) {
@@ -2291,6 +2425,11 @@ function rowToHubUser(row) {
         approvedAt: row.approved_at,
         lastIp: row.last_ip || "",
         lastActiveAt: row.last_active_at ?? null,
+        identityKey: row.identity_key || "",
+        leftAt: row.left_at ?? null,
+        removedAt: row.removed_at ?? null,
+        rejectedAt: row.rejected_at ?? null,
+        rejoinRequestedAt: row.rejoin_requested_at ?? null,
     };
 }
 function rowToHubTask(row) {