SubMiner/stats/src/lib/sentence-search.tsx

import { Fragment, type ReactNode } from 'react';
import type { SentenceSearchResult } from '../types/stats';
import { getStatsMineCardUnavailableReason } from './mining';

export interface SentenceMatchRange {
  start: number;
  end: number;
}

export interface SentenceSearchMineAvailability {
  canMineSentence: boolean;
  canMineWordAudio: boolean;
  exactMatch: boolean;
  unavailableReason: string | null;
}

function normalizedSearchWord(query: string): string {
  return query.trim();
}

function buildFoldedSearchIndex(text: string): {
  text: string;
  sourceStartByIndex: number[];
  sourceEndByIndex: number[];
} {
  let foldedText = '';
  const sourceStartByIndex: number[] = [];
  const sourceEndByIndex: number[] = [];

  for (let sourceStart = 0; sourceStart < text.length; ) {
    const codePoint = text.codePointAt(sourceStart);
    if (codePoint == null) break;

    const char = String.fromCodePoint(codePoint);
    const sourceEnd = sourceStart + char.length;
    const foldedChar = char.toLocaleLowerCase();
    for (let index = 0; index < foldedChar.length; index++) {
      sourceStartByIndex.push(sourceStart);
      sourceEndByIndex.push(sourceEnd);
    }
    foldedText += foldedChar;
    sourceStart = sourceEnd;
  }

  return { text: foldedText, sourceStartByIndex, sourceEndByIndex };
}

export function findExactSentenceMatches(text: string, query: string): SentenceMatchRange[] {
  const needle = normalizedSearchWord(query);
  if (!needle) return [];

  const ranges: SentenceMatchRange[] = [];
  const haystack = buildFoldedSearchIndex(text);
  const normalizedNeedle = needle.toLocaleLowerCase();
  let searchFrom = 0;

  while (searchFrom < haystack.text.length) {
    const index = haystack.text.indexOf(normalizedNeedle, searchFrom);
    if (index < 0) break;
    const endIndex = index + normalizedNeedle.length - 1;
    ranges.push({
      start: haystack.sourceStartByIndex[index] ?? index,
      end: haystack.sourceEndByIndex[endIndex] ?? index + normalizedNeedle.length,
    });
    searchFrom = index + normalizedNeedle.length;
  }

  return ranges;
}

export function getSentenceSearchMineAvailability(
  result: Pick<SentenceSearchResult, 'sourcePath' | 'segmentStartMs' | 'segmentEndMs' | 'text'>,
  query: string,
): SentenceSearchMineAvailability {
  const exactMatch = findExactSentenceMatches(result.text, query).length > 0;
  const unavailableReason = getStatsMineCardUnavailableReason(result);

  return {
    canMineSentence: unavailableReason === null,
    canMineWordAudio: unavailableReason === null && exactMatch,
    exactMatch,
    unavailableReason,
  };
}

export function renderSentenceWithMatches(text: string, query: string): ReactNode {
  const ranges = findExactSentenceMatches(text, query);
  if (ranges.length === 0) return text;

  const parts: ReactNode[] = [];
  let cursor = 0;
  ranges.forEach((range, index) => {
    if (range.start > cursor) {
      parts.push(<Fragment key={`text-${cursor}`}>{text.slice(cursor, range.start)}</Fragment>);
    }
    parts.push(
      <mark
        key={`${range.start}-${index}`}
        className="rounded bg-ctp-yellow/15 px-0.5 text-ctp-yellow underline decoration-ctp-yellow/60 underline-offset-2"
      >
        {text.slice(range.start, range.end)}
      </mark>,
    );
    cursor = range.end;
  });

  if (cursor < text.length) {
    parts.push(<Fragment key={`text-${cursor}`}>{text.slice(cursor)}</Fragment>);
  }

  return parts;
}