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/create.go | 37 +++++++++++++++++++++++++++++++++++-
 1 file changed, 36 insertions(+), 1 deletion(-)

(limited to 'internal/processing/status/create.go')

diff --git a/internal/processing/status/create.go b/internal/processing/status/create.go
index ee4466b1b..40b3f2df2 100644
--- a/internal/processing/status/create.go
+++ b/internal/processing/status/create.go
@@ -70,6 +70,10 @@ func (p *Processor) Create(ctx context.Context, requestingAccount *gtsmodel.Acco
 		return nil, errWithCode
 	}
 
+	if errWithCode := p.processThreadID(ctx, status); errWithCode != nil {
+		return nil, errWithCode
+	}
+
 	if errWithCode := p.processMediaIDs(ctx, form, requestingAccount.ID, status); errWithCode != nil {
 		return nil, errWithCode
 	}
@@ -99,7 +103,7 @@ func (p *Processor) Create(ctx context.Context, requestingAccount *gtsmodel.Acco
 		OriginAccount:  requestingAccount,
 	})
 
-	return p.apiStatus(ctx, status, requestingAccount)
+	return p.c.GetAPIStatus(ctx, requestingAccount, status)
 }
 
 func (p *Processor) processReplyToID(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, thisAccountID string, status *gtsmodel.Status) gtserror.WithCode {
@@ -141,12 +145,43 @@ func (p *Processor) processReplyToID(ctx context.Context, form *apimodel.Advance
 
 	// Set status fields from inReplyTo.
 	status.InReplyToID = inReplyTo.ID
+	status.InReplyTo = inReplyTo
 	status.InReplyToURI = inReplyTo.URI
 	status.InReplyToAccountID = inReplyTo.AccountID
 
 	return nil
 }
 
+func (p *Processor) processThreadID(ctx context.Context, status *gtsmodel.Status) gtserror.WithCode {
+	// Status takes the thread ID
+	// of whatever it replies to.
+	if status.InReplyTo != nil {
+		status.ThreadID = status.InReplyTo.ThreadID
+		return nil
+	}
+
+	// Status doesn't reply to anything,
+	// so it's a new local top-level status
+	// and therefore needs a thread ID.
+	threadID := id.NewULID()
+
+	if err := p.state.DB.PutThread(
+		ctx,
+		&gtsmodel.Thread{
+			ID: threadID,
+		},
+	); err != nil {
+		err := gtserror.Newf("error inserting new thread in db: %w", err)
+		return gtserror.NewErrorInternalError(err)
+	}
+
+	// Future replies to this status
+	// (if any) will inherit this thread ID.
+	status.ThreadID = threadID
+
+	return nil
+}
+
 func (p *Processor) processMediaIDs(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, thisAccountID string, status *gtsmodel.Status) gtserror.WithCode {
 	if form.MediaIDs == nil {
 		return nil
-- 
cgit v1.2.3