nastech-tui 0.0.1

package/packages/nastech-ink/src/ink/searchHighlight.ts

@@ -0,0 +1,91 @@
+import { cellAtIndex, CellWidth, type Screen, setCellStyleId, type StylePool } from './screen.js'
+
+/**
+ * Highlight all visible occurrences of `query` in the screen buffer by
+ * inverting cell styles (SGR 7). Post-render, same damage-tracking machinery
+ * as applySelectionOverlay — the diff picks up highlighted cells as ordinary
+ * changes, LogUpdate stays a pure diff engine.
+ *
+ * Case-insensitive. Handles wide characters (CJK, emoji) by building a
+ * col-of-char map per row — the Nth character isn't at col N when wide chars
+ * are present (each occupies 2 cells: head + SpacerTail).
+ *
+ * This ONLY inverts — there is no "current match" logic here. The yellow
+ * current-match overlay is handled separately by applyPositionedHighlight
+ * (render-to-screen.ts), which writes on top using positions scanned from
+ * the target message's DOM subtree.
+ *
+ * Returns true if any match was highlighted (damage gate — caller forces
+ * full-frame damage when true).
+ */
+export function applySearchHighlight(screen: Screen, query: string, stylePool: StylePool): boolean {
+  if (!query) {
+    return false
+  }
+
+  const lq = query.toLowerCase()
+  const qlen = lq.length
+  const w = screen.width
+  const noSelect = screen.noSelect
+  const height = screen.height
+
+  let applied = false
+
+  for (let row = 0; row < height; row++) {
+    const rowOff = row * w
+    // Build row text (already lowercased) + code-unit→cell-index map.
+    // Three skip conditions, all aligned with setCellStyleId /
+    // extractRowText (selection.ts):
+    //   - SpacerTail: 2nd cell of a wide char, no char of its own
+    //   - SpacerHead: end-of-line padding when a wide char wraps
+    //   - noSelect: gutters (⎿, line numbers) — same exclusion as
+    //     applySelectionOverlay. "Highlight what you see" still holds for
+    //     content; gutters aren't search targets.
+    // Lowercasing per-char (not on the joined string at the end) means
+    // codeUnitToCell maps positions in the LOWERCASED text — U+0130
+    // (Turkish İ) lowercases to 2 code units, so lowering the joined
+    // string would desync indexOf positions from the map.
+    let text = ''
+    const colOf: number[] = []
+    const codeUnitToCell: number[] = []
+
+    for (let col = 0; col < w; col++) {
+      const idx = rowOff + col
+      const cell = cellAtIndex(screen, idx)
+
+      if (cell.width === CellWidth.SpacerTail || cell.width === CellWidth.SpacerHead || noSelect[idx] === 1) {
+        continue
+      }
+
+      const lc = cell.char.toLowerCase()
+      const cellIdx = colOf.length
+
+      for (let i = 0; i < lc.length; i++) {
+        codeUnitToCell.push(cellIdx)
+      }
+
+      text += lc
+      colOf.push(col)
+    }
+
+    let pos = text.indexOf(lq)
+
+    while (pos >= 0) {
+      applied = true
+      const startCi = codeUnitToCell[pos]!
+      const endCi = codeUnitToCell[pos + qlen - 1]!
+
+      for (let ci = startCi; ci <= endCi; ci++) {
+        const col = colOf[ci]!
+        const cell = cellAtIndex(screen, rowOff + col)
+        setCellStyleId(screen, col, row, stylePool.withInverse(cell.styleId))
+      }
+
+      // Non-overlapping advance (less/vim/grep/Ctrl+F). pos+1 would find
+      // 'aa' at 0 AND 1 in 'aaa' → double-invert cell 1.
+      pos = text.indexOf(lq, pos + qlen)
+    }
+  }
+
+  return applied
+}

package/packages/nastech-ink/src/ink/selection.test.ts

@@ -0,0 +1,82 @@
+import { describe, expect, it } from 'vitest'
+
+import { cellAt, CellWidth, CharPool, createScreen, HyperlinkPool, setCellAt, StylePool } from './screen.js'
+import {
+  applySelectionOverlay,
+  createSelectionState,
+  getSelectedText,
+  startSelection,
+  updateSelection
+} from './selection.js'
+
+const screenWithText = () => {
+  const styles = new StylePool()
+  const screen = createScreen(10, 3, styles, new CharPool(), new HyperlinkPool())
+
+  setCellAt(screen, 2, 1, { char: 'h', hyperlink: undefined, styleId: screen.emptyStyleId, width: CellWidth.Narrow })
+  setCellAt(screen, 3, 1, { char: 'i', hyperlink: undefined, styleId: screen.emptyStyleId, width: CellWidth.Narrow })
+
+  return { screen, styles }
+}
+
+describe('selection whitespace handling', () => {
+  it('does not copy whitespace-only selections', () => {
+    const { screen } = screenWithText()
+    const selection = createSelectionState()
+
+    startSelection(selection, 0, 0)
+    updateSelection(selection, 9, 0)
+
+    expect(getSelectedText(selection, screen)).toBe('')
+  })
+
+  it('trims outer drag padding while preserving selected content', () => {
+    const { screen } = screenWithText()
+    const selection = createSelectionState()
+
+    startSelection(selection, 0, 1)
+    updateSelection(selection, 9, 1)
+
+    expect(getSelectedText(selection, screen)).toBe('hi')
+  })
+
+  it('preserves selected indentation when spaces are rendered content', () => {
+    const styles = new StylePool()
+    const screen = createScreen(10, 1, styles, new CharPool(), new HyperlinkPool())
+    const selection = createSelectionState()
+
+    setCellAt(screen, 0, 0, { char: ' ', hyperlink: undefined, styleId: screen.emptyStyleId, width: CellWidth.Narrow })
+    setCellAt(screen, 1, 0, { char: ' ', hyperlink: undefined, styleId: screen.emptyStyleId, width: CellWidth.Narrow })
+    setCellAt(screen, 2, 0, { char: 'x', hyperlink: undefined, styleId: screen.emptyStyleId, width: CellWidth.Narrow })
+
+    startSelection(selection, 0, 0)
+    updateSelection(selection, 9, 0)
+
+    expect(getSelectedText(selection, screen)).toBe('  x')
+  })
+
+  it('clamps copied selection bounds to screen width', () => {
+    const { screen } = screenWithText()
+    const selection = createSelectionState()
+
+    startSelection(selection, 0, 1)
+    updateSelection(selection, 99, 1)
+
+    expect(getSelectedText(selection, screen)).toBe('hi')
+  })
+
+  it('does not paint selection background on leading/trailing empty cells or empty rows', () => {
+    const { screen, styles } = screenWithText()
+    const selection = createSelectionState()
+
+    startSelection(selection, 0, 0)
+    updateSelection(selection, 9, 2)
+    applySelectionOverlay(screen, selection, styles)
+
+    expect(cellAt(screen, 0, 0)?.styleId).toBe(screen.emptyStyleId)
+    expect(cellAt(screen, 0, 1)?.styleId).toBe(screen.emptyStyleId)
+    expect(cellAt(screen, 2, 1)?.styleId).not.toBe(screen.emptyStyleId)
+    expect(cellAt(screen, 4, 1)?.styleId).toBe(screen.emptyStyleId)
+    expect(cellAt(screen, 0, 2)?.styleId).toBe(screen.emptyStyleId)
+  })
+})