@pasko70/pibo 1.0.5 → 1.2.0

package/dist/apps/chat/read-model.js

@@ -1,218 +1 @@
-import { randomUUID } from "node:crypto";
-import { mkdirSync } from "node:fs";
-import { dirname, resolve } from "node:path";
-import { DatabaseSync } from "node:sqlite";
-export class ChatWebReadModel {
-    db;
-    constructor(path) {
-        const resolvedPath = path === ":memory:" ? path : resolve(path);
-        if (resolvedPath !== ":memory:") {
-            mkdirSync(dirname(resolvedPath), { recursive: true });
-        }
-        this.db = new DatabaseSync(resolvedPath);
-        this.db.exec("PRAGMA busy_timeout = 5000");
-        if (resolvedPath !== ":memory:")
-            this.db.exec("PRAGMA journal_mode = WAL");
-        this.dropLegacySchema();
-        this.db.exec(`
-			CREATE TABLE IF NOT EXISTS web_chat_sessions (
-				pibo_session_id TEXT PRIMARY KEY,
-				pi_session_id TEXT NOT NULL,
-				parent_id TEXT,
-				profile TEXT NOT NULL,
-				channel TEXT NOT NULL,
-				kind TEXT NOT NULL,
-				created_at TEXT NOT NULL,
-				updated_at TEXT NOT NULL,
-				last_activity_at TEXT,
-				status TEXT NOT NULL DEFAULT 'idle'
-			);
-
-			CREATE TABLE IF NOT EXISTS web_chat_events (
-				id TEXT PRIMARY KEY,
-				pibo_session_id TEXT NOT NULL,
-				event_sequence INTEGER,
-				event_id TEXT,
-				type TEXT NOT NULL,
-				created_at TEXT NOT NULL,
-				payload_json TEXT NOT NULL
-			);
-		`);
-        this.migrateEventSequence();
-        this.db.exec(`
-
-			CREATE INDEX IF NOT EXISTS idx_web_chat_events_session_created
-				ON web_chat_events(pibo_session_id, created_at, id);
-			CREATE INDEX IF NOT EXISTS idx_web_chat_events_session_sequence
-				ON web_chat_events(pibo_session_id, event_sequence, id);
-			CREATE INDEX IF NOT EXISTS idx_web_chat_events_event_id
-				ON web_chat_events(event_id);
-			CREATE INDEX IF NOT EXISTS idx_web_chat_sessions_parent
-				ON web_chat_sessions(parent_id);
-		`);
-        this.resetInterruptedSessions();
-    }
-    upsertSession(session, status = "idle") {
-        this.db
-            .prepare(`
-				INSERT INTO web_chat_sessions (
-					pibo_session_id,
-					pi_session_id,
-					parent_id,
-					profile,
-					channel,
-					kind,
-					created_at,
-					updated_at,
-					last_activity_at,
-					status
-				) VALUES (?, ?, ?, ?, ?, ?, ?, ?, NULL, ?)
-				ON CONFLICT(pibo_session_id) DO UPDATE SET
-					pi_session_id = excluded.pi_session_id,
-					parent_id = excluded.parent_id,
-					profile = excluded.profile,
-					channel = excluded.channel,
-					kind = excluded.kind,
-					updated_at = excluded.updated_at,
-					status = CASE
-						WHEN web_chat_sessions.status = 'running' AND excluded.status = 'idle' THEN web_chat_sessions.status
-						ELSE excluded.status
-					END
-			`)
-            .run(session.id, session.piSessionId, session.parentId ?? null, session.profile, session.channel, session.kind, session.createdAt, session.updatedAt, status);
-    }
-    recordEvent(event, session) {
-        if (session)
-            this.upsertSession(session, statusFromEvent(event));
-        const id = randomUUID();
-        const createdAt = new Date().toISOString();
-        const eventSequence = this.nextEventSequence(event.piboSessionId);
-        const eventId = "eventId" in event && typeof event.eventId === "string" ? event.eventId : undefined;
-        this.db
-            .prepare("INSERT INTO web_chat_events (id, pibo_session_id, event_sequence, event_id, type, created_at, payload_json) VALUES (?, ?, ?, ?, ?, ?, ?)")
-            .run(id, event.piboSessionId, eventSequence, eventId ?? null, event.type, createdAt, JSON.stringify(event));
-        this.db
-            .prepare("UPDATE web_chat_sessions SET last_activity_at = ?, status = ?, updated_at = ? WHERE pibo_session_id = ?")
-            .run(createdAt, statusFromEvent(event), createdAt, event.piboSessionId);
-        return { id, piboSessionId: event.piboSessionId, eventSequence, eventId, type: event.type, createdAt, payload: event };
-    }
-    listSessions() {
-        return this.db.prepare("SELECT * FROM web_chat_sessions ORDER BY updated_at DESC").all().map(sessionFromRow);
-    }
-    listEvents(piboSessionId, limit = 1000) {
-        const rows = this.db
-            .prepare(`
-					SELECT * FROM (
-						SELECT rowid AS _rowid, * FROM web_chat_events
-						WHERE pibo_session_id = ?
-						ORDER BY event_sequence DESC, rowid DESC
-						LIMIT ?
-					)
-					ORDER BY event_sequence ASC, _rowid ASC
-				`)
-            .all(piboSessionId, limit);
-        return rows.map(eventFromRow);
-    }
-    listAllEvents(piboSessionId) {
-        const rows = this.db
-            .prepare(`
-					SELECT rowid AS _rowid, * FROM web_chat_events
-					WHERE pibo_session_id = ?
-					ORDER BY event_sequence ASC, _rowid ASC
-				`)
-            .all(piboSessionId);
-        return rows.map(eventFromRow);
-    }
-    getLatestEventSequence(piboSessionId) {
-        const row = this.db
-            .prepare("SELECT COALESCE(MAX(event_sequence), 0) AS latest_sequence FROM web_chat_events WHERE pibo_session_id = ?")
-            .get(piboSessionId);
-        return row.latest_sequence;
-    }
-    deleteSessions(piboSessionIds) {
-        if (!piboSessionIds.length)
-            return 0;
-        const placeholders = piboSessionIds.map(() => "?").join(", ");
-        this.db.prepare(`DELETE FROM web_chat_events WHERE pibo_session_id IN (${placeholders})`).run(...piboSessionIds);
-        const result = this.db
-            .prepare(`DELETE FROM web_chat_sessions WHERE pibo_session_id IN (${placeholders})`)
-            .run(...piboSessionIds);
-        return Number(result.changes ?? 0);
-    }
-    close() {
-        this.db.close();
-    }
-    dropLegacySchema() {
-        const sessionColumns = tableColumns(this.db, "web_chat_sessions");
-        const eventColumns = tableColumns(this.db, "web_chat_events");
-        if ((sessionColumns.size > 0 && !sessionColumns.has("pibo_session_id")) ||
-            (eventColumns.size > 0 && !eventColumns.has("pibo_session_id"))) {
-            this.db.exec(`
-				DROP TABLE IF EXISTS web_chat_events;
-				DROP TABLE IF EXISTS web_chat_sessions;
-			`);
-        }
-    }
-    migrateEventSequence() {
-        const eventColumns = tableColumns(this.db, "web_chat_events");
-        if (!eventColumns.has("event_sequence")) {
-            this.db.exec("ALTER TABLE web_chat_events ADD COLUMN event_sequence INTEGER");
-        }
-        this.db.exec("UPDATE web_chat_events SET event_sequence = rowid WHERE event_sequence IS NULL");
-    }
-    nextEventSequence(piboSessionId) {
-        const row = this.db
-            .prepare("SELECT COALESCE(MAX(event_sequence), 0) + 1 AS next_sequence FROM web_chat_events WHERE pibo_session_id = ?")
-            .get(piboSessionId);
-        return row.next_sequence;
-    }
-    resetInterruptedSessions() {
-        this.db.prepare("UPDATE web_chat_sessions SET status = 'idle' WHERE status = 'running'").run();
-    }
-}
-export function createDefaultChatWebReadModel(cwd = process.cwd()) {
-    return new ChatWebReadModel(resolve(cwd, ".pibo/web-chat.sqlite"));
-}
-function statusFromEvent(event) {
-    if (event.type === "session_error")
-        return "error";
-    if (event.type === "message_started" ||
-        event.type === "assistant_delta" ||
-        event.type === "thinking_started" ||
-        event.type === "thinking_delta" ||
-        event.type === "thinking_finished" ||
-        event.type === "subagent_session" ||
-        event.type === "tool_execution_started" ||
-        event.type === "tool_execution_updated") {
-        return "running";
-    }
-    return "idle";
-}
-function sessionFromRow(row) {
-    return {
-        piboSessionId: row.pibo_session_id,
-        piSessionId: row.pi_session_id,
-        parentId: row.parent_id ?? undefined,
-        profile: row.profile,
-        channel: row.channel,
-        kind: row.kind,
-        createdAt: row.created_at,
-        updatedAt: row.updated_at,
-        lastActivityAt: row.last_activity_at ?? undefined,
-        status: row.status === "running" || row.status === "error" ? row.status : "idle",
-    };
-}
-function eventFromRow(row) {
-    return {
-        id: row.id,
-        piboSessionId: row.pibo_session_id,
-        eventSequence: row.event_sequence ?? undefined,
-        eventId: row.event_id ?? undefined,
-        type: row.type,
-        createdAt: row.created_at,
-        payload: JSON.parse(row.payload_json),
-    };
-}
-function tableColumns(db, tableName) {
-    return new Set(db.prepare(`PRAGMA table_info(${tableName})`).all().map((column) => column.name));
-}
+export {};

package/dist/apps/chat/rooms.js

@@ -1,321 +1 @@
-import { randomUUID } from "node:crypto";
-import { mkdirSync } from "node:fs";
-import { dirname, resolve } from "node:path";
-import { DatabaseSync } from "node:sqlite";
-export const CHAT_ROOM_ID_METADATA_KEY = "chatRoomId";
-const CHAT_ROOM_ARCHIVED_AT_METADATA_KEY = "chatRoomArchivedAt";
-const CHAT_ROOM_WORKSPACE_METADATA_KEY = "workspace";
-export class PiboRoomStore {
-    db;
-    constructor(path) {
-        const resolvedPath = path === ":memory:" ? path : resolve(path);
-        if (resolvedPath !== ":memory:") {
-            mkdirSync(dirname(resolvedPath), { recursive: true });
-        }
-        this.db = new DatabaseSync(resolvedPath);
-        this.db.exec("PRAGMA busy_timeout = 5000");
-        if (resolvedPath !== ":memory:")
-            this.db.exec("PRAGMA journal_mode = WAL");
-        this.db.exec(`
-			CREATE TABLE IF NOT EXISTS pibo_rooms (
-				id TEXT PRIMARY KEY,
-				owner_scope TEXT NOT NULL,
-				name TEXT NOT NULL,
-				topic TEXT,
-				type TEXT NOT NULL,
-				parent_room_id TEXT,
-				created_at TEXT NOT NULL,
-				updated_at TEXT NOT NULL,
-				retention_policy_id TEXT,
-				metadata_json TEXT,
-				FOREIGN KEY(parent_room_id) REFERENCES pibo_rooms(id)
-			);
-
-			CREATE TABLE IF NOT EXISTS pibo_room_members (
-				room_id TEXT NOT NULL,
-				principal_id TEXT NOT NULL,
-				role TEXT NOT NULL,
-				joined_at TEXT NOT NULL,
-				last_read_stream_id INTEGER,
-				PRIMARY KEY(room_id, principal_id),
-				FOREIGN KEY(room_id) REFERENCES pibo_rooms(id)
-			);
-
-			CREATE INDEX IF NOT EXISTS idx_pibo_rooms_owner
-				ON pibo_rooms(owner_scope, updated_at);
-			CREATE INDEX IF NOT EXISTS idx_pibo_rooms_parent
-				ON pibo_rooms(parent_room_id, updated_at);
-			CREATE INDEX IF NOT EXISTS idx_pibo_room_members_principal
-				ON pibo_room_members(principal_id, room_id);
-		`);
-    }
-    createRoom(input) {
-        const now = new Date().toISOString();
-        const room = {
-            id: input.id ?? `room_${randomUUID()}`,
-            ownerScope: input.ownerScope,
-            name: input.name,
-            topic: input.topic,
-            workspace: roomWorkspaceFromMetadata(input.metadata),
-            type: input.type ?? "chat",
-            parentRoomId: input.parentRoomId,
-            createdAt: now,
-            updatedAt: now,
-            retentionPolicyId: input.retentionPolicyId,
-            metadata: input.metadata ?? {},
-        };
-        this.db
-            .prepare(`
-				INSERT INTO pibo_rooms (
-					id,
-					owner_scope,
-					name,
-					topic,
-					type,
-					parent_room_id,
-					created_at,
-					updated_at,
-					retention_policy_id,
-					metadata_json
-				) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-			`)
-            .run(room.id, room.ownerScope, room.name, room.topic ?? null, room.type, room.parentRoomId ?? null, room.createdAt, room.updatedAt, room.retentionPolicyId ?? null, JSON.stringify(room.metadata));
-        return room;
-    }
-    updateRoom(id, input) {
-        const existing = this.getRoom(id);
-        if (!existing)
-            return undefined;
-        const updated = {
-            ...existing,
-            name: input.name ?? existing.name,
-            topic: input.topic === null ? undefined : input.topic ?? existing.topic,
-            workspace: roomWorkspaceFromMetadata(input.metadata ?? existing.metadata),
-            parentRoomId: input.parentRoomId === null ? undefined : input.parentRoomId ?? existing.parentRoomId,
-            retentionPolicyId: input.retentionPolicyId === null ? undefined : input.retentionPolicyId ?? existing.retentionPolicyId,
-            metadata: input.metadata ?? existing.metadata,
-            updatedAt: new Date().toISOString(),
-        };
-        this.db
-            .prepare(`
-				UPDATE pibo_rooms SET
-					name = ?,
-					topic = ?,
-					parent_room_id = ?,
-					updated_at = ?,
-					retention_policy_id = ?,
-					metadata_json = ?
-				WHERE id = ?
-			`)
-            .run(updated.name, updated.topic ?? null, updated.parentRoomId ?? null, updated.updatedAt, updated.retentionPolicyId ?? null, JSON.stringify(updated.metadata), id);
-        return this.getRoom(id);
-    }
-    deleteRooms(ids) {
-        if (!ids.length)
-            return 0;
-        const placeholders = ids.map(() => "?").join(", ");
-        this.db.prepare(`DELETE FROM pibo_room_members WHERE room_id IN (${placeholders})`).run(...ids);
-        const result = this.db.prepare(`DELETE FROM pibo_rooms WHERE id IN (${placeholders})`).run(...ids);
-        return Number(result.changes ?? 0);
-    }
-    getRoom(id) {
-        const row = this.db.prepare("SELECT * FROM pibo_rooms WHERE id = ?").get(id);
-        return row ? roomFromRow(row) : undefined;
-    }
-    listRooms(ownerScope) {
-        return this.db
-            .prepare("SELECT * FROM pibo_rooms WHERE owner_scope = ? ORDER BY updated_at DESC, id ASC")
-            .all(ownerScope).map(roomFromRow);
-    }
-    listRoomTree(ownerScope) {
-        const rooms = this.listRooms(ownerScope);
-        const byId = new Map();
-        for (const room of rooms)
-            byId.set(room.id, { ...room, children: [] });
-        const roots = [];
-        for (const node of byId.values()) {
-            const parent = node.parentRoomId ? byId.get(node.parentRoomId) : undefined;
-            if (parent) {
-                parent.children.push(node);
-            }
-            else {
-                roots.push(node);
-            }
-        }
-        sortRoomNodes(roots);
-        return roots;
-    }
-    listRoomSubtree(roomId) {
-        const root = this.getRoom(roomId);
-        if (!root)
-            return [];
-        const rooms = this.listRooms(root.ownerScope);
-        const byParent = new Map();
-        for (const room of rooms) {
-            if (!room.parentRoomId)
-                continue;
-            const children = byParent.get(room.parentRoomId) ?? [];
-            children.push(room);
-            byParent.set(room.parentRoomId, children);
-        }
-        const result = [];
-        const visit = (room) => {
-            result.push(room);
-            for (const child of byParent.get(room.id) ?? [])
-                visit(child);
-        };
-        visit(root);
-        return result;
-    }
-    ensureDefaultRoom(input) {
-        const existing = this.listRooms(input.ownerScope).find((room) => room.metadata.default === true);
-        if (existing) {
-            this.ensureMember({
-                roomId: existing.id,
-                principalId: input.principalId,
-                role: "owner",
-            });
-            return existing;
-        }
-        const room = this.createRoom({
-            ownerScope: input.ownerScope,
-            name: input.name ?? "Personal Chat",
-            type: "chat",
-            metadata: { default: true },
-        });
-        this.ensureMember({ roomId: room.id, principalId: input.principalId, role: "owner" });
-        return room;
-    }
-    ensureMember(input) {
-        const now = new Date().toISOString();
-        this.db
-            .prepare(`
-				INSERT INTO pibo_room_members (room_id, principal_id, role, joined_at, last_read_stream_id)
-				VALUES (?, ?, ?, ?, NULL)
-				ON CONFLICT(room_id, principal_id) DO UPDATE SET
-					role = CASE
-						WHEN pibo_room_members.role = 'owner' THEN pibo_room_members.role
-						ELSE excluded.role
-					END
-			`)
-            .run(input.roomId, input.principalId, input.role, now);
-        const member = this.getMember(input.roomId, input.principalId);
-        if (!member)
-            throw new Error(`Failed to create room member for "${input.roomId}"`);
-        return member;
-    }
-    getMember(roomId, principalId) {
-        const row = this.db
-            .prepare("SELECT * FROM pibo_room_members WHERE room_id = ? AND principal_id = ?")
-            .get(roomId, principalId);
-        return row ? memberFromRow(row) : undefined;
-    }
-    updateReadCursor(roomId, principalId, lastReadStreamId) {
-        this.db
-            .prepare(`
-				UPDATE pibo_room_members
-				SET last_read_stream_id = MAX(COALESCE(last_read_stream_id, 0), ?)
-				WHERE room_id = ? AND principal_id = ?
-			`)
-            .run(lastReadStreamId, roomId, principalId);
-        return this.getMember(roomId, principalId);
-    }
-    requireRoomAccess(roomId, principalId, action = "read") {
-        const room = this.getRoom(roomId);
-        const member = room ? this.getMember(roomId, principalId) : undefined;
-        if (!room || !member)
-            throw new Error("Room is not available for this user");
-        if (action === "write" && member.role === "viewer")
-            throw new Error("Room write access is required");
-        if (action === "admin" && member.role !== "owner" && member.role !== "admin") {
-            throw new Error("Room admin access is required");
-        }
-        return room;
-    }
-    close() {
-        this.db.close();
-    }
-}
-export function createDefaultPiboRoomStore(cwd = process.cwd()) {
-    return new PiboRoomStore(resolve(cwd, ".pibo/web-chat.sqlite"));
-}
-export function chatRoomIdFromMetadata(metadata) {
-    const value = metadata?.[CHAT_ROOM_ID_METADATA_KEY];
-    return typeof value === "string" && value.length > 0 ? value : undefined;
-}
-export function withChatRoomId(metadata, roomId) {
-    return { ...(metadata ?? {}), [CHAT_ROOM_ID_METADATA_KEY]: roomId };
-}
-export function isDefaultPiboRoom(room) {
-    return room.metadata.default === true;
-}
-export function isPiboRoomArchived(room) {
-    return typeof room.metadata[CHAT_ROOM_ARCHIVED_AT_METADATA_KEY] === "string";
-}
-export function roomWorkspaceFromMetadata(metadata) {
-    const value = metadata?.[CHAT_ROOM_WORKSPACE_METADATA_KEY];
-    return typeof value === "string" && value.length > 0 ? value : undefined;
-}
-export function withPiboRoomArchived(metadata, archived) {
-    const next = { ...(metadata ?? {}) };
-    if (archived) {
-        next[CHAT_ROOM_ARCHIVED_AT_METADATA_KEY] = new Date().toISOString();
-    }
-    else {
-        delete next[CHAT_ROOM_ARCHIVED_AT_METADATA_KEY];
-    }
-    return next;
-}
-export function withPiboRoomWorkspace(metadata, workspace) {
-    const next = { ...(metadata ?? {}) };
-    if (workspace) {
-        next[CHAT_ROOM_WORKSPACE_METADATA_KEY] = workspace;
-    }
-    else {
-        delete next[CHAT_ROOM_WORKSPACE_METADATA_KEY];
-    }
-    return next;
-}
-function roomFromRow(row) {
-    const metadata = parseMetadata(row.metadata_json);
-    return {
-        id: row.id,
-        ownerScope: row.owner_scope,
-        name: row.name,
-        topic: row.topic ?? undefined,
-        workspace: roomWorkspaceFromMetadata(metadata),
-        type: row.type,
-        parentRoomId: row.parent_room_id ?? undefined,
-        createdAt: row.created_at,
-        updatedAt: row.updated_at,
-        retentionPolicyId: row.retention_policy_id ?? undefined,
-        metadata,
-    };
-}
-function memberFromRow(row) {
-    return {
-        roomId: row.room_id,
-        principalId: row.principal_id,
-        role: row.role,
-        joinedAt: row.joined_at,
-        lastReadStreamId: row.last_read_stream_id ?? undefined,
-    };
-}
-function parseMetadata(value) {
-    if (!value)
-        return {};
-    try {
-        const parsed = JSON.parse(value);
-        if (!parsed || typeof parsed !== "object" || Array.isArray(parsed))
-            return {};
-        return parsed;
-    }
-    catch {
-        return {};
-    }
-}
-function sortRoomNodes(nodes) {
-    nodes.sort((left, right) => right.updatedAt.localeCompare(left.updatedAt) || left.name.localeCompare(right.name));
-    for (const node of nodes)
-        sortRoomNodes(node.children);
-}
+export { CHAT_ROOM_ID_METADATA_KEY, chatRoomIdFromMetadata, isDefaultPiboRoom, isPiboRoomArchived, roomWorkspaceFromMetadata, withChatRoomId, withPiboRoomArchived, withPiboRoomWorkspace, } from "./types/rooms.js";