pi2web 0.2.1 → 0.2.8

package/dist/metadata-store.js

@@ -1,8 +1,31 @@
 import Database from 'better-sqlite3';
 import { chmodSync, mkdirSync, statSync } from 'node:fs';
 import path from 'node:path';
-const asTime = (value) => { const time = value instanceof Date ? value.getTime() : typeof value === 'number' ? value : Date.parse(value ?? ''); return Number.isFinite(time) ? time : Date.now(); };
-/** SQLite-backed Remote Pi metadata. Pi conversation JSONL files remain the source of truth. */
+import { METADATA_DB_FILE } from './paths.js';
+const asTime = (value) => {
+    const time = value instanceof Date ? value.getTime() : typeof value === 'number' ? value : Date.parse(value ?? '');
+    return Number.isFinite(time) ? time : Date.now();
+};
+const parseTemplate = (value) => {
+    if (typeof value !== 'string' || !value)
+        return;
+    try {
+        const parsed = JSON.parse(value);
+        if (parsed &&
+            typeof parsed.id === 'string' &&
+            typeof parsed.systemPrompt === 'string' &&
+            (parsed.mode === 'replace' || parsed.mode === 'append'))
+            return {
+                id: parsed.id,
+                name: String(parsed.name ?? parsed.id),
+                mode: parsed.mode,
+                systemPrompt: parsed.systemPrompt
+            };
+    }
+    catch { }
+    return;
+};
+/** SQLite-backed pi2web metadata. Pi conversation JSONL files remain the source of truth. */
 export class MetadataStore {
     dataDir;
     db;
@@ -12,7 +35,9 @@ export class MetadataStore {
     constructor(dataDir) {
         this.dataDir = dataDir;
     }
-    get file() { return path.join(this.dataDir, 'remote-pi.db'); }
+    get file() {
+        return path.join(this.dataDir, METADATA_DB_FILE);
+    }
     init() {
         if (this.db)
             return;
@@ -21,7 +46,7 @@ export class MetadataStore {
             chmodSync(this.dataDir, 0o700);
         }
         catch { }
-        const db = this.db = new Database(this.file);
+        const db = (this.db = new Database(this.file));
         try {
             chmodSync(this.file, 0o600);
         }
@@ -46,6 +71,7 @@ export class MetadataStore {
         cwd TEXT NOT NULL,
         status TEXT NOT NULL,
         profile TEXT NOT NULL DEFAULT 'default',
+        template TEXT,
         created_at INTEGER NOT NULL,
         last_active_at INTEGER NOT NULL,
         archived_at INTEGER,
@@ -82,67 +108,188 @@ export class MetadataStore {
         const agentColumns = db.prepare('PRAGMA table_info(agents)').all();
         if (!agentColumns.some(column => column.name === 'profile'))
             db.exec("ALTER TABLE agents ADD COLUMN profile TEXT NOT NULL DEFAULT 'default'");
+        // System prompt template snapshot (JSON). Agents created before templates existed use Pi's prompt.
+        if (!agentColumns.some(column => column.name === 'template'))
+            db.exec('ALTER TABLE agents ADD COLUMN template TEXT');
         // Only stamp the initial version: later migrations (collaboration tables) raise it and must not be reset on restart.
         if (Number(db.pragma('user_version', { simple: true })) < 1)
             db.pragma('user_version = 1');
         this.archivedAgents = new Set(db.prepare('SELECT id FROM agents WHERE archived_at IS NOT NULL').all().map(row => row.id));
     }
-    get database() { if (!this.db)
-        throw new Error('Metadata store is not initialized'); return this.db; }
+    get database() {
+        if (!this.db)
+            throw new Error('Metadata store is not initialized');
+        return this.db;
+    }
     /** Shared handle so collaboration tables live in the same file and the same backup. */
-    get connection() { return this.database; }
-    getSetting(key) { const row = this.database.prepare('SELECT value FROM settings WHERE key=?').get(key); if (!row)
-        return; try {
-        return JSON.parse(row.value);
+    get connection() {
+        return this.database;
+    }
+    getSetting(key) {
+        const row = this.database.prepare('SELECT value FROM settings WHERE key=?').get(key);
+        if (!row)
+            return;
+        try {
+            return JSON.parse(row.value);
+        }
+        catch {
+            return;
+        }
+    }
+    setSetting(key, value) {
+        this.database
+            .prepare('INSERT INTO settings(key,value,updated_at) VALUES(?,?,?) ON CONFLICT(key) DO UPDATE SET value=excluded.value,updated_at=excluded.updated_at')
+            .run(key, JSON.stringify(value), Date.now());
+    }
+    listWorkspaces() {
+        return this.database
+            .prepare('SELECT id,label,root_path,created_at FROM workspaces ORDER BY created_at,id')
+            .all().map(row => ({
+            id: row.id,
+            label: row.label,
+            rootPath: row.root_path,
+            createdAt: new Date(row.created_at).toISOString()
+        }));
+    }
+    saveWorkspace(workspace) {
+        this.database
+            .prepare(`INSERT INTO workspaces(id,label,root_path,created_at) VALUES(?,?,?,?) ON CONFLICT(id) DO UPDATE SET label=excluded.label,root_path=excluded.root_path`)
+            .run(workspace.id, workspace.label, workspace.rootPath, asTime(workspace.createdAt));
+    }
+    listAgents() {
+        return this.database
+            .prepare('SELECT * FROM agents WHERE archived_at IS NULL ORDER BY last_active_at DESC,id DESC')
+            .all().map(row => ({
+            agentId: row.id,
+            workspaceId: row.workspace_id,
+            sessionId: row.session_id,
+            sessionFile: row.session_file ?? undefined,
+            sessionName: row.session_name ?? undefined,
+            cwd: row.cwd,
+            status: row.status,
+            profile: row.profile === 'collab' ? 'collab' : 'default',
+            ...(parseTemplate(row.template) ? { template: parseTemplate(row.template) } : {}),
+            createdAt: new Date(row.created_at).toISOString(),
+            lastActiveAt: new Date(row.last_active_at).toISOString()
+        }));
+    }
+    writeAgent(record, archivedAt) {
+        this.database
+            .prepare(
+        // Ordinary state/index writes must not overwrite the conversation's template snapshot.
+        // Only commitAgentTemplate may change it after creation.
+        `INSERT INTO agents(id,workspace_id,session_id,session_file,session_name,cwd,status,profile,template,created_at,last_active_at,archived_at) VALUES(?,?,?,?,?,?,?,?,?,?,?,?) ON CONFLICT(id) DO UPDATE SET workspace_id=excluded.workspace_id,session_id=excluded.session_id,session_file=excluded.session_file,session_name=excluded.session_name,cwd=excluded.cwd,status=excluded.status,profile=excluded.profile,last_active_at=excluded.last_active_at,archived_at=excluded.archived_at`)
+            .run(record.agentId, record.workspaceId, record.sessionId, record.sessionFile ?? null, record.sessionName ?? null, record.cwd, record.status, record.profile ?? 'default', record.template ? JSON.stringify(record.template) : null, asTime(record.createdAt), asTime(record.lastActiveAt), archivedAt);
+    }
+    saveAgent(record) {
+        if (this.archivedAgents.has(record.agentId))
+            return;
+        this.pendingAgents.delete(record.agentId);
+        this.writeAgent(record, null);
+    }
+    /** Commit the validated pre-conversation choice and its session identity together. */
+    commitAgentTemplate(record) {
+        this.database.transaction(() => {
+            const updated = this.database
+                .prepare('UPDATE agents SET template=? WHERE id=? AND archived_at IS NULL')
+                .run(record.template ? JSON.stringify(record.template) : null, record.agentId);
+            if (!updated.changes)
+                throw Object.assign(new Error('Agent not found'), { code: 'AGENT_NOT_FOUND' });
+            this.writeAgent(record, null);
+        })();
+        this.pendingAgents.delete(record.agentId);
+    }
+    scheduleAgent(record) {
+        if (this.archivedAgents.has(record.agentId))
+            return;
+        this.pendingAgents.set(record.agentId, { ...record });
+        if (!this.flushTimer)
+            this.flushTimer = setTimeout(() => this.flush(), 1000);
+    }
+    archiveAgent(record) {
+        this.archivedAgents.add(record.agentId);
+        this.pendingAgents.delete(record.agentId);
+        this.writeAgent(record, Date.now());
+    }
+    flush() {
+        if (this.flushTimer) {
+            clearTimeout(this.flushTimer);
+            this.flushTimer = undefined;
+        }
+        if (!this.pendingAgents.size)
+            return;
+        const records = [...this.pendingAgents.values()];
+        this.pendingAgents.clear();
+        this.database.transaction((items) => {
+            for (const item of items)
+                this.writeAgent(item, null);
+        })(records);
     }
-    catch {
-        return;
-    } }
-    setSetting(key, value) { this.database.prepare('INSERT INTO settings(key,value,updated_at) VALUES(?,?,?) ON CONFLICT(key) DO UPDATE SET value=excluded.value,updated_at=excluded.updated_at').run(key, JSON.stringify(value), Date.now()); }
-    listWorkspaces() { return this.database.prepare('SELECT id,label,root_path,created_at FROM workspaces ORDER BY created_at,id').all().map(row => ({ id: row.id, label: row.label, rootPath: row.root_path, createdAt: new Date(row.created_at).toISOString() })); }
-    saveWorkspace(workspace) { this.database.prepare(`INSERT INTO workspaces(id,label,root_path,created_at) VALUES(?,?,?,?) ON CONFLICT(id) DO UPDATE SET label=excluded.label,root_path=excluded.root_path`).run(workspace.id, workspace.label, workspace.rootPath, asTime(workspace.createdAt)); }
-    listAgents() { return this.database.prepare('SELECT * FROM agents WHERE archived_at IS NULL ORDER BY last_active_at DESC,id DESC').all().map(row => ({ agentId: row.id, workspaceId: row.workspace_id, sessionId: row.session_id, sessionFile: row.session_file ?? undefined, sessionName: row.session_name ?? undefined, cwd: row.cwd, status: row.status, profile: row.profile === 'collab' ? 'collab' : 'default', createdAt: new Date(row.created_at).toISOString(), lastActiveAt: new Date(row.last_active_at).toISOString() })); }
-    writeAgent(record, archivedAt) { this.database.prepare(`INSERT INTO agents(id,workspace_id,session_id,session_file,session_name,cwd,status,profile,created_at,last_active_at,archived_at) VALUES(?,?,?,?,?,?,?,?,?,?,?) ON CONFLICT(id) DO UPDATE SET workspace_id=excluded.workspace_id,session_id=excluded.session_id,session_file=excluded.session_file,session_name=excluded.session_name,cwd=excluded.cwd,status=excluded.status,profile=excluded.profile,last_active_at=excluded.last_active_at,archived_at=excluded.archived_at`).run(record.agentId, record.workspaceId, record.sessionId, record.sessionFile ?? null, record.sessionName ?? null, record.cwd, record.status, record.profile ?? 'default', asTime(record.createdAt), asTime(record.lastActiveAt), archivedAt); }
-    saveAgent(record) { if (this.archivedAgents.has(record.agentId))
-        return; this.pendingAgents.delete(record.agentId); this.writeAgent(record, null); }
-    scheduleAgent(record) { if (this.archivedAgents.has(record.agentId))
-        return; this.pendingAgents.set(record.agentId, { ...record }); if (!this.flushTimer)
-        this.flushTimer = setTimeout(() => this.flush(), 1000); }
-    archiveAgent(record) { this.archivedAgents.add(record.agentId); this.pendingAgents.delete(record.agentId); this.writeAgent(record, Date.now()); }
-    flush() { if (this.flushTimer) {
-        clearTimeout(this.flushTimer);
-        this.flushTimer = undefined;
-    } if (!this.pendingAgents.size)
-        return; const records = [...this.pendingAgents.values()]; this.pendingAgents.clear(); this.database.transaction((items) => { for (const item of items)
-        this.writeAgent(item, null); })(records); }
     syncSessions(workspaceId, cwd, sessions) {
         const db = this.database, upsert = db.prepare(`INSERT INTO sessions(session_id,session_file,workspace_id,cwd,name,parent_session_path,created_at,last_active_at,message_count,first_message,file_size,file_mtime,archived_at) VALUES(?,?,?,?,?,?,?,?,?,?,?,?,NULL) ON CONFLICT(session_id) DO UPDATE SET session_file=excluded.session_file,workspace_id=excluded.workspace_id,cwd=excluded.cwd,name=excluded.name,parent_session_path=excluded.parent_session_path,last_active_at=excluded.last_active_at,message_count=excluded.message_count,first_message=excluded.first_message,file_size=excluded.file_size,file_mtime=excluded.file_mtime,archived_at=NULL`);
         const remove = db.prepare('DELETE FROM sessions WHERE session_file=?');
-        db.transaction(() => { const paths = new Set(); for (const session of sessions) {
-            paths.add(session.path);
-            let size = 0, mtime = asTime(session.modified);
-            try {
-                const stat = statSync(session.path);
-                size = stat.size;
-                mtime = stat.mtimeMs;
+        db.transaction(() => {
+            const paths = new Set();
+            for (const session of sessions) {
+                paths.add(session.path);
+                let size = 0, mtime = asTime(session.modified);
+                try {
+                    const stat = statSync(session.path);
+                    size = stat.size;
+                    mtime = stat.mtimeMs;
+                }
+                catch { }
+                upsert.run(session.id, session.path, workspaceId, cwd, session.name ?? null, session.parentSessionPath ?? null, asTime(session.created), asTime(session.modified), session.messageCount, session.firstMessage, size, mtime);
             }
-            catch { }
-            upsert.run(session.id, session.path, workspaceId, cwd, session.name ?? null, session.parentSessionPath ?? null, asTime(session.created), asTime(session.modified), session.messageCount, session.firstMessage, size, mtime);
-        } for (const row of db.prepare('SELECT session_file FROM sessions WHERE cwd=? AND archived_at IS NULL').all(cwd))
-            if (!paths.has(row.session_file))
-                remove.run(row.session_file); })();
-    }
-    upsertSessionFromAgent(record, summary) { const sessionFile = typeof summary.sessionFile === 'string' ? summary.sessionFile : record.sessionFile; if (!sessionFile)
-        return; let size = 0, mtime = asTime(record.lastActiveAt); try {
-        const stat = statSync(sessionFile);
-        size = stat.size;
-        mtime = stat.mtimeMs;
-    }
-    catch {
-        return;
-    } const sessionId = typeof summary.sessionId === 'string' ? summary.sessionId : record.sessionId, name = typeof summary.sessionName === 'string' ? summary.sessionName : record.sessionName, messageCount = Number(summary.stats?.totalMessages) || 0; this.database.prepare(`INSERT INTO sessions(session_id,session_file,workspace_id,cwd,name,created_at,last_active_at,message_count,first_message,file_size,file_mtime,archived_at) VALUES(?,?,?,?,?,?,?,?,?,?,?,NULL) ON CONFLICT(session_id) DO UPDATE SET session_file=excluded.session_file,workspace_id=excluded.workspace_id,cwd=excluded.cwd,name=excluded.name,last_active_at=excluded.last_active_at,message_count=excluded.message_count,file_size=excluded.file_size,file_mtime=excluded.file_mtime,archived_at=NULL`).run(sessionId, sessionFile, record.workspaceId, record.cwd, name ?? null, asTime(record.createdAt), Math.max(asTime(record.lastActiveAt), mtime), messageCount, '', size, mtime); }
-    listSessions(cwd, limit, offset = 0) { const bounded = limit === undefined ? undefined : Math.min(200, Math.max(1, Math.trunc(limit))); const sql = `SELECT session_file AS path,session_id AS id,cwd,name,parent_session_path AS parentSessionPath,created_at,last_active_at,message_count AS messageCount,first_message AS firstMessage FROM sessions WHERE cwd=? AND archived_at IS NULL ORDER BY last_active_at DESC,session_id DESC${bounded === undefined ? '' : ' LIMIT ? OFFSET ?'}`; const rows = (bounded === undefined ? this.database.prepare(sql).all(cwd) : this.database.prepare(sql).all(cwd, bounded, Math.max(0, Math.trunc(offset)))); return rows.map(row => ({ ...row, created: new Date(row.created_at).toISOString(), modified: new Date(row.last_active_at).toISOString(), allMessagesText: '' })); }
-    hasSession(cwd, sessionFile) { return !!this.database.prepare('SELECT 1 FROM sessions WHERE cwd=? AND session_file=? AND archived_at IS NULL').get(cwd, sessionFile); }
-    close() { if (!this.db)
-        return; this.flush(); this.db.pragma('wal_checkpoint(TRUNCATE)'); this.db.close(); this.db = undefined; }
+            for (const row of db
+                .prepare('SELECT session_file FROM sessions WHERE cwd=? AND archived_at IS NULL')
+                .all(cwd))
+                if (!paths.has(row.session_file))
+                    remove.run(row.session_file);
+        })();
+    }
+    upsertSessionFromAgent(record, summary) {
+        const sessionFile = typeof summary.sessionFile === 'string' ? summary.sessionFile : record.sessionFile;
+        if (!sessionFile)
+            return;
+        let size = 0, mtime = asTime(record.lastActiveAt);
+        try {
+            const stat = statSync(sessionFile);
+            size = stat.size;
+            mtime = stat.mtimeMs;
+        }
+        catch {
+            return;
+        }
+        const sessionId = typeof summary.sessionId === 'string' ? summary.sessionId : record.sessionId, name = typeof summary.sessionName === 'string' ? summary.sessionName : record.sessionName, messageCount = Number(summary.stats?.totalMessages) || 0;
+        this.database
+            .prepare(`INSERT INTO sessions(session_id,session_file,workspace_id,cwd,name,created_at,last_active_at,message_count,first_message,file_size,file_mtime,archived_at) VALUES(?,?,?,?,?,?,?,?,?,?,?,NULL) ON CONFLICT(session_id) DO UPDATE SET session_file=excluded.session_file,workspace_id=excluded.workspace_id,cwd=excluded.cwd,name=excluded.name,last_active_at=excluded.last_active_at,message_count=excluded.message_count,file_size=excluded.file_size,file_mtime=excluded.file_mtime,archived_at=NULL`)
+            .run(sessionId, sessionFile, record.workspaceId, record.cwd, name ?? null, asTime(record.createdAt), Math.max(asTime(record.lastActiveAt), mtime), messageCount, '', size, mtime);
+    }
+    listSessions(cwd, limit, offset = 0) {
+        const bounded = limit === undefined ? undefined : Math.min(200, Math.max(1, Math.trunc(limit)));
+        const sql = `SELECT session_file AS path,session_id AS id,cwd,name,parent_session_path AS parentSessionPath,created_at,last_active_at,message_count AS messageCount,first_message AS firstMessage FROM sessions WHERE cwd=? AND archived_at IS NULL ORDER BY last_active_at DESC,session_id DESC${bounded === undefined ? '' : ' LIMIT ? OFFSET ?'}`;
+        const rows = (bounded === undefined
+            ? this.database.prepare(sql).all(cwd)
+            : this.database.prepare(sql).all(cwd, bounded, Math.max(0, Math.trunc(offset))));
+        return rows.map(row => ({
+            ...row,
+            created: new Date(row.created_at).toISOString(),
+            modified: new Date(row.last_active_at).toISOString(),
+            allMessagesText: ''
+        }));
+    }
+    hasSession(cwd, sessionFile) {
+        return !!this.database
+            .prepare('SELECT 1 FROM sessions WHERE cwd=? AND session_file=? AND archived_at IS NULL')
+            .get(cwd, sessionFile);
+    }
+    close() {
+        if (!this.db)
+            return;
+        this.flush();
+        this.db.pragma('wal_checkpoint(TRUNCATE)');
+        this.db.close();
+        this.db = undefined;
+    }
 }

package/dist/paths.js

@@ -0,0 +1,10 @@
+import { homedir } from 'node:os';
+import path from 'node:path';
+/**
+ * pi2web keeps its own state next to Pi's (`~/.pi/agent`): `~/.pi/pi2web` holds the metadata database,
+ * pairing-code hashes and the shareable Agent templates. `PI2WEB_DIR` or `--data-dir` override it.
+ */
+export const defaultDataDir = () => process.env.PI2WEB_DIR || path.join(homedir(), '.pi', 'pi2web');
+export const METADATA_DB_FILE = 'pi2web.db';
+/** Shareable Agent templates: one Markdown file per template, see `agent-templates.ts`. */
+export const AGENT_TEMPLATES_DIR = 'agent-templates';