feat: added highlight update

mbret · Nov 19, 2024 · 9efc38d · 9efc38d
1 parent ce069ca
commit 9efc38d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/packages/enhancer-annotations/src/Commands.ts b/packages/enhancer-annotations/src/Commands.ts
@@ -8,14 +8,15 @@ type Command =
   | { type: "highlight"; data: HighlightParams }
   | { type: "add"; data: Highlight | Highlight[] }
   | { type: "delete"; id: string }
+  | { type: "update"; id: string; data: Pick<HighlightParams, "color" | "contents"> }
 
 export class Commands extends DestroyableClass {
   private commandSubject = new Subject<Command>()
 
   public readonly highlight$ = this.commandSubject.pipe(filter((command) => command.type === "highlight"))
   public readonly add$ = this.commandSubject.pipe(filter((command) => command.type === "add"))
   public readonly delete$ = this.commandSubject.pipe(filter((command) => command.type === "delete"))
-
+  public readonly update$ = this.commandSubject.pipe(filter((command) => command.type === "update"))
   highlight = (params: HighlightParams) => {
     this.commandSubject.next({ type: "highlight", data: params })
   }
@@ -28,6 +29,10 @@ export class Commands extends DestroyableClass {
     this.commandSubject.next({ type: "delete", id })
   }
 
+  update = (id: string, data: Pick<HighlightParams, "color" | "contents">) => {
+    this.commandSubject.next({ type: "update", id, data })
+  }
+
   destroy() {
     super.destroy()
     this.commandSubject.complete()

diff --git a/packages/enhancer-annotations/src/annotationsEnhancer.ts b/packages/enhancer-annotations/src/annotationsEnhancer.ts
@@ -16,6 +16,7 @@ export const annotationsEnhancer =
       highlight: Commands["highlight"]
       add: Commands["add"]
       delete: Commands["delete"]
+      update: Commands["update"]
       getHighlightsForTarget: (target: EventTarget) => RuntimeHighlight[]
     }
   } => {
@@ -47,6 +48,12 @@ export const annotationsEnhancer =
       }),
     )
 
+    const update$ = commands.update$.pipe(
+      tap(({ id, data }) => {
+        highlightsSubject.next(highlightsSubject.getValue().map((highlight) => (highlight.id === id ? { ...highlight, ...data } : highlight)))
+      }),
+    ) 
+
     const annotations$ = highlightsSubject.asObservable()
 
     const readerHighlights = new ReaderHighlights(reader, highlightsSubject)
@@ -55,6 +62,7 @@ export const annotationsEnhancer =
       highlight$,
       add$,
       delete$,
+      update$,
       annotations$.pipe(
         tap((annotations) => {
           report.debug("annotations", annotations)
@@ -79,6 +87,7 @@ export const annotationsEnhancer =
         highlight: commands.highlight,
         add: commands.add,
         delete: commands.delete,
+        update: commands.update,
       },
     }
   }