95 lines
3.0 KiB
TypeScript
95 lines
3.0 KiB
TypeScript
import { JobCategory } from '../../../../shared'
|
|
import { buildSignedActivity } from '../../../helpers/activitypub'
|
|
import { logger } from '../../../helpers/logger'
|
|
import { getServerActor } from '../../../helpers/utils'
|
|
import { ACTIVITY_PUB } from '../../../initializers'
|
|
import { ActorModel } from '../../../models/activitypub/actor'
|
|
import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
|
|
import { JobHandler, JobScheduler } from '../job-scheduler'
|
|
|
|
import * as activitypubHttpBroadcastHandler from './activitypub-http-broadcast-handler'
|
|
import * as activitypubHttpFetcherHandler from './activitypub-http-fetcher-handler'
|
|
import * as activitypubHttpUnicastHandler from './activitypub-http-unicast-handler'
|
|
|
|
type ActivityPubHttpPayload = {
|
|
uris: string[]
|
|
signatureActorId?: number
|
|
body?: any
|
|
attemptNumber?: number
|
|
}
|
|
|
|
const jobHandlers: { [ handlerName: string ]: JobHandler<ActivityPubHttpPayload, void> } = {
|
|
activitypubHttpBroadcastHandler,
|
|
activitypubHttpUnicastHandler,
|
|
activitypubHttpFetcherHandler
|
|
}
|
|
const jobCategory: JobCategory = 'activitypub-http'
|
|
|
|
const activitypubHttpJobScheduler = new JobScheduler(jobCategory, jobHandlers)
|
|
|
|
async function maybeRetryRequestLater (err: Error, payload: ActivityPubHttpPayload, uri: string) {
|
|
logger.warn('Cannot make request to %s.', uri, err)
|
|
|
|
let attemptNumber = payload.attemptNumber || 1
|
|
attemptNumber += 1
|
|
|
|
if (attemptNumber < ACTIVITY_PUB.MAX_HTTP_ATTEMPT) {
|
|
logger.debug('Retrying request to %s (attempt %d/%d).', uri, attemptNumber, ACTIVITY_PUB.MAX_HTTP_ATTEMPT, err)
|
|
|
|
const actor = await ActorFollowModel.loadByFollowerInbox(uri, undefined)
|
|
if (!actor) {
|
|
logger.debug('Actor %s is not a follower, do not retry the request.', uri)
|
|
return false
|
|
}
|
|
|
|
const newPayload = Object.assign(payload, {
|
|
uris: [ uri ],
|
|
attemptNumber
|
|
})
|
|
await activitypubHttpJobScheduler.createJob(undefined, 'activitypubHttpUnicastHandler', newPayload)
|
|
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
async function computeBody (payload: ActivityPubHttpPayload) {
|
|
let body = payload.body
|
|
|
|
if (payload.signatureActorId) {
|
|
const actorSignature = await ActorModel.load(payload.signatureActorId)
|
|
if (!actorSignature) throw new Error('Unknown signature actor id.')
|
|
body = await buildSignedActivity(actorSignature, payload.body)
|
|
}
|
|
|
|
return body
|
|
}
|
|
|
|
async function buildSignedRequestOptions (payload: ActivityPubHttpPayload) {
|
|
let actor: ActorModel
|
|
if (payload.signatureActorId) {
|
|
actor = await ActorModel.load(payload.signatureActorId)
|
|
if (!actor) throw new Error('Unknown signature actor id.')
|
|
} else {
|
|
// We need to sign the request, so use the server
|
|
actor = await getServerActor()
|
|
}
|
|
|
|
const keyId = actor.getWebfingerUrl()
|
|
return {
|
|
algorithm: 'rsa-sha256',
|
|
authorizationHeaderName: 'Signature',
|
|
keyId,
|
|
key: actor.privateKey
|
|
}
|
|
}
|
|
|
|
export {
|
|
ActivityPubHttpPayload,
|
|
activitypubHttpJobScheduler,
|
|
maybeRetryRequestLater,
|
|
computeBody,
|
|
buildSignedRequestOptions
|
|
}
|