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