server and web app
This commit is contained in:
304
apps/server/internal/collaboration/repository.go
Normal file
304
apps/server/internal/collaboration/repository.go
Normal file
@@ -0,0 +1,304 @@
|
||||
package collaboration
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Repository struct {
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
func NewRepository(db *sql.DB) *Repository {
|
||||
return &Repository{db: db}
|
||||
}
|
||||
|
||||
// Comments
|
||||
|
||||
func (r *Repository) CreateComment(ctx context.Context, comment Comment) error {
|
||||
_, err := r.db.ExecContext(ctx, `
|
||||
INSERT INTO comments (id, document_id, version_hash, parent_id, author_id, content, anchor_hash, anchor_line, created_at, resolved_at, resolved_by)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
`, comment.ID, comment.DocumentID, comment.VersionHash, comment.ParentID, comment.AuthorID, comment.Content, comment.AnchorHash, comment.AnchorLine, formatTime(comment.CreatedAt), comment.ResolvedAt, comment.ResolvedBy)
|
||||
if err != nil {
|
||||
return fmt.Errorf("insert comment: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetComment(ctx context.Context, id string) (*Comment, error) {
|
||||
var c Comment
|
||||
var parentID, anchorHash sql.NullString
|
||||
var anchorLine sql.NullInt64
|
||||
var resolvedAt, resolvedBy sql.NullString
|
||||
var createdAt string
|
||||
|
||||
err := r.db.QueryRowContext(ctx, `
|
||||
SELECT c.id, c.document_id, c.version_hash, c.parent_id, c.author_id, COALESCE(u.display_name, u.email), c.content, c.anchor_hash, c.anchor_line, c.created_at, c.resolved_at, c.resolved_by
|
||||
FROM comments c
|
||||
JOIN users u ON u.id = c.author_id
|
||||
WHERE c.id = ?
|
||||
`, id).Scan(
|
||||
&c.ID, &c.DocumentID, &c.VersionHash, &parentID, &c.AuthorID, &c.AuthorName,
|
||||
&c.Content, &anchorHash, &anchorLine, &createdAt, &resolvedAt, &resolvedBy,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
c.ParentID = nullString(parentID.String)
|
||||
c.AnchorHash = nullString(anchorHash.String)
|
||||
if anchorLine.Valid {
|
||||
c.AnchorLine = ptr(int(anchorLine.Int64))
|
||||
}
|
||||
c.CreatedAt, _ = parseTime(createdAt)
|
||||
if resolvedAt.Valid && resolvedAt.String != "" {
|
||||
t, _ := parseTime(resolvedAt.String)
|
||||
c.ResolvedAt = &t
|
||||
c.ResolvedBy = nullString(resolvedBy.String)
|
||||
}
|
||||
|
||||
return &c, nil
|
||||
}
|
||||
|
||||
func (r *Repository) ListCommentsByDocument(ctx context.Context, documentID string) ([]Comment, error) {
|
||||
rows, err := r.db.QueryContext(ctx, `
|
||||
SELECT c.id, c.document_id, c.version_hash, c.parent_id, c.author_id, COALESCE(u.display_name, u.email), c.content, c.anchor_hash, c.anchor_line, c.created_at, c.resolved_at, c.resolved_by
|
||||
FROM comments c
|
||||
JOIN users u ON u.id = c.author_id
|
||||
WHERE c.document_id = ?
|
||||
ORDER BY c.created_at ASC
|
||||
`, documentID)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("list comments: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
return scanComments(rows)
|
||||
}
|
||||
|
||||
func (r *Repository) ListCommentsByAnchor(ctx context.Context, documentID string, anchorHash string) ([]Comment, error) {
|
||||
rows, err := r.db.QueryContext(ctx, `
|
||||
SELECT c.id, c.document_id, c.version_hash, c.parent_id, c.author_id, COALESCE(u.display_name, u.email), c.content, c.anchor_hash, c.anchor_line, c.created_at, c.resolved_at, c.resolved_by
|
||||
FROM comments c
|
||||
JOIN users u ON u.id = c.author_id
|
||||
WHERE c.document_id = ? AND c.anchor_hash = ? AND c.resolved_at IS NULL
|
||||
ORDER BY c.created_at ASC
|
||||
`, documentID, anchorHash)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("list comments by anchor: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
return scanComments(rows)
|
||||
}
|
||||
|
||||
func (r *Repository) ResolveComment(ctx context.Context, commentID string, userID string) error {
|
||||
_, err := r.db.ExecContext(ctx, `
|
||||
UPDATE comments SET resolved_at = ?, resolved_by = ?
|
||||
WHERE id = ?
|
||||
`, formatTime(time.Now().UTC()), userID, commentID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("resolve comment: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func scanComments(rows *sql.Rows) ([]Comment, error) {
|
||||
var comments []Comment
|
||||
for rows.Next() {
|
||||
var c Comment
|
||||
var parentID, anchorHash sql.NullString
|
||||
var anchorLine sql.NullInt64
|
||||
var resolvedAt, resolvedBy sql.NullString
|
||||
var createdAt string
|
||||
|
||||
if err := rows.Scan(
|
||||
&c.ID, &c.DocumentID, &c.VersionHash, &parentID, &c.AuthorID, &c.AuthorName,
|
||||
&c.Content, &anchorHash, &anchorLine, &createdAt, &resolvedAt, &resolvedBy,
|
||||
); err != nil {
|
||||
return nil, fmt.Errorf("scan comment: %w", err)
|
||||
}
|
||||
|
||||
c.ParentID = nullString(parentID.String)
|
||||
c.AnchorHash = nullString(anchorHash.String)
|
||||
if anchorLine.Valid {
|
||||
c.AnchorLine = ptr(int(anchorLine.Int64))
|
||||
}
|
||||
c.CreatedAt, _ = parseTime(createdAt)
|
||||
if resolvedAt.Valid && resolvedAt.String != "" {
|
||||
t, _ := parseTime(resolvedAt.String)
|
||||
c.ResolvedAt = &t
|
||||
c.ResolvedBy = nullString(resolvedBy.String)
|
||||
}
|
||||
comments = append(comments, c)
|
||||
}
|
||||
return comments, rows.Err()
|
||||
}
|
||||
|
||||
// Notifications
|
||||
|
||||
func (r *Repository) CreateNotification(ctx context.Context, n Notification) error {
|
||||
_, err := r.db.ExecContext(ctx, `
|
||||
INSERT INTO notifications (id, user_id, type, resource_type, resource_id, message, read_at, emailed_at, created_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
`, n.ID, n.UserID, n.Type, n.ResourceType, n.ResourceID, n.Message, n.ReadAt, n.EmailedAt, formatTime(n.CreatedAt))
|
||||
if err != nil {
|
||||
return fmt.Errorf("insert notification: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *Repository) ListNotifications(ctx context.Context, userID string, unreadOnly bool, limit int) ([]Notification, error) {
|
||||
query := `
|
||||
SELECT id, user_id, type, resource_type, resource_id, message, read_at, emailed_at, created_at
|
||||
FROM notifications
|
||||
WHERE user_id = ?`
|
||||
args := []any{userID}
|
||||
if unreadOnly {
|
||||
query += ` AND read_at IS NULL`
|
||||
}
|
||||
query += ` ORDER BY created_at DESC`
|
||||
if limit > 0 {
|
||||
query += ` LIMIT ?`
|
||||
args = append(args, limit)
|
||||
}
|
||||
|
||||
rows, err := r.db.QueryContext(ctx, query, args...)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("list notifications: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var notifications []Notification
|
||||
for rows.Next() {
|
||||
var n Notification
|
||||
var readAt, emailedAt sql.NullString
|
||||
var createdAt string
|
||||
if err := rows.Scan(&n.ID, &n.UserID, &n.Type, &n.ResourceType, &n.ResourceID, &n.Message, &readAt, &emailedAt, &createdAt); err != nil {
|
||||
return nil, fmt.Errorf("scan notification: %w", err)
|
||||
}
|
||||
if readAt.Valid && readAt.String != "" {
|
||||
t, _ := parseTime(readAt.String)
|
||||
n.ReadAt = &t
|
||||
}
|
||||
if emailedAt.Valid && emailedAt.String != "" {
|
||||
t, _ := parseTime(emailedAt.String)
|
||||
n.EmailedAt = &t
|
||||
}
|
||||
n.CreatedAt, _ = parseTime(createdAt)
|
||||
notifications = append(notifications, n)
|
||||
}
|
||||
return notifications, rows.Err()
|
||||
}
|
||||
|
||||
func (r *Repository) CountUnreadNotifications(ctx context.Context, userID string) (int, error) {
|
||||
var count int
|
||||
err := r.db.QueryRowContext(ctx, `
|
||||
SELECT COUNT(*) FROM notifications WHERE user_id = ? AND read_at IS NULL
|
||||
`, userID).Scan(&count)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("count unread notifications: %w", err)
|
||||
}
|
||||
return count, nil
|
||||
}
|
||||
|
||||
func (r *Repository) MarkNotificationRead(ctx context.Context, notificationID string, userID string) error {
|
||||
_, err := r.db.ExecContext(ctx, `
|
||||
UPDATE notifications SET read_at = ? WHERE id = ? AND user_id = ?
|
||||
`, formatTime(time.Now().UTC()), notificationID, userID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("mark notification read: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *Repository) MarkAllNotificationsRead(ctx context.Context, userID string) error {
|
||||
_, err := r.db.ExecContext(ctx, `
|
||||
UPDATE notifications SET read_at = ? WHERE user_id = ? AND read_at IS NULL
|
||||
`, formatTime(time.Now().UTC()), userID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("mark all notifications read: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Watchers
|
||||
|
||||
func (r *Repository) AddWatcher(ctx context.Context, w Watcher) error {
|
||||
_, err := r.db.ExecContext(ctx, `
|
||||
INSERT INTO watchers (user_id, document_id, folder_path, created_at)
|
||||
VALUES (?, ?, ?, ?)
|
||||
ON CONFLICT(user_id, document_id, folder_path) DO NOTHING
|
||||
`, w.UserID, w.DocumentID, w.FolderPath, formatTime(w.CreatedAt))
|
||||
if err != nil {
|
||||
return fmt.Errorf("add watcher: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *Repository) RemoveWatcher(ctx context.Context, userID string, documentID *string, folderPath *string) error {
|
||||
_, err := r.db.ExecContext(ctx, `
|
||||
DELETE FROM watchers WHERE user_id = ? AND document_id IS ? AND folder_path IS ?
|
||||
`, userID, documentID, folderPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("remove watcher: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *Repository) ListWatchersByDocument(ctx context.Context, documentID string) ([]Watcher, error) {
|
||||
rows, err := r.db.QueryContext(ctx, `
|
||||
SELECT user_id, document_id, folder_path, created_at
|
||||
FROM watchers
|
||||
WHERE document_id = ?
|
||||
`, documentID)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("list document watchers: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
return scanWatchers(rows)
|
||||
}
|
||||
|
||||
func (r *Repository) ListWatchersByFolder(ctx context.Context, folderPath string) ([]Watcher, error) {
|
||||
rows, err := r.db.QueryContext(ctx, `
|
||||
SELECT user_id, document_id, folder_path, created_at
|
||||
FROM watchers
|
||||
WHERE folder_path = ? OR ? || '/' = substr(folder_path, 1, length(?) + 1)
|
||||
`, folderPath, folderPath, folderPath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("list folder watchers: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
return scanWatchers(rows)
|
||||
}
|
||||
|
||||
func (r *Repository) IsWatching(ctx context.Context, userID string, documentID string) (bool, error) {
|
||||
var count int
|
||||
err := r.db.QueryRowContext(ctx, `
|
||||
SELECT COUNT(*) FROM watchers WHERE user_id = ? AND document_id = ?
|
||||
`, userID, documentID).Scan(&count)
|
||||
if err != nil {
|
||||
return false, fmt.Errorf("check watching: %w", err)
|
||||
}
|
||||
return count > 0, nil
|
||||
}
|
||||
|
||||
func scanWatchers(rows *sql.Rows) ([]Watcher, error) {
|
||||
var watchers []Watcher
|
||||
for rows.Next() {
|
||||
var w Watcher
|
||||
var docID, folderPath sql.NullString
|
||||
var createdAt string
|
||||
if err := rows.Scan(&w.UserID, &docID, &folderPath, &createdAt); err != nil {
|
||||
return nil, fmt.Errorf("scan watcher: %w", err)
|
||||
}
|
||||
w.DocumentID = nullString(docID.String)
|
||||
w.FolderPath = nullString(folderPath.String)
|
||||
w.CreatedAt, _ = parseTime(createdAt)
|
||||
watchers = append(watchers, w)
|
||||
}
|
||||
return watchers, rows.Err()
|
||||
}
|
||||
Reference in New Issue
Block a user