mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-10-22 09:53:41 +00:00
* tidy up streaming * cut down code duplication * test get followers/following * test streaming processor * fix some test models * add TimeMustParse * fix uri / url typo * make trace logging less verbose * make logging more consistent * disable quote on logging * remove context.Background * remove many extraneous mastodon references * regenerate swagger * don't log query on no rows result * log latency first for easier reading
281 lines
9.8 KiB
Go
281 lines
9.8 KiB
Go
/*
|
|
GoToSocial
|
|
Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org
|
|
|
|
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 status
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
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"
|
|
)
|
|
|
|
func (p *processor) ProcessVisibility(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, accountDefaultVis gtsmodel.Visibility, status *gtsmodel.Status) error {
|
|
// by default all flags are set to true
|
|
federated := true
|
|
boostable := true
|
|
replyable := true
|
|
likeable := true
|
|
|
|
var vis gtsmodel.Visibility
|
|
// If visibility isn't set on the form, then just take the account default.
|
|
// If that's also not set, take the default for the whole instance.
|
|
if form.Visibility != "" {
|
|
vis = p.tc.APIVisToVis(form.Visibility)
|
|
} else if accountDefaultVis != "" {
|
|
vis = accountDefaultVis
|
|
} else {
|
|
vis = gtsmodel.VisibilityDefault
|
|
}
|
|
|
|
switch vis {
|
|
case gtsmodel.VisibilityPublic:
|
|
// for public, there's no need to change any of the advanced flags from true regardless of what the user filled out
|
|
break
|
|
case gtsmodel.VisibilityUnlocked:
|
|
// for unlocked the user can set any combination of flags they like so look at them all to see if they're set and then apply them
|
|
if form.Federated != nil {
|
|
federated = *form.Federated
|
|
}
|
|
|
|
if form.Boostable != nil {
|
|
boostable = *form.Boostable
|
|
}
|
|
|
|
if form.Replyable != nil {
|
|
replyable = *form.Replyable
|
|
}
|
|
|
|
if form.Likeable != nil {
|
|
likeable = *form.Likeable
|
|
}
|
|
|
|
case gtsmodel.VisibilityFollowersOnly, gtsmodel.VisibilityMutualsOnly:
|
|
// for followers or mutuals only, boostable will *always* be false, but the other fields can be set so check and apply them
|
|
boostable = false
|
|
|
|
if form.Federated != nil {
|
|
federated = *form.Federated
|
|
}
|
|
|
|
if form.Replyable != nil {
|
|
replyable = *form.Replyable
|
|
}
|
|
|
|
if form.Likeable != nil {
|
|
likeable = *form.Likeable
|
|
}
|
|
|
|
case gtsmodel.VisibilityDirect:
|
|
// direct is pretty easy: there's only one possible setting so return it
|
|
federated = true
|
|
boostable = false
|
|
replyable = true
|
|
likeable = true
|
|
}
|
|
|
|
status.Visibility = vis
|
|
status.Federated = federated
|
|
status.Boostable = boostable
|
|
status.Replyable = replyable
|
|
status.Likeable = likeable
|
|
return nil
|
|
}
|
|
|
|
func (p *processor) ProcessReplyToID(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, thisAccountID string, status *gtsmodel.Status) error {
|
|
if form.InReplyToID == "" {
|
|
return nil
|
|
}
|
|
|
|
// If this status is a reply to another status, we need to do a bit of work to establish whether or not this status can be posted:
|
|
//
|
|
// 1. Does the replied status exist in the database?
|
|
// 2. Is the replied status marked as replyable?
|
|
// 3. Does a block exist between either the current account or the account that posted the status it's replying to?
|
|
//
|
|
// If this is all OK, then we fetch the repliedStatus and the repliedAccount for later processing.
|
|
repliedStatus := >smodel.Status{}
|
|
repliedAccount := >smodel.Account{}
|
|
// check replied status exists + is replyable
|
|
if err := p.db.GetByID(ctx, form.InReplyToID, repliedStatus); err != nil {
|
|
if err == db.ErrNoEntries {
|
|
return fmt.Errorf("status with id %s not replyable because it doesn't exist", form.InReplyToID)
|
|
}
|
|
return fmt.Errorf("status with id %s not replyable: %s", form.InReplyToID, err)
|
|
}
|
|
if !repliedStatus.Replyable {
|
|
return fmt.Errorf("status with id %s is marked as not replyable", form.InReplyToID)
|
|
}
|
|
|
|
// check replied account is known to us
|
|
if err := p.db.GetByID(ctx, repliedStatus.AccountID, repliedAccount); err != nil {
|
|
if err == db.ErrNoEntries {
|
|
return fmt.Errorf("status with id %s not replyable because account id %s is not known", form.InReplyToID, repliedStatus.AccountID)
|
|
}
|
|
return fmt.Errorf("status with id %s not replyable: %s", form.InReplyToID, err)
|
|
}
|
|
// check if a block exists
|
|
if blocked, err := p.db.IsBlocked(ctx, thisAccountID, repliedAccount.ID, true); err != nil {
|
|
if err != db.ErrNoEntries {
|
|
return fmt.Errorf("status with id %s not replyable: %s", form.InReplyToID, err)
|
|
}
|
|
} else if blocked {
|
|
return fmt.Errorf("status with id %s not replyable", form.InReplyToID)
|
|
}
|
|
status.InReplyToID = repliedStatus.ID
|
|
status.InReplyToAccountID = repliedAccount.ID
|
|
|
|
return nil
|
|
}
|
|
|
|
func (p *processor) ProcessMediaIDs(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, thisAccountID string, status *gtsmodel.Status) error {
|
|
if form.MediaIDs == nil {
|
|
return nil
|
|
}
|
|
|
|
gtsMediaAttachments := []*gtsmodel.MediaAttachment{}
|
|
attachments := []string{}
|
|
for _, mediaID := range form.MediaIDs {
|
|
// check these attachments exist
|
|
a := >smodel.MediaAttachment{}
|
|
if err := p.db.GetByID(ctx, mediaID, a); err != nil {
|
|
return fmt.Errorf("invalid media type or media not found for media id %s", mediaID)
|
|
}
|
|
// check they belong to the requesting account id
|
|
if a.AccountID != thisAccountID {
|
|
return fmt.Errorf("media with id %s does not belong to account %s", mediaID, thisAccountID)
|
|
}
|
|
// check they're not already used in a status
|
|
if a.StatusID != "" || a.ScheduledStatusID != "" {
|
|
return fmt.Errorf("media with id %s is already attached to a status", mediaID)
|
|
}
|
|
gtsMediaAttachments = append(gtsMediaAttachments, a)
|
|
attachments = append(attachments, a.ID)
|
|
}
|
|
status.Attachments = gtsMediaAttachments
|
|
status.AttachmentIDs = attachments
|
|
return nil
|
|
}
|
|
|
|
func (p *processor) ProcessLanguage(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, accountDefaultLanguage string, status *gtsmodel.Status) error {
|
|
if form.Language != "" {
|
|
status.Language = form.Language
|
|
} else {
|
|
status.Language = accountDefaultLanguage
|
|
}
|
|
if status.Language == "" {
|
|
return errors.New("no language given either in status create form or account default")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
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()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
menchie.ID = menchieID
|
|
|
|
if err := p.db.Put(ctx, menchie); err != nil {
|
|
return fmt.Errorf("error putting mentions in db: %s", err)
|
|
}
|
|
menchies = append(menchies, menchie.ID)
|
|
}
|
|
// add full populated gts menchies to the status for passing them around conveniently
|
|
status.Mentions = gtsMenchies
|
|
// add just the ids of the mentioned accounts to the status for putting in the db
|
|
status.MentionIDs = menchies
|
|
return nil
|
|
}
|
|
|
|
func (p *processor) ProcessTags(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, accountID string, status *gtsmodel.Status) error {
|
|
tags := []string{}
|
|
gtsTags, err := p.db.TagStringsToTags(ctx, util.DeriveHashtagsFromText(form.Status), accountID)
|
|
if err != nil {
|
|
return fmt.Errorf("error generating hashtags from status: %s", err)
|
|
}
|
|
for _, tag := range gtsTags {
|
|
if err := p.db.Put(ctx, tag); err != nil && err != db.ErrAlreadyExists {
|
|
return fmt.Errorf("error putting tags in db: %s", err)
|
|
}
|
|
tags = append(tags, tag.ID)
|
|
}
|
|
// add full populated gts tags to the status for passing them around conveniently
|
|
status.Tags = gtsTags
|
|
// add just the ids of the used tags to the status for putting in the db
|
|
status.TagIDs = tags
|
|
return nil
|
|
}
|
|
|
|
func (p *processor) ProcessEmojis(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, accountID string, status *gtsmodel.Status) error {
|
|
emojis := []string{}
|
|
gtsEmojis, err := p.db.EmojiStringsToEmojis(ctx, util.DeriveEmojisFromText(form.Status))
|
|
if err != nil {
|
|
return fmt.Errorf("error generating emojis from status: %s", err)
|
|
}
|
|
for _, e := range gtsEmojis {
|
|
emojis = append(emojis, e.ID)
|
|
}
|
|
// add full populated gts emojis to the status for passing them around conveniently
|
|
status.Emojis = gtsEmojis
|
|
// add just the ids of the used emojis to the status for putting in the db
|
|
status.EmojiIDs = emojis
|
|
return nil
|
|
}
|
|
|
|
func (p *processor) ProcessContent(ctx context.Context, form *apimodel.AdvancedStatusCreateForm, accountID string, status *gtsmodel.Status) error {
|
|
// if there's nothing in the status at all we can just return early
|
|
if form.Status == "" {
|
|
status.Content = ""
|
|
return nil
|
|
}
|
|
|
|
// if format wasn't specified we should set the default
|
|
if form.Format == "" {
|
|
form.Format = apimodel.StatusFormatDefault
|
|
}
|
|
|
|
// remove any existing html from the status
|
|
content := text.RemoveHTML(form.Status)
|
|
|
|
// parse content out of the status depending on what format has been submitted
|
|
var formatted string
|
|
switch form.Format {
|
|
case apimodel.StatusFormatPlain:
|
|
formatted = p.formatter.FromPlain(ctx, content, status.Mentions, status.Tags)
|
|
case apimodel.StatusFormatMarkdown:
|
|
formatted = p.formatter.FromMarkdown(ctx, content, status.Mentions, status.Tags)
|
|
default:
|
|
return fmt.Errorf("format %s not recognised as a valid status format", form.Format)
|
|
}
|
|
|
|
status.Content = formatted
|
|
return nil
|
|
}
|