init

internal/tenant/search.go

@@ -0,0 +1,67 @@
+package tenant
+
+import (
+	"context"
+)
+
+type SearchResult struct {
+	Slug       string
+	Collection string
+	Title      string
+	Snippet    string
+	Type       string
+	URL        string
+	Date       string
+}
+
+func (q *Queries) Search(ctx context.Context, query string, limit int) ([]SearchResult, error) {
+	if limit <= 0 {
+		limit = 20
+	}
+
+	rows, err := q.db.QueryContext(ctx,
+		`SELECT slug, collection_slug, title, snippet(search_index, 4, '<mark>', '</mark>', '...', 32), type, url, date
+		FROM search_index
+		WHERE search_index MATCH ?
+		ORDER BY rank
+		LIMIT ?`, query, limit)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var results []SearchResult
+	for rows.Next() {
+		var r SearchResult
+		if err := rows.Scan(&r.Slug, &r.Collection, &r.Title, &r.Snippet, &r.Type, &r.URL, &r.Date); err != nil {
+			return nil, err
+		}
+		results = append(results, r)
+	}
+	return results, rows.Err()
+}
+
+func (q *Queries) IndexPost(ctx context.Context, p *Post) error {
+	q.db.ExecContext(ctx, `DELETE FROM search_index WHERE slug = ? AND type = 'post'`, p.Slug)
+
+	if !p.IsPublished {
+		return nil
+	}
+
+	dateStr := ""
+	if p.PublishedAt != nil {
+		dateStr = p.PublishedAt.Format("2006-01-02")
+	}
+
+	_, err := q.db.ExecContext(ctx,
+		`INSERT INTO search_index (slug, collection_slug, title, description, content, type, url, date)
+		VALUES (?, ?, ?, ?, ?, 'post', ?, ?)`,
+		p.Slug, "", p.Title, p.Description, p.ContentMD,
+		"/posts/"+p.Slug, dateStr)
+	return err
+}
+
+func (q *Queries) RemoveFromIndex(ctx context.Context, slug, itemType string) error {
+	_, err := q.db.ExecContext(ctx, `DELETE FROM search_index WHERE slug = ? AND type = ?`, slug, itemType)
+	return err
+}