293 lines
9.4 KiB
Go
293 lines
9.4 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 mastotypes
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/superseriousbusiness/gotosocial/internal/config"
|
|
"github.com/superseriousbusiness/gotosocial/internal/db"
|
|
"github.com/superseriousbusiness/gotosocial/internal/db/gtsmodel"
|
|
mastotypes "github.com/superseriousbusiness/gotosocial/internal/mastotypes/mastomodel"
|
|
"github.com/superseriousbusiness/gotosocial/internal/util"
|
|
)
|
|
|
|
// Converter is an interface for the common action of converting between mastotypes (frontend, serializable) models and internal gts models used in the database.
|
|
// It requires access to the database because many of the conversions require pulling out database entries and counting them etc.
|
|
type Converter interface {
|
|
// AccountToMastoSensitive takes a db model account as a param, and returns a populated mastotype account, or an error
|
|
// if something goes wrong. The returned account should be ready to serialize on an API level, and may have sensitive fields,
|
|
// so serve it only to an authorized user who should have permission to see it.
|
|
AccountToMastoSensitive(account *gtsmodel.Account) (*mastotypes.Account, error)
|
|
|
|
// AccountToMastoPublic takes a db model account as a param, and returns a populated mastotype account, or an error
|
|
// if something goes wrong. The returned account should be ready to serialize on an API level, and may NOT have sensitive fields.
|
|
// In other words, this is the public record that the server has of an account.
|
|
AccountToMastoPublic(account *gtsmodel.Account) (*mastotypes.Account, error)
|
|
|
|
// AppToMastoSensitive takes a db model application as a param, and returns a populated mastotype application, or an error
|
|
// if something goes wrong. The returned application should be ready to serialize on an API level, and may have sensitive fields
|
|
// (such as client id and client secret), so serve it only to an authorized user who should have permission to see it.
|
|
AppToMastoSensitive(application *gtsmodel.Application) (*mastotypes.Application, error)
|
|
|
|
// AppToMastoPublic takes a db model application as a param, and returns a populated mastotype application, or an error
|
|
// if something goes wrong. The returned application should be ready to serialize on an API level, and has sensitive
|
|
// fields sanitized so that it can be served to non-authorized accounts without revealing any private information.
|
|
AppToMastoPublic(application *gtsmodel.Application) (*mastotypes.Application, error)
|
|
|
|
AttachmentToMasto(attachment *gtsmodel.MediaAttachment) (mastotypes.Attachment, error)
|
|
|
|
MentionToMasto(m *gtsmodel.Mention) (mastotypes.Mention, error)
|
|
}
|
|
|
|
type converter struct {
|
|
config *config.Config
|
|
db db.DB
|
|
}
|
|
|
|
func New(config *config.Config, db db.DB) Converter {
|
|
return &converter{
|
|
config: config,
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (c *converter) AccountToMastoSensitive(a *gtsmodel.Account) (*mastotypes.Account, error) {
|
|
// we can build this sensitive account easily by first getting the public account....
|
|
mastoAccount, err := c.AccountToMastoPublic(a)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// then adding the Source object to it...
|
|
|
|
// check pending follow requests aimed at this account
|
|
fr := []gtsmodel.FollowRequest{}
|
|
if err := c.db.GetFollowRequestsForAccountID(a.ID, &fr); err != nil {
|
|
if _, ok := err.(db.ErrNoEntries); !ok {
|
|
return nil, fmt.Errorf("error getting follow requests: %s", err)
|
|
}
|
|
}
|
|
var frc int
|
|
if fr != nil {
|
|
frc = len(fr)
|
|
}
|
|
|
|
mastoAccount.Source = &mastotypes.Source{
|
|
Privacy: util.ParseMastoVisFromGTSVis(a.Privacy),
|
|
Sensitive: a.Sensitive,
|
|
Language: a.Language,
|
|
Note: a.Note,
|
|
Fields: mastoAccount.Fields,
|
|
FollowRequestsCount: frc,
|
|
}
|
|
|
|
return mastoAccount, nil
|
|
}
|
|
|
|
func (c *converter) AccountToMastoPublic(a *gtsmodel.Account) (*mastotypes.Account, error) {
|
|
// count followers
|
|
followers := []gtsmodel.Follow{}
|
|
if err := c.db.GetFollowersByAccountID(a.ID, &followers); err != nil {
|
|
if _, ok := err.(db.ErrNoEntries); !ok {
|
|
return nil, fmt.Errorf("error getting followers: %s", err)
|
|
}
|
|
}
|
|
var followersCount int
|
|
if followers != nil {
|
|
followersCount = len(followers)
|
|
}
|
|
|
|
// count following
|
|
following := []gtsmodel.Follow{}
|
|
if err := c.db.GetFollowingByAccountID(a.ID, &following); err != nil {
|
|
if _, ok := err.(db.ErrNoEntries); !ok {
|
|
return nil, fmt.Errorf("error getting following: %s", err)
|
|
}
|
|
}
|
|
var followingCount int
|
|
if following != nil {
|
|
followingCount = len(following)
|
|
}
|
|
|
|
// count statuses
|
|
statuses := []gtsmodel.Status{}
|
|
if err := c.db.GetStatusesByAccountID(a.ID, &statuses); err != nil {
|
|
if _, ok := err.(db.ErrNoEntries); !ok {
|
|
return nil, fmt.Errorf("error getting last statuses: %s", err)
|
|
}
|
|
}
|
|
var statusesCount int
|
|
if statuses != nil {
|
|
statusesCount = len(statuses)
|
|
}
|
|
|
|
// check when the last status was
|
|
lastStatus := >smodel.Status{}
|
|
if err := c.db.GetLastStatusForAccountID(a.ID, lastStatus); err != nil {
|
|
if _, ok := err.(db.ErrNoEntries); !ok {
|
|
return nil, fmt.Errorf("error getting last status: %s", err)
|
|
}
|
|
}
|
|
var lastStatusAt string
|
|
if lastStatus != nil {
|
|
lastStatusAt = lastStatus.CreatedAt.Format(time.RFC3339)
|
|
}
|
|
|
|
// build the avatar and header URLs
|
|
avi := >smodel.MediaAttachment{}
|
|
if err := c.db.GetAvatarForAccountID(avi, a.ID); err != nil {
|
|
if _, ok := err.(db.ErrNoEntries); !ok {
|
|
return nil, fmt.Errorf("error getting avatar: %s", err)
|
|
}
|
|
}
|
|
aviURL := avi.File.Path
|
|
aviURLStatic := avi.Thumbnail.Path
|
|
|
|
header := >smodel.MediaAttachment{}
|
|
if err := c.db.GetHeaderForAccountID(avi, a.ID); err != nil {
|
|
if _, ok := err.(db.ErrNoEntries); !ok {
|
|
return nil, fmt.Errorf("error getting header: %s", err)
|
|
}
|
|
}
|
|
headerURL := header.File.Path
|
|
headerURLStatic := header.Thumbnail.Path
|
|
|
|
// get the fields set on this account
|
|
fields := []mastotypes.Field{}
|
|
for _, f := range a.Fields {
|
|
mField := mastotypes.Field{
|
|
Name: f.Name,
|
|
Value: f.Value,
|
|
}
|
|
if !f.VerifiedAt.IsZero() {
|
|
mField.VerifiedAt = f.VerifiedAt.Format(time.RFC3339)
|
|
}
|
|
fields = append(fields, mField)
|
|
}
|
|
|
|
var acct string
|
|
if a.Domain != "" {
|
|
// this is a remote user
|
|
acct = fmt.Sprintf("%s@%s", a.Username, a.Domain)
|
|
} else {
|
|
// this is a local user
|
|
acct = a.Username
|
|
}
|
|
|
|
return &mastotypes.Account{
|
|
ID: a.ID,
|
|
Username: a.Username,
|
|
Acct: acct,
|
|
DisplayName: a.DisplayName,
|
|
Locked: a.Locked,
|
|
Bot: a.Bot,
|
|
CreatedAt: a.CreatedAt.Format(time.RFC3339),
|
|
Note: a.Note,
|
|
URL: a.URL,
|
|
Avatar: aviURL,
|
|
AvatarStatic: aviURLStatic,
|
|
Header: headerURL,
|
|
HeaderStatic: headerURLStatic,
|
|
FollowersCount: followersCount,
|
|
FollowingCount: followingCount,
|
|
StatusesCount: statusesCount,
|
|
LastStatusAt: lastStatusAt,
|
|
Emojis: nil, // TODO: implement this
|
|
Fields: fields,
|
|
}, nil
|
|
}
|
|
|
|
func (c *converter) AppToMastoSensitive(a *gtsmodel.Application) (*mastotypes.Application, error) {
|
|
return &mastotypes.Application{
|
|
ID: a.ID,
|
|
Name: a.Name,
|
|
Website: a.Website,
|
|
RedirectURI: a.RedirectURI,
|
|
ClientID: a.ClientID,
|
|
ClientSecret: a.ClientSecret,
|
|
VapidKey: a.VapidKey,
|
|
}, nil
|
|
}
|
|
|
|
func (c *converter) AppToMastoPublic(a *gtsmodel.Application) (*mastotypes.Application, error) {
|
|
return &mastotypes.Application{
|
|
Name: a.Name,
|
|
Website: a.Website,
|
|
}, nil
|
|
}
|
|
|
|
func (c *converter) AttachmentToMasto(a *gtsmodel.MediaAttachment) (mastotypes.Attachment, error) {
|
|
return mastotypes.Attachment{
|
|
ID: a.ID,
|
|
Type: string(a.Type),
|
|
URL: a.URL,
|
|
PreviewURL: a.Thumbnail.URL,
|
|
RemoteURL: a.RemoteURL,
|
|
PreviewRemoteURL: a.Thumbnail.RemoteURL,
|
|
Meta: mastotypes.MediaMeta{
|
|
Original: mastotypes.MediaDimensions{
|
|
Width: a.FileMeta.Original.Width,
|
|
Height: a.FileMeta.Original.Height,
|
|
Size: fmt.Sprintf("%dx%d", a.FileMeta.Original.Width, a.FileMeta.Original.Height),
|
|
Aspect: float32(a.FileMeta.Original.Aspect),
|
|
},
|
|
Small: mastotypes.MediaDimensions{
|
|
Width: a.FileMeta.Small.Width,
|
|
Height: a.FileMeta.Small.Height,
|
|
Size: fmt.Sprintf("%dx%d", a.FileMeta.Small.Width, a.FileMeta.Small.Height),
|
|
Aspect: float32(a.FileMeta.Small.Aspect),
|
|
},
|
|
Focus: mastotypes.MediaFocus{
|
|
X: a.FileMeta.Focus.X,
|
|
Y: a.FileMeta.Focus.Y,
|
|
},
|
|
},
|
|
Description: a.Description,
|
|
Blurhash: a.Blurhash,
|
|
}, nil
|
|
}
|
|
|
|
func (c *converter) MentionToMasto(m *gtsmodel.Mention) (mastotypes.Mention, error) {
|
|
target := >smodel.Account{}
|
|
if err := c.db.GetByID(m.TargetAccountID, target); err != nil {
|
|
return mastotypes.Mention{}, err
|
|
}
|
|
|
|
var local bool
|
|
if target.Domain == "" {
|
|
local = true
|
|
}
|
|
|
|
var acct string
|
|
if local {
|
|
acct = fmt.Sprintf("@%s", target.Username)
|
|
} else {
|
|
acct = fmt.Sprintf("@%s@%s", target.Username, target.Domain)
|
|
}
|
|
|
|
return mastotypes.Mention{
|
|
ID: m.ID,
|
|
Username: target.Username,
|
|
URL: target.URL,
|
|
Acct: acct,
|
|
}, nil
|
|
}
|