Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280package xrpc
import ( "encoding/json" "net/http" "strconv"
"github.com/bluesky-social/indigo/atproto/syntax" "tangled.org/core/api/tangled" "tangled.org/core/appview/db" "tangled.org/core/appview/models" "tangled.org/core/appview/pagination" "tangled.org/core/orm" xrpcerr "tangled.org/core/xrpc/errors")
func (x *Xrpc) NotificationList(w http.ResponseWriter, r *http.Request) { l := x.Logger.With("handler", "NotificationList")
did, ok := actorDid(r) if !ok { writeError(w, xrpcerr.MissingActorDidError, http.StatusForbidden) return }
q := r.URL.Query() readFilter := q.Get("read") categoryFilter := q.Get("category")
filters := []orm.Filter{orm.FilterEq("recipient_did", did)} if readFilter == "unread" { filters = append(filters, orm.FilterEq("read", 0)) } switch categoryFilter { case "social": filters = append(filters, orm.FilterIn("type", models.SocialNotificationTypes)) case "work": filters = append(filters, orm.FilterIn("type", models.WorkNotificationTypes)) }
limit := 50 if s := q.Get("limit"); s != "" { if n, err := strconv.Atoi(s); err == nil && n > 0 && n <= 100 { limit = n } }
notifications, err := db.GetNotificationsWithEntities(x.DB, pagination.Page{Limit: limit}, filters...) if err != nil { l.Error("failed to get notifications", "err", err) writeError(w, errInternal, http.StatusInternalServerError) return }
unreadBase := []orm.Filter{ orm.FilterEq("recipient_did", did), orm.FilterEq("read", 0), } workUnread, _ := db.CountNotifications(x.DB, append(unreadBase, orm.FilterIn("type", models.WorkNotificationTypes))...) socialUnread, _ := db.CountNotifications(x.DB, append(unreadBase, orm.FilterIn("type", models.SocialNotificationTypes))...)
items := make([]*tangled.TempNotificationListNotifications_Notification, 0, len(notifications)) for _, n := range notifications { item := &tangled.TempNotificationListNotifications_Notification{ Id: n.ID, Type: string(n.Type), Category: notificationCategory(n.Type), ActorDid: n.ActorDid, Read: n.Read, CreatedAt: n.Created.UTC().Format("2006-01-02T15:04:05.000Z"), } if n.Repo != nil { s := n.Repo.RepoDid item.RepoDid = &s } if n.Issue != nil { s := n.Issue.AtUri().String() item.IssueAt = &s } if n.Pull != nil { s := n.Pull.AtUri().String() item.PullAt = &s } items = append(items, item) }
x.writeJSON(w, &tangled.TempNotificationListNotifications_Output{ Notifications: items, WorkUnreadCount: workUnread, SocialUnreadCount: socialUnread, })}
func (x *Xrpc) NotificationGetUnreadCount(w http.ResponseWriter, r *http.Request) { l := x.Logger.With("handler", "NotificationGetUnreadCount")
did, ok := actorDid(r) if !ok { writeError(w, xrpcerr.MissingActorDidError, http.StatusForbidden) return }
count, err := db.CountNotifications(x.DB, orm.FilterEq("recipient_did", did), orm.FilterEq("read", 0), ) if err != nil { l.Error("failed to count unread notifications", "err", err) writeError(w, errInternal, http.StatusInternalServerError) return }
x.writeJSON(w, &tangled.TempNotificationGetUnreadCount_Output{Count: count})}
func (x *Xrpc) NotificationUpdateSeen(w http.ResponseWriter, r *http.Request) { l := x.Logger.With("handler", "NotificationUpdateSeen")
did, ok := actorDid(r) if !ok { writeError(w, xrpcerr.MissingActorDidError, http.StatusForbidden) return }
var input tangled.TempNotificationUpdateSeen_Input if err := json.NewDecoder(r.Body).Decode(&input); err != nil { writeError(w, errBadRequestBody, http.StatusBadRequest) return }
var err error if input.Read { err = db.MarkNotificationRead(x.DB, input.Id, did) } else { err = db.MarkNotificationUnread(x.DB, input.Id, did) } if err != nil { l.Error("failed to update notification read state", "err", err) writeError(w, errInternal, http.StatusInternalServerError) return }
w.WriteHeader(http.StatusOK)}
func (x *Xrpc) NotificationMarkAllRead(w http.ResponseWriter, r *http.Request) { l := x.Logger.With("handler", "NotificationMarkAllRead")
did, ok := actorDid(r) if !ok { writeError(w, xrpcerr.MissingActorDidError, http.StatusForbidden) return }
if err := db.MarkAllNotificationsRead(x.DB, did); err != nil { l.Error("failed to mark all notifications read", "err", err) writeError(w, errInternal, http.StatusInternalServerError) return }
w.WriteHeader(http.StatusOK)}
func (x *Xrpc) NotificationDelete(w http.ResponseWriter, r *http.Request) { l := x.Logger.With("handler", "NotificationDelete")
did, ok := actorDid(r) if !ok { writeError(w, xrpcerr.MissingActorDidError, http.StatusForbidden) return }
var input tangled.TempNotificationDeleteNotification_Input if err := json.NewDecoder(r.Body).Decode(&input); err != nil { writeError(w, errBadRequestBody, http.StatusBadRequest) return }
if err := db.DeleteNotification(x.DB, input.Id, did); err != nil { l.Error("failed to delete notification", "err", err) writeError(w, errInternal, http.StatusInternalServerError) return }
w.WriteHeader(http.StatusOK)}
func (x *Xrpc) NotificationGetPreferences(w http.ResponseWriter, r *http.Request) { l := x.Logger.With("handler", "NotificationGetPreferences")
did, ok := actorDid(r) if !ok { writeError(w, xrpcerr.MissingActorDidError, http.StatusForbidden) return }
prefs, err := db.GetNotificationPreference(x.DB, did) if err != nil { l.Error("failed to get notification preferences", "err", err) writeError(w, errInternal, http.StatusInternalServerError) return }
x.writeJSON(w, &tangled.TempNotificationGetPreferences_Preferences{ EmailNotifications: prefs.EmailNotifications, Followed: prefs.Followed, IssueClosed: prefs.IssueClosed, IssueCommented: prefs.IssueCommented, IssueCreated: prefs.IssueCreated, PullCommented: prefs.PullCommented, PullCreated: prefs.PullCreated, PullMerged: prefs.PullMerged, RepoStarred: prefs.RepoStarred, UserMentioned: prefs.UserMentioned, })}
func (x *Xrpc) NotificationUpdatePreferences(w http.ResponseWriter, r *http.Request) { l := x.Logger.With("handler", "NotificationUpdatePreferences")
did, ok := actorDid(r) if !ok { writeError(w, xrpcerr.MissingActorDidError, http.StatusForbidden) return }
var input tangled.TempNotificationUpdatePreferences_Input if err := json.NewDecoder(r.Body).Decode(&input); err != nil { writeError(w, errBadRequestBody, http.StatusBadRequest) return }
existing, err := db.GetNotificationPreference(x.DB, did) if err != nil { l.Error("failed to get existing notification preferences", "err", err) writeError(w, errInternal, http.StatusInternalServerError) return }
prefs := &models.NotificationPreferences{ UserDid: syntax.DID(did), RepoStarred: applyBoolPtr(existing.RepoStarred, input.RepoStarred), IssueCreated: applyBoolPtr(existing.IssueCreated, input.IssueCreated), IssueCommented: applyBoolPtr(existing.IssueCommented, input.IssueCommented), IssueClosed: applyBoolPtr(existing.IssueClosed, input.IssueClosed), PullCreated: applyBoolPtr(existing.PullCreated, input.PullCreated), PullCommented: applyBoolPtr(existing.PullCommented, input.PullCommented), PullMerged: applyBoolPtr(existing.PullMerged, input.PullMerged), Followed: applyBoolPtr(existing.Followed, input.Followed), UserMentioned: applyBoolPtr(existing.UserMentioned, input.UserMentioned), EmailNotifications: applyBoolPtr(existing.EmailNotifications, input.EmailNotifications), }
if err := x.DB.UpdateNotificationPreferences(r.Context(), prefs); err != nil { l.Error("failed to update notification preferences", "err", err) writeError(w, errInternal, http.StatusInternalServerError) return }
w.WriteHeader(http.StatusOK)}
func notificationCategory(t models.NotificationType) string { for _, st := range models.SocialNotificationTypes { if st == t { return "social" } } return "work"}
func applyBoolPtr(existing bool, update *bool) bool { if update != nil { return *update } return existing}