195 lines
6.3 KiB
Python
195 lines
6.3 KiB
Python
from __future__ import annotations
|
|
|
|
import re
|
|
from typing import Iterable, Tuple
|
|
|
|
from PySide6.QtCore import Qt, Signal
|
|
from PySide6.QtGui import QFont, QTextCharFormat, QTextCursor, QTextDocument
|
|
from PySide6.QtWidgets import (
|
|
QLabel,
|
|
QLineEdit,
|
|
QListWidget,
|
|
QListWidgetItem,
|
|
QHBoxLayout,
|
|
QVBoxLayout,
|
|
QWidget,
|
|
)
|
|
|
|
# type: rows are (date_iso, content)
|
|
Row = Tuple[str, str]
|
|
|
|
|
|
class Search(QWidget):
|
|
"""Encapsulates the search UI + logic and emits a signal when a result is chosen."""
|
|
|
|
openDateRequested = Signal(str)
|
|
|
|
def __init__(self, db, parent: QWidget | None = None):
|
|
super().__init__(parent)
|
|
self._db = db
|
|
|
|
self.search = QLineEdit()
|
|
self.search.setPlaceholderText("Search for notes here")
|
|
self.search.textChanged.connect(self._search)
|
|
|
|
self.results = QListWidget()
|
|
self.results.setUniformItemSizes(False)
|
|
self.results.setSelectionMode(self.results.SelectionMode.SingleSelection)
|
|
self.results.itemClicked.connect(self._open_selected)
|
|
self.results.hide()
|
|
|
|
lay = QVBoxLayout(self)
|
|
lay.setContentsMargins(0, 0, 0, 0)
|
|
lay.setSpacing(6)
|
|
lay.addWidget(self.search)
|
|
lay.addWidget(self.results)
|
|
|
|
def _open_selected(self, item: QListWidgetItem):
|
|
date_str = item.data(Qt.ItemDataRole.UserRole)
|
|
if date_str:
|
|
self.openDateRequested.emit(date_str)
|
|
|
|
def _search(self, text: str):
|
|
"""
|
|
Search for the supplied text in the database.
|
|
For all rows found, populate the results widget with a clickable preview.
|
|
"""
|
|
q = text.strip()
|
|
if not q:
|
|
self.results.clear()
|
|
self.results.hide()
|
|
return
|
|
|
|
try:
|
|
rows: Iterable[Row] = self._db.search_entries(q)
|
|
except Exception:
|
|
# be quiet on DB errors here; caller can surface if desired
|
|
rows = []
|
|
|
|
self._populate_results(q, rows)
|
|
|
|
def _populate_results(self, query: str, rows: Iterable[Row]):
|
|
self.results.clear()
|
|
rows = list(rows)
|
|
if not rows:
|
|
self.results.hide()
|
|
return
|
|
|
|
self.results.show()
|
|
|
|
for date_str, content in rows:
|
|
# Build an HTML fragment around the match and whether to show ellipses
|
|
frag_html, left_ell, right_ell = self._make_html_snippet(
|
|
content, query, radius=30, maxlen=90
|
|
)
|
|
|
|
# ---- Per-item widget: date on top, preview row below (with ellipses) ----
|
|
container = QWidget()
|
|
outer = QVBoxLayout(container)
|
|
outer.setContentsMargins(8, 6, 8, 6)
|
|
outer.setSpacing(2)
|
|
|
|
# Date label (plain text)
|
|
date_lbl = QLabel(date_str)
|
|
date_lbl.setTextFormat(Qt.TextFormat.PlainText)
|
|
date_f = date_lbl.font()
|
|
date_f.setPointSizeF(date_f.pointSizeF() - 1)
|
|
date_lbl.setFont(date_f)
|
|
date_lbl.setStyleSheet("color:#666;")
|
|
outer.addWidget(date_lbl)
|
|
|
|
# Preview row with optional ellipses
|
|
row = QWidget()
|
|
h = QHBoxLayout(row)
|
|
h.setContentsMargins(0, 0, 0, 0)
|
|
h.setSpacing(4)
|
|
|
|
if left_ell:
|
|
left = QLabel("…")
|
|
left.setStyleSheet("color:#888;")
|
|
h.addWidget(left, 0, Qt.AlignmentFlag.AlignTop)
|
|
|
|
preview = QLabel()
|
|
preview.setTextFormat(Qt.TextFormat.RichText)
|
|
preview.setWordWrap(True)
|
|
preview.setOpenExternalLinks(True)
|
|
preview.setText(
|
|
frag_html
|
|
if frag_html
|
|
else "<span style='color:#888'>(no preview)</span>"
|
|
)
|
|
h.addWidget(preview, 1)
|
|
|
|
if right_ell:
|
|
right = QLabel("…")
|
|
right.setStyleSheet("color:#888;")
|
|
h.addWidget(right, 0, Qt.AlignmentFlag.AlignBottom)
|
|
|
|
outer.addWidget(row)
|
|
|
|
# ---- Add to list ----
|
|
item = QListWidgetItem()
|
|
item.setData(Qt.ItemDataRole.UserRole, date_str)
|
|
item.setSizeHint(container.sizeHint())
|
|
|
|
self.results.addItem(item)
|
|
self.results.setItemWidget(item, container)
|
|
|
|
# --- Snippet/highlight helpers -----------------------------------------
|
|
def _make_html_snippet(self, html_src: str, query: str, *, radius=60, maxlen=180):
|
|
doc = QTextDocument()
|
|
doc.setHtml(html_src)
|
|
plain = doc.toPlainText()
|
|
if not plain:
|
|
return "", False, False
|
|
|
|
tokens = [t for t in re.split(r"\s+", query.strip()) if t]
|
|
L = len(plain)
|
|
|
|
# Find first occurrence (phrase first, then earliest token)
|
|
idx, mlen = -1, 0
|
|
if tokens:
|
|
lower = plain.lower()
|
|
phrase = " ".join(tokens).lower()
|
|
j = lower.find(phrase)
|
|
if j >= 0:
|
|
idx, mlen = j, len(phrase)
|
|
else:
|
|
for t in tokens:
|
|
tj = lower.find(t.lower())
|
|
if tj >= 0 and (idx < 0 or tj < idx):
|
|
idx, mlen = tj, len(t)
|
|
# Compute window
|
|
if idx < 0:
|
|
start, end = 0, min(L, maxlen)
|
|
else:
|
|
start = max(0, min(idx - radius, max(0, L - maxlen)))
|
|
end = min(L, max(idx + mlen + radius, start + maxlen))
|
|
|
|
# Bold all token matches that fall inside [start, end)
|
|
if tokens:
|
|
lower = plain.lower()
|
|
fmt = QTextCharFormat()
|
|
fmt.setFontWeight(QFont.Weight.Bold)
|
|
for t in tokens:
|
|
t_low = t.lower()
|
|
pos = start
|
|
while True:
|
|
k = lower.find(t_low, pos)
|
|
if k == -1 or k >= end:
|
|
break
|
|
c = QTextCursor(doc)
|
|
c.setPosition(k)
|
|
c.setPosition(k + len(t), QTextCursor.MoveMode.KeepAnchor)
|
|
c.mergeCharFormat(fmt)
|
|
pos = k + len(t)
|
|
|
|
# Select the window and export as HTML fragment
|
|
c = QTextCursor(doc)
|
|
c.setPosition(start)
|
|
c.setPosition(end, QTextCursor.MoveMode.KeepAnchor)
|
|
fragment_html = (
|
|
c.selection().toHtml()
|
|
) # preserves original styles + our bolding
|
|
|
|
return fragment_html, start > 0, end < L
|