First cut.

Author: nedtwigg

browser-extension/src/entrypoints/background.ts

@@ -0,0 +1,43 @@
+import type { BackgroundMessage, CommentDraft, CommentSpot } from '../lib/enhancer'
+import { JsonMap } from '../lib/jsonmap'
+
+interface Tab {
+  tabId: number
+  windowId: number
+}
+interface TabAndSpot {
+  tab: Tab
+  spot: CommentSpot
+}
+interface CommentState {
+  tab: Tab
+  spot: CommentSpot
+  drafts: [number, CommentDraft][]
+}
+
+const _states = new JsonMap<TabAndSpot, CommentState>()
+
+browser.runtime.onMessage.addListener((message: BackgroundMessage, sender) => {
+  if (message.action === 'COMMENT_EVENT' && sender.tab?.id && sender.tab?.windowId) {
+    const tab: Tab = {
+      tabId: sender.tab.id,
+      windowId: sender.tab.windowId,
+    }
+
+    const tabAndSpot: TabAndSpot = {
+      spot: message.event.spot,
+      tab,
+    }
+
+    if (message.event.type === 'ENHANCED') {
+      const commentState: CommentState = {
+        drafts: [],
+        spot: message.event.spot,
+        tab,
+      }
+      _states.set(tabAndSpot, commentState)
+    } else if (message.event.type === 'DESTROYED') {
+      _states.delete(tabAndSpot)
+    }
+  }
+})

browser-extension/src/entrypoints/content.ts

@@ -1,11 +1,27 @@
 import { CONFIG, type ModeType } from '../lib/config'
+import type { BackgroundMessage, CommentSpot } from '../lib/enhancer'
 import { logger } from '../lib/logger'
 import { EnhancerRegistry, TextareaRegistry } from '../lib/registries'
 import { githubPrNewCommentContentScript } from '../playgrounds/github-playground'
 
 const enhancers = new EnhancerRegistry()
 const enhancedTextareas = new TextareaRegistry()
 
+function sendEventToBackground(type: 'ENHANCED' | 'DESTROYED', spot: CommentSpot): void {
+  const message: BackgroundMessage = {
+    action: 'COMMENT_EVENT',
+    event: { spot, type },
+  }
+  browser.runtime.sendMessage(message).catch((error) => {
+    logger.debug('Failed to send event to background:', error)
+  })
+}
+
+enhancedTextareas.setEventHandlers(
+  (spot) => sendEventToBackground('ENHANCED', spot),
+  (spot) => sendEventToBackground('DESTROYED', spot),
+)
+
 export default defineContentScript({
   main() {
     if ((CONFIG.MODE as ModeType) === 'PLAYGROUNDS_PR') {

browser-extension/src/lib/enhancer.ts

@@ -23,6 +23,14 @@ export interface CommentEvent {
   draft: CommentDraft | undefined
 }
 
+export interface BackgroundMessage {
+  action: 'COMMENT_EVENT'
+  event: {
+    type: Extract<CommentEventType, 'ENHANCED' | 'DESTROYED'>
+    spot: CommentSpot
+  }
+}
+
 /** Wraps the textareas of a given platform with Gitcasso's enhancements. */
 export interface CommentEnhancer<Spot extends CommentSpot = CommentSpot> {
   /** Guarantees to only return a type within this list. */

browser-extension/src/lib/registries.ts

@@ -59,15 +59,26 @@ export class EnhancerRegistry {
 
 export class TextareaRegistry {
   private textareas = new Map<HTMLTextAreaElement, EnhancedTextarea>()
+  private onEnhanced?: (spot: CommentSpot) => void
+  private onDestroyed?: (spot: CommentSpot) => void
+
+  setEventHandlers(
+    onEnhanced: (spot: CommentSpot) => void,
+    onDestroyed: (spot: CommentSpot) => void,
+  ): void {
+    this.onEnhanced = onEnhanced
+    this.onDestroyed = onDestroyed
+  }
 
   register<T extends CommentSpot>(textareaInfo: EnhancedTextarea<T>): void {
     this.textareas.set(textareaInfo.textarea, textareaInfo)
-    // TODO: register as a draft in progress with the global list
+    this.onEnhanced?.(textareaInfo.spot)
   }
 
   unregisterDueToModification(textarea: HTMLTextAreaElement): void {
-    if (this.textareas.has(textarea)) {
-      // TODO: register as abandoned or maybe submitted with the global list
+    const textareaInfo = this.textareas.get(textarea)
+    if (textareaInfo) {
+      this.onDestroyed?.(textareaInfo.spot)
       this.textareas.delete(textarea)
     }
   }