Split files in activitypub server

1 files changed, 113 insertions, 0 deletions

diff --git a/server/lib/activitypub/send/utils.ts b/server/lib/activitypub/send/utils.ts
new file mode 100644
index 000000000..80d4463ff
--- /dev/null
+++ b/server/lib/activitypub/send/utils.ts
@@ -0,0 +1,113 @@
+import { Transaction } from 'sequelize'
+import { Activity } from '../../../../shared/models/activitypub'
+import { logger } from '../../../helpers/logger'
+import { ActorModel } from '../../../models/activitypub/actor'
+import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
+import { JobQueue } from '../../job-queue'
+
+async function forwardActivity (
+  activity: Activity,
+  t: Transaction,
+  followersException: ActorModel[] = [],
+  additionalFollowerUrls: string[] = []
+) {
+  const to = activity.to || []
+  const cc = activity.cc || []
+
+  const followersUrls = additionalFollowerUrls
+  for (const dest of to.concat(cc)) {
+    if (dest.endsWith('/followers')) {
+      followersUrls.push(dest)
+    }
+  }
+
+  const toActorFollowers = await ActorModel.listByFollowersUrls(followersUrls, t)
+  const uris = await computeFollowerUris(toActorFollowers, followersException, t)
+
+  if (uris.length === 0) {
+    logger.info('0 followers for %s, no forwarding.', toActorFollowers.map(a => a.id).join(', '))
+    return undefined
+  }
+
+  logger.debug('Creating forwarding job.', { uris })
+
+  const payload = {
+    uris,
+    body: activity
+  }
+  return JobQueue.Instance.createJob({ type: 'activitypub-http-broadcast', payload })
+}
+
+async function broadcastToFollowers (
+  data: any,
+  byActor: ActorModel,
+  toActorFollowers: ActorModel[],
+  t: Transaction,
+  actorsException: ActorModel[] = []
+) {
+  const uris = await computeFollowerUris(toActorFollowers, actorsException, t)
+  return broadcastTo(uris, data, byActor)
+}
+
+async function broadcastToActors (
+  data: any,
+  byActor: ActorModel,
+  toActors: ActorModel[],
+  actorsException: ActorModel[] = []
+) {
+  const uris = await computeUris(toActors, actorsException)
+  return broadcastTo(uris, data, byActor)
+}
+
+async function broadcastTo (uris: string[], data: any, byActor: ActorModel) {
+  if (uris.length === 0) return undefined
+
+  logger.debug('Creating broadcast job.', { uris })
+
+  const payload = {
+    uris,
+    signatureActorId: byActor.id,
+    body: data
+  }
+
+  return JobQueue.Instance.createJob({ type: 'activitypub-http-broadcast', payload })
+}
+
+async function unicastTo (data: any, byActor: ActorModel, toActorUrl: string) {
+  logger.debug('Creating unicast job.', { uri: toActorUrl })
+
+  const payload = {
+    uri: toActorUrl,
+    signatureActorId: byActor.id,
+    body: data
+  }
+
+  return JobQueue.Instance.createJob({ type: 'activitypub-http-unicast', payload })
+}
+
+// ---------------------------------------------------------------------------
+
+export {
+  broadcastToFollowers,
+  unicastTo,
+  forwardActivity,
+  broadcastToActors
+}
+
+// ---------------------------------------------------------------------------
+
+async function computeFollowerUris (toActorFollower: ActorModel[], actorsException: ActorModel[], t: Transaction) {
+  const toActorFollowerIds = toActorFollower.map(a => a.id)
+
+  const result = await ActorFollowModel.listAcceptedFollowerSharedInboxUrls(toActorFollowerIds, t)
+  const sharedInboxesException = actorsException.map(f => f.sharedInboxUrl || f.inboxUrl)
+  return result.data.filter(sharedInbox => sharedInboxesException.indexOf(sharedInbox) === -1)
+}
+
+async function computeUris (toActors: ActorModel[], actorsException: ActorModel[] = []) {
+  const toActorSharedInboxesSet = new Set(toActors.map(a => a.sharedInboxUrl || a.inboxUrl))
+
+  const sharedInboxesException = actorsException.map(f => f.sharedInboxUrl || f.inboxUrl)
+  return Array.from(toActorSharedInboxesSet)
+              .filter(sharedInbox => sharedInboxesException.indexOf(sharedInbox) === -1)
+}