@agentshouse/mdruntime 0.1.0-cd3486bd400ed → 0.1.0-cd37f80239af0

package/dist/memory.js

@@ -1,7 +1,11 @@
 import { sameChange } from './overlay.js';
+const ATTACHMENTS = 'attachments';
+function lastName(path) {
+    return path.slice(path.lastIndexOf('/') + 1);
+}
 const FIXTURE_RETENTION = {
     history: { kind: 'duration', seconds: 1 },
-    outcome: 1,
+    outcome: 604_800,
     refreshHorizon: 1,
 };
 function declaredRetention(value) {
@@ -15,12 +19,7 @@ function declaredRetention(value) {
 }
 const MODEL_ABSENT = { kind: 'model-unavailable', reason: 'not-readable' };
 function modelsOf(room) {
-    return {
-        collection: room.model ?? MODEL_ABSENT,
-        ...(room.typedSelection === undefined
-            ? {}
-            : { selection: room.selectionModel ?? MODEL_ABSENT }),
-    };
+    return { collection: room.model ?? MODEL_ABSENT };
 }
 function changedModels(room, from) {
     const now = modelsOf(room);
@@ -53,12 +52,16 @@ export function createMemoryAdapter(options = {}) {
     const listeners = new Set();
     const modeObservers = new Set();
     const scripted = new Map();
-    const unsealed = new Map();
-    const sealed = new Set();
+    const sentTurns = new Map();
     const historyPageSize = options.historyPageSize;
-    let held = [];
+    const attached = new Map();
+    const linked = new WeakMap();
+    const carried = new Map();
     let expiring = [];
+    let openings = 0;
+    const declaredRooms = new Set();
     let replicated = [];
+    let heldCursor = null;
     let issued = 0;
     function scriptConversations(entries) {
         scripted.clear();
@@ -86,6 +89,19 @@ export function createMemoryAdapter(options = {}) {
         const wanted = ordinals.get(`${locator}\u0000${target}`);
         return held !== undefined && wanted !== undefined && held >= wanted;
     }
+    function heldBlob(room, path) {
+        return attached.get(`${room.locator}\u0000${path}`) ?? room.blobs?.[path];
+    }
+    function freePath(room, name) {
+        const dot = name.lastIndexOf('.');
+        const stem = dot > 0 ? name.slice(0, dot) : name;
+        const extension = dot > 0 ? name.slice(dot) : '';
+        for (let count = 1;; count += 1) {
+            const path = `${ATTACHMENTS}/${count === 1 ? stem : `${stem}-${count}`}${extension}`;
+            if (heldBlob(room, path) === undefined && room.documents?.[path] === undefined)
+                return path;
+        }
+    }
     function retain(entries) {
         for (const room of entries) {
             note(room.locator, room);
@@ -114,30 +130,44 @@ export function createMemoryAdapter(options = {}) {
         const size = pageSize ?? paths.length;
         const taken = paths.slice(from, from + size);
         const next = from + taken.length;
-        const targets = opening ? (room.binaryTargets ?? []) : [];
+        const targets = opening ? Object.keys(room.binaryFiles ?? {}) : [];
         return {
             kind: 'bootstrap',
             roomVersion: room.roomVersion,
             ...(opening ? { models: modelsOf(room) } : {}),
             documents: named(room.documents ?? {}, taken),
-            revisions: named(room.revisions ?? {}, taken),
+            revisions: named(room.revisions ?? {}, [...taken, ...targets]),
+            ...(room.fileIds === undefined
+                ? {}
+                : { fileIds: named(room.fileIds, [...taken, ...targets]) }),
             changes: named(room.changes ?? {}, [...taken, ...targets]),
-            binaryTargets: targets,
-            ...(opening && room.attachments !== undefined ? { attachments: room.attachments } : {}),
+            binaryFiles: named(room.binaryFiles ?? {}, targets),
+            ...(opening && room.folders !== undefined ? { folders: room.folders } : {}),
             ...(next < paths.length ? { pageToken: taken.at(-1) } : {}),
         };
     }
-    function advancement(room, from, at) {
+    function foldersMoved(room, from) {
+        const now = new Set(room.folders ?? []);
+        const before = new Set(from.folders ?? []);
+        const folders = [...now].filter((folder) => !before.has(folder)).sort();
+        const folderRemovals = [...before].filter((folder) => !now.has(folder)).sort();
+        return {
+            ...(folders.length === 0 ? {} : { folders }),
+            ...(folderRemovals.length === 0 ? {} : { folderRemovals }),
+        };
+    }
+    function advancement(room, from, at, whole = false) {
         const current = room.documents ?? {};
         const previous = from.documents ?? {};
-        const currentTargets = new Set(room.binaryTargets ?? []);
-        const previousTargets = new Set(from.binaryTargets ?? []);
+        const currentTargets = new Set(Object.keys(room.binaryFiles ?? {}));
+        const previousTargets = new Set(Object.keys(from.binaryFiles ?? {}));
         const headMoved = (path) => room.revisions?.[path] !== from.revisions?.[path] ||
+            room.fileIds?.[path] !== from.fileIds?.[path] ||
             !sameChange(room.changes?.[path], from.changes?.[path]);
         const changed = Object.keys(current)
             .filter((path) => current[path] !== previous[path] || headMoved(path))
             .sort();
-        const size = pageSize ?? changed.length;
+        const size = whole ? changed.length : (pageSize ?? changed.length);
         const taken = changed.slice(at, at + size);
         const next = at + taken.length;
         const opening = at === 0;
@@ -147,27 +177,48 @@ export function createMemoryAdapter(options = {}) {
         const targets = opening
             ? [...currentTargets].filter((target) => !previousTargets.has(target) || headMoved(target))
             : [];
-        const attachments = room.attachments ?? [];
-        const declared = opening && JSON.stringify(attachments) !== JSON.stringify(from.attachments ?? []);
         const models = opening ? changedModels(room, from) : undefined;
         return {
             kind: 'net-advancement',
             roomVersion: room.roomVersion,
             ...(models === undefined ? {} : { models }),
             documents: named(current, taken),
-            revisions: named(room.revisions ?? {}, taken),
+            revisions: named(room.revisions ?? {}, [...taken, ...targets]),
+            ...(room.fileIds === undefined
+                ? {}
+                : { fileIds: named(room.fileIds, [...taken, ...targets]) }),
             changes: named(room.changes ?? {}, [...taken, ...targets]),
-            binaryTargets: targets,
+            binaryFiles: named(room.binaryFiles ?? {}, targets),
+            ...(opening ? foldersMoved(room, from) : {}),
             removals: opening
                 ? [
                     ...Object.keys(previous).filter((path) => !(path in current) && !currentTargets.has(path)),
                     ...[...previousTargets].filter((target) => !currentTargets.has(target) && !(target in current)),
                 ].sort()
                 : [],
-            ...(declared ? { attachments } : {}),
             ...(pageToken === undefined ? {} : { pageToken }),
         };
     }
+    function carry(locator) {
+        const at = carried.get(locator);
+        const current = rooms.find((entry) => entry.locator === locator);
+        if (at === undefined || current === undefined || current.roomVersion === at)
+            return;
+        const from = served.get(locator)?.get(at);
+        carried.set(locator, current.roomVersion);
+        if (from === undefined)
+            return;
+        const page = advancement(current, from, 0, true);
+        for (const observer of observers) {
+            observer({
+                kind: 'room-change',
+                locator,
+                from: at,
+                roomVersion: current.roomVersion,
+                read: async () => page,
+            });
+        }
+    }
     function pageOf(conversation, after) {
         const messages = conversation.messages ?? [];
         const first = typeof after === 'string' ? messages.findIndex((message) => message.position > after) : 0;
@@ -182,16 +233,13 @@ export function createMemoryAdapter(options = {}) {
                 : {}),
         };
     }
-    function historiesOf(input) {
-        const asked = input?.conversations;
-        if (!Array.isArray(asked))
-            return { kind: 'rejected', code: 'invalid_request', fields: {}, retryable: false };
+    function historiesOf(request) {
         const histories = [];
-        for (const entry of asked) {
-            const conversation = typeof entry.conversation === 'string' ? scripted.get(entry.conversation) : undefined;
+        for (const entry of request.conversations) {
+            const conversation = scripted.get(entry.conversation);
             if (conversation === undefined)
                 continue;
-            const outcome = conversation.outcomes?.['conversation.history'];
+            const outcome = conversation.outcomes?.history;
             if (outcome !== undefined)
                 return outcome;
             histories.push(pageOf(conversation, entry.after));
@@ -202,15 +250,13 @@ export function createMemoryAdapter(options = {}) {
     function conversationAction(request) {
         const name = request.name;
         const input = (request.input ?? {});
-        if (name === 'conversation.history')
-            return historiesOf(request.input);
-        if (name === 'conversation.create') {
+        if (name === 'conversations/create') {
             issued += 1;
             const conversation = `conversation-${issued}`;
             scripted.set(conversation, { conversation });
-            return { kind: 'action', result: { conversation } };
+            return { kind: 'action', result: { conversation_id: conversation } };
         }
-        const named = typeof input.conversation === 'string' ? input.conversation : '';
+        const named = typeof input.conversation_id === 'string' ? input.conversation_id : '';
         const conversation = scripted.get(named);
         if (conversation === undefined) {
             return {
@@ -223,28 +269,28 @@ export function createMemoryAdapter(options = {}) {
         const outcome = conversation.outcomes?.[name];
         if (outcome !== undefined)
             return outcome;
-        if (name === 'conversation.send') {
+        if (name === 'conversations/send') {
+            const operation = typeof input.operation_id === 'string' ? input.operation_id : null;
+            const answered = operation === null ? undefined : sentTurns.get(operation);
+            if (answered !== undefined)
+                return { kind: 'action', result: { message_id: answered } };
             issued += 1;
             const turn = `turn-${issued}`;
-            unsealed.set(turn, named);
-            return { kind: 'action', result: { turn } };
+            if (operation !== null)
+                sentTurns.set(operation, turn);
+            return { kind: 'action', result: { message_id: turn } };
         }
-        if (name === 'conversation.cancel') {
-            const turn = typeof input.turn === 'string' ? input.turn : '';
-            if (sealed.has(turn) || !unsealed.has(turn)) {
-                return { kind: 'rejected', code: 'turn-sealed', fields: { turn }, retryable: false };
-            }
-            unsealed.delete(turn);
-            return { kind: 'action', result: { turn, cancelled: true } };
-        }
-        if (name === 'conversation.delete')
+        if (name === 'conversations/delete')
             scripted.delete(named);
-        return { kind: 'action', result: { conversation: named } };
+        return { kind: 'action', result: { conversation_id: named } };
     }
     function announceList() {
-        const entries = [...scripted.values()].flatMap(({ conversation, listed }) => listed === undefined ? [] : [{ conversation, ...listed }]);
-        for (const listener of listeners)
-            listener({ kind: 'list', entries });
+        const entries = [...scripted.values()].flatMap(({ conversation, listed, messages = [] }) => listed === undefined
+            ? []
+            : [{ conversation, ...listed, newest: messages.at(-1)?.position ?? null }]);
+        for (const listener of listeners) {
+            listener({ kind: 'list', entries, whole: true, more: false, cursor: null });
+        }
     }
     function discovery() {
         return {
@@ -256,7 +302,41 @@ export function createMemoryAdapter(options = {}) {
             ...(options.member === undefined ? {} : { member: options.member }),
         };
     }
+    function discovered() {
+        return resolveScript(scripts.discover, discovery, undefined).then((answer) => {
+            if (answer.kind === 'discovery')
+                for (const room of answer.rooms)
+                    note(room.locator, room);
+            return answer;
+        });
+    }
+    function opened(notify) {
+        const opening = ++openings;
+        notify({ kind: 'opened' });
+        void discovered()
+            .catch(() => ({ kind: 'transport-uncertainty' }))
+            .then((answer) => {
+            if (opening !== openings)
+                return;
+            if (answer.kind !== 'discovery') {
+                openings += 1;
+                carried.clear();
+                const revoked = answer.kind === 'revoked-authorization';
+                notify({ kind: 'closed', revoked, lost: !revoked });
+                return;
+            }
+            declaredRooms.clear();
+            for (const room of answer.rooms)
+                declaredRooms.add(room.locator);
+            notify({ kind: 'discovered', discovery: answer });
+        });
+    }
+    function emitAll(event) {
+        for (const observer of observers)
+            observer(event);
+    }
     return {
+        ...(options.roomTypes === undefined ? {} : { roomTypes: options.roomTypes }),
         setRooms(next) {
             rooms = [...next];
             retain(rooms);
@@ -265,26 +345,37 @@ export function createMemoryAdapter(options = {}) {
             scriptConversations(next);
         },
         open() {
+            carried.clear();
             announceList();
-            for (const observer of observers)
-                observer({ kind: 'opened' });
+            opened(emitAll);
         },
-        close(revoked = false) {
-            for (const observer of observers)
-                observer({ kind: 'closed', revoked });
+        close({ revoked = false, lost = !revoked } = {}) {
+            openings += 1;
+            carried.clear();
+            emitAll({ kind: 'closed', revoked, lost });
         },
         expire(locators) {
             expiring = [...locators];
         },
-        notifyRoomVersion({ locator, roomVersion }) {
-            for (const observer of observers)
-                observer({ kind: 'room-version', locator, roomVersion });
+        notifyRoomVersion({ locator }) {
+            carry(locator);
+        },
+        notifyAuthorityChanged(locators) {
+            const named = new Set(locators ?? [...declaredRooms, ...rooms.map((room) => room.locator)]);
+            const listed = rooms.filter((room) => named.has(room.locator));
+            const left = [...named].filter((locator) => !listed.some((room) => room.locator === locator));
+            for (const room of listed)
+                declaredRooms.add(room.locator);
+            for (const locator of left)
+                declaredRooms.delete(locator);
+            emitAll({
+                kind: 'authority-changed',
+                rooms: listed.map(declared),
+                left,
+                ...(options.member === undefined ? {} : { member: options.member }),
+            });
         },
         notifyConversation(event) {
-            if (event.kind === 'accepted-message') {
-                sealed.add(event.message.turn);
-                unsealed.delete(event.message.turn);
-            }
             for (const listener of listeners)
                 listener(event);
         },
@@ -293,21 +384,31 @@ export function createMemoryAdapter(options = {}) {
                 observer(mode);
         },
         heldRooms() {
-            return held;
+            return [...carried.keys()].sort();
         },
         replicatedConversations() {
             return replicated;
         },
+        heldConversationCursor() {
+            return heldCursor;
+        },
         conversations: {
             replicate(next) {
                 replicated = [...next];
             },
+            hold(cursor) {
+                heldCursor = cursor;
+            },
             observe(notify) {
                 listeners.add(notify);
                 return () => {
                     listeners.delete(notify);
                 };
             },
+            history: (request) => Promise.resolve(historiesOf(request)),
+            ...(options.agents === undefined
+                ? {}
+                : { agents: () => Promise.resolve(structuredClone(options.agents)) }),
         },
         serviceMode: {
             observe(notify) {
@@ -319,29 +420,29 @@ export function createMemoryAdapter(options = {}) {
         },
         roomStream: {
             hold(asked) {
-                held = asked.map((room) => room.locator);
-                const expired = expiring.filter((locator) => held.includes(locator));
-                expiring = [];
-                const moved = asked.flatMap((room) => {
+                const named = asked.map((room) => room.locator);
+                const unknown = asked.flatMap((room) => {
                     const current = rooms.find((entry) => entry.locator === room.locator);
                     return current === undefined ||
-                        expired.includes(room.locator) ||
-                        current.roomVersion === room.roomVersion
+                        current.roomVersion === room.roomVersion ||
+                        served.get(room.locator)?.has(room.roomVersion) === true
                         ? []
-                        : [{ locator: room.locator, roomVersion: current.roomVersion }];
+                        : [room.locator];
                 });
+                const expired = [
+                    ...new Set([...expiring.filter((locator) => named.includes(locator)), ...unknown]),
+                ];
+                expiring = expiring.filter((locator) => !named.includes(locator));
                 queueMicrotask(() => {
-                    for (const observer of observers) {
-                        for (const room of moved)
-                            observer({ kind: 'room-version', ...room });
-                        observer({ kind: 'opening', expired });
+                    for (const room of asked) {
+                        if (expired.includes(room.locator))
+                            carried.delete(room.locator);
+                        else {
+                            carried.set(room.locator, room.roomVersion);
+                            carry(room.locator);
+                        }
                     }
-                });
-            },
-            reopen() {
-                queueMicrotask(() => {
-                    for (const observer of observers)
-                        observer({ kind: 'closed', revoked: false });
+                    emitAll({ kind: 'caught-up', expired });
                 });
             },
             observe(notify) {
@@ -350,11 +451,14 @@ export function createMemoryAdapter(options = {}) {
                     if (!observers.has(notify))
                         return;
                     if (options.unopened === true) {
-                        notify({ kind: 'closed', revoked: false });
+                        notify({ kind: 'closed', revoked: false, lost: true });
                         return;
                     }
                     announceList();
-                    notify({ kind: 'opened' });
+                    opened((event) => {
+                        if (observers.has(notify))
+                            notify(event);
+                    });
                 });
                 return () => {
                     observers.delete(notify);
@@ -371,12 +475,7 @@ export function createMemoryAdapter(options = {}) {
             return resolveScript(scripts.authorize, () => ({}), undefined);
         },
         async discover(_authorization) {
-            return resolveScript(scripts.discover, discovery, undefined).then((answer) => {
-                if (answer.kind === 'discovery')
-                    for (const room of answer.rooms)
-                        note(room.locator, room);
-                return answer;
-            });
+            return discovered();
         },
         reached,
         follows(locator, roomVersion, next) {
@@ -433,7 +532,13 @@ export function createMemoryAdapter(options = {}) {
                     retryable: false,
                     affected: {},
                 };
-            }, request).then((answer) => note(request.locator, answer));
+            }, request).then((answer) => {
+                const destination = request.changeBatch.between?.destination.room;
+                if (answer.kind === 'accepted' && answer.destination !== undefined && destination) {
+                    note(destination, answer.destination);
+                }
+                return note(request.locator, answer);
+            });
         },
         async history(_authorization, request) {
             return resolveScript(scripts.history, () => {
@@ -447,9 +552,44 @@ export function createMemoryAdapter(options = {}) {
                 };
             }, request);
         },
+        async binary(_authorization, request) {
+            return resolveScript(scripts.binary, () => {
+                const room = rooms.find((entry) => entry.locator === request.locator);
+                if (room === undefined)
+                    return { kind: 'room-not-readable', locator: request.locator };
+                const blob = heldBlob(room, request.path);
+                return blob === undefined
+                    ? { kind: 'rejected', code: 'file_not_found', fields: {}, retryable: false }
+                    : { kind: 'binary', body: blob.stream(), mediaType: blob.type };
+            }, request);
+        },
+        async link(_authorization, request) {
+            return resolveScript(scripts.link, () => {
+                const room = rooms.find((entry) => entry.locator === request.locator);
+                if (room === undefined)
+                    return { kind: 'room-not-readable', locator: request.locator };
+                const blob = heldBlob(room, request.path);
+                if (blob === undefined) {
+                    return { kind: 'rejected', code: 'file_not_found', fields: {}, retryable: false };
+                }
+                const url = linked.get(blob) ?? URL.createObjectURL(blob);
+                linked.set(blob, url);
+                return { kind: 'link', url, name: lastName(request.path), mediaType: blob.type };
+            }, request);
+        },
+        async attach(_authorization, request) {
+            return resolveScript(scripts.attach, () => {
+                const room = rooms.find((entry) => entry.locator === request.locator);
+                if (room === undefined)
+                    return { kind: 'room-not-readable', locator: request.locator };
+                const path = freePath(room, request.name);
+                attached.set(`${room.locator}\u0000${path}`, request.blob);
+                return { kind: 'attached', path, revision: `attached-${attached.size}` };
+            }, request);
+        },
         async action(_authorization, request) {
             return resolveScript(scripts.action, () => {
-                if (request.name.startsWith('conversation.'))
+                if (request.name.startsWith('conversations/'))
                     return conversationAction(request);
                 return { kind: 'rejected', code: 'unscripted-action', fields: {}, retryable: false };
             }, request).then((answer) => {

package/dist/messages.d.ts

@@ -0,0 +1,28 @@
+import type { AcceptedMessage } from './adapter.js';
+import type { Connection } from './tables.js';
+export declare const MESSAGE_TABLES: readonly string[];
+export type HeldConversation = {
+    readonly conversation: string;
+    readonly read: string | null | undefined;
+    readonly partial: boolean;
+};
+export type MessageWindow = {
+    readonly messages: AcceptedMessage[];
+    readonly more: boolean;
+};
+export type ConversationWrite = {
+    readonly conversation: string;
+    readonly messages: readonly AcceptedMessage[];
+    readonly read?: {
+        readonly at: string | null;
+        readonly partial: boolean;
+    };
+};
+export declare function heldConversations(connection: Connection): HeldConversation[];
+export declare function keptMessages(connection: Connection, { conversation, messages, read }: ConversationWrite): null;
+export declare function messageWindow(connection: Connection, conversation: string, limit: number, before: readonly [string, string] | null): MessageWindow;
+export declare function messageCount(connection: Connection, conversation: string): number;
+export declare function turnArrived(connection: Connection, conversation: string, turn: string): boolean;
+export declare function droppedMessages(connection: Connection, conversations: readonly string[]): void;
+export declare function forgottenMessages(connection: Connection): void;
+//# sourceMappingURL=messages.d.ts.map

package/dist/messages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../src/messages.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAEpD,OAAO,KAAK,EAAE,UAAU,EAAc,MAAM,aAAa,CAAC;AAE1D,eAAO,MAAM,cAAc,EAAE,SAAS,MAAM,EAM3C,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IACzC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAAE,QAAQ,CAAC,QAAQ,EAAE,eAAe,EAAE,CAAC;IAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAA;CAAE,CAAC;AAE7F,MAAM,MAAM,iBAAiB,GAAG;IAC9B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,QAAQ,EAAE,SAAS,eAAe,EAAE,CAAC;IAC9C,QAAQ,CAAC,IAAI,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;CAC3E,CAAC;AAMF,wBAAgB,iBAAiB,CAAC,UAAU,EAAE,UAAU,GAAG,gBAAgB,EAAE,CAmB5E;AAED,wBAAgB,YAAY,CAC1B,UAAU,EAAE,UAAU,EACtB,EAAE,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,iBAAiB,GAClD,IAAI,CAoCN;AAYD,wBAAgB,aAAa,CAC3B,UAAU,EAAE,UAAU,EACtB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,GACvC,aAAa,CAUf;AAED,wBAAgB,YAAY,CAAC,UAAU,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,GAAG,MAAM,CAMjF;AAED,wBAAgB,WAAW,CAAC,UAAU,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAO/F;AAED,wBAAgB,eAAe,CAAC,UAAU,EAAE,UAAU,EAAE,aAAa,EAAE,SAAS,MAAM,EAAE,GAAG,IAAI,CAW9F;AAED,wBAAgB,iBAAiB,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI,CAG9D"}

package/dist/messages.js

@@ -0,0 +1,90 @@
+import { indexMessage, unindexMessages } from './fulltext.js';
+export const MESSAGE_TABLES = [
+    'CREATE TABLE conversation_reads (conversation TEXT PRIMARY KEY, read TEXT, partial INTEGER NOT NULL) WITHOUT ROWID',
+    'CREATE TABLE conversation_messages (id INTEGER PRIMARY KEY, conversation TEXT NOT NULL, identity TEXT NOT NULL, turn TEXT NOT NULL, position TEXT NOT NULL, message TEXT NOT NULL, text TEXT NOT NULL)',
+    'CREATE UNIQUE INDEX conversation_messages_identity ON conversation_messages (conversation, identity)',
+    'CREATE INDEX conversation_messages_position ON conversation_messages (conversation, position, identity)',
+    'CREATE INDEX conversation_messages_turn ON conversation_messages (conversation, turn)',
+];
+function text(value) {
+    return typeof value === 'string' ? value : String(value);
+}
+export function heldConversations(connection) {
+    const held = new Map();
+    for (const row of connection.rows('SELECT conversation, read, partial FROM conversation_reads')) {
+        held.set(text(row.conversation), {
+            conversation: text(row.conversation),
+            read: row.read === null ? null : text(row.read),
+            partial: Number(row.partial) === 1,
+        });
+    }
+    for (const row of connection.rows('SELECT DISTINCT conversation FROM conversation_messages WHERE conversation NOT IN (SELECT conversation FROM conversation_reads)')) {
+        held.set(text(row.conversation), {
+            conversation: text(row.conversation),
+            read: undefined,
+            partial: true,
+        });
+    }
+    return [...held.values()];
+}
+export function keptMessages(connection, { conversation, messages, read }) {
+    for (const message of messages) {
+        const [held] = connection.rows('SELECT id FROM conversation_messages WHERE conversation = ? AND identity = ?', [conversation, message.identity]);
+        const record = [
+            message.turn,
+            message.position,
+            JSON.stringify(message.message ?? null),
+            message.text,
+        ];
+        let id;
+        if (held === undefined) {
+            connection.run('INSERT INTO conversation_messages (conversation, identity, turn, position, message, text) VALUES (?, ?, ?, ?, ?, ?)', [conversation, message.identity, ...record]);
+            id = Number(connection.rows('SELECT last_insert_rowid() AS id')[0].id);
+        }
+        else {
+            id = Number(held.id);
+            unindexMessages(connection, [id]);
+            connection.run('UPDATE conversation_messages SET turn = ?, position = ?, message = ?, text = ? WHERE id = ?', [...record, id]);
+        }
+        indexMessage(connection, id, message.text);
+    }
+    if (read !== undefined) {
+        connection.run('INSERT OR REPLACE INTO conversation_reads (conversation, read, partial) VALUES (?, ?, ?)', [conversation, read.at, read.partial ? 1 : 0]);
+    }
+    return null;
+}
+function accepted(row) {
+    return {
+        identity: text(row.identity),
+        turn: text(row.turn),
+        position: text(row.position),
+        text: text(row.text),
+        message: JSON.parse(text(row.message)),
+    };
+}
+export function messageWindow(connection, conversation, limit, before) {
+    const rows = connection.rows(`SELECT identity, turn, position, message, text FROM conversation_messages WHERE conversation = ?${before === null ? '' : ' AND (position < ? OR (position = ? AND identity < ?))'} ORDER BY position DESC, identity DESC LIMIT ?`, before === null
+        ? [conversation, limit + 1]
+        : [conversation, before[0], before[0], before[1], limit + 1]);
+    return { messages: rows.slice(0, limit).map(accepted).reverse(), more: rows.length > limit };
+}
+export function messageCount(connection, conversation) {
+    const [counted] = connection.rows('SELECT count(*) AS held FROM conversation_messages WHERE conversation = ?', [conversation]);
+    return Number(counted?.held ?? 0);
+}
+export function turnArrived(connection, conversation, turn) {
+    return (connection.rows('SELECT 1 FROM conversation_messages WHERE conversation = ? AND turn = ? LIMIT 1', [conversation, turn]).length > 0);
+}
+export function droppedMessages(connection, conversations) {
+    for (const conversation of conversations) {
+        unindexMessages(connection, connection
+            .rows('SELECT id FROM conversation_messages WHERE conversation = ?', [conversation])
+            .map((row) => Number(row.id)));
+        connection.run('DELETE FROM conversation_messages WHERE conversation = ?', [conversation]);
+        connection.run('DELETE FROM conversation_reads WHERE conversation = ?', [conversation]);
+    }
+}
+export function forgottenMessages(connection) {
+    connection.run('DELETE FROM conversation_messages');
+    connection.run('DELETE FROM conversation_reads');
+}

package/dist/model.d.ts

@@ -28,13 +28,19 @@ export type ModelState = {
     verifiedWith: ModelIdentity;
     types: VerifiedModel['types'];
 };
-export type TypesResult = {
+export type StoredModel = Extract<ModelState, {
     status: 'unavailable';
+}> | Omit<Extract<ModelState, {
+    status: 'current' | 'cached';
+}>, 'types'>;
+export type TypesResult = {
+    status: Open<'unavailable'>;
     reason: ModelUnavailableReason;
+    types: null;
     schema?: string | null;
     revision?: string | null;
 } | {
-    status: 'current' | 'cached';
+    status: Open<'current' | 'cached'>;
     identity: string;
     version: string;
     verifiedWith: ModelIdentity;
@@ -50,6 +56,7 @@ export type ValidationReceipt = {
 export declare const NO_MODEL: ModelState;
 export declare function definitionOf(state: ModelState, type: string | undefined): Definition | undefined;
 export declare function modelTypes(state: ModelState): TypesResult;
+export declare function storedModel(state: ModelState): StoredModel;
 export declare function cached(state: ModelState): ModelState;
 export declare function sameModel(left: ModelIdentity, right: ModelIdentity): boolean;
 export declare function referenceField(facts: ValueFacts | undefined): boolean;