diff options
| author | 2022-03-29 11:54:56 +0200 | |
|---|---|---|
| committer | 2022-03-29 11:54:56 +0200 | |
| commit | 37d310f981f3e26bc643aeabac134b591c84455a (patch) | |
| tree | 041f5db16254b5257471bcac0730ac9d93ce13d5 /internal/processing/status | |
| parent | [feature/security] Add systemd sandboxing options to harden security (#440) (diff) | |
| download | gotosocial-37d310f981f3e26bc643aeabac134b591c84455a.tar.xz | |
[feature] Dereference remote mentions when the account is not already known (#442)v0.2.2
* remove mention util function from db
* add ParseMentionFunc to gtsmodel
* add parseMentionFunc to processor
* refactor search to simplify it a bit
* add parseMentionFunc to account
* add parseMentionFunc to status
* some renaming for clarity
* test dereference of unknown mentioned account
Diffstat (limited to 'internal/processing/status')
| -rw-r--r-- | internal/processing/status/status.go | 4 | ||||
| -rw-r--r-- | internal/processing/status/status_test.go | 21 | ||||
| -rw-r--r-- | internal/processing/status/util.go | 33 | 
3 files changed, 39 insertions, 19 deletions
diff --git a/internal/processing/status/status.go b/internal/processing/status/status.go index 3118b23ad..421ab5bbe 100644 --- a/internal/processing/status/status.go +++ b/internal/processing/status/status.go @@ -74,15 +74,17 @@ type processor struct {  	filter        visibility.Filter  	formatter     text.Formatter  	fromClientAPI chan messages.FromClientAPI +	parseMention  gtsmodel.ParseMentionFunc  }  // New returns a new status processor. -func New(db db.DB, tc typeutils.TypeConverter, fromClientAPI chan messages.FromClientAPI) Processor { +func New(db db.DB, tc typeutils.TypeConverter, fromClientAPI chan messages.FromClientAPI, parseMention gtsmodel.ParseMentionFunc) Processor {  	return &processor{  		tc:            tc,  		db:            db,  		filter:        visibility.NewFilter(db),  		formatter:     text.NewFormatter(db),  		fromClientAPI: fromClientAPI, +		parseMention:  parseMention,  	}  } diff --git a/internal/processing/status/status_test.go b/internal/processing/status/status_test.go index 803e7a0a5..25794070e 100644 --- a/internal/processing/status/status_test.go +++ b/internal/processing/status/status_test.go @@ -19,11 +19,16 @@  package status_test  import ( +	"codeberg.org/gruf/go-store/kv"  	"github.com/stretchr/testify/suite"  	"github.com/superseriousbusiness/gotosocial/internal/db" +	"github.com/superseriousbusiness/gotosocial/internal/federation"  	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" +	"github.com/superseriousbusiness/gotosocial/internal/media"  	"github.com/superseriousbusiness/gotosocial/internal/messages" +	"github.com/superseriousbusiness/gotosocial/internal/processing"  	"github.com/superseriousbusiness/gotosocial/internal/processing/status" +	"github.com/superseriousbusiness/gotosocial/internal/transport"  	"github.com/superseriousbusiness/gotosocial/internal/typeutils"  	"github.com/superseriousbusiness/gotosocial/testrig"  ) @@ -32,6 +37,10 @@ type StatusStandardTestSuite struct {  	suite.Suite  	db                db.DB  	typeConverter     typeutils.TypeConverter +	tc                transport.Controller +	storage           *kv.KVStore +	mediaManager      media.Manager +	federator         federation.Federator  	fromClientAPIChan chan messages.FromClientAPI  	// standard suite models @@ -62,17 +71,23 @@ func (suite *StatusStandardTestSuite) SetupSuite() {  }  func (suite *StatusStandardTestSuite) SetupTest() { -	testrig.InitTestLog()  	testrig.InitTestConfig() +	testrig.InitTestLog()  	suite.db = testrig.NewTestDB()  	suite.typeConverter = testrig.NewTestTypeConverter(suite.db)  	suite.fromClientAPIChan = make(chan messages.FromClientAPI, 100) -	suite.status = status.New(suite.db, suite.typeConverter, suite.fromClientAPIChan) +	suite.tc = testrig.NewTestTransportController(testrig.NewMockHTTPClient(nil), suite.db) +	suite.storage = testrig.NewTestStorage() +	suite.mediaManager = testrig.NewTestMediaManager(suite.db, suite.storage) +	suite.federator = testrig.NewTestFederator(suite.db, suite.tc, suite.storage, suite.mediaManager) +	suite.status = status.New(suite.db, suite.typeConverter, suite.fromClientAPIChan, processing.GetParseMentionFunc(suite.db, suite.federator)) -	testrig.StandardDBSetup(suite.db, nil) +	testrig.StandardDBSetup(suite.db, suite.testAccounts) +	testrig.StandardStorageSetup(suite.storage, "../../../testrig/media")  }  func (suite *StatusStandardTestSuite) TearDownTest() {  	testrig.StandardDBTeardown(suite.db) +	testrig.StandardStorageTeardown(suite.storage)  } diff --git a/internal/processing/status/util.go b/internal/processing/status/util.go index f2640929d..5de66af8a 100644 --- a/internal/processing/status/util.go +++ b/internal/processing/status/util.go @@ -23,10 +23,10 @@ import (  	"errors"  	"fmt" +	"github.com/sirupsen/logrus"  	apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"  	"github.com/superseriousbusiness/gotosocial/internal/db"  	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" -	"github.com/superseriousbusiness/gotosocial/internal/id"  	"github.com/superseriousbusiness/gotosocial/internal/text"  	"github.com/superseriousbusiness/gotosocial/internal/util"  ) @@ -192,27 +192,30 @@ func (p *processor) ProcessLanguage(ctx context.Context, form *apimodel.Advanced  }  func (p *processor) ProcessMentions(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, accountID string, status *gtsmodel.Status) error { -	menchies := []string{} -	gtsMenchies, err := p.db.MentionStringsToMentions(ctx, util.DeriveMentionsFromText(form.Status), accountID, status.ID) -	if err != nil { -		return fmt.Errorf("error generating mentions from status: %s", err) -	} -	for _, menchie := range gtsMenchies { -		menchieID, err := id.NewRandomULID() +	mentionedAccountNames := util.DeriveMentionNamesFromText(form.Status) +	mentions := []*gtsmodel.Mention{} +	mentionIDs := []string{} + +	for _, mentionedAccountName := range mentionedAccountNames { +		gtsMention, err := p.parseMention(ctx, mentionedAccountName, accountID, status.ID)  		if err != nil { -			return err +			logrus.Errorf("ProcessMentions: error parsing mention %s from status: %s", mentionedAccountName, err) +			continue  		} -		menchie.ID = menchieID -		if err := p.db.Put(ctx, menchie); err != nil { -			return fmt.Errorf("error putting mentions in db: %s", err) +		if err := p.db.Put(ctx, gtsMention); err != nil { +			logrus.Errorf("ProcessMentions: error putting mention in db: %s", err)  		} -		menchies = append(menchies, menchie.ID) + +		mentions = append(mentions, gtsMention) +		mentionIDs = append(mentionIDs, gtsMention.ID)  	} +  	// add full populated gts menchies to the status for passing them around conveniently -	status.Mentions = gtsMenchies +	status.Mentions = mentions  	// add just the ids of the mentioned accounts to the status for putting in the db -	status.MentionIDs = menchies +	status.MentionIDs = mentionIDs +  	return nil  }  | 
