PeerTube/server/lib/activitypub/send/shared/send-utils.ts

281 lines
8.4 KiB
TypeScript
Raw Normal View History

2017-11-20 09:43:39 +01:00
import { Transaction } from 'sequelize'
import { ActorFollowHealthCache } from '@server/lib/actor-follow-health-cache'
2021-05-11 11:15:29 +02:00
import { getServerActor } from '@server/models/application/application'
2022-03-23 16:14:33 +01:00
import { Activity, ActivityAudience, ActivitypubHttpBroadcastPayload } from '@shared/models'
2021-05-11 11:15:29 +02:00
import { ContextType } from '@shared/models/activitypub/context'
import { afterCommitIfTransaction } from '../../../../helpers/database-utils'
import { logger } from '../../../../helpers/logger'
import { ActorModel } from '../../../../models/actor/actor'
import { ActorFollowModel } from '../../../../models/actor/actor-follow'
import { MActor, MActorId, MActorLight, MActorWithInboxes, MVideoAccountLight, MVideoId, MVideoImmutable } from '../../../../types/models'
import { JobQueue } from '../../../job-queue'
import { getActorsInvolvedInVideo, getAudienceFromFollowersOf, getOriginVideoAudience } from './audience-utils'
2018-09-14 16:51:35 +02:00
async function sendVideoRelatedActivity (activityBuilder: (audience: ActivityAudience) => Activity, options: {
2020-01-31 16:56:52 +01:00
byActor: MActorLight
2020-02-04 16:14:33 +01:00
video: MVideoImmutable | MVideoAccountLight
2022-03-23 16:14:33 +01:00
contextType: ContextType
2020-02-04 09:12:08 +01:00
transaction?: Transaction
2018-09-14 16:51:35 +02:00
}) {
const { byActor, video, transaction, contextType } = options
const actorsInvolvedInVideo = await getActorsInvolvedInVideo(video, transaction)
2018-09-14 16:51:35 +02:00
// Send to origin
if (video.isOwned() === false) {
return sendVideoActivityToOrigin(activityBuilder, options)
2018-09-14 16:51:35 +02:00
}
// Send to followers
const audience = getAudienceFromFollowersOf(actorsInvolvedInVideo)
const activity = activityBuilder(audience)
const actorsException = [ byActor ]
2019-07-29 11:59:29 +02:00
2022-03-23 16:14:33 +01:00
return broadcastToFollowers({
data: activity,
byActor,
toFollowersOf: actorsInvolvedInVideo,
transaction,
actorsException,
contextType
})
2018-09-14 16:51:35 +02:00
}
async function sendVideoActivityToOrigin (activityBuilder: (audience: ActivityAudience) => Activity, options: {
byActor: MActorLight
video: MVideoImmutable | MVideoAccountLight
2022-03-23 16:14:33 +01:00
contextType: ContextType
actorsInvolvedInVideo?: MActorLight[]
transaction?: Transaction
}) {
const { byActor, video, actorsInvolvedInVideo, transaction, contextType } = options
if (video.isOwned()) throw new Error('Cannot send activity to owned video origin ' + video.url)
let accountActor: MActorLight = (video as MVideoAccountLight).VideoChannel?.Account?.Actor
if (!accountActor) accountActor = await ActorModel.loadAccountActorByVideoId(video.id, transaction)
const audience = getOriginVideoAudience(accountActor, actorsInvolvedInVideo)
const activity = activityBuilder(audience)
return afterCommitIfTransaction(transaction, () => {
2022-03-23 16:14:33 +01:00
return unicastTo({
data: activity,
byActor,
toActorUrl: accountActor.getSharedInbox(),
contextType
})
})
}
// ---------------------------------------------------------------------------
async function forwardVideoRelatedActivity (
activity: Activity,
t: Transaction,
2020-02-28 16:03:39 +01:00
followersException: MActorWithInboxes[],
video: MVideoId
) {
// Mastodon does not add our announces in audience, so we forward to them manually
const additionalActors = await getActorsInvolvedInVideo(video, t)
const additionalFollowerUrls = additionalActors.map(a => a.followersUrl)
return forwardActivity(activity, t, followersException, additionalFollowerUrls)
}
async function forwardActivity (
activity: Activity,
t: Transaction,
followersException: MActorWithInboxes[] = [],
2018-01-08 10:00:35 +01:00
additionalFollowerUrls: string[] = []
) {
2018-07-30 17:02:40 +02:00
logger.info('Forwarding activity %s.', activity.id)
const to = activity.to || []
const cc = activity.cc || []
2018-01-08 10:00:35 +01:00
const followersUrls = additionalFollowerUrls
for (const dest of to.concat(cc)) {
if (dest.endsWith('/followers')) {
followersUrls.push(dest)
}
}
2017-12-14 17:38:41 +01:00
const toActorFollowers = await ActorModel.listByFollowersUrls(followersUrls, t)
const uris = await computeFollowerUris(toActorFollowers, followersException, t)
if (uris.length === 0) {
2017-12-14 17:38:41 +01:00
logger.info('0 followers for %s, no forwarding.', toActorFollowers.map(a => a.id).join(', '))
2017-11-24 15:00:10 +01:00
return undefined
}
logger.debug('Creating forwarding job.', { uris })
2022-03-23 16:14:33 +01:00
const payload: ActivitypubHttpBroadcastPayload = {
uris,
2022-03-23 16:14:33 +01:00
body: activity,
contextType: null
}
2019-07-29 11:59:29 +02:00
return afterCommitIfTransaction(t, () => JobQueue.Instance.createJob({ type: 'activitypub-http-broadcast', payload }))
}
2017-11-20 09:43:39 +01:00
// ---------------------------------------------------------------------------
2022-03-23 16:14:33 +01:00
async function broadcastToFollowers (options: {
data: any
byActor: MActorId
toFollowersOf: MActorId[]
transaction: Transaction
contextType: ContextType
actorsException?: MActorWithInboxes[]
}) {
const { data, byActor, toFollowersOf, transaction, contextType, actorsException = [] } = options
2019-07-29 11:59:29 +02:00
2022-03-23 16:14:33 +01:00
const uris = await computeFollowerUris(toFollowersOf, actorsException, transaction)
return afterCommitIfTransaction(transaction, () => {
return broadcastTo({
uris,
data,
byActor,
contextType
})
})
2018-01-08 10:00:35 +01:00
}
2022-03-23 16:14:33 +01:00
async function broadcastToActors (options: {
data: any
byActor: MActorId
toActors: MActor[]
transaction: Transaction
contextType: ContextType
actorsException?: MActorWithInboxes[]
}) {
const { data, byActor, toActors, transaction, contextType, actorsException = [] } = options
2018-01-08 10:00:35 +01:00
const uris = await computeUris(toActors, actorsException)
2022-03-23 16:14:33 +01:00
return afterCommitIfTransaction(transaction, () => {
return broadcastTo({
uris,
data,
byActor,
contextType
})
})
2018-01-08 10:00:35 +01:00
}
2022-03-23 16:14:33 +01:00
function broadcastTo (options: {
uris: string[]
data: any
byActor: MActorId
contextType: ContextType
}) {
const { uris, data, byActor, contextType } = options
2018-01-08 10:00:35 +01:00
if (uris.length === 0) return undefined
2017-11-20 09:43:39 +01:00
const broadcastUris: string[] = []
const unicastUris: string[] = []
2017-11-22 16:25:03 +01:00
// Bad URIs could be slow to respond, prefer to process them in a dedicated queue
for (const uri of uris) {
if (ActorFollowHealthCache.Instance.isBadInbox(uri)) {
unicastUris.push(uri)
} else {
broadcastUris.push(uri)
}
}
logger.debug('Creating broadcast job.', { broadcastUris, unicastUris })
if (broadcastUris.length !== 0) {
const payload = {
uris: broadcastUris,
signatureActorId: byActor.id,
body: data,
contextType
}
JobQueue.Instance.createJob({ type: 'activitypub-http-broadcast', payload })
2017-11-20 09:43:39 +01:00
}
for (const unicastUri of unicastUris) {
const payload = {
uri: unicastUri,
signatureActorId: byActor.id,
body: data,
contextType
}
JobQueue.Instance.createJob({ type: 'activitypub-http-unicast', payload })
}
2017-11-20 09:43:39 +01:00
}
2022-03-23 16:14:33 +01:00
function unicastTo (options: {
data: any
byActor: MActorId
toActorUrl: string
contextType: ContextType
}) {
const { data, byActor, toActorUrl, contextType } = options
2017-12-14 17:38:41 +01:00
logger.debug('Creating unicast job.', { uri: toActorUrl })
const payload = {
uri: toActorUrl,
2017-12-14 17:38:41 +01:00
signatureActorId: byActor.id,
body: data,
contextType
2017-11-20 09:43:39 +01:00
}
2019-07-29 11:59:29 +02:00
JobQueue.Instance.createJob({ type: 'activitypub-http-unicast', payload })
2017-11-20 09:43:39 +01:00
}
2018-05-25 11:32:36 +02:00
// ---------------------------------------------------------------------------
2017-11-20 09:43:39 +01:00
2018-05-25 11:32:36 +02:00
export {
broadcastToFollowers,
unicastTo,
forwardActivity,
broadcastToActors,
sendVideoActivityToOrigin,
2018-09-14 16:51:35 +02:00
forwardVideoRelatedActivity,
sendVideoRelatedActivity
2017-11-20 09:43:39 +01:00
}
2018-05-25 11:32:36 +02:00
// ---------------------------------------------------------------------------
2017-12-19 10:34:56 +01:00
async function computeFollowerUris (toFollowersOf: MActorId[], actorsException: MActorWithInboxes[], t: Transaction) {
2018-09-11 16:27:07 +02:00
const toActorFollowerIds = toFollowersOf.map(a => a.id)
2017-12-14 17:38:41 +01:00
const result = await ActorFollowModel.listAcceptedFollowerSharedInboxUrls(toActorFollowerIds, t)
const sharedInboxesException = await buildSharedInboxesException(actorsException)
2020-02-28 16:03:39 +01:00
return result.data.filter(sharedInbox => sharedInboxesException.includes(sharedInbox) === false)
2018-01-08 10:00:35 +01:00
}
async function computeUris (toActors: MActor[], actorsException: MActorWithInboxes[] = []) {
const serverActor = await getServerActor()
const targetUrls = toActors
.filter(a => a.id !== serverActor.id) // Don't send to ourselves
.map(a => a.getSharedInbox())
const toActorSharedInboxesSet = new Set(targetUrls)
2018-01-08 10:00:35 +01:00
const sharedInboxesException = await buildSharedInboxesException(actorsException)
2018-01-08 10:00:35 +01:00
return Array.from(toActorSharedInboxesSet)
2020-02-28 16:03:39 +01:00
.filter(sharedInbox => sharedInboxesException.includes(sharedInbox) === false)
2017-11-20 09:43:39 +01:00
}
async function buildSharedInboxesException (actorsException: MActorWithInboxes[]) {
const serverActor = await getServerActor()
return actorsException
.map(f => f.getSharedInbox())
.concat([ serverActor.sharedInboxUrl ])
}