feat: add notifications table to db

internal/db/notification.go

···
+package db
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/bluesky-social/indigo/atproto/syntax"
+)
+
+type NotificationType string
+
+const (
+	NotificationTypeFollow   NotificationType = "follow"
+	NotificationTypeReaction NotificationType = "reaction"
+)
+
+type NotificationState string
+
+const (
+	NotificationStateUnread NotificationState = "unread"
+	NotificationStateRead   NotificationState = "read"
+)
+
+type NotificationWithBskyHandle struct {
+	Notification
+	ActorBskyHandle string
+}
+
+type Notification struct {
+	ID           int
+	RecipientDid string
+	ActorDid     string
+	SubjectRkey  string
+	State        NotificationState
+	Type         NotificationType
+	CreatedAt    time.Time
+}
+
+func CreateNotification(e Execer, recipientDid, actorDid, subjectUri string, notificationType NotificationType) error {
+	query := `
+		insert into notifications 
+			(recipient_did, actor_did, subject_uri, type)
+		values (?, ?, ?, ?)
+	`
+
+	_, err := e.Exec(query, recipientDid, actorDid, subjectUri, notificationType)
+	if err != nil {
+		return fmt.Errorf("failed to insert notification: %w", err)
+	}
+
+	return nil
+}
+
+func GetNotificationsByDid(e Execer, did string, limit, offset int) ([]Notification, error) {
+	query := `
+		select
+			id,
+			recipient_did,
+			actor_did,
+			subject_uri,
+			state,
+			type,
+			created_at
+		from
+			notifications
+		where
+			recipient_did = ?
+		order by created_at desc
+		limit ? offset ?
+	`
+
+	rows, err := e.Query(query, did, limit, offset)
+	if err != nil {
+		return nil, fmt.Errorf("failed to query notifications: %w", err)
+	}
+	defer rows.Close()
+
+	var notifications []Notification
+	for rows.Next() {
+		var notification Notification
+		var createdAtStr string
+		var subjectUriStr string
+
+		err := rows.Scan(
+			&notification.ID,
+			&notification.RecipientDid,
+			&notification.ActorDid,
+			&subjectUriStr,
+			&notification.State,
+			&notification.Type,
+			&createdAtStr,
+		)
+		if err != nil {
+			return nil, fmt.Errorf("failed to scan notification row: %w", err)
+		}
+
+		createdAt, err := time.Parse(time.RFC3339, createdAtStr)
+		if err != nil {
+			return nil, fmt.Errorf("failed to parse createdAt string '%s': %w", createdAtStr, err)
+		}
+		notification.CreatedAt = createdAt
+
+		subjectUri, err := syntax.ParseATURI(subjectUriStr)
+		if err != nil {
+			return nil, fmt.Errorf("failed to parse at-uri: %w", err)
+		}
+		notification.SubjectRkey = subjectUri.RecordKey().String()
+
+		notifications = append(notifications, notification)
+	}
+	if err = rows.Err(); err != nil {
+		return nil, err
+	}
+
+	return notifications, nil
+}
+
+func GetUnreadNotificationCount(e Execer, recipientDid string) (int, error) {
+	query := `select count(*) from notifications where recipient_did = ? and state = 'unread';`
+
+	var count int
+	row := e.QueryRow(query, recipientDid)
+	if err := row.Scan(&count); err != nil {
+		return 0, fmt.Errorf("failed to get unread notification count: %w", err)
+	}
+
+	return count, nil
+}
+
+func MarkAllNotificationsAsRead(e Execer, did string) error {
+	query := `
+		update notifications
+		set state = 'read'
+		where recipient_did = ? and state = 'unread';
+	`
+
+	_, err := e.Exec(query, did)
+	if err != nil {
+		return fmt.Errorf("failed to mark notifications as read for did %s: %w", did, err)
+	}
+
+	return nil
+}