Init

1 files changed, 62 insertions, 0 deletions

diff --git a/internal/db/search.go b/internal/db/search.go
new file mode 100644
index 0000000..1214e2d
--- /dev/null
+++ b/internal/db/search.go
@@ -0,0 +1,62 @@
+package db
+
+type SearchPage struct {
+	Path    string
+	Title   string
+	Content string
+}
+
+type SearchResult struct {
+	Path    string
+	Title   string
+	Snippet string
+}
+
+func (d *DB) IndexPage(p SearchPage) error {
+	tx, err := d.db.Begin()
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	if _, err = tx.Exec(`DELETE FROM pages_fts WHERE path = ?`, p.Path); err != nil {
+		return err
+	}
+	if _, err = tx.Exec(
+		`INSERT INTO pages_fts (path, title, content) VALUES (?, ?, ?)`,
+		p.Path, p.Title, p.Content,
+	); err != nil {
+		return err
+	}
+	return tx.Commit()
+}
+
+func (d *DB) UnindexPage(path string) error {
+	_, err := d.db.Exec(`DELETE FROM pages_fts WHERE path = ?`, path)
+	return err
+}
+
+// Search runs a full-text query and returns up to 20 results with snippets.
+func (d *DB) Search(query string) ([]SearchResult, error) {
+	rows, err := d.db.Query(`
+		SELECT path, title,
+			snippet(pages_fts, 2, '<mark>', '</mark>', '...', 20)
+		FROM pages_fts
+		WHERE pages_fts MATCH ?
+		ORDER BY rank
+		LIMIT 20
+	`, query)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var results []SearchResult
+	for rows.Next() {
+		var r SearchResult
+		if err := rows.Scan(&r.Path, &r.Title, &r.Snippet); err != nil {
+			return nil, err
+		}
+		results = append(results, r)
+	}
+	return results, rows.Err()
+}