server: implement messages and gRPC support for creating messages

server/api/db.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"log/slog"
+	"time"
 
 	"git.dubyatp.xyz/dubyatp/scannerbot/server/db"
 	"github.com/jackc/pgx/v5"
@@ -14,6 +15,7 @@ var ErrUserNotFound = errors.New("db: user not found")
 var ErrSessionNotFound = errors.New("db: session not found")
 var ErrChannelNotFound = errors.New("db: channel not found")
 var ErrFileNotFound = errors.New("db: file not found")
+var ErrMessageNotFound = errors.New("db: message not found")
 
 func dbGetUser(id string) (*User, error) {
 	query := `SELECT id, name, password FROM users WHERE id = $1`
@@ -233,3 +235,112 @@ func dbGetFile(id string) (*File, error) {
 	slog.Debug("db: file found", "fileid", file.ID, "filename", file.Name)
 	return &file, nil
 }
+
+func DBGetChannel(id string) (*Channel, error) { return dbGetChannel(id) }
+func DBAddFile(file *File) error               { return dbAddFile(file) }
+
+func DBAddMessage(msg *Message) error {
+	query := `INSERT INTO messages (id, channel, created, content, audio) VALUES ($1, $2, $3, $4, $5)`
+	_, err := db.Pool.Exec(context.Background(), query, msg.ID, msg.Channel.ID, msg.Created, msg.Content, msg.Audio.ID)
+	if err != nil {
+		slog.Error("db: failed to add message", "error", err, "messageid", msg.ID)
+		return fmt.Errorf("failed to add message")
+	}
+	slog.Debug("db: message added", "messageid", msg.ID)
+	return nil
+}
+
+func dbGetMessage(id string) (*Message, error) {
+	query := `SELECT id, channel, created, content, audio FROM messages WHERE id = $1`
+	var channelID, audioID string
+	var msg Message
+	err := db.Pool.QueryRow(context.Background(), query, id).Scan(&msg.ID, &channelID, &msg.Created, &msg.Content, &audioID)
+	if errors.Is(err, pgx.ErrNoRows) {
+		slog.Debug("db: message not found", "messageid", id)
+		return nil, ErrMessageNotFound
+	} else if err != nil {
+		slog.Error("db: failed to query message", "error", err)
+		return nil, fmt.Errorf("failed to query message")
+	}
+
+	channel, err := dbGetChannel(channelID)
+	if err != nil {
+		slog.Error("db: failed to fetch channel for message", "messageid", id, "channelid", channelID, "error", err)
+		return nil, fmt.Errorf("failed to fetch channel for message")
+	}
+	audio, err := dbGetFile(audioID)
+	if err != nil {
+		slog.Error("db: failed to fetch audio for message", "messageid", id, "audioid", audioID, "error", err)
+		return nil, fmt.Errorf("failed to fetch audio for message")
+	}
+
+	msg.Channel = *channel
+	msg.Audio = *audio
+	slog.Debug("db: message found", "messageid", msg.ID)
+	return &msg, nil
+}
+
+func dbGetMessagesByChannel(channelID string, from, to *time.Time) ([]*Message, error) {
+	query := `SELECT id, channel, created, content, audio FROM messages WHERE channel = $1`
+	args := []any{channelID}
+	if from != nil {
+		args = append(args, *from)
+		query += fmt.Sprintf(" AND created >= $%d", len(args))
+	}
+	if to != nil {
+		args = append(args, *to)
+		query += fmt.Sprintf(" AND created <= $%d", len(args))
+	}
+	query += " ORDER BY created DESC"
+	rows, err := db.Pool.Query(context.Background(), query, args...)
+	if err != nil {
+		slog.Error("db: failed to query messages", "error", err)
+		return nil, fmt.Errorf("failed to query messages")
+	}
+	defer rows.Close()
+
+	type messageRow struct {
+		msg       Message
+		channelID string
+		audioID   string
+	}
+
+	var rows_ []messageRow
+	for rows.Next() {
+		var mr messageRow
+		if err := rows.Scan(&mr.msg.ID, &mr.channelID, &mr.msg.Created, &mr.msg.Content, &mr.audioID); err != nil {
+			slog.Error("db: failed to scan message", "error", err)
+			return nil, fmt.Errorf("failed to scan message")
+		}
+		rows_ = append(rows_, mr)
+	}
+	if err := rows.Err(); err != nil {
+		slog.Error("db: row iteration error", "error", err)
+		return nil, fmt.Errorf("failed to iterate messages")
+	}
+	if len(rows_) == 0 {
+		slog.Debug("db: no messages found", "channelid", channelID)
+		return nil, ErrMessageNotFound
+	}
+
+	channel, err := dbGetChannel(channelID)
+	if err != nil {
+		slog.Error("db: failed to fetch channel for messages", "channelid", channelID, "error", err)
+		return nil, fmt.Errorf("failed to fetch channel for messages")
+	}
+
+	var messages []*Message
+	for _, mr := range rows_ {
+		audio, err := dbGetFile(mr.audioID)
+		if err != nil {
+			slog.Error("db: failed to fetch audio for message", "messageid", mr.msg.ID, "audioid", mr.audioID, "error", err)
+			return nil, fmt.Errorf("failed to fetch audio for message")
+		}
+		mr.msg.Channel = *channel
+		mr.msg.Audio = *audio
+		messages = append(messages, &mr.msg)
+	}
+
+	slog.Debug("db: message list returned", "channelid", channelID, "count", len(messages))
+	return messages, nil
+}