@cuboapp/crdt 1.0.5 → 1.0.7

package/src/index.ts

@@ -1,132 +1,4 @@
-import { type CuboApiEntitiesMap } from '@cuboapp/types'
-import { EventEmitter } from '@cuboapp/utils'
-
-import { CuboCrdtOptions } from './types'
-import { CuboCrdtDocument } from './yjs'
-
 export * from './client'
+export * from './server'
 export * from './types'
 export * from './utils'
-export * from './yjs'
-
-export const CUBO_CRDT_EVENT = {
-  DOCUMENT_UPDATED: 'document:updated',
-  DOCUMENT_DELETED: 'document:deleted',
-  DOCUMENT_CREATED: 'document:created'
-}
-
-export class CuboCrdt<
-  T extends CuboApiEntitiesMap<T>,
-  A = {},
-  K extends string = Extract<keyof T, string>,
-  ID extends string = `${K}:${number | string}`
-> {
-  public documents = new Map<ID, CuboCrdtDocument>()
-  private loading = new Map<ID, Promise<CuboCrdtDocument>>()
-
-  public ee = new EventEmitter()
-
-  constructor(private options: CuboCrdtOptions<A>) {}
-
-  public async start() {
-    // setInterval(() => {
-    // console.log('documents', this.documents.size)
-    // }, 1000)
-    // const doc = new Doc()
-    // const map = doc.getMap('store')
-    // doc.on('update', (upd) => {
-    //   const map1 = doc.getMap('store')
-    //   console.log('new state', map1.toJSON())
-    // })
-    // doc.transact(() => {
-    //   map.set('key', 'value')
-    // })
-  }
-
-  public async stop() {}
-
-  public async getDocument(name: ID, autoCreate = true) {
-    // существующий документ
-    if (!this.documents.has(name)) {
-      if (!this.loading.has(name) && autoCreate) {
-        if (this.options.debug) {
-          console.log('CRDT create document load start', name)
-        }
-
-        this.loading.set(
-          name,
-          new Promise(async (resolve) => {
-            const document = new CuboCrdtDocument({
-              name,
-              autoRemove: true,
-              debug: this.options.debug,
-              initialState: () => {
-                return this.options.beforeLoadDocument?.(name)
-              },
-              onUpdate: ({ data, document }) => {
-                this.ee.emit(CUBO_CRDT_EVENT.DOCUMENT_UPDATED, { name, data, document })
-              }
-            })
-
-            await document.init()
-
-            resolve(document)
-          })
-        )
-      }
-
-      const document = await this.loading.get(name)
-
-      this.loading.delete(name)
-
-      if (document) {
-        this.documents.set(name, document)
-      }
-    }
-
-    if (this.options.debug) {
-      console.log('CRDT create document load completed', name, this.documents.size)
-    }
-
-    return this.documents.get(name)
-  }
-
-  public async removeDocument(name: ID) {
-    if (this.options.debug) {
-      console.log('CRDT remove document', name)
-    }
-
-    if (this.loading.has(name)) {
-      try {
-        Promise.reject(this.loading.get(name))
-        this.loading.delete(name)
-      } catch {}
-    }
-
-    if (this.documents.has(name)) {
-      this.documents.delete(name)
-    }
-  }
-
-  public async updateDocument(name: ID, dto: object) {
-    const document = await this.getDocument(name)
-
-    if (document) {
-      await document.write(dto)
-    }
-
-    await this.checkDocumentUnload(name)
-  }
-
-  public async checkDocumentUnload(name: ID) {
-    const doc = this.documents.get(name)
-
-    if (doc?.opts?.autoRemove && doc.clients.size === 0) {
-      if (this.options.debug) {
-        console.log('CRDT destroy document', name)
-      }
-
-      this.documents.delete(name)
-    }
-  }
-}

package/src/server/document/index.ts

@@ -0,0 +1,84 @@
+import { debounce, pick } from '@cuboapp/utils'
+import { applyUpdate, Doc, encodeStateAsUpdate } from 'yjs'
+
+import { CuboCrdtServerDocumentOptions, CuboCrdtServerDocumentOrigin } from '../types'
+
+export class CuboCrdtServerDocument {
+  private ydoc: Doc
+
+  // private firstUpdate = false
+
+  constructor(public opts: CuboCrdtServerDocumentOptions) {
+    this.ydoc = new Doc({
+      ...(this.opts.yjsOptions || {}),
+      autoLoad: false
+    })
+
+    // записываем исходное состояние
+    this.write(opts.initialState ?? {})
+
+    // подписываемся на апдейты
+    this.ydoc.on('update', (data, origin: CuboCrdtServerDocumentOrigin) => {
+      // if (!this.firstUpdate) {
+      //   console.log('first update', this.name)
+      //   this.firstUpdate = true
+      //   // this.opts?.onCreate?.(data, origin)
+      // } else {
+      this.opts?.onUpdate?.(data, origin)
+      // }
+
+      // если нужно сохранять стейт документа - вызываем store по дебаунсу
+      if (origin?.store) {
+        const row = this.getJson()
+        const body = pick(row, origin.keys ?? Object.keys(row))
+
+        if (Object.keys(body)) {
+          this.debounceStore(body, origin)
+        }
+      }
+    })
+  }
+
+  public get name() {
+    return this.opts.name
+  }
+
+  public get stateAsUpdate() {
+    return encodeStateAsUpdate(this.ydoc)
+  }
+
+  applyUpdate(update: Uint8Array, origin?: CuboCrdtServerDocumentOrigin) {
+    return applyUpdate(this.ydoc, update, origin)
+  }
+
+  getMap() {
+    return this.ydoc.getMap()
+  }
+
+  getJson() {
+    return this.getMap().toJSON()
+  }
+
+  async write(dto: object, origin?: CuboCrdtServerDocumentOrigin) {
+    const map = this.getMap()
+
+    this.ydoc.transact(() => {
+      Object.entries(dto).forEach(([key, value]) => {
+        map.set(key, value)
+      })
+    }, origin)
+  }
+
+  get id() {
+    return this.ydoc.guid
+  }
+
+  destroy() {
+    this.ydoc.destroy()
+  }
+
+  public store(body: object, origin: CuboCrdtServerDocumentOrigin) {
+    this.opts?.onStore?.(body, origin)
+  }
+  public debounceStore = debounce(this.store.bind(this), 300)
+}