diff options
author | 2023-03-20 19:10:08 +0100 | |
---|---|---|
committer | 2023-03-20 18:10:08 +0000 | |
commit | e8595f0c64f527af0913d1a426b697e67ff74ac9 (patch) | |
tree | a5d45b1ad8b96318944408a23fda91f008643900 /internal/processing/followrequest.go | |
parent | [chore]: Bump github.com/miekg/dns from 1.1.51 to 1.1.52 (#1636) (diff) | |
download | gotosocial-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/processing/followrequest.go')
-rw-r--r-- | internal/processing/followrequest.go | 82 |
1 files changed, 28 insertions, 54 deletions
diff --git a/internal/processing/followrequest.go b/internal/processing/followrequest.go index 2b45108c6..7b8c36050 100644 --- a/internal/processing/followrequest.go +++ b/internal/processing/followrequest.go @@ -19,34 +19,33 @@ package processing import ( "context" + "errors" "github.com/superseriousbusiness/gotosocial/internal/ap" apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model" "github.com/superseriousbusiness/gotosocial/internal/db" "github.com/superseriousbusiness/gotosocial/internal/gtserror" + "github.com/superseriousbusiness/gotosocial/internal/log" "github.com/superseriousbusiness/gotosocial/internal/messages" "github.com/superseriousbusiness/gotosocial/internal/oauth" ) func (p *Processor) FollowRequestsGet(ctx context.Context, auth *oauth.Auth) ([]apimodel.Account, gtserror.WithCode) { - frs, err := p.state.DB.GetAccountFollowRequests(ctx, auth.Account.ID) - if err != nil { - if err != db.ErrNoEntries { - return nil, gtserror.NewErrorInternalError(err) - } + followRequests, err := p.state.DB.GetFollowRequests(ctx, "", auth.Account.ID) + if err != nil && !errors.Is(err, db.ErrNoEntries) { + return nil, gtserror.NewErrorInternalError(err) } - accts := []apimodel.Account{} - for _, fr := range frs { - if fr.Account == nil { - frAcct, err := p.state.DB.GetAccountByID(ctx, fr.AccountID) - if err != nil { - return nil, gtserror.NewErrorInternalError(err) - } - fr.Account = frAcct + accts := make([]apimodel.Account, 0, len(followRequests)) + for _, followRequest := range followRequests { + if followRequest.Account == nil { + // The creator of the follow doesn't exist, + // just skip this one. + log.WithContext(ctx).WithField("followRequest", followRequest).Warn("follow request had no associated account") + continue } - apiAcct, err := p.tc.AccountToAPIAccountPublic(ctx, fr.Account) + apiAcct, err := p.tc.AccountToAPIAccountPublic(ctx, followRequest.Account) if err != nil { return nil, gtserror.NewErrorInternalError(err) } @@ -62,19 +61,10 @@ func (p *Processor) FollowRequestAccept(ctx context.Context, auth *oauth.Auth, a } if follow.Account == nil { - followAccount, err := p.state.DB.GetAccountByID(ctx, follow.AccountID) - if err != nil { - return nil, gtserror.NewErrorInternalError(err) - } - follow.Account = followAccount - } - - if follow.TargetAccount == nil { - followTargetAccount, err := p.state.DB.GetAccountByID(ctx, follow.TargetAccountID) - if err != nil { - return nil, gtserror.NewErrorInternalError(err) - } - follow.TargetAccount = followTargetAccount + // The creator of the follow doesn't exist, + // so we can't do further processing. + log.WithContext(ctx).WithField("follow", follow).Warn("follow had no associated account") + return p.relationship(ctx, auth.Account.ID, accountID) } p.state.Workers.EnqueueClientAPI(ctx, messages.FromClientAPI{ @@ -85,17 +75,7 @@ func (p *Processor) FollowRequestAccept(ctx context.Context, auth *oauth.Auth, a TargetAccount: follow.TargetAccount, }) - gtsR, err := p.state.DB.GetRelationship(ctx, auth.Account.ID, accountID) - if err != nil { - return nil, gtserror.NewErrorInternalError(err) - } - - r, err := p.tc.RelationshipToAPIRelationship(ctx, gtsR) - if err != nil { - return nil, gtserror.NewErrorInternalError(err) - } - - return r, nil + return p.relationship(ctx, auth.Account.ID, accountID) } func (p *Processor) FollowRequestReject(ctx context.Context, auth *oauth.Auth, accountID string) (*apimodel.Relationship, gtserror.WithCode) { @@ -105,19 +85,9 @@ func (p *Processor) FollowRequestReject(ctx context.Context, auth *oauth.Auth, a } if followRequest.Account == nil { - a, err := p.state.DB.GetAccountByID(ctx, followRequest.AccountID) - if err != nil { - return nil, gtserror.NewErrorInternalError(err) - } - followRequest.Account = a - } - - if followRequest.TargetAccount == nil { - a, err := p.state.DB.GetAccountByID(ctx, followRequest.TargetAccountID) - if err != nil { - return nil, gtserror.NewErrorInternalError(err) - } - followRequest.TargetAccount = a + // The creator of the request doesn't exist, + // so we can't do further processing. + return p.relationship(ctx, auth.Account.ID, accountID) } p.state.Workers.EnqueueClientAPI(ctx, messages.FromClientAPI{ @@ -128,15 +98,19 @@ func (p *Processor) FollowRequestReject(ctx context.Context, auth *oauth.Auth, a TargetAccount: followRequest.TargetAccount, }) - gtsR, err := p.state.DB.GetRelationship(ctx, auth.Account.ID, accountID) + return p.relationship(ctx, auth.Account.ID, accountID) +} + +func (p *Processor) relationship(ctx context.Context, accountID string, targetAccountID string) (*apimodel.Relationship, gtserror.WithCode) { + relationship, err := p.state.DB.GetRelationship(ctx, accountID, targetAccountID) if err != nil { return nil, gtserror.NewErrorInternalError(err) } - r, err := p.tc.RelationshipToAPIRelationship(ctx, gtsR) + apiRelationship, err := p.tc.RelationshipToAPIRelationship(ctx, relationship) if err != nil { return nil, gtserror.NewErrorInternalError(err) } - return r, nil + return apiRelationship, nil } |