server/lib/job-queue/handlers/activitypub-http-broadcast.ts

   1 import * as Bull from 'bull'
   2 import * as Bluebird from 'bluebird'
   3 import { logger } from '../../../helpers/logger'
   4 import { doRequest } from '../../../helpers/requests'
   5 import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
   6 import { buildGlobalHeaders, buildSignedRequestOptions, computeBody } from './utils/activitypub-http-utils'
   7 import { BROADCAST_CONCURRENCY, JOB_REQUEST_TIMEOUT } from '../../../initializers'
   8 import { ActorFollowScoreCache } from '../../cache'
   9
  10 export type ActivitypubHttpBroadcastPayload = {
  11   uris: string[]
  12   signatureActorId?: number
  13   body: any
  14 }
  15
  16 async function processActivityPubHttpBroadcast (job: Bull.Job) {
  17   logger.info('Processing ActivityPub broadcast in job %d.', job.id)
  18
  19   const payload = job.data as ActivitypubHttpBroadcastPayload
  20
  21   const body = await computeBody(payload)
  22   const httpSignatureOptions = await buildSignedRequestOptions(payload)
  23
  24   const options = {
  25     method: 'POST',
  26     uri: '',
  27     json: body,
  28     httpSignature: httpSignatureOptions,
  29     timeout: JOB_REQUEST_TIMEOUT,
  30     headers: buildGlobalHeaders(body)
  31   }
  32
  33   const badUrls: string[] = []
  34   const goodUrls: string[] = []
  35
  36   await Bluebird.map(payload.uris, uri => {
  37     return doRequest(Object.assign({}, options, { uri }))
  38       .then(() => goodUrls.push(uri))
  39       .catch(() => badUrls.push(uri))
  40   }, { concurrency: BROADCAST_CONCURRENCY })
  41
  42   return ActorFollowScoreCache.Instance.updateActorFollowsScore(goodUrls, badUrls)
  43 }
  44
  45 // ---------------------------------------------------------------------------
  46
  47 export {
  48   processActivityPubHttpBroadcast
  49 }