summaryrefslogtreecommitdiff
path: root/internal/db/bundb/notification.go
diff options
context:
space:
mode:
authorLibravatar tobi <31960611+tsmethurst@users.noreply.github.com>2023-03-20 19:10:08 +0100
committerLibravatar GitHub <noreply@github.com>2023-03-20 18:10:08 +0000
commite8595f0c64f527af0913d1a426b697e67ff74ac9 (patch)
treea5d45b1ad8b96318944408a23fda91f008643900 /internal/db/bundb/notification.go
parent[chore]: Bump github.com/miekg/dns from 1.1.51 to 1.1.52 (#1636) (diff)
downloadgotosocial-e8595f0c64f527af0913d1a426b697e67ff74ac9.tar.xz
[chore] Refactor account deleting/block logic, tidy up some other processing things (#1599)
* start refactoring account deletion * update to use state.DB * further messing about * some more tidying up * more tidying, cleaning, nice-making * further adventures in refactoring and the woes of technical debt * update fr accept/reject * poking + prodding * fix up deleting * create fave uri * don't log using requestingAccount.ID because it might be nil * move getBookmarks function * use exists query to check for status bookmark * use deletenotifications func * fiddle * delete follow request notif * split up some db functions * Fix possible nil pointer panic * fix more possible nil pointers * fix license headers * warn when follow missing (target) account * return wrapped err when bookmark/fave models can't be retrieved * simplify self account delete * warn log likely race condition * de-sillify status delete loop * move error check due north * warn when unfollowSideEffects has no target account * warn when no boost account is found * warn + dump follow when no account * more warnings * warn on fave account not set * move for loop inside anonymous function * fix funky logic * don't remove mutual account items on block; do make sure unfollow occurs in both directions!
Diffstat (limited to 'internal/db/bundb/notification.go')
-rw-r--r--internal/db/bundb/notification.go62
1 files changed, 59 insertions, 3 deletions
diff --git a/internal/db/bundb/notification.go b/internal/db/bundb/notification.go
index 627dc1783..b1e7f45ff 100644
--- a/internal/db/bundb/notification.go
+++ b/internal/db/bundb/notification.go
@@ -19,6 +19,7 @@ package bundb
import (
"context"
+ "errors"
"github.com/superseriousbusiness/gotosocial/internal/db"
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
@@ -104,15 +105,70 @@ func (n *notificationDB) GetNotifications(ctx context.Context, accountID string,
return notifs, nil
}
-func (n *notificationDB) ClearNotifications(ctx context.Context, accountID string) db.Error {
+func (n *notificationDB) DeleteNotification(ctx context.Context, id string) db.Error {
if _, err := n.conn.
NewDelete().
TableExpr("? AS ?", bun.Ident("notifications"), bun.Ident("notification")).
- Where("? = ?", bun.Ident("notification.target_account_id"), accountID).
+ Where("? = ?", bun.Ident("notification.id"), id).
Exec(ctx); err != nil {
return n.conn.ProcessError(err)
}
- n.state.Caches.GTS.Notification().Clear()
+ n.state.Caches.GTS.Notification().Invalidate("ID", id)
+ return nil
+}
+
+func (n *notificationDB) DeleteNotifications(ctx context.Context, targetAccountID string, originAccountID string) db.Error {
+ if targetAccountID == "" && originAccountID == "" {
+ return errors.New("DeleteNotifications: one of targetAccountID or originAccountID must be set")
+ }
+
+ // Capture notification IDs in a RETURNING statement.
+ ids := []string{}
+
+ q := n.conn.
+ NewDelete().
+ TableExpr("? AS ?", bun.Ident("notifications"), bun.Ident("notification")).
+ Returning("?", bun.Ident("id"))
+
+ if targetAccountID != "" {
+ q = q.Where("? = ?", bun.Ident("notification.target_account_id"), targetAccountID)
+ }
+
+ if originAccountID != "" {
+ q = q.Where("? = ?", bun.Ident("notification.origin_account_id"), originAccountID)
+ }
+
+ if _, err := q.Exec(ctx, &ids); err != nil {
+ return n.conn.ProcessError(err)
+ }
+
+ // Invalidate each returned ID.
+ for _, id := range ids {
+ n.state.Caches.GTS.Notification().Invalidate("ID", id)
+ }
+
+ return nil
+}
+
+func (n *notificationDB) DeleteNotificationsForStatus(ctx context.Context, statusID string) db.Error {
+ // Capture notification IDs in a RETURNING statement.
+ ids := []string{}
+
+ q := n.conn.
+ NewDelete().
+ TableExpr("? AS ?", bun.Ident("notifications"), bun.Ident("notification")).
+ Where("? = ?", bun.Ident("notification.status_id"), statusID).
+ Returning("?", bun.Ident("id"))
+
+ if _, err := q.Exec(ctx, &ids); err != nil {
+ return n.conn.ProcessError(err)
+ }
+
+ // Invalidate each returned ID.
+ for _, id := range ids {
+ n.state.Caches.GTS.Notification().Invalidate("ID", id)
+ }
+
return nil
}