Changeset 0.29.0 (#195)

2026-03-17 16:28:47 +00:00
parent 128cbb8174
commit 5d637d3a90
129 changed files with 9418 additions and 3016 deletions
--- a/server/store/postgres/note.go
+++ b/server/store/postgres/note.go
@@ -219,3 +219,100 @@ func (s *NoteStore) SearchTitles(ctx context.Context, userID, query string, limi
 	}
 	return results, rows.Err()
 }
+
+// ── CS5c additions (v0.29.0) ──────────────────────────────────────────
+
+func (s *NoteStore) SetEmbedding(ctx context.Context, noteID, vecStr string) error {
+	_, err := DB.ExecContext(ctx,
+		`UPDATE notes SET embedding = $1::vector WHERE id = $2`,
+		vecStr, noteID)
+	return err
+}
+
+func (s *NoteStore) SearchKeyword(ctx context.Context, userID, query string, limit int) ([]store.NoteSearchResult, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT id, title, folder_path, tags, LEFT(content, 500),
+		       ts_rank(search_vector, plainto_tsquery('english', $2)) AS rank,
+		       ts_headline('english', content, plainto_tsquery('english', $2),
+		                   'MaxWords=60, MinWords=20, StartSel=**, StopSel=**') AS headline
+		FROM notes
+		WHERE user_id = $1
+		  AND search_vector @@ plainto_tsquery('english', $2)
+		ORDER BY rank DESC
+		LIMIT $3
+	`, userID, query, limit)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	results := make([]store.NoteSearchResult, 0)
+	for rows.Next() {
+		var r store.NoteSearchResult
+		var dbTags pq.StringArray
+		if err := rows.Scan(&r.ID, &r.Title, &r.FolderPath, &dbTags, &r.Excerpt, &r.Rank, &r.Headline); err != nil {
+			continue
+		}
+		r.Tags = []string(dbTags)
+		if r.Tags == nil {
+			r.Tags = []string{}
+		}
+		results = append(results, r)
+	}
+	return results, rows.Err()
+}
+
+func (s *NoteStore) SearchSemantic(ctx context.Context, userID, vecStr string, limit int) ([]store.NoteSearchResult, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT id, title, folder_path, tags, LEFT(content, 500),
+		       1 - (embedding <=> $2::vector) AS similarity
+		FROM notes
+		WHERE user_id = $1
+		  AND embedding IS NOT NULL
+		  AND 1 - (embedding <=> $2::vector) > 0.3
+		ORDER BY embedding <=> $2::vector
+		LIMIT $3
+	`, userID, vecStr, limit)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	results := make([]store.NoteSearchResult, 0)
+	for rows.Next() {
+		var r store.NoteSearchResult
+		var dbTags pq.StringArray
+		if err := rows.Scan(&r.ID, &r.Title, &r.FolderPath, &dbTags, &r.Excerpt, &r.Rank); err != nil {
+			continue
+		}
+		r.Tags = []string(dbTags)
+		if r.Tags == nil {
+			r.Tags = []string{}
+		}
+		results = append(results, r)
+	}
+	return results, rows.Err()
+}
+
+func (s *NoteStore) ListFolders(ctx context.Context, userID string) ([]store.FolderInfo, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT DISTINCT folder_path, COUNT(*) AS count
+		FROM notes WHERE user_id = $1
+		GROUP BY folder_path
+		ORDER BY folder_path
+	`, userID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	results := make([]store.FolderInfo, 0)
+	for rows.Next() {
+		var f store.FolderInfo
+		if err := rows.Scan(&f.Path, &f.Count); err != nil {
+			continue
+		}
+		results = append(results, f)
+	}
+	return results, rows.Err()
+}