]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - server/lib/activitypub/send/utils.ts
Add subscriptions endpoints to REST API
[github/Chocobozzz/PeerTube.git] / server / lib / activitypub / send / utils.ts
1 import { Transaction } from 'sequelize'
2 import { Activity } from '../../../../shared/models/activitypub'
3 import { logger } from '../../../helpers/logger'
4 import { ActorModel } from '../../../models/activitypub/actor'
5 import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
6 import { JobQueue } from '../../job-queue'
7 import { VideoModel } from '../../../models/video/video'
8 import { getActorsInvolvedInVideo } from '../audience'
9 import { getServerActor } from '../../../helpers/utils'
10
11 async function forwardVideoRelatedActivity (
12 activity: Activity,
13 t: Transaction,
14 followersException: ActorModel[] = [],
15 video: VideoModel
16 ) {
17 // Mastodon does not add our announces in audience, so we forward to them manually
18 const additionalActors = await getActorsInvolvedInVideo(video, t)
19 const additionalFollowerUrls = additionalActors.map(a => a.followersUrl)
20
21 return forwardActivity(activity, t, followersException, additionalFollowerUrls)
22 }
23
24 async function forwardActivity (
25 activity: Activity,
26 t: Transaction,
27 followersException: ActorModel[] = [],
28 additionalFollowerUrls: string[] = []
29 ) {
30 logger.info('Forwarding activity %s.', activity.id)
31
32 const to = activity.to || []
33 const cc = activity.cc || []
34
35 const followersUrls = additionalFollowerUrls
36 for (const dest of to.concat(cc)) {
37 if (dest.endsWith('/followers')) {
38 followersUrls.push(dest)
39 }
40 }
41
42 const toActorFollowers = await ActorModel.listByFollowersUrls(followersUrls, t)
43 const uris = await computeFollowerUris(toActorFollowers, followersException, t)
44
45 if (uris.length === 0) {
46 logger.info('0 followers for %s, no forwarding.', toActorFollowers.map(a => a.id).join(', '))
47 return undefined
48 }
49
50 logger.debug('Creating forwarding job.', { uris })
51
52 const payload = {
53 uris,
54 body: activity
55 }
56 return JobQueue.Instance.createJob({ type: 'activitypub-http-broadcast', payload })
57 }
58
59 async function broadcastToFollowers (
60 data: any,
61 byActor: ActorModel,
62 toActorFollowers: ActorModel[],
63 t: Transaction,
64 actorsException: ActorModel[] = []
65 ) {
66 const uris = await computeFollowerUris(toActorFollowers, actorsException, t)
67 return broadcastTo(uris, data, byActor)
68 }
69
70 async function broadcastToActors (
71 data: any,
72 byActor: ActorModel,
73 toActors: ActorModel[],
74 actorsException: ActorModel[] = []
75 ) {
76 const uris = await computeUris(toActors, actorsException)
77 return broadcastTo(uris, data, byActor)
78 }
79
80 async function broadcastTo (uris: string[], data: any, byActor: ActorModel) {
81 if (uris.length === 0) return undefined
82
83 logger.debug('Creating broadcast job.', { uris })
84
85 const payload = {
86 uris,
87 signatureActorId: byActor.id,
88 body: data
89 }
90
91 return JobQueue.Instance.createJob({ type: 'activitypub-http-broadcast', payload })
92 }
93
94 async function unicastTo (data: any, byActor: ActorModel, toActorUrl: string) {
95 logger.debug('Creating unicast job.', { uri: toActorUrl })
96
97 const payload = {
98 uri: toActorUrl,
99 signatureActorId: byActor.id,
100 body: data
101 }
102
103 return JobQueue.Instance.createJob({ type: 'activitypub-http-unicast', payload })
104 }
105
106 // ---------------------------------------------------------------------------
107
108 export {
109 broadcastToFollowers,
110 unicastTo,
111 forwardActivity,
112 broadcastToActors,
113 forwardVideoRelatedActivity
114 }
115
116 // ---------------------------------------------------------------------------
117
118 async function computeFollowerUris (toActorFollower: ActorModel[], actorsException: ActorModel[], t: Transaction) {
119 const toActorFollowerIds = toActorFollower.map(a => a.id)
120
121 const result = await ActorFollowModel.listAcceptedFollowerSharedInboxUrls(toActorFollowerIds, t)
122 const sharedInboxesException = await buildSharedInboxesException(actorsException)
123
124 return result.data.filter(sharedInbox => sharedInboxesException.indexOf(sharedInbox) === -1)
125 }
126
127 async function computeUris (toActors: ActorModel[], actorsException: ActorModel[] = []) {
128 const serverActor = await getServerActor()
129 const targetUrls = toActors
130 .filter(a => a.id !== serverActor.id) // Don't send to ourselves
131 .map(a => a.sharedInboxUrl || a.inboxUrl)
132
133 const toActorSharedInboxesSet = new Set(targetUrls)
134
135 const sharedInboxesException = await buildSharedInboxesException(actorsException)
136 return Array.from(toActorSharedInboxesSet)
137 .filter(sharedInbox => sharedInboxesException.indexOf(sharedInbox) === -1)
138 }
139
140 async function buildSharedInboxesException (actorsException: ActorModel[]) {
141 const serverActor = await getServerActor()
142
143 return actorsException
144 .map(f => f.sharedInboxUrl || f.inboxUrl)
145 .concat([ serverActor.sharedInboxUrl ])
146 }