From c7b6cd7770cad9bfdc23decffa7c4068752dbbbd Mon Sep 17 00:00:00 2001 From: tobi <31960611+tsmethurst@users.noreply.github.com> Date: Wed, 25 Oct 2023 16:04:53 +0200 Subject: [feature] Status thread mute/unmute functionality (#2278) * add db models + functions for keeping track of threads * give em the old linty testy * create, remove, check mutes * swagger * testerino * test mute/unmute via api * add info log about new index creation * thread + allow muting of any remote statuses that mention a local account * IsStatusThreadMutedBy -> IsThreadMutedByAccount * use common processing functions in status processor * set = NULL * favee! * get rekt darlings, darlings get rekt * testrig please, have mercy muy liege --- internal/processing/status/mute.go | 146 +++++++++++++++++++++++++++++++++++++ 1 file changed, 146 insertions(+) create mode 100644 internal/processing/status/mute.go (limited to 'internal/processing/status/mute.go') diff --git a/internal/processing/status/mute.go b/internal/processing/status/mute.go new file mode 100644 index 000000000..1663ee0bc --- /dev/null +++ b/internal/processing/status/mute.go @@ -0,0 +1,146 @@ +// GoToSocial +// Copyright (C) GoToSocial Authors admin@gotosocial.org +// SPDX-License-Identifier: AGPL-3.0-or-later +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU Affero General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Affero General Public License for more details. +// +// You should have received a copy of the GNU Affero General Public License +// along with this program. If not, see . + +package status + +import ( + "context" + "errors" + + 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/gtsmodel" + "github.com/superseriousbusiness/gotosocial/internal/id" +) + +// getMuteableStatus fetches targetStatusID status and +// ensures that requestingAccount can mute or unmute it. +// +// It checks: +// - Status exists and is visible to requester. +// - Status belongs to or mentions requesting account. +// - Status is not a boost. +// - Status has a thread ID. +func (p *Processor) getMuteableStatus( + ctx context.Context, + requestingAccount *gtsmodel.Account, + targetStatusID string, +) (*gtsmodel.Status, gtserror.WithCode) { + targetStatus, errWithCode := p.c.GetVisibleTargetStatus(ctx, requestingAccount, targetStatusID) + if errWithCode != nil { + return nil, errWithCode + } + + if !targetStatus.BelongsToAccount(requestingAccount.ID) && + !targetStatus.MentionsAccount(requestingAccount.ID) { + err := gtserror.Newf("status %s does not belong to or mention account %s", targetStatusID, requestingAccount.ID) + return nil, gtserror.NewErrorNotFound(err) + } + + if targetStatus.BoostOfID != "" { + err := gtserror.New("cannot mute or unmute boosts") + return nil, gtserror.NewErrorBadRequest(err, err.Error()) + } + + if targetStatus.ThreadID == "" { + err := gtserror.New("cannot mute or unmute status with no threadID") + return nil, gtserror.NewErrorBadRequest(err, err.Error()) + } + + return targetStatus, nil +} + +func (p *Processor) MuteCreate( + ctx context.Context, + requestingAccount *gtsmodel.Account, + targetStatusID string, +) (*apimodel.Status, gtserror.WithCode) { + targetStatus, errWithCode := p.getMuteableStatus(ctx, requestingAccount, targetStatusID) + if errWithCode != nil { + return nil, errWithCode + } + + var ( + threadID = targetStatus.ThreadID + accountID = requestingAccount.ID + ) + + // Check if mute already exists for this thread ID. + threadMute, err := p.state.DB.GetThreadMutedByAccount(ctx, threadID, accountID) + if err != nil && !errors.Is(err, db.ErrNoEntries) { + // Real db error. + err := gtserror.Newf("db error fetching mute of thread %s for account %s", threadID, accountID) + return nil, gtserror.NewErrorInternalError(err) + } + + if threadMute != nil { + // Thread mute already exists. + // Our job here is done ("but you didn't do anything!"). + return p.c.GetAPIStatus(ctx, requestingAccount, targetStatus) + } + + // Gotta create a mute. + if err := p.state.DB.PutThreadMute(ctx, >smodel.ThreadMute{ + ID: id.NewULID(), + ThreadID: threadID, + AccountID: accountID, + }); err != nil { + err := gtserror.Newf("db error putting mute of thread %s for account %s", threadID, accountID) + return nil, gtserror.NewErrorInternalError(err) + } + + return p.c.GetAPIStatus(ctx, requestingAccount, targetStatus) +} + +func (p *Processor) MuteRemove( + ctx context.Context, + requestingAccount *gtsmodel.Account, + targetStatusID string, +) (*apimodel.Status, gtserror.WithCode) { + targetStatus, errWithCode := p.getMuteableStatus(ctx, requestingAccount, targetStatusID) + if errWithCode != nil { + return nil, errWithCode + } + + var ( + threadID = targetStatus.ThreadID + accountID = requestingAccount.ID + ) + + // Check if mute exists for this thread ID. + threadMute, err := p.state.DB.GetThreadMutedByAccount(ctx, threadID, accountID) + if err != nil && !errors.Is(err, db.ErrNoEntries) { + // Real db error. + err := gtserror.Newf("db error fetching mute of thread %s for account %s", threadID, accountID) + return nil, gtserror.NewErrorInternalError(err) + } + + if threadMute == nil { + // Thread mute doesn't exist. + // Our job here is done ("but you didn't do anything!"). + return p.c.GetAPIStatus(ctx, requestingAccount, targetStatus) + } + + // Gotta remove the mute. + if err := p.state.DB.DeleteThreadMute(ctx, threadMute.ID); err != nil { + err := gtserror.Newf("db error deleting mute of thread %s for account %s", threadID, accountID) + return nil, gtserror.NewErrorInternalError(err) + } + + return p.c.GetAPIStatus(ctx, requestingAccount, targetStatus) +} -- cgit v1.2.3