server/lib/jobs/activitypub-http-job-scheduler/activitypub-http-job-scheduler.ts

   1 import { JobCategory } from '../../../../shared'
   2 import { buildSignedActivity, getServerActor, logger } from '../../../helpers'
   3 import { ACTIVITY_PUB } from '../../../initializers'
   4 import { ActorModel } from '../../../models/activitypub/actor'
   5 import { JobHandler, JobScheduler } from '../job-scheduler'
   6
   7 import * as activitypubHttpBroadcastHandler from './activitypub-http-broadcast-handler'
   8 import * as activitypubHttpFetcherHandler from './activitypub-http-fetcher-handler'
   9 import * as activitypubHttpUnicastHandler from './activitypub-http-unicast-handler'
  10
  11 type ActivityPubHttpPayload = {
  12   uris: string[]
  13   signatureActorId?: number
  14   body?: any
  15   attemptNumber?: number
  16 }
  17
  18 const jobHandlers: { [ handlerName: string ]: JobHandler<ActivityPubHttpPayload, void> } = {
  19   activitypubHttpBroadcastHandler,
  20   activitypubHttpUnicastHandler,
  21   activitypubHttpFetcherHandler
  22 }
  23 const jobCategory: JobCategory = 'activitypub-http'
  24
  25 const activitypubHttpJobScheduler = new JobScheduler(jobCategory, jobHandlers)
  26
  27 function maybeRetryRequestLater (err: Error, payload: ActivityPubHttpPayload, uri: string) {
  28   logger.warn('Cannot make request to %s.', uri, err)
  29
  30   let attemptNumber = payload.attemptNumber || 1
  31   attemptNumber += 1
  32
  33   if (attemptNumber < ACTIVITY_PUB.MAX_HTTP_ATTEMPT) {
  34     logger.debug('Retrying request to %s (attempt %d/%d).', uri, attemptNumber, ACTIVITY_PUB.MAX_HTTP_ATTEMPT, err)
  35
  36     const newPayload = Object.assign(payload, {
  37       uris: [ uri ],
  38       attemptNumber
  39     })
  40     return activitypubHttpJobScheduler.createJob(undefined, 'activitypubHttpUnicastHandler', newPayload)
  41   }
  42 }
  43
  44 async function computeBody (payload: ActivityPubHttpPayload) {
  45   let body = payload.body
  46
  47   if (payload.signatureActorId) {
  48     const actorSignature = await ActorModel.load(payload.signatureActorId)
  49     if (!actorSignature) throw new Error('Unknown signature actor id.')
  50     body = await buildSignedActivity(actorSignature, payload.body)
  51   }
  52
  53   return body
  54 }
  55
  56 async function buildSignedRequestOptions (payload: ActivityPubHttpPayload) {
  57   let actor: ActorModel
  58   if (payload.signatureActorId) {
  59     actor = await ActorModel.load(payload.signatureActorId)
  60     if (!actor) throw new Error('Unknown signature actor id.')
  61   } else {
  62     // We need to sign the request, so use the server
  63     actor = await getServerActor()
  64   }
  65
  66   const keyId = actor.getWebfingerUrl()
  67   return {
  68     algorithm: 'rsa-sha256',
  69     authorizationHeaderName: 'Signature',
  70     keyId,
  71     key: actor.privateKey
  72   }
  73 }
  74
  75 export {
  76   ActivityPubHttpPayload,
  77   activitypubHttpJobScheduler,
  78   maybeRetryRequestLater,
  79   computeBody,
  80   buildSignedRequestOptions
  81 }