tinker-agent 2.12.0 → 3.0.0

package/src/remote/service-store.ts

@@ -1,6 +1,7 @@
 import { Database } from "bun:sqlite";
 import { chmod, mkdir } from "node:fs/promises";
 import path from "node:path";
+import type { RemoteWorkspaceConfig } from "./config";
 import { stableJsonStringify, sha256 } from "../model/model-request-preflight";
 import { parseSessionId } from "../ids/runtime-id";
 import { SessionLease } from "../session/session-lock";
@@ -14,6 +15,8 @@ import {
 export type ManagedSessionRecord = RemoteSessionInfo & {
   workspacePath: string;
   initialized: boolean;
+  profileName?: string;
+  residentState?: { reasoning?: string; yolo?: boolean };
 };
 type ReceiptRow = {
   device: string;
@@ -21,7 +24,7 @@ type ReceiptRow = {
   receipt: string;
   input: string;
 };
-const SERVICE_LEASE_ID = parseSessionId("00000000-0000-7000-8000-000000000001");
+export const SERVICE_LEASE_ID = parseSessionId("00000000-0000-7000-8000-000000000001");
 
 /** Durable acceptance receipts, separate from the canonical conversation databases. */
 export class RemoteServiceStore {
@@ -48,12 +51,13 @@ export class RemoteServiceStore {
       const version = (
         db.query("PRAGMA user_version").get() as { user_version: number }
       ).user_version;
-      if (version !== 0 && version !== 1)
+      if (version !== 0 && version !== 1 && version !== 2)
         throw new Error("Unsupported remote state schema.");
       db.exec(`CREATE TABLE IF NOT EXISTS managed_sessions (id TEXT PRIMARY KEY, record TEXT NOT NULL) STRICT;
         CREATE TABLE IF NOT EXISTS operations (id TEXT PRIMARY KEY, session_id TEXT NOT NULL, device TEXT NOT NULL, fingerprint TEXT NOT NULL, input TEXT NOT NULL, receipt TEXT NOT NULL) STRICT;
         CREATE INDEX IF NOT EXISTS operations_session ON operations(session_id);
-        PRAGMA user_version=1;`);
+        CREATE TABLE IF NOT EXISTS registered_workspaces (id TEXT PRIMARY KEY, path TEXT NOT NULL UNIQUE, record TEXT NOT NULL) STRICT;
+        PRAGMA user_version=2;`);
       const store = new RemoteServiceStore(db, lease);
       for (const row of db.query("SELECT receipt FROM operations").all() as {
         receipt: string;
@@ -81,6 +85,18 @@ export class RemoteServiceStore {
       this.db.query("SELECT record FROM managed_sessions").all() as { record: string }[]
     ).map((row) => JSON.parse(row.record) as ManagedSessionRecord);
   }
+  workspaces(): RemoteWorkspaceConfig[] {
+    return (
+      this.db.query("SELECT record FROM registered_workspaces ORDER BY id").all() as {
+        record: string;
+      }[]
+    ).map((row) => JSON.parse(row.record) as RemoteWorkspaceConfig);
+  }
+  registerWorkspace(workspace: RemoteWorkspaceConfig): void {
+    this.db
+      .query("INSERT INTO registered_workspaces (id, path, record) VALUES (?, ?, ?)")
+      .run(workspace.id, workspace.path, JSON.stringify(workspace));
+  }
   session(id: string): ManagedSessionRecord | undefined {
     const row = this.db
       .query("SELECT record FROM managed_sessions WHERE id = ?")
@@ -94,6 +110,9 @@ export class RemoteServiceStore {
       )
       .run(record.id, JSON.stringify(record));
   }
+  releaseSession(id: string): void {
+    this.db.query("DELETE FROM managed_sessions WHERE id = ?").run(id);
+  }
   existing(input: RemoteOperationInput, device: string): OperationReceipt | undefined {
     const row = this.db
       .query("SELECT * FROM operations WHERE id = ?")
@@ -121,7 +140,9 @@ export class RemoteServiceStore {
       status: "accepted",
       createdAt: now,
       updatedAt: now,
-      ...(input.kind === "prompt" ? { prompt: input.prompt } : {}),
+      ...(input.kind === "prompt"
+        ? { prompt: input.prompt, attachments: input.attachments }
+        : {}),
     };
     this.db.transaction(() => {
       this.db
@@ -150,11 +171,12 @@ export class RemoteServiceStore {
       );
     return JSON.parse(row.receipt) as OperationReceipt;
   }
-  operations(sessionId: string): OperationReceipt[] {
+  operations(sessionId: string, includeResults = true): OperationReceipt[] {
     return (
       this.db
         .query(
-          "SELECT receipt FROM operations WHERE session_id = ? ORDER BY rowid DESC LIMIT 100",
+          `SELECT ${includeResults ? "receipt" : "json_remove(receipt, '$.result', '$.sessionResult') AS receipt"}
+          FROM operations WHERE session_id = ? ORDER BY rowid DESC LIMIT 100`,
         )
         .all(sessionId) as { receipt: string }[]
     )