Merge pull request #1 from zouhangwithsweet/feat/support-select-layers

zouhangwithsweet · Feb 27, 2024 · 5ad6c7e · 5ad6c7e
2 parents c7becfc + 2312455
commit 5ad6c7e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/entrypoints/injected/components/Code.tsx b/entrypoints/injected/components/Code.tsx
@@ -97,6 +97,20 @@ export const CodeArea = memo((props: { minimized?: boolean }) => {
     }
   }, [handleSelectionChange])
 
+  useEffect(() => {
+    const layersClickHandle = (e: Event) => {
+      // hard code specially class prefix
+      if (/dev_handoff_layers.*/.test((e.target as HTMLElement).className)) {
+        handleSelectionChange()
+      }
+    }
+
+    document?.addEventListener('click', layersClickHandle)
+    return () => {
+      document?.removeEventListener('click', layersClickHandle)
+    }
+  }, [handleSelectionChange])
+
   const [_, copy] = useCopyToClipboard()
 
   const handleCopy = (text: string) => () => {