mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-11-03 19:32:26 -06:00 
			
		
		
		
	
		
			
				
	
	
		
			237 lines
		
	
	
	
		
			7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			237 lines
		
	
	
	
		
			7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// 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 <http://www.gnu.org/licenses/>.
 | 
						|
 | 
						|
package federatingdb
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"errors"
 | 
						|
	"fmt"
 | 
						|
 | 
						|
	"codeberg.org/gruf/go-logger/v2/level"
 | 
						|
	"github.com/superseriousbusiness/activity/streams/vocab"
 | 
						|
	"github.com/superseriousbusiness/gotosocial/internal/ap"
 | 
						|
	"github.com/superseriousbusiness/gotosocial/internal/db"
 | 
						|
	"github.com/superseriousbusiness/gotosocial/internal/gtscontext"
 | 
						|
	"github.com/superseriousbusiness/gotosocial/internal/gtserror"
 | 
						|
	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
 | 
						|
	"github.com/superseriousbusiness/gotosocial/internal/log"
 | 
						|
)
 | 
						|
 | 
						|
func (f *federatingDB) Undo(ctx context.Context, undo vocab.ActivityStreamsUndo) error {
 | 
						|
	l := log.WithContext(ctx)
 | 
						|
 | 
						|
	if log.Level() >= level.DEBUG {
 | 
						|
		i, err := marshalItem(undo)
 | 
						|
		if err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
		l = l.WithField("undo", i)
 | 
						|
		l.Debug("entering Undo")
 | 
						|
	}
 | 
						|
 | 
						|
	activityContext := getActivityContext(ctx)
 | 
						|
	if activityContext.internal {
 | 
						|
		return nil // Already processed.
 | 
						|
	}
 | 
						|
 | 
						|
	requestingAcct := activityContext.requestingAcct
 | 
						|
	receivingAcct := activityContext.receivingAcct
 | 
						|
 | 
						|
	var errs gtserror.MultiError
 | 
						|
	for _, object := range ap.ExtractObjects(undo) {
 | 
						|
		// Try to get object as vocab.Type,
 | 
						|
		// else skip handling (likely) IRI.
 | 
						|
		objType := object.GetType()
 | 
						|
		if objType == nil {
 | 
						|
			continue
 | 
						|
		}
 | 
						|
 | 
						|
		switch objType.GetTypeName() {
 | 
						|
		case ap.ActivityFollow:
 | 
						|
			if err := f.undoFollow(ctx, receivingAcct, requestingAcct, undo, objType); err != nil {
 | 
						|
				errs.Appendf("error undoing follow: %w", err)
 | 
						|
			}
 | 
						|
		case ap.ActivityLike:
 | 
						|
			if err := f.undoLike(ctx, receivingAcct, requestingAcct, undo, objType); err != nil {
 | 
						|
				errs.Appendf("error undoing like: %w", err)
 | 
						|
			}
 | 
						|
		case ap.ActivityAnnounce:
 | 
						|
			// TODO: actually handle this !
 | 
						|
			log.Warn(ctx, "skipped undo announce")
 | 
						|
		case ap.ActivityBlock:
 | 
						|
			if err := f.undoBlock(ctx, receivingAcct, requestingAcct, undo, objType); err != nil {
 | 
						|
				errs.Appendf("error undoing block: %w", err)
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return errs.Combine()
 | 
						|
}
 | 
						|
 | 
						|
func (f *federatingDB) undoFollow(
 | 
						|
	ctx context.Context,
 | 
						|
	receivingAccount *gtsmodel.Account,
 | 
						|
	requestingAccount *gtsmodel.Account,
 | 
						|
	undo vocab.ActivityStreamsUndo,
 | 
						|
	t vocab.Type,
 | 
						|
) error {
 | 
						|
	Follow, ok := t.(vocab.ActivityStreamsFollow)
 | 
						|
	if !ok {
 | 
						|
		return errors.New("undoFollow: couldn't parse vocab.Type into vocab.ActivityStreamsFollow")
 | 
						|
	}
 | 
						|
 | 
						|
	// Make sure the undo actor owns the target.
 | 
						|
	if !sameActor(undo.GetActivityStreamsActor(), Follow.GetActivityStreamsActor()) {
 | 
						|
		// Ignore this Activity.
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	follow, err := f.converter.ASFollowToFollow(ctx, Follow)
 | 
						|
	if err != nil {
 | 
						|
		return fmt.Errorf("undoFollow: error converting ActivityStreams Follow to follow: %w", err)
 | 
						|
	}
 | 
						|
 | 
						|
	// Ensure addressee is follow target.
 | 
						|
	if follow.TargetAccountID != receivingAccount.ID {
 | 
						|
		// Ignore this Activity.
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	// Ensure requester is follow origin.
 | 
						|
	if follow.AccountID != requestingAccount.ID {
 | 
						|
		// Ignore this Activity.
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	// Delete any existing follow with this URI.
 | 
						|
	if err := f.state.DB.DeleteFollowByURI(ctx, follow.URI); err != nil && !errors.Is(err, db.ErrNoEntries) {
 | 
						|
		return fmt.Errorf("undoFollow: db error removing follow: %w", err)
 | 
						|
	}
 | 
						|
 | 
						|
	// Delete any existing follow request with this URI.
 | 
						|
	if err := f.state.DB.DeleteFollowRequestByURI(ctx, follow.URI); err != nil && !errors.Is(err, db.ErrNoEntries) {
 | 
						|
		return fmt.Errorf("undoFollow: db error removing follow request: %w", err)
 | 
						|
	}
 | 
						|
 | 
						|
	log.Debug(ctx, "Follow undone")
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
func (f *federatingDB) undoLike(
 | 
						|
	ctx context.Context,
 | 
						|
	receivingAccount *gtsmodel.Account,
 | 
						|
	requestingAccount *gtsmodel.Account,
 | 
						|
	undo vocab.ActivityStreamsUndo,
 | 
						|
	t vocab.Type,
 | 
						|
) error {
 | 
						|
	Like, ok := t.(vocab.ActivityStreamsLike)
 | 
						|
	if !ok {
 | 
						|
		return errors.New("undoLike: couldn't parse vocab.Type into vocab.ActivityStreamsLike")
 | 
						|
	}
 | 
						|
 | 
						|
	// Make sure the undo actor owns the target.
 | 
						|
	if !sameActor(undo.GetActivityStreamsActor(), Like.GetActivityStreamsActor()) {
 | 
						|
		// Ignore this Activity.
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	fave, err := f.converter.ASLikeToFave(ctx, Like)
 | 
						|
	if err != nil {
 | 
						|
		return fmt.Errorf("undoLike: error converting ActivityStreams Like to fave: %w", err)
 | 
						|
	}
 | 
						|
 | 
						|
	// Ensure addressee is fave target.
 | 
						|
	if fave.TargetAccountID != receivingAccount.ID {
 | 
						|
		// Ignore this Activity.
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	// Ensure requester is fave origin.
 | 
						|
	if fave.AccountID != requestingAccount.ID {
 | 
						|
		// Ignore this Activity.
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	// Ignore URI on Likes, since we often get multiple Likes
 | 
						|
	// with the same target and account ID, but differing URIs.
 | 
						|
	// Instead, we'll select using account and target status.
 | 
						|
	// Regardless of the URI, we can read an Undo Like to mean
 | 
						|
	// "I don't want to fave this post anymore".
 | 
						|
	fave, err = f.state.DB.GetStatusFave(gtscontext.SetBarebones(ctx), fave.AccountID, fave.StatusID)
 | 
						|
	if err != nil {
 | 
						|
		if errors.Is(err, db.ErrNoEntries) {
 | 
						|
			// We didn't have a like/fave
 | 
						|
			// for this combo anyway, ignore.
 | 
						|
			return nil
 | 
						|
		}
 | 
						|
		// Real error.
 | 
						|
		return fmt.Errorf("undoLike: db error getting fave from %s targeting %s: %w", fave.AccountID, fave.StatusID, err)
 | 
						|
	}
 | 
						|
 | 
						|
	// Delete the status fave.
 | 
						|
	if err := f.state.DB.DeleteStatusFaveByID(ctx, fave.ID); err != nil {
 | 
						|
		return fmt.Errorf("undoLike: db error deleting fave %s: %w", fave.ID, err)
 | 
						|
	}
 | 
						|
 | 
						|
	log.Debug(ctx, "Like undone")
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
func (f *federatingDB) undoBlock(
 | 
						|
	ctx context.Context,
 | 
						|
	receivingAccount *gtsmodel.Account,
 | 
						|
	requestingAccount *gtsmodel.Account,
 | 
						|
	undo vocab.ActivityStreamsUndo,
 | 
						|
	t vocab.Type,
 | 
						|
) error {
 | 
						|
	Block, ok := t.(vocab.ActivityStreamsBlock)
 | 
						|
	if !ok {
 | 
						|
		return errors.New("undoBlock: couldn't parse vocab.Type into vocab.ActivityStreamsBlock")
 | 
						|
	}
 | 
						|
 | 
						|
	// Make sure the undo actor owns the target.
 | 
						|
	if !sameActor(undo.GetActivityStreamsActor(), Block.GetActivityStreamsActor()) {
 | 
						|
		// Ignore this Activity.
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	block, err := f.converter.ASBlockToBlock(ctx, Block)
 | 
						|
	if err != nil {
 | 
						|
		return fmt.Errorf("undoBlock: error converting ActivityStreams Block to block: %w", err)
 | 
						|
	}
 | 
						|
 | 
						|
	// Ensure addressee is block target.
 | 
						|
	if block.TargetAccountID != receivingAccount.ID {
 | 
						|
		// Ignore this Activity.
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	// Ensure requester is block origin.
 | 
						|
	if block.AccountID != requestingAccount.ID {
 | 
						|
		// Ignore this Activity.
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	// Delete any existing BLOCK
 | 
						|
	if err := f.state.DB.DeleteBlockByURI(ctx, block.URI); err != nil && !errors.Is(err, db.ErrNoEntries) {
 | 
						|
		return fmt.Errorf("undoBlock: db error removing block: %w", err)
 | 
						|
	}
 | 
						|
 | 
						|
	log.Debug(ctx, "Block undone")
 | 
						|
	return nil
 | 
						|
}
 |