feat(files): improve editor recovery and editing controls

Author: waleedlatif1

apps/realtime/src/handlers/file-doc-store.test.ts

@@ -63,12 +63,19 @@ vi.mock('redis', () => {
         for (let i = 0; i < backing.readDelayTicks; i++) await Promise.resolve()
         return (backing.streams.get(key) ?? []).map((e) => ({ ...e }))
       },
+      xRevRange: async (key: string) =>
+        [...(backing.streams.get(key) ?? [])]
+          .reverse()
+          .slice(0, 1)
+          .map((entry) => ({ ...entry })),
       xLen: async (key: string) => (backing.streams.get(key) ?? []).length,
-      xRead: async (streams: { key: string; id: string }[]) => {
+      xRead: async (streams: { key: string; id: string }[], options?: { COUNT?: number }) => {
         const res: { name: string; messages: { id: string; message: Record<string, string> }[] }[] =
           []
         for (const { key, id } of streams) {
-          const after = (backing.streams.get(key) ?? []).filter((e) => seqOf(e.id) > seqOf(id))
+          const after = (backing.streams.get(key) ?? [])
+            .filter((e) => seqOf(e.id) > seqOf(id))
+            .slice(0, options?.COUNT)
           if (after.length) res.push({ name: key, messages: after.map((e) => ({ ...e })) })
         }
         if (res.length) return res

apps/realtime/src/handlers/file-doc-store.ts

@@ -25,6 +25,7 @@ const fakeStore = {
   versions: new Map<string, number>(),
   acquireMergeSlot: vi.fn(async () => 'token'),
   releaseMergeSlot: vi.fn(async () => {}),
+  getDocumentGeneration: vi.fn(async () => 'shared-generation'),
   getStreamState: vi.fn(async () => new Uint8Array([1])),
   publishAndWait: vi.fn(async () => {}),
   getSyncedVersion: vi.fn(async (name: string) => fakeStore.versions.get(name) ?? null),
@@ -67,7 +68,13 @@ describe('applyMarkdownToLiveFileDoc — multi-replica (store-enabled) ordering'
     expect(await applyMarkdownToLiveFileDoc('file-1', '# durable', { version: 100 })).toBe(
       'applied'
     )
-    expect(fakeStore.setSyncedVersion).toHaveBeenCalledWith(ROOM_NAME, 100)
+    expect(fakeStore.setSyncedVersion).toHaveBeenCalledWith(ROOM_NAME, 100, 'shared-generation')
+    expect(fakeStore.getStreamState).toHaveBeenCalledWith(ROOM_NAME, 'shared-generation')
+    expect(fakeStore.publishAndWait).toHaveBeenCalledWith(
+      ROOM_NAME,
+      expect.any(Uint8Array),
+      'shared-generation'
+    )
     mockFetchFileDocMerge.mockClear()
 
     // A durable write with an OLDER version than the SHARED synced version is stale — rejected under the
@@ -81,7 +88,7 @@ describe('applyMarkdownToLiveFileDoc — multi-replica (store-enabled) ordering'
     expect(await applyMarkdownToLiveFileDoc('file-1', '# durable again', { version: 150 })).toBe(
       'applied'
     )
-    expect(fakeStore.setSyncedVersion).toHaveBeenCalledWith(ROOM_NAME, 150)
+    expect(fakeStore.setSyncedVersion).toHaveBeenCalledWith(ROOM_NAME, 150, 'shared-generation')
     // setSyncedVersion fired only for the two applied durable writes, never for the stale one.
     expect(fakeStore.setSyncedVersion).toHaveBeenCalledTimes(2)
   })
@@ -98,7 +105,7 @@ describe('applyMarkdownToLiveFileDoc — multi-replica (store-enabled) ordering'
     ).toBe('applied')
     expect(mockFetchFileDocMerge).not.toHaveBeenCalled() // content deferred to the client
     expect(fakeStore.publishAndWait).not.toHaveBeenCalled()
-    expect(fakeStore.setSyncedVersion).toHaveBeenCalledWith(ROOM_NAME, 100) // version still recorded
+    expect(fakeStore.setSyncedVersion).toHaveBeenCalledWith(ROOM_NAME, 100, 'shared-generation') // version still recorded
 
     // Once streaming stops the flag clears and the (now near-noop) durable merge resumes normally.
     fakeStore.isAgentStreaming.mockResolvedValue(false)