@agentshouse/mdruntime 0.1.0-ce531b51ff088 → 0.1.0-ce6bc9239a3ee

package/dist/memory.js

@@ -1,9 +1,12 @@
-import { ROOM_CONTRACT_VERSION, } from './adapter.js';
-import { MD_MODEL } from './collection.js';
+import { documentFacts, parse } from '@agentshouse/mdmodel';
+import { NO_ROOM_TYPES, schemaTypes } from './attached-model.js';
 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,
     refreshHorizon: 1,
 };
 function declaredRetention(value) {
@@ -15,6 +18,14 @@ function declaredRetention(value) {
     }
     throw new Error('history retention duration must be a finite positive number of seconds');
 }
+const MODEL_ABSENT = { kind: 'model-unavailable', reason: 'not-readable' };
+function modelsOf(room) {
+    return { collection: room.model ?? MODEL_ABSENT };
+}
+function changedModels(room, from) {
+    const now = modelsOf(room);
+    return JSON.stringify(now) === JSON.stringify(modelsOf(from)) ? undefined : now;
+}
 function named(source, paths) {
     const taken = {};
     for (const path of paths) {
@@ -31,8 +42,6 @@ async function resolveScript(answer, fallback, request) {
 }
 export function createMemoryAdapter(options = {}) {
     let rooms = options.rooms === undefined ? [] : [...options.rooms];
-    const mdModel = options.mdModel ?? MD_MODEL;
-    const supportedVersions = options.supportedVersions ?? [ROOM_CONTRACT_VERSION];
     const capabilities = options.capabilities ?? [];
     const providerLimits = options.providerLimits ?? [];
     const retention = declaredRetention(options.retention);
@@ -44,11 +53,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 selected = [];
+    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();
@@ -76,8 +90,48 @@ 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;
+        }
+    }
+    const saved = new WeakMap();
+    function factsAt(room, path, content) {
+        const types = schemaTypes(room.model?.kind === 'model' ? room.model.snapshot.schema : null, NO_ROOM_TYPES);
+        let held = saved.get(types);
+        if (held === undefined) {
+            held = new Map();
+            saved.set(types, held);
+        }
+        const key = `${path}\u0000${content}`;
+        if (!held.has(key)) {
+            const facts = documentFacts(path, parse(content, path), { types });
+            held.set(key, facts === null ? null : JSON.parse(JSON.stringify(facts)));
+        }
+        return held.get(key);
+    }
+    function factsIn(room, paths) {
+        const facts = {};
+        for (const path of paths) {
+            const content = room.documents?.[path];
+            const held = room.facts?.[path] ?? (content === undefined ? null : factsAt(room, path, content));
+            if (held !== null)
+                facts[path] = held;
+        }
+        return facts;
+    }
     function retain(entries) {
         for (const room of entries) {
+            for (const [path, content] of Object.entries(room.documents ?? {}))
+                factsAt(room, path, content);
             note(room.locator, room);
             const versions = served.get(room.locator) ?? new Map();
             versions.set(room.roomVersion, room);
@@ -89,42 +143,64 @@ export function createMemoryAdapter(options = {}) {
         return {
             name: room.name,
             locator: room.locator,
+            handle: room.handle,
             roomVersion: room.roomVersion,
             operations: room.operations,
             replica: room.replica,
-            attachedModel: room.attachedModel,
             ...(room.typedSelection === undefined ? {} : { typedSelection: room.typedSelection }),
             providerOwnedPaths: room.providerOwnedPaths,
         };
     }
-    function page(room, from) {
+    function page(room, after) {
         const paths = Object.keys(room.documents ?? {}).sort();
+        const opening = after === undefined;
+        const from = opening ? 0 : paths.filter((path) => path <= after).length;
         const size = pageSize ?? paths.length;
         const taken = paths.slice(from, from + size);
         const next = from + taken.length;
-        const targets = from === 0 ? (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),
+            facts: factsIn(room, taken),
+            revisions: named(room.revisions ?? {}, [...taken, ...targets]),
+            ...(room.fileIds === undefined
+                ? {}
+                : { fileIds: named(room.fileIds, [...taken, ...targets]) }),
+            ...(room.created === undefined
+                ? {}
+                : { created: named(room.created, [...taken, ...targets]) }),
             changes: named(room.changes ?? {}, [...taken, ...targets]),
-            binaryTargets: targets,
-            ...(from === 0 && room.attachments !== undefined ? { attachments: room.attachments } : {}),
-            ...(next < paths.length ? { pageToken: String(next) } : {}),
+            binaryFiles: named(room.binaryFiles ?? {}, targets),
+            ...(opening && room.folders !== undefined ? { folders: room.folders } : {}),
+            ...(next < paths.length ? { pageToken: taken.at(-1) } : {}),
+        };
+    }
+    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) {
+    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] ||
+            room.created?.[path] !== from.created?.[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;
@@ -134,25 +210,52 @@ 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),
+            facts: factsIn(room, taken),
+            revisions: named(room.revisions ?? {}, [...taken, ...targets]),
+            ...(room.fileIds === undefined
+                ? {}
+                : { fileIds: named(room.fileIds, [...taken, ...targets]) }),
+            ...(room.created === undefined
+                ? {}
+                : { created: named(room.created, [...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;
@@ -167,16 +270,13 @@ export function createMemoryAdapter(options = {}) {
                 : {}),
         };
     }
-    function historiesOf(input) {
-        const asked = input?.conversations;
-        if (!Array.isArray(asked))
-            return { kind: 'rejected', code: 'invalid_request', fields: {} };
+    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));
@@ -187,52 +287,94 @@ 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 { kind: 'rejected', code: 'unknown-conversation', fields: { conversation: named } };
+            return {
+                kind: 'rejected',
+                code: 'unknown-conversation',
+                fields: { conversation: named },
+                retryable: false,
+            };
         }
         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 (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 } };
-            }
-            unsealed.delete(turn);
-            return { kind: 'action', result: { turn, cancelled: true } };
+            if (operation !== null)
+                sentTurns.set(operation, turn);
+            return { kind: 'action', result: { message_id: turn } };
         }
-        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, 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 {
             kind: 'discovery',
-            supportedVersions,
-            mdModel,
             rooms: rooms.map(declared),
             capabilities,
             providerLimits,
             retention,
+            ...(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, authorityChanged: false });
+                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 }),
+        ...(options.typedPaths === undefined ? {} : { typedPaths: options.typedPaths }),
         setRooms(next) {
             rooms = [...next];
             retain(rooms);
@@ -240,15 +382,38 @@ export function createMemoryAdapter(options = {}) {
         setConversations(next) {
             scriptConversations(next);
         },
-        notifyRoomVersion(notification) {
-            for (const observer of observers)
-                observer(notification);
+        open() {
+            carried.clear();
+            announceList();
+            opened(emitAll);
+        },
+        close({ revoked = false, lost = !revoked, authorityChanged = false } = {}) {
+            openings += 1;
+            carried.clear();
+            emitAll({ kind: 'closed', revoked, lost, authorityChanged });
+        },
+        expire(locators) {
+            expiring = [...locators];
+        },
+        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);
         },
@@ -256,22 +421,32 @@ export function createMemoryAdapter(options = {}) {
             for (const observer of modeObservers)
                 observer(mode);
         },
-        selectedRooms() {
-            return selected;
+        heldRooms() {
+            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) {
@@ -281,12 +456,49 @@ export function createMemoryAdapter(options = {}) {
                 };
             },
         },
-        roomVersions: {
-            select(locators) {
-                selected = [...locators];
+        roomStream: {
+            hold(asked) {
+                const named = asked.map((room) => room.locator);
+                const unknown = asked.flatMap((room) => {
+                    const current = rooms.find((entry) => entry.locator === room.locator);
+                    const version = room.roomVersion;
+                    return current === undefined ||
+                        (version !== null &&
+                            (current.roomVersion === version || served.get(room.locator)?.has(version) === true))
+                        ? []
+                        : [room.locator];
+                });
+                const expired = [
+                    ...new Set([...expiring.filter((locator) => named.includes(locator)), ...unknown]),
+                ];
+                expiring = expiring.filter((locator) => !named.includes(locator));
+                queueMicrotask(() => {
+                    for (const room of asked) {
+                        if (expired.includes(room.locator) || room.roomVersion === null)
+                            carried.delete(room.locator);
+                        else {
+                            carried.set(room.locator, room.roomVersion);
+                            carry(room.locator);
+                        }
+                    }
+                    emitAll({ kind: 'caught-up', expired });
+                });
             },
             observe(notify) {
                 observers.add(notify);
+                queueMicrotask(() => {
+                    if (!observers.has(notify))
+                        return;
+                    if (options.unopened === true) {
+                        notify({ kind: 'closed', revoked: false, lost: true, authorityChanged: false });
+                        return;
+                    }
+                    announceList();
+                    opened((event) => {
+                        if (observers.has(notify))
+                            notify(event);
+                    });
+                });
                 return () => {
                     observers.delete(notify);
                 };
@@ -298,32 +510,28 @@ export function createMemoryAdapter(options = {}) {
             else
                 scripts[operation] = answer;
         },
-        async model(_authorization, request) {
-            return resolveScript(scripts.model, () => {
-                const room = rooms.find((entry) => entry.locator === request.locator);
-                const answer = request.attached === 'collection' ? room?.model : room?.selectionModel;
-                return answer ?? { kind: 'model-unavailable', reason: 'not-readable' };
-            }, request);
-        },
         async authorize() {
             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) {
+            const scripted = scripts.follows;
+            if (typeof scripted === 'boolean')
+                return scripted;
+            if (scripted !== undefined)
+                return scripted(locator, roomVersion, next);
+            return true;
+        },
         async bootstrap(_authorization, request) {
             return resolveScript(scripts.bootstrap, () => {
                 const room = rooms.find((entry) => entry.locator === request.locator);
                 if (room === undefined) {
                     return { kind: 'room-not-readable', locator: request.locator };
                 }
-                return page(room, request.pageToken === undefined ? 0 : Number(request.pageToken));
+                return page(room, request.pageToken);
             }, request).then((answer) => note(request.locator, answer));
         },
         async refresh(_authorization, request) {
@@ -341,10 +549,6 @@ export function createMemoryAdapter(options = {}) {
                     }
                     return advancement(continuation.room, continuation.from, continuation.at);
                 }
-                if (request.minPosition !== undefined &&
-                    !reached(room.locator, room.roomVersion, request.minPosition)) {
-                    return { kind: 'retryable-refusal', code: 'replica_behind', fields: {} };
-                }
                 if (request.fromRoomVersion === room.roomVersion) {
                     return { kind: 'still-current' };
                 }
@@ -360,8 +564,20 @@ export function createMemoryAdapter(options = {}) {
                 if (room === undefined) {
                     return { kind: 'room-not-readable', locator: request.locator };
                 }
-                return { kind: 'rejected', code: 'unscripted-mutation', fields: {}, affected: {} };
-            }, request).then((answer) => note(request.locator, answer));
+                return {
+                    kind: 'rejected',
+                    code: 'unscripted-mutation',
+                    fields: {},
+                    retryable: false,
+                    affected: {},
+                };
+            }, 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, () => {
@@ -375,11 +591,46 @@ 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: {} };
+                return { kind: 'rejected', code: 'unscripted-action', fields: {}, retryable: false };
             }, request).then((answer) => {
                 if (answer.kind === 'action' && answer.at !== undefined)
                     note(answer.at.locator, answer.at);