2023-03-12 15:00:57 +00:00
// 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/>.
2021-08-10 12:32:39 +01:00
package dereferencing
import (
"context"
"encoding/json"
"errors"
"fmt"
2022-01-16 17:52:55 +00:00
"io"
2021-08-10 12:32:39 +01:00
"net/url"
2022-01-24 17:12:04 +00:00
"time"
2021-08-10 12:32:39 +01:00
2021-11-13 16:29:43 +00:00
"github.com/superseriousbusiness/activity/streams"
"github.com/superseriousbusiness/activity/streams/vocab"
2021-08-10 12:32:39 +01:00
"github.com/superseriousbusiness/gotosocial/internal/ap"
2022-08-20 21:47:19 +01:00
"github.com/superseriousbusiness/gotosocial/internal/config"
2022-06-11 10:01:34 +01:00
"github.com/superseriousbusiness/gotosocial/internal/db"
2021-08-10 12:32:39 +01:00
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
"github.com/superseriousbusiness/gotosocial/internal/id"
2022-07-19 09:47:55 +01:00
"github.com/superseriousbusiness/gotosocial/internal/log"
2022-01-09 17:41:22 +00:00
"github.com/superseriousbusiness/gotosocial/internal/media"
2023-02-09 08:27:07 +00:00
"github.com/superseriousbusiness/gotosocial/internal/transport"
2021-08-10 12:32:39 +01:00
)
2023-05-12 10:15:54 +01:00
// accountUpToDate returns whether the given account model is both updateable (i.e.
// non-instance remote account) and whether it needs an update based on `fetched_at`.
func accountUpToDate ( account * gtsmodel . Account ) bool {
if account . IsLocal ( ) {
// Can't update local accounts.
return true
}
if ! account . CreatedAt . IsZero ( ) && account . IsInstance ( ) {
// Existing instance account. No need for update.
return true
}
// If this account was updated recently (last interval), we return as-is.
if next := account . FetchedAt . Add ( 6 * time . Hour ) ; time . Now ( ) . Before ( next ) {
return true
}
return false
}
// GetAccountByURI: implements Dereferencer{}.GetAccountByURI.
func ( d * deref ) GetAccountByURI ( ctx context . Context , requestUser string , uri * url . URL ) ( * gtsmodel . Account , ap . Accountable , error ) {
// Fetch and dereference account if necessary.
account , apubAcc , err := d . getAccountByURI ( ctx ,
requestUser ,
uri ,
)
if err != nil {
return nil , nil , err
}
if apubAcc != nil {
// This account was updated, enqueue re-dereference featured posts.
d . state . Workers . Federator . MustEnqueueCtx ( ctx , func ( ctx context . Context ) {
if err := d . dereferenceAccountFeatured ( ctx , requestUser , account ) ; err != nil {
log . Errorf ( ctx , "error fetching account featured collection: %v" , err )
}
} )
}
return account , apubAcc , nil
}
// getAccountByURI is a package internal form of .GetAccountByURI() that doesn't bother dereferencing featured posts on update.
func ( d * deref ) getAccountByURI ( ctx context . Context , requestUser string , uri * url . URL ) ( * gtsmodel . Account , ap . Accountable , error ) {
2023-02-03 20:03:05 +00:00
var (
account * gtsmodel . Account
uriStr = uri . String ( )
err error
)
2022-11-29 09:24:55 +00:00
2023-02-03 20:03:05 +00:00
// Search the database for existing account with ID URI.
2023-05-12 10:15:54 +01:00
account , err = d . state . DB . GetAccountByURI ( ctx , uriStr )
2023-02-03 20:03:05 +00:00
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
2023-05-12 10:15:54 +01:00
return nil , nil , fmt . Errorf ( "GetAccountByURI: error checking database for account %s by uri: %w" , uriStr , err )
2023-02-03 20:03:05 +00:00
}
2022-11-29 09:24:55 +00:00
2023-02-03 20:03:05 +00:00
if account == nil {
// Else, search the database for existing by ID URL.
2023-05-12 10:15:54 +01:00
account , err = d . state . DB . GetAccountByURL ( ctx , uriStr )
2023-02-03 20:03:05 +00:00
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
2023-05-12 10:15:54 +01:00
return nil , nil , fmt . Errorf ( "GetAccountByURI: error checking database for account %s by url: %w" , uriStr , err )
2022-11-29 09:24:55 +00:00
}
}
2023-02-03 20:03:05 +00:00
if account == nil {
// Ensure that this is isn't a search for a local account.
if uri . Host == config . GetHost ( ) || uri . Host == config . GetAccountDomain ( ) {
2023-05-12 10:15:54 +01:00
return nil , nil , NewErrNotRetrievable ( err ) // this will be db.ErrNoEntries
2022-08-20 21:47:19 +01:00
}
2023-02-03 20:03:05 +00:00
// Create and pass-through a new bare-bones model for dereferencing.
return d . enrichAccount ( ctx , requestUser , uri , & gtsmodel . Account {
ID : id . NewULID ( ) ,
Domain : uri . Host ,
URI : uriStr ,
2023-05-12 10:15:54 +01:00
} , nil )
}
// Check whether needs update.
if accountUpToDate ( account ) {
return account , nil , nil
2022-06-11 10:01:34 +01:00
}
2023-05-12 10:15:54 +01:00
// Try to update existing account model.
latest , apubAcc , err := d . enrichAccount ( ctx ,
requestUser ,
uri ,
account ,
nil ,
)
2022-06-11 10:01:34 +01:00
if err != nil {
2023-02-17 11:02:29 +00:00
log . Errorf ( ctx , "error enriching remote account: %v" , err )
2023-05-12 10:15:54 +01:00
// Update fetch-at to slow re-attempts.
account . FetchedAt = time . Now ( )
_ = d . state . DB . UpdateAccount ( ctx , account , "fetched_at" )
// Fallback to existing.
return account , nil , nil
2022-06-11 10:01:34 +01:00
}
2023-05-12 10:15:54 +01:00
return latest , apubAcc , nil
2023-02-03 20:03:05 +00:00
}
2022-06-11 10:01:34 +01:00
2023-05-12 10:15:54 +01:00
// GetAccountByUsernameDomain: implements Dereferencer{}.GetAccountByUsernameDomain.
func ( d * deref ) GetAccountByUsernameDomain ( ctx context . Context , requestUser string , username string , domain string ) ( * gtsmodel . Account , ap . Accountable , error ) {
2023-02-03 20:03:05 +00:00
if domain == config . GetHost ( ) || domain == config . GetAccountDomain ( ) {
// We do local lookups using an empty domain,
// else it will fail the db search below.
domain = ""
}
2022-11-29 09:24:55 +00:00
2023-05-12 10:15:54 +01:00
// Search the database for existing account with USERNAME@DOMAIN.
account , err := d . state . DB . GetAccountByUsernameDomain ( ctx , username , domain )
2023-02-03 20:03:05 +00:00
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
2023-05-12 10:15:54 +01:00
return nil , nil , fmt . Errorf ( "GetAccountByUsernameDomain: error checking database for account %s@%s: %w" , username , domain , err )
2023-02-03 20:03:05 +00:00
}
2022-01-24 17:12:04 +00:00
2023-02-03 20:03:05 +00:00
if account == nil {
if domain == "" {
2023-05-12 10:15:54 +01:00
// failed local lookup, will be db.ErrNoEntries.
return nil , nil , NewErrNotRetrievable ( err )
2021-08-10 12:32:39 +01:00
}
2023-02-03 20:03:05 +00:00
// Create and pass-through a new bare-bones model for dereferencing.
2023-05-12 10:15:54 +01:00
account , apubAcc , err := d . enrichAccount ( ctx , requestUser , nil , & gtsmodel . Account {
2023-02-03 20:03:05 +00:00
ID : id . NewULID ( ) ,
Username : username ,
Domain : domain ,
2023-05-12 10:15:54 +01:00
} , nil )
if err != nil {
return nil , nil , err
2023-05-09 11:16:10 +01:00
}
2023-05-12 10:15:54 +01:00
// This account was updated, enqueue dereference featured posts.
d . state . Workers . Federator . MustEnqueueCtx ( ctx , func ( ctx context . Context ) {
if err := d . dereferenceAccountFeatured ( ctx , requestUser , account ) ; err != nil {
log . Errorf ( ctx , "error fetching account featured collection: %v" , err )
}
} )
return account , apubAcc , nil
2021-08-10 12:32:39 +01:00
}
2023-05-12 10:15:54 +01:00
// Try to update existing account model.
latest , apubAcc , err := d . RefreshAccount ( ctx ,
requestUser ,
account ,
nil ,
false ,
)
2023-02-03 20:03:05 +00:00
if err != nil {
2023-05-12 10:15:54 +01:00
// Fallback to existing.
return account , nil , nil //nolint
2022-06-11 10:01:34 +01:00
}
2023-05-12 10:15:54 +01:00
return latest , apubAcc , nil
2023-02-03 20:03:05 +00:00
}
2023-05-12 10:15:54 +01:00
// RefreshAccount: implements Dereferencer{}.RefreshAccount.
func ( d * deref ) RefreshAccount ( ctx context . Context , requestUser string , account * gtsmodel . Account , apubAcc ap . Accountable , force bool ) ( * gtsmodel . Account , ap . Accountable , error ) {
// Check whether needs update (and not forced).
if accountUpToDate ( account ) && ! force {
return account , nil , nil
2023-05-09 11:16:10 +01:00
}
2023-05-12 10:15:54 +01:00
// Parse the URI from account.
uri , err := url . Parse ( account . URI )
2023-05-09 11:16:10 +01:00
if err != nil {
2023-05-12 10:15:54 +01:00
return nil , nil , fmt . Errorf ( "RefreshAccount: invalid account uri %q: %w" , account . URI , err )
2023-05-09 11:16:10 +01:00
}
2023-05-12 10:15:54 +01:00
// Try to update + deref existing account model.
latest , apubAcc , err := d . enrichAccount ( ctx ,
requestUser ,
uri ,
account ,
apubAcc ,
2023-05-09 11:16:10 +01:00
)
if err != nil {
2023-05-12 10:15:54 +01:00
log . Errorf ( ctx , "error enriching remote account: %v" , err )
// Update fetch-at to slow re-attempts.
account . FetchedAt = time . Now ( )
_ = d . state . DB . UpdateAccount ( ctx , account , "fetched_at" )
return nil , nil , err
2023-05-09 11:16:10 +01:00
}
2023-05-12 10:15:54 +01:00
// This account was updated, enqueue re-dereference featured posts.
d . state . Workers . Federator . MustEnqueueCtx ( ctx , func ( ctx context . Context ) {
if err := d . dereferenceAccountFeatured ( ctx , requestUser , account ) ; err != nil {
log . Errorf ( ctx , "error fetching account featured collection: %v" , err )
}
} )
return latest , apubAcc , nil
2023-02-03 20:03:05 +00:00
}
2023-05-12 10:15:54 +01:00
// RefreshAccountAsync: implements Dereferencer{}.RefreshAccountAsync.
func ( d * deref ) RefreshAccountAsync ( ctx context . Context , requestUser string , account * gtsmodel . Account , apubAcc ap . Accountable , force bool ) {
// Check whether needs update (and not forced).
if accountUpToDate ( account ) && ! force {
return
2022-11-29 09:24:55 +00:00
}
2023-05-12 10:15:54 +01:00
// Parse the URI from account.
uri , err := url . Parse ( account . URI )
if err != nil {
log . Errorf ( ctx , "RefreshAccountAsync: invalid account uri %q: %v" , account . URI , err )
return
2022-06-11 10:01:34 +01:00
}
2023-05-12 10:15:54 +01:00
// Enqueue a worker function to enrich this account async.
d . state . Workers . Federator . MustEnqueueCtx ( ctx , func ( ctx context . Context ) {
latest , _ , err := d . enrichAccount ( ctx , requestUser , uri , account , apubAcc )
if err != nil {
log . Errorf ( ctx , "error enriching remote account: %v" , err )
return
}
2023-02-03 20:03:05 +00:00
2023-05-12 10:15:54 +01:00
// This account was updated, re-dereference account featured posts.
if err := d . dereferenceAccountFeatured ( ctx , requestUser , latest ) ; err != nil {
log . Errorf ( ctx , "error fetching account featured collection: %v" , err )
2022-06-11 10:01:34 +01:00
}
2023-05-12 10:15:54 +01:00
} )
}
2022-06-11 10:01:34 +01:00
2023-05-12 10:15:54 +01:00
// enrichAccount will enrich the given account, whether a new barebones model, or existing model from the database. It handles necessary dereferencing, webfingering etc.
func ( d * deref ) enrichAccount ( ctx context . Context , requestUser string , uri * url . URL , account * gtsmodel . Account , apubAcc ap . Accountable ) ( * gtsmodel . Account , ap . Accountable , error ) {
2023-02-10 20:15:23 +00:00
// Pre-fetch a transport for requesting username, used by later deref procedures.
2023-05-12 10:15:54 +01:00
tsport , err := d . transportController . NewTransportForUsername ( ctx , requestUser )
2023-02-09 08:27:07 +00:00
if err != nil {
2023-05-12 10:15:54 +01:00
return nil , nil , fmt . Errorf ( "enrichAccount: couldn't create transport: %w" , err )
2023-02-09 08:27:07 +00:00
}
2023-02-03 20:03:05 +00:00
if account . Username != "" {
// A username was provided so we can attempt a webfinger, this ensures up-to-date accountdomain info.
2023-05-12 10:15:54 +01:00
accDomain , accURI , err := d . fingerRemoteAccount ( ctx , tsport , account . Username , account . Domain )
if err != nil {
if account . URI == "" {
// this is a new account (to us) with username@domain but failed webfinger, nothing more we can do.
return nil , nil , & ErrNotRetrievable { fmt . Errorf ( "enrichAccount: error webfingering account: %w" , err ) }
}
2021-08-10 12:32:39 +01:00
2023-05-12 10:15:54 +01:00
// Simply log this error and move on, we already have an account URI.
2023-03-08 17:19:49 +00:00
log . Errorf ( ctx , "error webfingering[1] remote account %s@%s: %v" , account . Username , account . Domain , err )
2023-05-12 10:15:54 +01:00
}
2023-03-08 17:19:49 +00:00
2023-05-12 10:15:54 +01:00
if err == nil {
2023-02-09 09:34:44 +00:00
if account . Domain != accDomain {
2023-05-12 10:15:54 +01:00
// Domain has changed, assume the activitypub
// account data provided may not be the latest.
apubAcc = nil
2023-02-10 20:15:23 +00:00
// After webfinger, we now have correct account domain from which we can do a final DB check.
2023-05-12 10:15:54 +01:00
alreadyAccount , err := d . state . DB . GetAccountByUsernameDomain ( ctx , account . Username , accDomain )
2023-02-09 09:34:44 +00:00
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
2023-05-12 10:15:54 +01:00
return nil , nil , fmt . Errorf ( "enrichAccount: db err looking for account again after webfinger: %w" , err )
2023-02-09 09:34:44 +00:00
}
2023-05-12 10:15:54 +01:00
if alreadyAccount != nil {
2023-02-10 20:15:23 +00:00
// Enrich existing account.
2023-02-09 09:34:44 +00:00
account = alreadyAccount
}
}
2023-02-03 20:03:05 +00:00
// Update account with latest info.
account . URI = accURI . String ( )
account . Domain = accDomain
uri = accURI
2021-08-10 12:32:39 +01:00
}
2023-02-03 20:03:05 +00:00
}
2021-08-10 12:32:39 +01:00
2023-02-03 20:03:05 +00:00
if uri == nil {
// No URI provided / found, must parse from account.
uri , err = url . Parse ( account . URI )
if err != nil {
2023-05-12 10:15:54 +01:00
return nil , nil , fmt . Errorf ( "enrichAccount: invalid uri %q: %w" , account . URI , err )
2021-08-10 12:32:39 +01:00
}
2023-02-03 20:03:05 +00:00
}
2021-08-10 12:32:39 +01:00
2023-02-10 20:15:23 +00:00
// Check whether this account URI is a blocked domain / subdomain.
2023-05-12 10:15:54 +01:00
if blocked , err := d . state . DB . IsDomainBlocked ( ctx , uri . Host ) ; err != nil {
return nil , nil , fmt . Errorf ( "enrichAccount: error checking blocked domain: %w" , err )
2023-02-03 20:03:05 +00:00
} else if blocked {
2023-05-12 10:15:54 +01:00
return nil , nil , fmt . Errorf ( "enrichAccount: %s is blocked" , uri . Host )
2023-02-03 20:03:05 +00:00
}
2022-06-11 10:01:34 +01:00
2023-02-10 20:15:23 +00:00
// Mark deref+update handshake start.
2023-02-03 20:03:05 +00:00
d . startHandshake ( requestUser , uri )
defer d . stopHandshake ( requestUser , uri )
2022-01-24 12:12:17 +00:00
2023-05-12 10:15:54 +01:00
// By default we assume that apubAcc has been passed,
// indicating that the given account is already latest.
latestAcc := account
if apubAcc == nil {
// Dereference latest version of the account.
b , err := tsport . Dereference ( ctx , uri )
if err != nil {
return nil , nil , & ErrNotRetrievable { fmt . Errorf ( "enrichAccount: error deferencing %s: %w" , uri , err ) }
}
// Attempt to resolve ActivityPub account from data.
apubAcc , err = ap . ResolveAccountable ( ctx , b )
if err != nil {
return nil , nil , fmt . Errorf ( "enrichAccount: error resolving accountable from data for account %s: %w" , uri , err )
}
// Convert the dereferenced AP account object to our GTS model.
latestAcc , err = d . typeConverter . ASRepresentationToAccount ( ctx ,
apubAcc ,
account . Domain ,
)
if err != nil {
return nil , nil , fmt . Errorf ( "enrichAccount: error converting accountable to gts model for account %s: %w" , uri , err )
}
2021-08-10 12:32:39 +01:00
}
2023-02-03 20:03:05 +00:00
if account . Username == "" {
// No username was provided, so no webfinger was attempted earlier.
//
2023-03-19 15:45:13 +00:00
// Now we have a username we can attempt again, to ensure up-to-date
// accountDomain info. For this final attempt we should use the domain
// of the ID of the dereffed account, rather than the URI we were given.
//
// This avoids cases where we were given a URI like
// https://example.org/@someone@somewhere.else and we've been redirected
// from example.org to somewhere.else: we want to take somewhere.else
// as the accountDomain then, not the example.org we were redirected from.
// Assume the host from the returned ActivityPub representation.
idProp := apubAcc . GetJSONLDId ( )
if idProp == nil || ! idProp . IsIRI ( ) {
2023-05-12 10:15:54 +01:00
return nil , nil , errors . New ( "enrichAccount: no id property found on person, or id was not an iri" )
2023-03-19 15:45:13 +00:00
}
2023-05-12 10:15:54 +01:00
// Get IRI host value.
2023-03-19 15:45:13 +00:00
accHost := idProp . GetIRI ( ) . Host
2023-03-08 17:19:49 +00:00
2023-05-12 10:15:54 +01:00
latestAcc . Domain , _ , err = d . fingerRemoteAccount ( ctx ,
tsport ,
latestAcc . Username ,
accHost ,
)
2023-03-19 15:45:13 +00:00
if err != nil {
// We still couldn't webfinger the account, so we're not certain
// what the accountDomain actually is. Still, we can make a solid
// guess that it's the Host of the ActivityPub URI of the account.
// If we're wrong, we can just try again in a couple days.
log . Errorf ( ctx , "error webfingering[2] remote account %s@%s: %v" , latestAcc . Username , accHost , err )
latestAcc . Domain = accHost
2022-09-23 20:27:35 +01:00
}
}
2023-02-03 20:03:05 +00:00
// Ensure ID is set and update fetch time.
latestAcc . ID = account . ID
latestAcc . FetchedAt = time . Now ( )
2023-05-09 11:16:10 +01:00
// Reuse the existing account media attachments by default.
2023-02-10 20:15:23 +00:00
latestAcc . AvatarMediaAttachmentID = account . AvatarMediaAttachmentID
latestAcc . HeaderMediaAttachmentID = account . HeaderMediaAttachmentID
2023-05-12 10:15:54 +01:00
if ( latestAcc . AvatarMediaAttachmentID == "" ) ||
( latestAcc . AvatarRemoteURL != account . AvatarRemoteURL ) {
2023-02-13 20:19:51 +00:00
// Reset the avatar media ID (handles removed).
latestAcc . AvatarMediaAttachmentID = ""
if latestAcc . AvatarRemoteURL != "" {
// Avatar has changed to a new one, fetch up-to-date copy and use new ID.
latestAcc . AvatarMediaAttachmentID , err = d . fetchRemoteAccountAvatar ( ctx ,
2023-05-12 10:15:54 +01:00
tsport ,
2023-02-13 20:19:51 +00:00
latestAcc . AvatarRemoteURL ,
latestAcc . ID ,
)
if err != nil {
2023-02-17 11:02:29 +00:00
log . Errorf ( ctx , "error fetching remote avatar for account %s: %v" , uri , err )
2023-02-13 20:19:51 +00:00
// Keep old avatar for now, we'll try again in $interval.
latestAcc . AvatarMediaAttachmentID = account . AvatarMediaAttachmentID
latestAcc . AvatarRemoteURL = account . AvatarRemoteURL
}
2023-02-09 08:27:07 +00:00
}
}
2023-05-12 10:15:54 +01:00
if ( latestAcc . HeaderMediaAttachmentID == "" ) ||
( latestAcc . HeaderRemoteURL != account . HeaderRemoteURL ) {
2023-02-13 20:19:51 +00:00
// Reset the header media ID (handles removed).
latestAcc . HeaderMediaAttachmentID = ""
if latestAcc . HeaderRemoteURL != "" {
// Header has changed to a new one, fetch up-to-date copy and use new ID.
latestAcc . HeaderMediaAttachmentID , err = d . fetchRemoteAccountHeader ( ctx ,
2023-05-12 10:15:54 +01:00
tsport ,
2023-02-13 20:19:51 +00:00
latestAcc . HeaderRemoteURL ,
latestAcc . ID ,
)
if err != nil {
2023-02-17 11:02:29 +00:00
log . Errorf ( ctx , "error fetching remote header for account %s: %v" , uri , err )
2023-02-13 20:19:51 +00:00
// Keep old header for now, we'll try again in $interval.
latestAcc . HeaderMediaAttachmentID = account . HeaderMediaAttachmentID
latestAcc . HeaderRemoteURL = account . HeaderRemoteURL
}
2023-02-09 08:27:07 +00:00
}
2022-01-24 12:12:17 +00:00
}
2023-02-03 20:03:05 +00:00
// Fetch the latest remote account emoji IDs used in account display name/bio.
2023-03-01 17:52:44 +00:00
if _ , err = d . fetchRemoteAccountEmojis ( ctx , latestAcc , requestUser ) ; err != nil {
2023-02-17 11:02:29 +00:00
log . Errorf ( ctx , "error fetching remote emojis for account %s: %v" , uri , err )
2022-01-24 12:12:17 +00:00
}
2023-02-03 20:03:05 +00:00
if account . CreatedAt . IsZero ( ) {
// CreatedAt will be zero if no local copy was
// found in one of the GetAccountBy___() functions.
//
// Set time of creation from the last-fetched date.
latestAcc . CreatedAt = latestAcc . FetchedAt
latestAcc . UpdatedAt = latestAcc . FetchedAt
2023-03-03 08:34:34 +00:00
// This is new, put it in the database.
2023-05-12 10:15:54 +01:00
err := d . state . DB . PutAccount ( ctx , latestAcc )
2023-03-03 08:34:34 +00:00
if errors . Is ( err , db . ErrAlreadyExists ) {
// TODO: replace this quick fix with per-URI deref locks.
2023-05-12 10:15:54 +01:00
latestAcc , err = d . state . DB . GetAccountByURI ( ctx , latestAcc . URI )
return latestAcc , nil , err
2023-03-03 08:34:34 +00:00
}
if err != nil {
2023-05-12 10:15:54 +01:00
return nil , nil , fmt . Errorf ( "enrichAccount: error putting in database: %w" , err )
2023-02-03 20:03:05 +00:00
}
} else {
// Set time of update from the last-fetched date.
latestAcc . UpdatedAt = latestAcc . FetchedAt
// Use existing account values.
latestAcc . CreatedAt = account . CreatedAt
latestAcc . Language = account . Language
// This is an existing account, update the model in the database.
2023-05-12 10:15:54 +01:00
if err := d . state . DB . UpdateAccount ( ctx , latestAcc ) ; err != nil {
return nil , nil , fmt . Errorf ( "enrichAccount: error updating database: %w" , err )
2022-01-25 10:21:22 +00:00
}
2022-01-24 12:12:17 +00:00
}
2023-05-12 10:15:54 +01:00
return latestAcc , apubAcc , nil
2021-08-10 12:32:39 +01:00
}
2023-02-10 20:15:23 +00:00
func ( d * deref ) fetchRemoteAccountAvatar ( ctx context . Context , tsport transport . Transport , avatarURL string , accountID string ) ( string , error ) {
// Parse and validate provided media URL.
avatarURI , err := url . Parse ( avatarURL )
if err != nil {
return "" , err
}
// Acquire lock for derefs map.
unlock := d . derefAvatarsMu . Lock ( )
defer unlock ( )
2023-02-13 20:19:51 +00:00
// Look for an existing dereference in progress.
processing , ok := d . derefAvatars [ avatarURL ]
2023-02-10 20:15:23 +00:00
2023-02-13 20:19:51 +00:00
if ! ok {
var err error
2023-02-10 20:15:23 +00:00
2023-02-13 20:19:51 +00:00
// Set the media data function to dereference avatar from URI.
data := func ( ctx context . Context ) ( io . ReadCloser , int64 , error ) {
return tsport . DereferenceMedia ( ctx , avatarURI )
}
// Create new media processing request from the media manager instance.
processing , err = d . mediaManager . PreProcessMedia ( ctx , data , nil , accountID , & media . AdditionalMediaInfo {
Avatar : func ( ) * bool { v := true ; return & v } ( ) ,
RemoteURL : & avatarURL ,
} )
if err != nil {
return "" , err
}
2023-02-10 20:15:23 +00:00
2023-02-13 20:19:51 +00:00
// Store media in map to mark as processing.
d . derefAvatars [ avatarURL ] = processing
defer func ( ) {
// On exit safely remove media from map.
unlock := d . derefAvatarsMu . Lock ( )
delete ( d . derefAvatars , avatarURL )
unlock ( )
} ( )
}
2023-02-10 20:15:23 +00:00
// Unlock map.
unlock ( )
// Start media attachment loading (blocking call).
if _ , err := processing . LoadAttachment ( ctx ) ; err != nil {
return "" , err
}
return processing . AttachmentID ( ) , nil
}
func ( d * deref ) fetchRemoteAccountHeader ( ctx context . Context , tsport transport . Transport , headerURL string , accountID string ) ( string , error ) {
// Parse and validate provided media URL.
headerURI , err := url . Parse ( headerURL )
2023-02-09 08:27:07 +00:00
if err != nil {
return "" , err
2021-08-10 12:32:39 +01:00
}
2023-02-10 20:15:23 +00:00
// Acquire lock for derefs map.
unlock := d . derefHeadersMu . Lock ( )
defer unlock ( )
2023-02-13 20:19:51 +00:00
// Look for an existing dereference in progress.
processing , ok := d . derefHeaders [ headerURL ]
2023-02-10 20:15:23 +00:00
2023-02-13 20:19:51 +00:00
if ! ok {
var err error
2022-11-11 19:27:37 +00:00
2023-02-13 20:19:51 +00:00
// Set the media data function to dereference header from URI.
data := func ( ctx context . Context ) ( io . ReadCloser , int64 , error ) {
return tsport . DereferenceMedia ( ctx , headerURI )
}
// Create new media processing request from the media manager instance.
processing , err = d . mediaManager . PreProcessMedia ( ctx , data , nil , accountID , & media . AdditionalMediaInfo {
Header : func ( ) * bool { v := true ; return & v } ( ) ,
RemoteURL : & headerURL ,
} )
if err != nil {
return "" , err
}
// Store media in map to mark as processing.
d . derefHeaders [ headerURL ] = processing
2022-01-24 17:12:04 +00:00
2023-02-13 20:19:51 +00:00
defer func ( ) {
// On exit safely remove media from map.
unlock := d . derefHeadersMu . Lock ( )
delete ( d . derefHeaders , headerURL )
unlock ( )
} ( )
}
2023-02-10 20:15:23 +00:00
// Unlock map.
unlock ( )
// Start media attachment loading (blocking call).
if _ , err := processing . LoadAttachment ( ctx ) ; err != nil {
2023-02-09 08:27:07 +00:00
return "" , err
2021-08-10 12:32:39 +01:00
}
2022-01-24 12:12:17 +00:00
2023-02-10 20:15:23 +00:00
return processing . AttachmentID ( ) , nil
2021-08-10 12:32:39 +01:00
}
2022-01-24 12:12:17 +00:00
2022-09-26 10:56:01 +01:00
func ( d * deref ) fetchRemoteAccountEmojis ( ctx context . Context , targetAccount * gtsmodel . Account , requestingUsername string ) ( bool , error ) {
maybeEmojis := targetAccount . Emojis
maybeEmojiIDs := targetAccount . EmojiIDs
// It's possible that the account had emoji IDs set on it, but not Emojis
// themselves, depending on how it was fetched before being passed to us.
//
// If we only have IDs, fetch the emojis from the db. We know they're in
// there or else they wouldn't have IDs.
if len ( maybeEmojiIDs ) > len ( maybeEmojis ) {
2022-11-11 19:27:37 +00:00
maybeEmojis = make ( [ ] * gtsmodel . Emoji , 0 , len ( maybeEmojiIDs ) )
2022-09-26 10:56:01 +01:00
for _ , emojiID := range maybeEmojiIDs {
2023-05-12 10:15:54 +01:00
maybeEmoji , err := d . state . DB . GetEmojiByID ( ctx , emojiID )
2022-09-26 10:56:01 +01:00
if err != nil {
return false , err
}
maybeEmojis = append ( maybeEmojis , maybeEmoji )
}
}
// For all the maybe emojis we have, we either fetch them from the database
// (if we haven't already), or dereference them from the remote instance.
gotEmojis , err := d . populateEmojis ( ctx , maybeEmojis , requestingUsername )
if err != nil {
return false , err
}
// Extract the ID of each fetched or dereferenced emoji, so we can attach
// this to the account if necessary.
gotEmojiIDs := make ( [ ] string , 0 , len ( gotEmojis ) )
for _ , e := range gotEmojis {
gotEmojiIDs = append ( gotEmojiIDs , e . ID )
}
var (
changed = false // have the emojis for this account changed?
maybeLen = len ( maybeEmojis )
gotLen = len ( gotEmojis )
)
// if the length of everything is zero, this is simple:
// nothing has changed and there's nothing to do
if maybeLen == 0 && gotLen == 0 {
return changed , nil
}
// if the *amount* of emojis on the account has changed, then the got emojis
// are definitely different from the previous ones (if there were any) --
// the account has either more or fewer emojis set on it now, so take the
// discovered emojis as the new correct ones.
if maybeLen != gotLen {
changed = true
targetAccount . Emojis = gotEmojis
targetAccount . EmojiIDs = gotEmojiIDs
return changed , nil
}
// if the lengths are the same but not all of the slices are
// zero, something *might* have changed, so we have to check
// 1. did we have emojis before that we don't have now?
for _ , maybeEmoji := range maybeEmojis {
var stillPresent bool
for _ , gotEmoji := range gotEmojis {
if maybeEmoji . URI == gotEmoji . URI {
// the emoji we maybe had is still present now,
// so we can stop checking gotEmojis
stillPresent = true
break
}
}
if ! stillPresent {
// at least one maybeEmoji is no longer present in
// the got emojis, so we can stop checking now
changed = true
targetAccount . Emojis = gotEmojis
targetAccount . EmojiIDs = gotEmojiIDs
return changed , nil
}
}
// 2. do we have emojis now that we didn't have before?
for _ , gotEmoji := range gotEmojis {
var wasPresent bool
for _ , maybeEmoji := range maybeEmojis {
// check emoji IDs here as well, because unreferenced
// maybe emojis we didn't already have would not have
// had IDs set on them yet
if gotEmoji . URI == maybeEmoji . URI && gotEmoji . ID == maybeEmoji . ID {
// this got emoji was present already in the maybeEmoji,
// so we can stop checking through maybeEmojis
wasPresent = true
break
}
}
if ! wasPresent {
// at least one gotEmojis was not present in
// the maybeEmojis, so we can stop checking now
changed = true
targetAccount . Emojis = gotEmojis
targetAccount . EmojiIDs = gotEmojiIDs
return changed , nil
}
}
return changed , nil
}
2023-03-01 17:52:44 +00:00
2023-05-12 10:15:54 +01:00
// dereferenceAccountFeatured dereferences an account's featuredCollectionURI (if not empty). For each discovered status, this status will
// be dereferenced (if necessary) and marked as pinned (if necessary). Then, old pins will be removed if they're not included in new pins.
func ( d * deref ) dereferenceAccountFeatured ( ctx context . Context , requestUser string , account * gtsmodel . Account ) error {
uri , err := url . Parse ( account . FeaturedCollectionURI )
2023-03-01 17:52:44 +00:00
if err != nil {
return err
}
2023-05-12 10:15:54 +01:00
// Pre-fetch a transport for requesting username, used by later deref procedures.
tsport , err := d . transportController . NewTransportForUsername ( ctx , requestUser )
2023-03-01 17:52:44 +00:00
if err != nil {
2023-05-12 10:15:54 +01:00
return fmt . Errorf ( "enrichAccount: couldn't create transport: %w" , err )
2023-03-01 17:52:44 +00:00
}
b , err := tsport . Dereference ( ctx , uri )
if err != nil {
return err
}
m := make ( map [ string ] interface { } )
if err := json . Unmarshal ( b , & m ) ; err != nil {
return fmt . Errorf ( "error unmarshalling bytes into json: %w" , err )
}
t , err := streams . ToType ( ctx , m )
if err != nil {
return fmt . Errorf ( "error resolving json into ap vocab type: %w" , err )
}
if t . GetTypeName ( ) != ap . ObjectOrderedCollection {
2023-05-12 10:15:54 +01:00
return fmt . Errorf ( "%s was not an OrderedCollection" , uri )
2023-03-01 17:52:44 +00:00
}
collection , ok := t . ( vocab . ActivityStreamsOrderedCollection )
if ! ok {
return errors . New ( "couldn't coerce OrderedCollection" )
}
items := collection . GetActivityStreamsOrderedItems ( )
if items == nil {
return errors . New ( "nil orderedItems" )
}
// Get previous pinned statuses (we'll need these later).
2023-05-12 10:15:54 +01:00
wasPinned , err := d . state . DB . GetAccountPinnedStatuses ( ctx , account . ID )
2023-03-01 17:52:44 +00:00
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
return fmt . Errorf ( "error getting account pinned statuses: %w" , err )
}
statusURIs := make ( [ ] * url . URL , 0 , items . Len ( ) )
for iter := items . Begin ( ) ; iter != items . End ( ) ; iter = iter . Next ( ) {
var statusURI * url . URL
switch {
case iter . IsActivityStreamsNote ( ) :
// We got a whole Note. Extract the URI.
if note := iter . GetActivityStreamsNote ( ) ; note != nil {
if id := note . GetJSONLDId ( ) ; id != nil {
statusURI = id . GetIRI ( )
}
}
case iter . IsActivityStreamsArticle ( ) :
// We got a whole Article. Extract the URI.
if article := iter . GetActivityStreamsArticle ( ) ; article != nil {
if id := article . GetJSONLDId ( ) ; id != nil {
statusURI = id . GetIRI ( )
}
}
default :
// Try to get just the URI.
statusURI = iter . GetIRI ( )
}
if statusURI == nil {
continue
}
if statusURI . Host != uri . Host {
// If this status doesn't share a host with its featured
// collection URI, we shouldn't trust it. Just move on.
continue
}
// Already append this status URI to our slice.
// We do this here so that even if we can't get
// the status in the next part for some reason,
// we still know it was *meant* to be pinned.
statusURIs = append ( statusURIs , statusURI )
2023-05-12 10:15:54 +01:00
status , _ , err := d . getStatusByURI ( ctx , requestUser , statusURI )
2023-03-01 17:52:44 +00:00
if err != nil {
2023-05-12 10:15:54 +01:00
// We couldn't get the status, bummer. Just log + move on, we can try later.
log . Errorf ( ctx , "error getting status from featured collection %s: %v" , statusURI , err )
2023-03-01 17:52:44 +00:00
continue
}
// If the status was already pinned, we don't need to do anything.
if ! status . PinnedAt . IsZero ( ) {
continue
}
2023-05-12 10:15:54 +01:00
if status . AccountID != account . ID {
2023-03-01 17:52:44 +00:00
// Someone's pinned a status that doesn't
// belong to them, this doesn't work for us.
continue
}
if status . BoostOfID != "" {
// Someone's pinned a boost. This also
// doesn't work for us.
continue
}
// All conditions are met for this status to
// be pinned, so we can finally update it.
status . PinnedAt = time . Now ( )
2023-05-12 10:15:54 +01:00
if err := d . state . DB . UpdateStatus ( ctx , status , "pinned_at" ) ; err != nil {
log . Errorf ( ctx , "error updating status in featured collection %s: %v" , status . URI , err )
continue
2023-03-01 17:52:44 +00:00
}
}
// Now that we know which statuses are pinned, we should
// *unpin* previous pinned statuses that aren't included.
outerLoop :
for _ , status := range wasPinned {
for _ , statusURI := range statusURIs {
if status . URI == statusURI . String ( ) {
// This status is included in most recent
// pinned uris. No need to keep checking.
continue outerLoop
}
}
// Status was pinned before, but is not included
// in most recent pinned uris, so unpin it now.
status . PinnedAt = time . Time { }
2023-05-12 10:15:54 +01:00
if err := d . state . DB . UpdateStatus ( ctx , status , "pinned_at" ) ; err != nil {
log . Errorf ( ctx , "error unpinning status %s: %v" , status . URI , err )
continue
2023-03-01 17:52:44 +00:00
}
}
return nil
}