Changeset 0.29.0 (#195)

server/store/postgres/message.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"database/sql"
 	"encoding/json"
+	"fmt"
 	"time"
 
 	"git.gobha.me/xcaliber/chat-switchboard/models"
@@ -136,7 +137,9 @@ func (s *MessageStore) GetPathToRoot(ctx context.Context, messageID string) ([]m
 				m.parent_id, m.sibling_index, m.participant_type, m.participant_id, m.deleted_at, m.created_at
 			FROM messages m JOIN path p ON m.id = p.parent_id
 		)
-		SELECT * FROM path ORDER BY created_at ASC`, messageID)
+		SELECT id, channel_id, role, content, model, tokens_used, tool_calls, metadata,
+			parent_id, sibling_index, participant_type, participant_id, deleted_at, created_at
+		FROM path ORDER BY created_at ASC`, messageID)
 	if err != nil {
 		return nil, err
 	}
@@ -189,3 +192,115 @@ func scanMessages(rows *sql.Rows) ([]models.Message, error) {
 	}
 	return result, rows.Err()
 }
+
+// ── CS2 additions (v0.29.0) ─────────────────────────────────────────────
+
+func (s *MessageStore) CountAll(ctx context.Context) (int, error) {
+	var count int
+	err := DB.QueryRowContext(ctx, "SELECT COUNT(*) FROM messages").Scan(&count)
+	return count, err
+}
+
+// ── CS5c additions (v0.29.0) ─────────────────────────────────────────────
+
+func (s *MessageStore) SearchInChannel(ctx context.Context, channelID, query, roleFilter string, limit int) ([]store.ChannelSearchResult, error) {
+	roleClause := ""
+	queryArgs := []interface{}{channelID, query, limit}
+	if roleFilter == "user" || roleFilter == "assistant" {
+		roleClause = "AND m.role = $4"
+		queryArgs = append(queryArgs, roleFilter)
+	}
+
+	rows, err := DB.QueryContext(ctx, fmt.Sprintf(`
+		SELECT m.id, m.role,
+		       ts_headline('english', m.content, plainto_tsquery('english', $2),
+		                   'MaxWords=60, MinWords=20, StartSel=**, StopSel=**') AS headline,
+		       ts_rank(to_tsvector('english', m.content), plainto_tsquery('english', $2)) AS rank,
+		       m.created_at
+		FROM messages m
+		WHERE m.channel_id = $1
+		  AND m.deleted_at IS NULL
+		  AND m.role IN ('user', 'assistant')
+		  AND to_tsvector('english', m.content) @@ plainto_tsquery('english', $2)
+		  %s
+		ORDER BY rank DESC, m.created_at DESC
+		LIMIT $3
+	`, roleClause), queryArgs...)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	results := make([]store.ChannelSearchResult, 0)
+	for rows.Next() {
+		var r store.ChannelSearchResult
+		if err := rows.Scan(&r.MessageID, &r.Role, &r.Excerpt, &r.Rank, &r.Timestamp); err != nil {
+			continue
+		}
+		results = append(results, r)
+	}
+	return results, rows.Err()
+}
+
+// ── CS7a additions (v0.29.0) ────────────────────────────────────────────
+
+func (s *MessageStore) ListWithSenderInfo(ctx context.Context, channelID string, limit, offset int) ([]store.MessageWithSender, int, error) {
+	var total int
+	err := DB.QueryRowContext(ctx,
+		`SELECT COUNT(*) FROM messages WHERE channel_id = $1 AND deleted_at IS NULL`,
+		channelID).Scan(&total)
+	if err != nil {
+		return nil, 0, err
+	}
+
+	rows, err := DB.QueryContext(ctx, `
+		SELECT m.id, m.channel_id, m.role, m.content, m.model, m.tokens_used, m.parent_id,
+		       m.sibling_index, m.participant_type, m.participant_id,
+		       CASE WHEN m.participant_type = 'user' THEN COALESCE(u.display_name, u.username)
+		            WHEN m.participant_type = 'persona' THEN p.name
+		            ELSE NULL END AS sender_name,
+		       CASE WHEN m.participant_type = 'user' THEN u.avatar_url
+		            WHEN m.participant_type = 'persona' THEN p.avatar
+		            ELSE NULL END AS sender_avatar,
+		       m.created_at
+		FROM messages m
+		LEFT JOIN users u ON m.participant_type = 'user' AND m.participant_id = u.id::text
+		LEFT JOIN personas p ON m.participant_type = 'persona' AND m.participant_id = p.id::text
+		WHERE m.channel_id = $1 AND m.deleted_at IS NULL
+		ORDER BY m.created_at ASC
+		LIMIT $2 OFFSET $3
+	`, channelID, limit, offset)
+	if err != nil {
+		return nil, 0, err
+	}
+	defer rows.Close()
+
+	results := make([]store.MessageWithSender, 0)
+	for rows.Next() {
+		var m store.MessageWithSender
+		if err := rows.Scan(
+			&m.ID, &m.ChannelID, &m.Role, &m.Content,
+			&m.Model, &m.TokensUsed, &m.ParentID,
+			&m.SiblingIndex, &m.ParticipantType, &m.ParticipantID,
+			&m.SenderName, &m.SenderAvatar,
+			&m.CreatedAt,
+		); err != nil {
+			return nil, total, err
+		}
+		results = append(results, m)
+	}
+	return results, total, rows.Err()
+}
+
+func (s *MessageStore) GetParentAndRole(ctx context.Context, messageID, channelID string) (*string, string, error) {
+	var parentID sql.NullString
+	var role string
+	err := DB.QueryRowContext(ctx, `
+		SELECT parent_id, role FROM messages
+		WHERE id = $1 AND channel_id = $2 AND deleted_at IS NULL
+	`, messageID, channelID).Scan(&parentID, &role)
+	if err != nil {
+		return nil, "", err
+	}
+	return NullableStringPtr(parentID), role, nil
+}