mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-05 01:52:46 +00:00
8fdd358f4b
* Implement conversations API * Sort and page conversations by last status ID * Appease linter * Fix deleting conversations and statuses * Refactor to make migrations automatic * Lint * Update tests post-merge * Fixes from live-fire testing * Linter caught a format problem * Refactor tests, fix cache * Negative test for non-DMs * Run conversations advanced migration on testrig startup as well as regular server startup * Document (lack of) side effects of API method for deleting a conversation * Make not-found check less nested for readability * Rename PutConversation to UpsertConversation * Use util.Ptr instead of IIFE * Reduce cache used by conversations * Remove unnecessary TableExpr/ColumnExpr * Use struct tags for both unique constraints on Conversation * Make it clear how paging with GetDirectStatusIDsBatch should be used * Let conversation paging skip conversations it can't render * Use Bun NewDropTable * Convert delete raw query to Bun * Convert update raw query to Bun * Convert latestConversationStatusesTempTable raw query partially to Bun * Convert conversationStatusesTempTable raw query partially to Bun * Rename field used to store result of MaxDirectStatusID * Move advanced migrations to their own tiny processor * Catch up util function name with main * Remove json.… wrappers * Remove redundant check * Combine error checks * Replace map with slice of structs * Address processor/type converter comments - Add context info for errors - Extract some common processor code into shared methods - Move conversation eligibility check ahead of populating conversation * Add error context when dropping temp tables
243 lines
7.8 KiB
Go
243 lines
7.8 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 conversations
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
|
|
"github.com/superseriousbusiness/gotosocial/internal/db"
|
|
statusfilter "github.com/superseriousbusiness/gotosocial/internal/filter/status"
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
|
"github.com/superseriousbusiness/gotosocial/internal/id"
|
|
"github.com/superseriousbusiness/gotosocial/internal/log"
|
|
"github.com/superseriousbusiness/gotosocial/internal/util"
|
|
)
|
|
|
|
// ConversationNotification carries the arguments to processing/stream.Processor.Conversation.
|
|
type ConversationNotification struct {
|
|
// AccountID of a local account to deliver the notification to.
|
|
AccountID string
|
|
// Conversation as the notification payload.
|
|
Conversation *apimodel.Conversation
|
|
}
|
|
|
|
// UpdateConversationsForStatus updates all conversations related to a status,
|
|
// and returns a map from local account IDs to conversation notifications that should be sent to them.
|
|
func (p *Processor) UpdateConversationsForStatus(ctx context.Context, status *gtsmodel.Status) ([]ConversationNotification, error) {
|
|
if status.Visibility != gtsmodel.VisibilityDirect {
|
|
// Only DMs are considered part of conversations.
|
|
return nil, nil
|
|
}
|
|
if status.BoostOfID != "" {
|
|
// Boosts can't be part of conversations.
|
|
// FUTURE: This may change if we ever implement quote posts.
|
|
return nil, nil
|
|
}
|
|
if status.ThreadID == "" {
|
|
// If the status doesn't have a thread ID, it didn't mention a local account,
|
|
// and thus can't be part of a conversation.
|
|
return nil, nil
|
|
}
|
|
|
|
// We need accounts to be populated for this.
|
|
if err := p.state.DB.PopulateStatus(ctx, status); err != nil {
|
|
return nil, gtserror.Newf("DB error populating status %s: %w", status.ID, err)
|
|
}
|
|
|
|
// The account which authored the status plus all mentioned accounts.
|
|
allParticipantsSet := make(map[string]*gtsmodel.Account, 1+len(status.Mentions))
|
|
allParticipantsSet[status.AccountID] = status.Account
|
|
for _, mention := range status.Mentions {
|
|
allParticipantsSet[mention.TargetAccountID] = mention.TargetAccount
|
|
}
|
|
|
|
// Create or update conversations for and send notifications to each local participant.
|
|
notifications := make([]ConversationNotification, 0, len(allParticipantsSet))
|
|
for _, participant := range allParticipantsSet {
|
|
if participant.IsRemote() {
|
|
continue
|
|
}
|
|
localAccount := participant
|
|
|
|
// If the status is not visible to this account, skip processing it for this account.
|
|
visible, err := p.filter.StatusVisible(ctx, localAccount, status)
|
|
if err != nil {
|
|
log.Errorf(
|
|
ctx,
|
|
"error checking status %s visibility for account %s: %v",
|
|
status.ID,
|
|
localAccount.ID,
|
|
err,
|
|
)
|
|
continue
|
|
} else if !visible {
|
|
continue
|
|
}
|
|
|
|
// Is the status filtered or muted for this user?
|
|
// Converting the status to an API status runs the filter/mute checks.
|
|
filters, mutes, errWithCode := p.getFiltersAndMutes(ctx, localAccount)
|
|
if errWithCode != nil {
|
|
log.Error(ctx, errWithCode)
|
|
continue
|
|
}
|
|
_, err = p.converter.StatusToAPIStatus(
|
|
ctx,
|
|
status,
|
|
localAccount,
|
|
statusfilter.FilterContextNotifications,
|
|
filters,
|
|
mutes,
|
|
)
|
|
if err != nil {
|
|
// If the status matched a hide filter, skip processing it for this account.
|
|
// If there was another kind of error, log that and skip it anyway.
|
|
if !errors.Is(err, statusfilter.ErrHideStatus) {
|
|
log.Errorf(
|
|
ctx,
|
|
"error checking status %s filtering/muting for account %s: %v",
|
|
status.ID,
|
|
localAccount.ID,
|
|
err,
|
|
)
|
|
}
|
|
continue
|
|
}
|
|
|
|
// Collect other accounts participating in the conversation.
|
|
otherAccounts := make([]*gtsmodel.Account, 0, len(allParticipantsSet)-1)
|
|
otherAccountIDs := make([]string, 0, len(allParticipantsSet)-1)
|
|
for accountID, account := range allParticipantsSet {
|
|
if accountID != localAccount.ID {
|
|
otherAccounts = append(otherAccounts, account)
|
|
otherAccountIDs = append(otherAccountIDs, accountID)
|
|
}
|
|
}
|
|
|
|
// Check for a previously existing conversation, if there is one.
|
|
conversation, err := p.state.DB.GetConversationByThreadAndAccountIDs(
|
|
ctx,
|
|
status.ThreadID,
|
|
localAccount.ID,
|
|
otherAccountIDs,
|
|
)
|
|
if err != nil && !errors.Is(err, db.ErrNoEntries) {
|
|
log.Errorf(
|
|
ctx,
|
|
"error trying to find a previous conversation for status %s and account %s: %v",
|
|
status.ID,
|
|
localAccount.ID,
|
|
err,
|
|
)
|
|
continue
|
|
}
|
|
|
|
if conversation == nil {
|
|
// Create a new conversation.
|
|
conversation = >smodel.Conversation{
|
|
ID: id.NewULID(),
|
|
AccountID: localAccount.ID,
|
|
OtherAccountIDs: otherAccountIDs,
|
|
OtherAccounts: otherAccounts,
|
|
OtherAccountsKey: gtsmodel.ConversationOtherAccountsKey(otherAccountIDs),
|
|
ThreadID: status.ThreadID,
|
|
Read: util.Ptr(true),
|
|
}
|
|
}
|
|
|
|
// Assume that if the conversation owner posted the status, they've already read it.
|
|
statusAuthoredByConversationOwner := status.AccountID == conversation.AccountID
|
|
|
|
// Update the conversation.
|
|
// If there is no previous last status or this one is more recently created, set it as the last status.
|
|
if conversation.LastStatus == nil || conversation.LastStatus.CreatedAt.Before(status.CreatedAt) {
|
|
conversation.LastStatusID = status.ID
|
|
conversation.LastStatus = status
|
|
}
|
|
// If the conversation is unread, leave it marked as unread.
|
|
// If the conversation is read but this status might not have been, mark the conversation as unread.
|
|
if !statusAuthoredByConversationOwner {
|
|
conversation.Read = util.Ptr(false)
|
|
}
|
|
|
|
// Create or update the conversation.
|
|
err = p.state.DB.UpsertConversation(ctx, conversation)
|
|
if err != nil {
|
|
log.Errorf(
|
|
ctx,
|
|
"error creating or updating conversation %s for status %s and account %s: %v",
|
|
conversation.ID,
|
|
status.ID,
|
|
localAccount.ID,
|
|
err,
|
|
)
|
|
continue
|
|
}
|
|
|
|
// Link the conversation to the status.
|
|
if err := p.state.DB.LinkConversationToStatus(ctx, conversation.ID, status.ID); err != nil {
|
|
log.Errorf(
|
|
ctx,
|
|
"error linking conversation %s to status %s: %v",
|
|
conversation.ID,
|
|
status.ID,
|
|
err,
|
|
)
|
|
continue
|
|
}
|
|
|
|
// Convert the conversation to API representation.
|
|
apiConversation, err := p.converter.ConversationToAPIConversation(
|
|
ctx,
|
|
conversation,
|
|
localAccount,
|
|
filters,
|
|
mutes,
|
|
)
|
|
if err != nil {
|
|
// If the conversation's last status matched a hide filter, skip it.
|
|
// If there was another kind of error, log that and skip it anyway.
|
|
if !errors.Is(err, statusfilter.ErrHideStatus) {
|
|
log.Errorf(
|
|
ctx,
|
|
"error converting conversation %s to API representation for account %s: %v",
|
|
status.ID,
|
|
localAccount.ID,
|
|
err,
|
|
)
|
|
}
|
|
continue
|
|
}
|
|
|
|
// Generate a notification,
|
|
// unless the status was authored by the user who would be notified,
|
|
// in which case they already know.
|
|
if status.AccountID != localAccount.ID {
|
|
notifications = append(notifications, ConversationNotification{
|
|
AccountID: localAccount.ID,
|
|
Conversation: apiConversation,
|
|
})
|
|
}
|
|
}
|
|
|
|
return notifications, nil
|
|
}
|