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 11:32:39 +00:00
package dereferencing
import (
"context"
"encoding/json"
"errors"
2022-01-16 17:52:55 +00:00
"io"
2021-08-10 11:32:39 +00:00
"net/url"
2022-01-24 17:12:04 +00:00
"time"
2021-08-10 11:32:39 +00:00
2021-11-13 16:29:43 +00:00
"github.com/superseriousbusiness/activity/streams"
"github.com/superseriousbusiness/activity/streams/vocab"
2021-08-10 11:32:39 +00:00
"github.com/superseriousbusiness/gotosocial/internal/ap"
2022-08-20 20:47:19 +00:00
"github.com/superseriousbusiness/gotosocial/internal/config"
2022-06-11 09:01:34 +00:00
"github.com/superseriousbusiness/gotosocial/internal/db"
2023-05-31 08:39:54 +00:00
"github.com/superseriousbusiness/gotosocial/internal/gtscontext"
2023-05-28 12:08:35 +00:00
"github.com/superseriousbusiness/gotosocial/internal/gtserror"
2021-08-10 11:32:39 +00:00
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
"github.com/superseriousbusiness/gotosocial/internal/id"
2022-07-19 08:47:55 +00: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 11:32:39 +00:00
)
2023-05-12 09:15:54 +00: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`.
2024-01-09 09:42:39 +00:00
func accountUpToDate ( account * gtsmodel . Account , force bool ) bool {
2023-11-10 16:16:58 +00:00
if ! account . SuspendedAt . IsZero ( ) {
// Can't update suspended accounts.
return true
}
2023-05-12 09:15:54 +00:00
if account . IsLocal ( ) {
// Can't update local accounts.
return true
}
2024-01-26 13:17:10 +00:00
if account . IsInstance ( ) && ! account . IsNew ( ) {
2023-05-12 09:15:54 +00:00
// Existing instance account. No need for update.
return true
}
2024-01-09 09:42:39 +00:00
// Default limit we allow
// statuses to be refreshed.
limit := 6 * time . Hour
if force {
// We specifically allow the force flag
// to force an early refresh (on a much
// smaller cooldown period).
limit = 5 * time . Minute
}
// If account was updated recently (within limit), we return as-is.
if next := account . FetchedAt . Add ( limit ) ; time . Now ( ) . Before ( next ) {
2023-05-12 09:15:54 +00:00
return true
}
return false
}
2023-10-23 09:58:13 +00:00
// GetAccountByURI will attempt to fetch an accounts by its URI, first checking the database. In the case of a newly-met remote model, or a remote model
// whose last_fetched date is beyond a certain interval, the account will be dereferenced. In the case of dereferencing, some low-priority account information
// may be enqueued for asynchronous fetching, e.g. featured account statuses (pins). An ActivityPub object indicates the account was dereferenced.
func ( d * Dereferencer ) GetAccountByURI ( ctx context . Context , requestUser string , uri * url . URL ) ( * gtsmodel . Account , ap . Accountable , error ) {
2023-05-12 09:15:54 +00:00
// Fetch and dereference account if necessary.
2024-01-09 09:42:39 +00:00
account , accountable , err := d . getAccountByURI ( ctx ,
2023-05-12 09:15:54 +00:00
requestUser ,
uri ,
)
if err != nil {
return nil , nil , err
}
2024-01-09 09:42:39 +00:00
if accountable != nil {
2023-05-12 09:15:54 +00: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 )
}
} )
}
2024-01-09 09:42:39 +00:00
return account , accountable , nil
2023-05-12 09:15:54 +00:00
}
// getAccountByURI is a package internal form of .GetAccountByURI() that doesn't bother dereferencing featured posts on update.
2023-10-23 09:58:13 +00:00
func ( d * Dereferencer ) 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-05-31 08:39:54 +00:00
// Search the database for existing account with URI.
account , err = d . state . DB . GetAccountByURI (
// request a barebones object, it may be in the
// db but with related models not yet dereferenced.
gtscontext . SetBarebones ( ctx ) ,
uriStr ,
)
2023-02-03 20:03:05 +00:00
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . Newf ( "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 {
2023-05-31 08:39:54 +00:00
// Else, search the database for existing by URL.
account , err = d . state . DB . GetAccountByURL (
gtscontext . SetBarebones ( ctx ) ,
uriStr ,
)
2023-02-03 20:03:05 +00:00
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . Newf ( "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-06-22 19:46:36 +00:00
return nil , nil , gtserror . SetUnretrievable ( err ) // this will be db.ErrNoEntries
2022-08-20 20:47:19 +00:00
}
2023-02-03 20:03:05 +00:00
// Create and pass-through a new bare-bones model for dereferencing.
2023-10-31 11:12:22 +00:00
return d . enrichAccountSafely ( ctx , requestUser , uri , & gtsmodel . Account {
2023-02-03 20:03:05 +00:00
ID : id . NewULID ( ) ,
Domain : uri . Host ,
URI : uriStr ,
2023-05-12 09:15:54 +00:00
} , nil )
}
2024-01-09 09:42:39 +00:00
if accountUpToDate ( account , false ) {
// This is an existing account that is up-to-date,
// before returning ensure it is fully populated.
2023-05-31 08:39:54 +00:00
if err := d . state . DB . PopulateAccount ( ctx , account ) ; err != nil {
log . Errorf ( ctx , "error populating existing account: %v" , err )
}
2024-01-09 09:42:39 +00:00
2023-05-12 09:15:54 +00:00
return account , nil , nil
2022-06-11 09:01:34 +00:00
}
2023-05-12 09:15:54 +00:00
// Try to update existing account model.
2024-01-09 09:42:39 +00:00
latest , accountable , err := d . enrichAccountSafely ( ctx ,
2023-05-12 09:15:54 +00:00
requestUser ,
uri ,
account ,
nil ,
)
2022-06-11 09:01:34 +00:00
if err != nil {
2023-02-17 11:02:29 +00:00
log . Errorf ( ctx , "error enriching remote account: %v" , err )
2023-05-12 09:15:54 +00:00
// Fallback to existing.
return account , nil , nil
2022-06-11 09:01:34 +00:00
}
2024-01-09 09:42:39 +00:00
return latest , accountable , nil
2023-02-03 20:03:05 +00:00
}
2022-06-11 09:01:34 +00:00
2023-10-23 09:58:13 +00:00
// GetAccountByUsernameDomain will attempt to fetch an accounts by its username@domain, first checking the database. In the case of a newly-met remote model,
// or a remote model whose last_fetched date is beyond a certain interval, the account will be dereferenced. In the case of dereferencing, some low-priority
// account information may be enqueued for asynchronous fetching, e.g. featured account statuses (pins). An ActivityPub object indicates the account was dereferenced.
func ( d * Dereferencer ) GetAccountByUsernameDomain ( ctx context . Context , requestUser string , username string , domain string ) ( * gtsmodel . Account , ap . Accountable , error ) {
2024-01-09 09:42:39 +00:00
account , accountable , err := d . getAccountByUsernameDomain (
2023-10-31 11:05:17 +00:00
ctx ,
requestUser ,
username ,
domain ,
)
if err != nil {
return nil , nil , err
}
2024-01-09 09:42:39 +00:00
if accountable != nil {
2023-10-31 11:05:17 +00: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 )
}
} )
}
2024-01-09 09:42:39 +00:00
return account , accountable , nil
2023-10-31 11:05:17 +00:00
}
2024-01-09 09:42:39 +00:00
// getAccountByUsernameDomain is a package internal form of
// GetAccountByUsernameDomain() that doesn't bother deref of featured posts.
2023-10-31 11:05:17 +00:00
func ( d * Dereferencer ) 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 09:15:54 +00:00
// Search the database for existing account with USERNAME@DOMAIN.
2023-05-31 08:39:54 +00:00
account , err := d . state . DB . GetAccountByUsernameDomain (
// request a barebones object, it may be in the
// db but with related models not yet dereferenced.
gtscontext . SetBarebones ( ctx ) ,
username , domain ,
)
2023-02-03 20:03:05 +00:00
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . Newf ( "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 09:15:54 +00:00
// failed local lookup, will be db.ErrNoEntries.
2023-06-22 19:46:36 +00:00
return nil , nil , gtserror . SetUnretrievable ( err )
2021-08-10 11:32:39 +00:00
}
2023-02-03 20:03:05 +00:00
// Create and pass-through a new bare-bones model for dereferencing.
2024-01-09 09:42:39 +00:00
account , accountable , err := d . enrichAccountSafely ( ctx , requestUser , nil , & gtsmodel . Account {
2023-02-03 20:03:05 +00:00
ID : id . NewULID ( ) ,
Domain : domain ,
2024-01-26 13:17:10 +00:00
Username : username ,
2023-05-12 09:15:54 +00:00
} , nil )
if err != nil {
return nil , nil , err
2023-05-09 10:16:10 +00:00
}
2024-01-09 09:42:39 +00:00
return account , accountable , nil
2021-08-10 11:32:39 +00:00
}
2023-05-12 09:15:54 +00:00
// Try to update existing account model.
2024-01-09 09:42:39 +00:00
latest , accountable , err := d . RefreshAccount ( ctx ,
2023-05-12 09:15:54 +00:00
requestUser ,
account ,
nil ,
false ,
)
2023-02-03 20:03:05 +00:00
if err != nil {
2023-05-12 09:15:54 +00:00
// Fallback to existing.
return account , nil , nil //nolint
2022-06-11 09:01:34 +00:00
}
2024-01-09 09:42:39 +00:00
if accountable == nil {
2023-05-31 08:39:54 +00:00
// This is existing up-to-date account, ensure it is populated.
2023-10-31 11:12:22 +00:00
if err := d . state . DB . PopulateAccount ( ctx , latest ) ; err != nil {
2023-05-31 08:39:54 +00:00
log . Errorf ( ctx , "error populating existing account: %v" , err )
}
}
2024-01-09 09:42:39 +00:00
return latest , accountable , nil
2023-02-03 20:03:05 +00:00
}
2024-01-09 09:42:39 +00:00
// RefreshAccount updates the given account if remote and last_fetched is
// beyond fetch interval, or if force is set. An updated account model is
// returned, but in the case of dereferencing, some low-priority account info
// may be enqueued for asynchronous fetching, e.g. featured account statuses (pins).
2023-10-23 09:58:13 +00:00
// An ActivityPub object indicates the account was dereferenced (i.e. updated).
2024-01-09 09:42:39 +00:00
func ( d * Dereferencer ) RefreshAccount (
ctx context . Context ,
requestUser string ,
account * gtsmodel . Account ,
accountable ap . Accountable ,
force bool ,
) ( * gtsmodel . Account , ap . Accountable , error ) {
// If no incoming data is provided,
// check whether account needs update.
if accountable == nil &&
accountUpToDate ( account , force ) {
2023-05-12 09:15:54 +00:00
return account , nil , nil
2023-05-09 10:16:10 +00:00
}
2023-05-12 09:15:54 +00:00
// Parse the URI from account.
uri , err := url . Parse ( account . URI )
2023-05-09 10:16:10 +00:00
if err != nil {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . Newf ( "invalid account uri %q: %w" , account . URI , err )
2023-05-09 10:16:10 +00:00
}
2023-10-31 11:12:22 +00:00
// Try to update + deref passed account model.
2024-01-09 09:42:39 +00:00
latest , accountable , err := d . enrichAccountSafely ( ctx ,
2023-05-12 09:15:54 +00:00
requestUser ,
uri ,
account ,
2024-01-09 09:42:39 +00:00
accountable ,
2023-05-09 10:16:10 +00:00
)
if err != nil {
2023-05-12 09:15:54 +00:00
log . Errorf ( ctx , "error enriching remote account: %v" , err )
2023-10-31 11:12:22 +00:00
return nil , nil , gtserror . Newf ( "error enriching remote account: %w" , err )
2023-05-09 10:16:10 +00:00
}
2024-01-09 09:42:39 +00:00
if accountable != nil {
2023-10-31 11:12:22 +00: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 , latest ) ; err != nil {
log . Errorf ( ctx , "error fetching account featured collection: %v" , err )
}
} )
}
2023-05-12 09:15:54 +00:00
2024-01-09 09:42:39 +00:00
return latest , accountable , nil
2023-02-03 20:03:05 +00:00
}
2024-01-09 09:42:39 +00:00
// RefreshAccountAsync enqueues the given account for an asychronous
// update fetching, if last_fetched is beyond fetch interval, or if force
// is set. This is a more optimized form of manually enqueueing .UpdateAccount()
// to the federation worker, since it only enqueues update if necessary.
func ( d * Dereferencer ) RefreshAccountAsync (
ctx context . Context ,
requestUser string ,
account * gtsmodel . Account ,
accountable ap . Accountable ,
force bool ,
) {
// If no incoming data is provided,
// check whether account needs update.
if accountable == nil &&
accountUpToDate ( account , force ) {
2023-05-12 09:15:54 +00:00
return
2022-11-29 09:24:55 +00:00
}
2023-05-12 09:15:54 +00:00
// Parse the URI from account.
uri , err := url . Parse ( account . URI )
if err != nil {
2023-05-28 12:08:35 +00:00
log . Errorf ( ctx , "invalid account uri %q: %v" , account . URI , err )
2023-05-12 09:15:54 +00:00
return
2022-06-11 09:01:34 +00:00
}
2023-05-12 09:15:54 +00:00
// Enqueue a worker function to enrich this account async.
d . state . Workers . Federator . MustEnqueueCtx ( ctx , func ( ctx context . Context ) {
2024-01-09 09:42:39 +00:00
latest , accountable , err := d . enrichAccountSafely ( ctx , requestUser , uri , account , accountable )
2023-05-12 09:15:54 +00:00
if err != nil {
log . Errorf ( ctx , "error enriching remote account: %v" , err )
return
}
2023-02-03 20:03:05 +00:00
2024-01-09 09:42:39 +00:00
if accountable != nil {
2023-10-31 11:12:22 +00:00
// This account was updated, enqueue re-dereference featured posts.
2023-11-14 14:57:25 +00:00
if err := d . dereferenceAccountFeatured ( ctx , requestUser , latest ) ; err != nil {
log . Errorf ( ctx , "error fetching account featured collection: %v" , err )
}
2022-06-11 09:01:34 +00:00
}
2023-05-12 09:15:54 +00:00
} )
}
2022-06-11 09:01:34 +00:00
2023-10-31 11:12:22 +00:00
// enrichAccountSafely wraps enrichAccount() to perform
// it within the State{}.FedLocks mutexmap, which protects
// dereferencing actions with per-URI mutex locks.
func ( d * Dereferencer ) enrichAccountSafely (
ctx context . Context ,
requestUser string ,
uri * url . URL ,
account * gtsmodel . Account ,
2024-01-09 09:42:39 +00:00
accountable ap . Accountable ,
2023-10-31 11:12:22 +00:00
) ( * gtsmodel . Account , ap . Accountable , error ) {
2023-11-10 16:16:58 +00:00
// Noop if account has been suspended.
if ! account . SuspendedAt . IsZero ( ) {
return account , nil , nil
}
2023-10-31 11:12:22 +00:00
// By default use account.URI
// as the per-URI deref lock.
2024-01-31 13:29:47 +00:00
var uriStr string
2024-01-26 13:17:10 +00:00
if account . URI != "" {
2024-01-31 13:29:47 +00:00
uriStr = account . URI
2024-01-26 13:17:10 +00:00
} else {
2023-10-31 11:12:22 +00:00
// No URI is set yet, instead generate a faux-one from user+domain.
2024-01-31 13:29:47 +00:00
uriStr = "https://" + account . Domain + "/users/" + account . Username
2023-10-31 11:12:22 +00:00
}
// Acquire per-URI deref lock, wraping unlock
// to safely defer in case of panic, while still
// performing more granular unlocks when needed.
2024-01-31 13:29:47 +00:00
unlock := d . state . FedLocks . Lock ( uriStr )
2023-10-31 11:12:22 +00:00
unlock = doOnce ( unlock )
defer unlock ( )
// Perform status enrichment with passed vars.
latest , apubAcc , err := d . enrichAccount ( ctx ,
requestUser ,
uri ,
account ,
2024-01-09 09:42:39 +00:00
accountable ,
2023-10-31 11:12:22 +00:00
)
2024-01-31 13:29:47 +00:00
if gtserror . StatusCode ( err ) >= 400 {
2024-01-26 13:17:10 +00:00
if account . IsNew ( ) {
// This was a new account enrich
// attempt which failed before we
// got to store it, so we can't
// return anything useful.
return nil , nil , err
}
// We had this account stored already
// before this enrichment attempt.
//
// Update fetched_at to slow re-attempts
// but don't return early. We can still
// return the model we had stored already.
2023-10-31 11:12:22 +00:00
account . FetchedAt = time . Now ( )
2024-01-26 13:17:10 +00:00
if err := d . state . DB . UpdateAccount ( ctx , account , "fetched_at" ) ; err != nil {
2024-01-31 13:29:47 +00:00
log . Error ( ctx , "error updating %s fetched_at: %v" , uriStr , err )
2024-01-26 13:17:10 +00:00
}
2023-10-31 11:12:22 +00:00
}
// Unlock now
// we're done.
unlock ( )
if errors . Is ( err , db . ErrAlreadyExists ) {
// Ensure AP model isn't set,
// otherwise this indicates WE
// enriched the account.
apubAcc = nil
// DATA RACE! We likely lost out to another goroutine
// in a call to db.Put(Account). Look again in DB by URI.
latest , err = d . state . DB . GetAccountByURI ( ctx , account . URI )
if err != nil {
2024-01-31 13:29:47 +00:00
err = gtserror . Newf ( "error getting account %s from database after race: %w" , uriStr , err )
2023-10-31 11:12:22 +00:00
}
}
return latest , apubAcc , err
}
// 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 * Dereferencer ) 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 09:15:54 +00:00
tsport , err := d . transportController . NewTransportForUsername ( ctx , requestUser )
2023-02-09 08:27:07 +00:00
if err != nil {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . Newf ( "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 09:15:54 +00:00
accDomain , accURI , err := d . fingerRemoteAccount ( ctx , tsport , account . Username , account . Domain )
2024-01-26 13:17:10 +00:00
switch {
2021-08-10 11:32:39 +00:00
2024-01-26 13:17:10 +00:00
case err != nil && account . URI == "" :
// This is a new account (to us) with username@domain
// but failed webfinger, nothing more we can do.
err := gtserror . Newf ( "error webfingering account: %w" , err )
return nil , nil , gtserror . SetUnretrievable ( err )
case err != nil :
// Simply log this error and move on,
// we already have an account URI.
log . Errorf ( ctx ,
"error webfingering[1] remote account %s@%s: %v" ,
account . Username , account . Domain , err ,
)
case err == nil && account . Domain != accDomain :
// After webfinger, we now have correct account domain from which we can do a final DB check.
alreadyAcct , err := d . state . DB . GetAccountByUsernameDomain ( ctx , account . Username , accDomain )
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
return nil , nil , gtserror . Newf ( "db err looking for account again after webfinger: %w" , err )
}
2023-03-08 17:19:49 +00:00
2024-01-26 13:17:10 +00:00
if alreadyAcct != nil {
// We had this account stored under
// the discovered accountDomain.
// Proceed with this account.
account = alreadyAcct
2023-02-09 09:34:44 +00:00
}
2024-01-26 13:17:10 +00:00
// Whether we had the account or not, we
// now have webfinger info relevant to the
// account, so fallthrough to set webfinger
// info on either the account we just found,
// or the stub account we were passed.
fallthrough
case err == nil :
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 11:32:39 +00:00
}
2023-02-03 20:03:05 +00:00
}
2021-08-10 11:32:39 +00:00
2023-02-03 20:03:05 +00:00
if uri == nil {
2024-01-26 13:17:10 +00:00
// No URI provided / found,
// must parse from account.
2023-02-03 20:03:05 +00:00
uri , err = url . Parse ( account . URI )
if err != nil {
2024-01-26 13:17:10 +00:00
return nil , nil , gtserror . Newf (
"invalid uri %q: %w" ,
account . URI , gtserror . SetUnretrievable ( err ) ,
)
}
if uri . Scheme != "http" && uri . Scheme != "https" {
err = errors . New ( "account URI scheme must be http or https" )
return nil , nil , gtserror . Newf (
"invalid uri %q: %w" ,
account . URI , gtserror . SetUnretrievable ( err ) ,
)
2021-08-10 11:32:39 +00:00
}
2023-02-03 20:03:05 +00:00
}
2021-08-10 11:32:39 +00:00
2024-01-26 13:17:10 +00:00
/ *
BY THIS POINT we must have an account URI set ,
either provided , pinned to the account , or
obtained via webfinger call .
* /
2023-02-10 20:15:23 +00:00
// Check whether this account URI is a blocked domain / subdomain.
2023-05-12 09:15:54 +00:00
if blocked , err := d . state . DB . IsDomainBlocked ( ctx , uri . Host ) ; err != nil {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . Newf ( "error checking blocked domain: %w" , err )
2023-02-03 20:03:05 +00:00
} else if blocked {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . Newf ( "%s is blocked" , uri . Host )
2023-02-03 20:03:05 +00:00
}
2022-06-11 09:01:34 +00: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 09:15:54 +00:00
if apubAcc == nil {
2024-01-26 13:17:10 +00:00
// We were not given any (partial) ActivityPub
// version of this account as a parameter.
2023-05-12 09:15:54 +00:00
// Dereference latest version of the account.
b , err := tsport . Dereference ( ctx , uri )
if err != nil {
2024-01-26 13:17:10 +00:00
err := gtserror . Newf ( "error dereferencing %s: %w" , uri , err )
2023-06-22 19:46:36 +00:00
return nil , nil , gtserror . SetUnretrievable ( err )
2023-05-12 09:15:54 +00:00
}
2023-06-22 19:46:36 +00:00
// Attempt to resolve ActivityPub acc from data.
2023-05-12 09:15:54 +00:00
apubAcc , err = ap . ResolveAccountable ( ctx , b )
if err != nil {
2024-01-26 13:17:10 +00:00
// ResolveAccountable will set Unretrievable/WrongType
// on the returned error, so we don't need to do it here.
err = gtserror . Newf ( "error resolving accountable from data for account %s: %w" , uri , err )
return nil , nil , err
2023-05-12 09:15:54 +00:00
}
2023-06-22 19:46:36 +00:00
}
2023-05-12 09:15:54 +00:00
2024-01-26 13:17:10 +00:00
/ *
BY THIS POINT we must have the ActivityPub
representation of the account , either provided ,
or obtained via a dereference call .
* /
2023-06-22 19:46:36 +00:00
// Convert the dereferenced AP account object to our GTS model.
2024-01-26 13:17:10 +00:00
//
// We put this in the variable latestAcc because we might want
// to compare the provided account model with this fresh version,
// in order to check if anything changed since we last saw it.
2023-09-23 16:44:11 +00:00
latestAcc , err := d . converter . ASRepresentationToAccount ( ctx ,
2023-06-22 19:46:36 +00:00
apubAcc ,
account . Domain ,
)
if err != nil {
2024-01-26 13:17:10 +00:00
// ASRepresentationToAccount will set Malformed on the
// returned error, so we don't need to do it here.
err = gtserror . Newf ( "error converting accountable to gts model for account %s: %w" , uri , err )
return nil , nil , err
2021-08-10 11:32:39 +00: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.
2024-01-26 13:17:10 +00:00
// Assume the host from the returned
// ActivityPub representation.
id := ap . GetJSONLDId ( apubAcc )
if id == nil {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . New ( "no id property found on person, or id was not an iri" )
2023-03-19 15:45:13 +00:00
}
2023-05-12 09:15:54 +00:00
// Get IRI host value.
2024-01-26 13:17:10 +00:00
accHost := id . Host
2023-03-08 17:19:49 +00:00
2023-05-12 09:15:54 +00: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 19:27:35 +00:00
}
}
2024-01-09 09:42:39 +00:00
if latestAcc . Domain == "" {
2024-01-26 13:17:10 +00:00
// Ensure we have a domain set by this point,
2024-01-09 09:42:39 +00:00
// otherwise it gets stored as a local user!
//
// TODO: there is probably a more granular way
// way of checking this in each of the above parts,
// and honestly it could do with a smol refactor.
return nil , nil , gtserror . Newf ( "empty domain for %s" , uri )
}
2024-01-26 13:17:10 +00:00
/ *
BY THIS POINT we have more or less a fullly - formed
representation of the target account , derived from
a combination of webfinger lookups and dereferencing .
Further fetching beyond this point is for peripheral
things like account avatar , header , emojis .
* /
// Ensure internal db ID is
// set and update fetch time.
2023-02-03 20:03:05 +00:00
latestAcc . ID = account . ID
latestAcc . FetchedAt = time . Now ( )
2023-06-22 19:46:36 +00:00
// Ensure the account's avatar media is populated, passing in existing to check for chages.
if err := d . fetchRemoteAccountAvatar ( ctx , tsport , account , latestAcc ) ; err != nil {
log . Errorf ( ctx , "error fetching remote avatar for account %s: %v" , uri , err )
2023-02-09 08:27:07 +00:00
}
2023-06-22 19:46:36 +00:00
// Ensure the account's avatar media is populated, passing in existing to check for chages.
if err := d . fetchRemoteAccountHeader ( ctx , tsport , account , latestAcc ) ; err != nil {
log . Errorf ( ctx , "error fetching remote header for account %s: %v" , uri , err )
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
}
2024-01-26 13:17:10 +00:00
if account . IsNew ( ) {
// Prefer published/created time from
// apubAcc, fall back to FetchedAt value.
if latestAcc . CreatedAt . IsZero ( ) {
latestAcc . CreatedAt = latestAcc . FetchedAt
}
// Set time of update from the last-fetched date.
2023-02-03 20:03:05 +00:00
latestAcc . UpdatedAt = latestAcc . FetchedAt
2023-03-03 08:34:34 +00:00
// This is new, put it in the database.
2023-05-12 09:15:54 +00:00
err := d . state . DB . PutAccount ( ctx , latestAcc )
2023-03-03 08:34:34 +00:00
if err != nil {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . Newf ( "error putting in database: %w" , err )
2023-02-03 20:03:05 +00:00
}
} else {
2024-01-26 13:17:10 +00:00
// Prefer published time from apubAcc,
// fall back to previous stored value.
if latestAcc . CreatedAt . IsZero ( ) {
latestAcc . CreatedAt = account . CreatedAt
}
2023-02-03 20:03:05 +00:00
// Set time of update from the last-fetched date.
latestAcc . UpdatedAt = latestAcc . FetchedAt
2024-01-26 13:17:10 +00:00
// Carry over existing account language.
2023-02-03 20:03:05 +00:00
latestAcc . Language = account . Language
// This is an existing account, update the model in the database.
2023-05-12 09:15:54 +00:00
if err := d . state . DB . UpdateAccount ( ctx , latestAcc ) ; err != nil {
2023-05-28 12:08:35 +00:00
return nil , nil , gtserror . Newf ( "error updating database: %w" , err )
2022-01-25 10:21:22 +00:00
}
2022-01-24 12:12:17 +00:00
}
2023-05-12 09:15:54 +00:00
return latestAcc , apubAcc , nil
2021-08-10 11:32:39 +00:00
}
2023-10-23 09:58:13 +00:00
func ( d * Dereferencer ) fetchRemoteAccountAvatar ( ctx context . Context , tsport transport . Transport , existing , latestAcc * gtsmodel . Account ) error {
2023-07-12 11:20:15 +00:00
if latestAcc . AvatarRemoteURL == "" {
// No avatar set on newest model, leave
// latest avatar attachment ID empty.
2023-06-22 19:46:36 +00:00
return nil
}
2023-07-12 11:20:15 +00:00
// By default we keep the previous media attachment ID. This will only
// be changed if and when we have the new media loaded into storage.
latestAcc . AvatarMediaAttachmentID = existing . AvatarMediaAttachmentID
2023-06-22 19:46:36 +00:00
2023-07-12 11:20:15 +00:00
// If we had a media attachment ID already, and the URL
// of the attachment hasn't changed from existing -> latest,
// then we may be able to just keep our existing attachment
// without having to make any remote calls.
if latestAcc . AvatarMediaAttachmentID != "" &&
existing . AvatarRemoteURL == latestAcc . AvatarRemoteURL {
// Ensure we have media attachment with the known ID.
2023-06-22 19:46:36 +00:00
media , err := d . state . DB . GetAttachmentByID ( ctx , existing . AvatarMediaAttachmentID )
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
return gtserror . Newf ( "error getting attachment %s: %w" , existing . AvatarMediaAttachmentID , err )
}
2023-07-12 11:20:15 +00:00
// Ensure attachment has correct properties.
if media != nil && media . RemoteURL == latestAcc . AvatarRemoteURL {
// We already have the most up-to-date
// media attachment, keep using it.
2023-06-22 19:46:36 +00:00
return nil
}
}
2023-07-12 11:20:15 +00:00
// If we reach here, we know we need to fetch the most
// up-to-date version of the attachment from remote.
2023-06-22 19:46:36 +00:00
// Parse and validate the newly provided media URL.
2023-07-12 11:20:15 +00:00
avatarURI , err := url . Parse ( latestAcc . AvatarRemoteURL )
2023-02-10 20:15:23 +00:00
if err != nil {
2023-07-12 11:20:15 +00:00
return gtserror . Newf ( "error parsing url %s: %w" , latestAcc . AvatarRemoteURL , err )
2023-02-10 20:15:23 +00:00
}
// Acquire lock for derefs map.
2023-10-31 11:12:22 +00:00
unlock := d . state . FedLocks . Lock ( latestAcc . AvatarRemoteURL )
unlock = doOnce ( unlock )
2023-02-10 20:15:23 +00:00
defer unlock ( )
2023-02-13 20:19:51 +00:00
// Look for an existing dereference in progress.
2023-07-12 11:20:15 +00:00
processing , ok := d . derefAvatars [ latestAcc . AvatarRemoteURL ]
2023-02-10 20:15:23 +00:00
2023-02-13 20:19:51 +00:00
if ! ok {
// 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.
2023-11-10 18:29:26 +00:00
processing = d . mediaManager . PreProcessMedia ( data , latestAcc . ID , & media . AdditionalMediaInfo {
2023-02-13 20:19:51 +00:00
Avatar : func ( ) * bool { v := true ; return & v } ( ) ,
2023-07-12 11:20:15 +00:00
RemoteURL : & latestAcc . AvatarRemoteURL ,
2023-02-13 20:19:51 +00:00
} )
2023-02-10 20:15:23 +00:00
2023-02-13 20:19:51 +00:00
// Store media in map to mark as processing.
2023-07-12 11:20:15 +00:00
d . derefAvatars [ latestAcc . AvatarRemoteURL ] = processing
2023-02-13 20:19:51 +00:00
defer func ( ) {
// On exit safely remove media from map.
2023-10-31 11:12:22 +00:00
unlock := d . state . FedLocks . Lock ( latestAcc . AvatarRemoteURL )
2023-07-12 11:20:15 +00:00
delete ( d . derefAvatars , latestAcc . AvatarRemoteURL )
2023-02-13 20:19:51 +00:00
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-07-12 11:20:15 +00:00
return gtserror . Newf ( "error loading attachment %s: %w" , latestAcc . AvatarRemoteURL , err )
2023-02-10 20:15:23 +00:00
}
2023-06-22 19:46:36 +00:00
// Set the newly loaded avatar media attachment ID.
2023-07-12 11:20:15 +00:00
latestAcc . AvatarMediaAttachmentID = processing . AttachmentID ( )
2023-06-22 19:46:36 +00:00
return nil
2023-02-10 20:15:23 +00:00
}
2023-10-23 09:58:13 +00:00
func ( d * Dereferencer ) fetchRemoteAccountHeader ( ctx context . Context , tsport transport . Transport , existing , latestAcc * gtsmodel . Account ) error {
2023-07-12 11:20:15 +00:00
if latestAcc . HeaderRemoteURL == "" {
// No header set on newest model, leave
// latest header attachment ID empty.
2023-06-22 19:46:36 +00:00
return nil
}
2023-07-12 11:20:15 +00:00
// By default we keep the previous media attachment ID. This will only
// be changed if and when we have the new media loaded into storage.
latestAcc . HeaderMediaAttachmentID = existing . HeaderMediaAttachmentID
2023-06-22 19:46:36 +00:00
2023-07-12 11:20:15 +00:00
// If we had a media attachment ID already, and the URL
// of the attachment hasn't changed from existing -> latest,
// then we may be able to just keep our existing attachment
// without having to make any remote calls.
if latestAcc . HeaderMediaAttachmentID != "" &&
existing . HeaderRemoteURL == latestAcc . HeaderRemoteURL {
// Ensure we have media attachment with the known ID.
2023-06-22 19:46:36 +00:00
media , err := d . state . DB . GetAttachmentByID ( ctx , existing . HeaderMediaAttachmentID )
if err != nil && ! errors . Is ( err , db . ErrNoEntries ) {
return gtserror . Newf ( "error getting attachment %s: %w" , existing . HeaderMediaAttachmentID , err )
}
2023-07-12 11:20:15 +00:00
// Ensure attachment has correct properties.
if media != nil && media . RemoteURL == latestAcc . HeaderRemoteURL {
// We already have the most up-to-date
// media attachment, keep using it.
2023-06-22 19:46:36 +00:00
return nil
}
}
2023-07-12 11:20:15 +00:00
// If we reach here, we know we need to fetch the most
// up-to-date version of the attachment from remote.
2023-06-22 19:46:36 +00:00
// Parse and validate the newly provided media URL.
2023-07-12 11:20:15 +00:00
headerURI , err := url . Parse ( latestAcc . HeaderRemoteURL )
2023-02-09 08:27:07 +00:00
if err != nil {
2023-07-12 11:20:15 +00:00
return gtserror . Newf ( "error parsing url %s: %w" , latestAcc . HeaderRemoteURL , err )
2021-08-10 11:32:39 +00:00
}
2023-02-10 20:15:23 +00:00
// Acquire lock for derefs map.
2023-10-31 11:12:22 +00:00
unlock := d . state . FedLocks . Lock ( latestAcc . HeaderRemoteURL )
unlock = doOnce ( unlock )
2023-02-10 20:15:23 +00:00
defer unlock ( )
2023-02-13 20:19:51 +00:00
// Look for an existing dereference in progress.
2023-07-12 11:20:15 +00:00
processing , ok := d . derefHeaders [ latestAcc . HeaderRemoteURL ]
2023-02-10 20:15:23 +00:00
2023-02-13 20:19:51 +00:00
if ! ok {
2023-06-22 19:46:36 +00:00
// Set the media data function to dereference avatar from URI.
2023-02-13 20:19:51 +00:00
data := func ( ctx context . Context ) ( io . ReadCloser , int64 , error ) {
return tsport . DereferenceMedia ( ctx , headerURI )
}
// Create new media processing request from the media manager instance.
2023-11-10 18:29:26 +00:00
processing = d . mediaManager . PreProcessMedia ( data , latestAcc . ID , & media . AdditionalMediaInfo {
2023-02-13 20:19:51 +00:00
Header : func ( ) * bool { v := true ; return & v } ( ) ,
2023-07-12 11:20:15 +00:00
RemoteURL : & latestAcc . HeaderRemoteURL ,
2023-02-13 20:19:51 +00:00
} )
// Store media in map to mark as processing.
2023-07-12 11:20:15 +00:00
d . derefHeaders [ latestAcc . HeaderRemoteURL ] = 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.
2023-10-31 11:12:22 +00:00
unlock := d . state . FedLocks . Lock ( latestAcc . HeaderRemoteURL )
2023-07-12 11:20:15 +00:00
delete ( d . derefHeaders , latestAcc . HeaderRemoteURL )
2023-02-13 20:19:51 +00:00
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-07-12 11:20:15 +00:00
return gtserror . Newf ( "error loading attachment %s: %w" , latestAcc . HeaderRemoteURL , err )
2021-08-10 11:32:39 +00:00
}
2022-01-24 12:12:17 +00:00
2023-06-22 19:46:36 +00:00
// Set the newly loaded avatar media attachment ID.
2023-07-12 11:20:15 +00:00
latestAcc . HeaderMediaAttachmentID = processing . AttachmentID ( )
2023-06-22 19:46:36 +00:00
return nil
2021-08-10 11:32:39 +00:00
}
2022-01-24 12:12:17 +00:00
2023-10-23 09:58:13 +00:00
func ( d * Dereferencer ) fetchRemoteAccountEmojis ( ctx context . Context , targetAccount * gtsmodel . Account , requestingUsername string ) ( bool , error ) {
2022-09-26 09:56:01 +00:00
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 09:56:01 +00:00
for _ , emojiID := range maybeEmojiIDs {
2023-05-12 09:15:54 +00:00
maybeEmoji , err := d . state . DB . GetEmojiByID ( ctx , emojiID )
2022-09-26 09:56:01 +00: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 09:15:54 +00: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.
2023-10-23 09:58:13 +00:00
func ( d * Dereferencer ) dereferenceAccountFeatured ( ctx context . Context , requestUser string , account * gtsmodel . Account ) error {
2023-05-12 09:15:54 +00:00
uri , err := url . Parse ( account . FeaturedCollectionURI )
2023-03-01 17:52:44 +00:00
if err != nil {
return err
}
2023-05-12 09:15:54 +00: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-28 12:08:35 +00:00
return gtserror . Newf ( "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 {
2023-05-28 12:08:35 +00:00
return gtserror . Newf ( "error unmarshalling bytes into json: %w" , err )
2023-03-01 17:52:44 +00:00
}
t , err := streams . ToType ( ctx , m )
if err != nil {
2023-05-28 12:08:35 +00:00
return gtserror . Newf ( "error resolving json into ap vocab type: %w" , err )
2023-03-01 17:52:44 +00:00
}
if t . GetTypeName ( ) != ap . ObjectOrderedCollection {
2023-05-28 12:08:35 +00:00
return gtserror . Newf ( "%s was not an OrderedCollection" , uri )
2023-03-01 17:52:44 +00:00
}
collection , ok := t . ( vocab . ActivityStreamsOrderedCollection )
if ! ok {
2023-05-28 12:08:35 +00:00
return gtserror . New ( "couldn't coerce OrderedCollection" )
2023-03-01 17:52:44 +00:00
}
items := collection . GetActivityStreamsOrderedItems ( )
if items == nil {
2023-05-28 12:08:35 +00:00
return gtserror . New ( "nil orderedItems" )
2023-03-01 17:52:44 +00:00
}
// Get previous pinned statuses (we'll need these later).
2023-05-12 09:15:54 +00: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 ) {
2023-05-28 12:08:35 +00:00
return gtserror . Newf ( "error getting account pinned statuses: %w" , err )
2023-03-01 17:52:44 +00:00
}
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 )
2024-01-31 13:29:47 +00:00
// Search for status by URI. Note this may return an existing model
// we have stored with an error from attempted update, so check both.
2023-11-04 20:21:20 +00:00
status , _ , _ , err := d . getStatusByURI ( ctx , requestUser , statusURI )
2023-03-01 17:52:44 +00:00
if err != nil {
2023-05-12 09:15:54 +00:00
log . Errorf ( ctx , "error getting status from featured collection %s: %v" , statusURI , err )
2024-01-31 13:29:47 +00:00
if status == nil {
// This is only unactionable
// if no status was returned.
continue
}
2023-03-01 17:52:44 +00:00
}
// If the status was already pinned, we don't need to do anything.
if ! status . PinnedAt . IsZero ( ) {
continue
}
2023-05-12 09:15:54 +00: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 09:15:54 +00: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 09:15:54 +00: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
}