Something went wrong. Try again.
Monorepo for Tangled — https://tangled.org forked from tangled.org/core
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451package db
import ( "context" "database/sql" "errors" "fmt" "strings" "time"
"tangled.org/core/appview/models" "tangled.org/core/appview/pagination")
func (d *DB) CreateNotification(ctx context.Context, notification *models.Notification) error { query := ` INSERT INTO notifications (recipient_did, actor_did, type, entity_type, entity_id, read, repo_id, issue_id, pull_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?) `
result, err := d.DB.ExecContext(ctx, query, notification.RecipientDid, notification.ActorDid, string(notification.Type), notification.EntityType, notification.EntityId, notification.Read, notification.RepoId, notification.IssueId, notification.PullId, ) if err != nil { return fmt.Errorf("failed to create notification: %w", err) }
id, err := result.LastInsertId() if err != nil { return fmt.Errorf("failed to get notification ID: %w", err) }
notification.ID = id return nil}
// GetNotificationsPaginated retrieves notifications with filters and paginationfunc GetNotificationsPaginated(e Execer, page pagination.Page, filters ...filter) ([]*models.Notification, error) { var conditions []string var args []any
for _, filter := range filters { conditions = append(conditions, filter.Condition()) args = append(args, filter.Arg()...) }
whereClause := "" if len(conditions) > 0 { whereClause = "WHERE " + conditions[0] for _, condition := range conditions[1:] { whereClause += " AND " + condition } }
query := fmt.Sprintf(` select id, recipient_did, actor_did, type, entity_type, entity_id, read, created, repo_id, issue_id, pull_id from notifications %s order by created desc limit ? offset ? `, whereClause)
args = append(args, page.Limit, page.Offset)
rows, err := e.QueryContext(context.Background(), query, args...) if err != nil { return nil, fmt.Errorf("failed to query notifications: %w", err) } defer rows.Close()
var notifications []*models.Notification for rows.Next() { var n models.Notification var typeStr string var createdStr string err := rows.Scan( &n.ID, &n.RecipientDid, &n.ActorDid, &typeStr, &n.EntityType, &n.EntityId, &n.Read, &createdStr, &n.RepoId, &n.IssueId, &n.PullId, ) if err != nil { return nil, fmt.Errorf("failed to scan notification: %w", err) } n.Type = models.NotificationType(typeStr) n.Created, err = time.Parse(time.RFC3339, createdStr) if err != nil { return nil, fmt.Errorf("failed to parse created timestamp: %w", err) } notifications = append(notifications, &n) }
return notifications, nil}
// GetNotificationsWithEntities retrieves notifications with their related entitiesfunc GetNotificationsWithEntities(e Execer, page pagination.Page, filters ...filter) ([]*models.NotificationWithEntity, error) { var conditions []string var args []any
for _, filter := range filters { conditions = append(conditions, filter.Condition()) args = append(args, filter.Arg()...) }
whereClause := "" if len(conditions) > 0 { whereClause = "WHERE " + conditions[0] for _, condition := range conditions[1:] { whereClause += " AND " + condition } }
query := fmt.Sprintf(` select n.id, n.recipient_did, n.actor_did, n.type, n.entity_type, n.entity_id, n.read, n.created, n.repo_id, n.issue_id, n.pull_id, r.id as r_id, r.did as r_did, r.name as r_name, r.description as r_description, i.id as i_id, i.did as i_did, i.issue_id as i_issue_id, i.title as i_title, i.open as i_open, p.id as p_id, p.owner_did as p_owner_did, p.pull_id as p_pull_id, p.title as p_title, p.state as p_state from notifications n left join repos r on n.repo_id = r.id left join issues i on n.issue_id = i.id left join pulls p on n.pull_id = p.id %s order by n.created desc limit ? offset ? `, whereClause)
args = append(args, page.Limit, page.Offset)
rows, err := e.QueryContext(context.Background(), query, args...) if err != nil { return nil, fmt.Errorf("failed to query notifications with entities: %w", err) } defer rows.Close()
var notifications []*models.NotificationWithEntity for rows.Next() { var n models.Notification var typeStr string var createdStr string var repo models.Repo var issue models.Issue var pull models.Pull var rId, iId, pId sql.NullInt64 var rDid, rName, rDescription sql.NullString var iDid sql.NullString var iIssueId sql.NullInt64 var iTitle sql.NullString var iOpen sql.NullBool var pOwnerDid sql.NullString var pPullId sql.NullInt64 var pTitle sql.NullString var pState sql.NullInt64
err := rows.Scan( &n.ID, &n.RecipientDid, &n.ActorDid, &typeStr, &n.EntityType, &n.EntityId, &n.Read, &createdStr, &n.RepoId, &n.IssueId, &n.PullId, &rId, &rDid, &rName, &rDescription, &iId, &iDid, &iIssueId, &iTitle, &iOpen, &pId, &pOwnerDid, &pPullId, &pTitle, &pState, ) if err != nil { return nil, fmt.Errorf("failed to scan notification with entities: %w", err) }
n.Type = models.NotificationType(typeStr) n.Created, err = time.Parse(time.RFC3339, createdStr) if err != nil { return nil, fmt.Errorf("failed to parse created timestamp: %w", err) }
nwe := &models.NotificationWithEntity{Notification: &n}
// populate repo if present if rId.Valid { repo.Id = rId.Int64 if rDid.Valid { repo.Did = rDid.String } if rName.Valid { repo.Name = rName.String } if rDescription.Valid { repo.Description = rDescription.String } nwe.Repo = &repo }
// populate issue if present if iId.Valid { issue.Id = iId.Int64 if iDid.Valid { issue.Did = iDid.String } if iIssueId.Valid { issue.IssueId = int(iIssueId.Int64) } if iTitle.Valid { issue.Title = iTitle.String } if iOpen.Valid { issue.Open = iOpen.Bool } nwe.Issue = &issue }
// populate pull if present if pId.Valid { pull.ID = int(pId.Int64) if pOwnerDid.Valid { pull.OwnerDid = pOwnerDid.String } if pPullId.Valid { pull.PullId = int(pPullId.Int64) } if pTitle.Valid { pull.Title = pTitle.String } if pState.Valid { pull.State = models.PullState(pState.Int64) } nwe.Pull = &pull }
notifications = append(notifications, nwe) }
return notifications, nil}
// GetNotifications retrieves notifications with filtersfunc GetNotifications(e Execer, filters ...filter) ([]*models.Notification, error) { return GetNotificationsPaginated(e, pagination.FirstPage(), filters...)}
func CountNotifications(e Execer, filters ...filter) (int64, error) { var conditions []string var args []any for _, filter := range filters { conditions = append(conditions, filter.Condition()) args = append(args, filter.Arg()...) }
whereClause := "" if conditions != nil { whereClause = " where " + strings.Join(conditions, " and ") }
query := fmt.Sprintf(`select count(1) from notifications %s`, whereClause) var count int64 err := e.QueryRow(query, args...).Scan(&count)
if !errors.Is(err, sql.ErrNoRows) && err != nil { return 0, err }
return count, nil}
func (d *DB) MarkNotificationRead(ctx context.Context, notificationID int64, userDID string) error { idFilter := FilterEq("id", notificationID) recipientFilter := FilterEq("recipient_did", userDID)
query := fmt.Sprintf(` UPDATE notifications SET read = 1 WHERE %s AND %s `, idFilter.Condition(), recipientFilter.Condition())
args := append(idFilter.Arg(), recipientFilter.Arg()...)
result, err := d.DB.ExecContext(ctx, query, args...) if err != nil { return fmt.Errorf("failed to mark notification as read: %w", err) }
rowsAffected, err := result.RowsAffected() if err != nil { return fmt.Errorf("failed to get rows affected: %w", err) }
if rowsAffected == 0 { return fmt.Errorf("notification not found or access denied") }
return nil}
func (d *DB) MarkAllNotificationsRead(ctx context.Context, userDID string) error { recipientFilter := FilterEq("recipient_did", userDID) readFilter := FilterEq("read", 0)
query := fmt.Sprintf(` UPDATE notifications SET read = 1 WHERE %s AND %s `, recipientFilter.Condition(), readFilter.Condition())
args := append(recipientFilter.Arg(), readFilter.Arg()...)
_, err := d.DB.ExecContext(ctx, query, args...) if err != nil { return fmt.Errorf("failed to mark all notifications as read: %w", err) }
return nil}
func (d *DB) DeleteNotification(ctx context.Context, notificationID int64, userDID string) error { idFilter := FilterEq("id", notificationID) recipientFilter := FilterEq("recipient_did", userDID)
query := fmt.Sprintf(` DELETE FROM notifications WHERE %s AND %s `, idFilter.Condition(), recipientFilter.Condition())
args := append(idFilter.Arg(), recipientFilter.Arg()...)
result, err := d.DB.ExecContext(ctx, query, args...) if err != nil { return fmt.Errorf("failed to delete notification: %w", err) }
rowsAffected, err := result.RowsAffected() if err != nil { return fmt.Errorf("failed to get rows affected: %w", err) }
if rowsAffected == 0 { return fmt.Errorf("notification not found or access denied") }
return nil}
func (d *DB) GetNotificationPreferences(ctx context.Context, userDID string) (*models.NotificationPreferences, error) { userFilter := FilterEq("user_did", userDID)
query := fmt.Sprintf(` SELECT id, user_did, repo_starred, issue_created, issue_commented, pull_created, pull_commented, followed, pull_merged, issue_closed, email_notifications FROM notification_preferences WHERE %s `, userFilter.Condition())
var prefs models.NotificationPreferences err := d.DB.QueryRowContext(ctx, query, userFilter.Arg()...).Scan( &prefs.ID, &prefs.UserDid, &prefs.RepoStarred, &prefs.IssueCreated, &prefs.IssueCommented, &prefs.PullCreated, &prefs.PullCommented, &prefs.Followed, &prefs.PullMerged, &prefs.IssueClosed, &prefs.EmailNotifications, )
if err != nil { if err == sql.ErrNoRows { return &models.NotificationPreferences{ UserDid: userDID, RepoStarred: true, IssueCreated: true, IssueCommented: true, PullCreated: true, PullCommented: true, Followed: true, PullMerged: true, IssueClosed: true, EmailNotifications: false, }, nil } return nil, fmt.Errorf("failed to get notification preferences: %w", err) }
return &prefs, nil}
func (d *DB) UpdateNotificationPreferences(ctx context.Context, prefs *models.NotificationPreferences) error { query := ` INSERT OR REPLACE INTO notification_preferences (user_did, repo_starred, issue_created, issue_commented, pull_created, pull_commented, followed, pull_merged, issue_closed, email_notifications) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?) `
result, err := d.DB.ExecContext(ctx, query, prefs.UserDid, prefs.RepoStarred, prefs.IssueCreated, prefs.IssueCommented, prefs.PullCreated, prefs.PullCommented, prefs.Followed, prefs.PullMerged, prefs.IssueClosed, prefs.EmailNotifications, ) if err != nil { return fmt.Errorf("failed to update notification preferences: %w", err) }
if prefs.ID == 0 { id, err := result.LastInsertId() if err != nil { return fmt.Errorf("failed to get preferences ID: %w", err) } prefs.ID = id }
return nil}
func (d *DB) ClearOldNotifications(ctx context.Context, olderThan time.Duration) error { cutoff := time.Now().Add(-olderThan) createdFilter := FilterLte("created", cutoff)
query := fmt.Sprintf(` DELETE FROM notifications WHERE %s `, createdFilter.Condition())
_, err := d.DB.ExecContext(ctx, query, createdFilter.Arg()...) if err != nil { return fmt.Errorf("failed to cleanup old notifications: %w", err) }
return nil}