feat: add tracing

packages/backend/src/AutomergeCollaboration.ts

@@ -3,10 +3,24 @@ import * as Automerge from 'automerge'
 import { Node } from 'slate'
 import { Server } from 'http'
 import throttle from 'lodash/throttle'
+import flatten from 'lodash/flatten'
 import { SyncDoc, CollabAction, toJS } from '@hiveteams/collab-bridge'
 import { debugCollabBackend } from './utils/debug'
 import AutomergeBackend from './AutomergeBackend'
 
+export interface IAutomergeMetaData {
+  docId: string
+  userId: string
+  type: string
+  opText?: string
+  opCount?: number
+}
+
+export type ITraceFunction = (
+  metaData: IAutomergeMetaData,
+  computationFunction: () => void
+) => void
+
 export interface IAutomergeCollaborationOptions {
   entry: Server
   connectOpts?: SocketIO.ServerOptions
@@ -21,6 +35,11 @@ export interface IAutomergeCollaborationOptions {
   ) => Promise<void> | void
   onDisconnect?: (docId: string, user: any) => Promise<void> | void
   onError?: (error: Error, data: any) => Promise<void> | void
+  onTrace?: ITraceFunction
+}
+
+const defaultOnTrace: ITraceFunction = (metaData, computation) => {
+  computation()
 }
 
 export default class AutomergeCollaboration {
@@ -29,6 +48,7 @@ export default class AutomergeCollaboration {
   public backend: AutomergeBackend
   private userMap: { [key: string]: any | undefined }
   private autoSaveDoc: (socket: SocketIO.Socket, docId: string) => void
+  private onTrace: ITraceFunction
 
   /**
    * Constructor
@@ -45,6 +65,8 @@ export default class AutomergeCollaboration {
 
     this.userMap = {}
 
+    this.onTrace = options.onTrace || defaultOnTrace
+
     /**
      * Save document with throttle
      */
@@ -182,11 +204,20 @@ export default class AutomergeCollaboration {
         return
       }
 
-      // Emit the socket message needed for receiving the automerge document
-      // on connect and reconnect
-      socket.emit('msg', {
-        type: 'document',
-        payload: Automerge.save<SyncDoc>(doc)
+      const user = this.userMap[id]
+      const metaData: IAutomergeMetaData = {
+        type: 'connect',
+        userId: user?._id,
+        docId
+      }
+
+      this.onTrace(metaData, () => {
+        // Emit the socket message needed for receiving the automerge document
+        // on connect and reconnect
+        socket.emit('msg', {
+          type: 'document',
+          payload: Automerge.save<SyncDoc>(doc)
+        })
       })
 
       debugCollabBackend('Open connection %s', id)
@@ -205,18 +236,37 @@ export default class AutomergeCollaboration {
     data: any
   ) => {
     const { id } = socket
-    switch (data.type) {
-      case 'operation':
-        try {
-          this.backend.receiveOperation(id, data)
+    const user = this.userMap[id]
 
-          this.autoSaveDoc(socket, docId)
-
-          this.garbageCursors(socket)
-        } catch (err) {
-          this.handleError(socket, err, { onMessageData: data })
-        }
+    // parse out the changes contained in this automerge change
+    const collabActions = flatten(
+      data.payload.changes?.map((change: Automerge.Change) =>
+        change.ops.map(op => op.action)
+      )
+    )
+    const metaData: IAutomergeMetaData = {
+      type: 'operation',
+      userId: user?._id,
+      docId,
+      // cap max operation text by 1000 chars
+      opText: collabActions.join(',').slice(0, 1000),
+      opCount: collabActions.length
     }
+
+    this.onTrace(metaData, () => {
+      switch (data.type) {
+        case 'operation':
+          try {
+            this.backend.receiveOperation(id, data)
+
+            this.autoSaveDoc(socket, docId)
+
+            this.garbageCursors(socket)
+          } catch (err) {
+            this.handleError(socket, err, { onMessageData: data })
+          }
+      }
+    })
   }
 
   /**