From d52f9f2a92795d619d0f2d3d16c7d3aaafe8c4ac Mon Sep 17 00:00:00 2001
From: Elian Doran <contact@eliandoran.me>
Date: Sat, 30 Aug 2025 19:07:06 +0300
Subject: [PATCH] chore(react/collections): highlighting in grid title

---
 apps/client/src/widgets/collections/legacy/ListView.tsx | 3 +++
 apps/client/src/widgets/react/hooks.tsx                 | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/apps/client/src/widgets/collections/legacy/ListView.tsx b/apps/client/src/widgets/collections/legacy/ListView.tsx
index c25f4ac55..eb3ea1de0 100644
--- a/apps/client/src/widgets/collections/legacy/ListView.tsx
+++ b/apps/client/src/widgets/collections/legacy/ListView.tsx
@@ -87,11 +87,14 @@ function GridNoteCard({ note, parentNote, highlightedTokens }: { note: FNote, pa
     const titleRef = useRef<HTMLSpanElement>(null);
     const [ noteTitle, setNoteTitle ] = useState<string>();
     const notePath = getNotePath(parentNote, note);
+    const highlightSearch = useImperativeSearchHighlighlighting(highlightedTokens);
 
     useEffect(() => {
         tree.getNoteTitle(note.noteId, parentNote.noteId).then(setNoteTitle);
     }, [ note ]);
 
+    useEffect(() => highlightSearch(titleRef.current), [ noteTitle, highlightedTokens ]);
+
     return (
         <div
             className={`note-book-card no-tooltip-preview block-link`}
diff --git a/apps/client/src/widgets/react/hooks.tsx b/apps/client/src/widgets/react/hooks.tsx
index b4148197d..4a9f24ceb 100644
--- a/apps/client/src/widgets/react/hooks.tsx
+++ b/apps/client/src/widgets/react/hooks.tsx
@@ -558,7 +558,7 @@ export function useImperativeSearchHighlighlighting(highlightedTokens: string[]
         return new RegExp(regex, "gi")
     }, [ highlightedTokens ]);
 
-    return (el: HTMLElement) => {
+    return (el: HTMLElement | null | undefined) => {
         if (!el || !highlightRegex) return;
 
         if (!mark.current) {