Function applyHighlight

src/renderer/src/hooks/useGlobalSearchHighlight.ts:53–149

()

    let frameId = 0

    const applyHighlight = (): void => {
      const messageEl = container.querySelector(`[data-message-id="${highlightMatch.messageId}"]`)
      if (!messageEl) {
        if (retryCount < MAX_HIGHLIGHT_RETRY_COUNT) {
          retryCount++
          frameId = requestAnimationFrame(applyHighlight)
        }
        return
      }

      const contentEl = messageEl.querySelector('.message-item__body')
      if (!contentEl) {
        if (retryCount < MAX_HIGHLIGHT_RETRY_COUNT) {
          retryCount++
          frameId = requestAnimationFrame(applyHighlight)
        }
        return
      }

      // 获取消息内容
      const messageContent = messages.find((m) => m.id === highlightMatch.messageId)?.content ?? ''
      const matchText = messageContent.slice(highlightMatch.contentStart, highlightMatch.contentEnd)
      if (!matchText) return

      // 收集文本节点
      const textNodes: Text[] = []
      const walker = document.createTreeWalker(contentEl, NodeFilter.SHOW_TEXT, null)
      let node: Text | null
      while ((node = walker.nextNode() as Text)) {
        if (node.textContent) {
          textNodes.push(node)
        }
      }

      // 拼接所有文本节点的内容
      let fullText = ''
      const nodeOffsets: { node: Text; start: number; end: number }[] = []
      for (const textNode of textNodes) {
        const text = textNode.textContent ?? ''
        nodeOffsets.push({
          node: textNode,
          start: fullText.length,
          end: fullText.length + text.length
        })
        fullText += text
      }

      // 找到匹配位置（基于原始内容位置在渲染后的 DOM 中查找）
      // 由于 Markdown 渲染可能改变位置，这里按“该消息中的第 N 个命中”定位
      const idx = findNthMatchIndex(
        fullText,
        matchText,
        highlightMatch.occurrenceIndexInMessage ?? 0
      )
      if (idx === -1) {
        if (retryCount < MAX_HIGHLIGHT_RETRY_COUNT) {
          retryCount++
          frameId = requestAnimationFrame(applyHighlight)