Refactor

inokawa · May 19, 2024 · dea6690 · dea6690
1 parent f530bed
commit dea6690
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/renderers/regex/index.tsx b/src/renderers/regex/index.tsx
@@ -123,8 +123,8 @@ const CSSHighlighter = memo(
     useIsomorphicLayoutEffect(() => {
       const el = ref.current!.firstChild!;
 
-      const highlights = regexes.flatMap((r, i) => {
-        const highlight = new Highlight(
+      const highlights = regexes.flatMap((r) => {
+        return new Highlight(
           ...execReg(r, value).map((m) => {
             const start = m.index;
             const end = m.index + m[0]!.length;
@@ -135,8 +135,10 @@ const CSSHighlighter = memo(
             return range;
           })
         );
-        CSS.highlights.set(createCSSHighlightID(i), highlight);
-        return highlight;
+      });
+
+      highlights.forEach((h, i) => {
+        CSS.highlights.set(createCSSHighlightID(i), h);
       });
 
       return () => {