@c-rex/contexts 0.3.0-build.34 → 0.3.0-build.36

package/package.json

@@ -1,6 +1,6 @@
 {
     "name":  "@c-rex/contexts",
-    "version":  "0.3.0-build.34",
+    "version":  "0.3.0-build.36",
     "type":  "module",
     "files":  [
                   "src"

package/src/highlight-provider.tsx

@@ -15,6 +15,7 @@ type HighlightContextType = {
     next: () => void;
     prev: () => void;
     registerContainer: (el: HTMLElement | null) => void;
+    refreshMarks: () => void;
 };
 
 const HighlightContext = createContext<HighlightContextType | null>(null);
@@ -25,18 +26,24 @@ export const HighlightProvider = ({ children }: { children: React.ReactNode }) =
     const [marks, setMarks] = useState<NodeListOf<HTMLElement> | null>(null);
     const [currentIndex, setCurrentIndex] = useState(-1);
 
-    const registerContainer = useCallback((el: HTMLElement | null) => {
-        containerRef.current = el;
-        if (el) {
-            const found = el.querySelectorAll("mark");
-            setMarks(found);
-            setCurrentIndex(found.length > 0 ? 0 : -1);
-        } else {
+    const syncMarks = useCallback(() => {
+        const container = containerRef.current;
+        if (!container) {
             setMarks(null);
             setCurrentIndex(-1);
+            return;
         }
+
+        const found = container.querySelectorAll("mark");
+        setMarks(found);
+        setCurrentIndex(found.length > 0 ? 0 : -1);
     }, []);
 
+    const registerContainer = useCallback((el: HTMLElement | null) => {
+        containerRef.current = el;
+        syncMarks();
+    }, [syncMarks]);
+
     useEffect(() => {
         if (marks && currentIndex >= 0 && marks[currentIndex]) {
             marks.forEach((m) => m.classList.remove("bg-yellow-200", "bg-orange-300"));
@@ -63,6 +70,7 @@ export const HighlightProvider = ({ children }: { children: React.ReactNode }) =
                 next,
                 prev,
                 registerContainer,
+                refreshMarks: syncMarks,
             }}
         >
             {children}