diff options
author | Chocobozzz <me@florianbigard.com> | 2022-03-23 16:14:33 +0100 |
---|---|---|
committer | Chocobozzz <me@florianbigard.com> | 2022-03-24 09:40:46 +0100 |
commit | a219c9100b3ce8774d454497d46be87465bf664e (patch) | |
tree | caa869e47919a9e23cc86dcece1100e239683b8c /server/lib/job-queue | |
parent | 7e98a7df7d04e19ba67163a86c7b876d78d76839 (diff) | |
download | PeerTube-a219c9100b3ce8774d454497d46be87465bf664e.tar.gz PeerTube-a219c9100b3ce8774d454497d46be87465bf664e.tar.zst PeerTube-a219c9100b3ce8774d454497d46be87465bf664e.zip |
Refactor AP context builder
Diffstat (limited to 'server/lib/job-queue')
3 files changed, 2 insertions, 61 deletions
diff --git a/server/lib/job-queue/handlers/activitypub-http-broadcast.ts b/server/lib/job-queue/handlers/activitypub-http-broadcast.ts index fbf01d276..709e8501f 100644 --- a/server/lib/job-queue/handlers/activitypub-http-broadcast.ts +++ b/server/lib/job-queue/handlers/activitypub-http-broadcast.ts | |||
@@ -1,11 +1,11 @@ | |||
1 | import { map } from 'bluebird' | 1 | import { map } from 'bluebird' |
2 | import { Job } from 'bull' | 2 | import { Job } from 'bull' |
3 | import { buildGlobalHeaders, buildSignedRequestOptions, computeBody } from '@server/lib/activitypub/send' | ||
3 | import { ActorFollowHealthCache } from '@server/lib/actor-follow-health-cache' | 4 | import { ActorFollowHealthCache } from '@server/lib/actor-follow-health-cache' |
4 | import { ActivitypubHttpBroadcastPayload } from '@shared/models' | 5 | import { ActivitypubHttpBroadcastPayload } from '@shared/models' |
5 | import { logger } from '../../../helpers/logger' | 6 | import { logger } from '../../../helpers/logger' |
6 | import { doRequest } from '../../../helpers/requests' | 7 | import { doRequest } from '../../../helpers/requests' |
7 | import { BROADCAST_CONCURRENCY } from '../../../initializers/constants' | 8 | import { BROADCAST_CONCURRENCY } from '../../../initializers/constants' |
8 | import { buildGlobalHeaders, buildSignedRequestOptions, computeBody } from './utils/activitypub-http-utils' | ||
9 | 9 | ||
10 | async function processActivityPubHttpBroadcast (job: Job) { | 10 | async function processActivityPubHttpBroadcast (job: Job) { |
11 | logger.info('Processing ActivityPub broadcast in job %d.', job.id) | 11 | logger.info('Processing ActivityPub broadcast in job %d.', job.id) |
diff --git a/server/lib/job-queue/handlers/activitypub-http-unicast.ts b/server/lib/job-queue/handlers/activitypub-http-unicast.ts index 673583d2b..99bcd3e8d 100644 --- a/server/lib/job-queue/handlers/activitypub-http-unicast.ts +++ b/server/lib/job-queue/handlers/activitypub-http-unicast.ts | |||
@@ -1,9 +1,9 @@ | |||
1 | import { Job } from 'bull' | 1 | import { Job } from 'bull' |
2 | import { buildGlobalHeaders, buildSignedRequestOptions, computeBody } from '@server/lib/activitypub/send' | ||
2 | import { ActivitypubHttpUnicastPayload } from '@shared/models' | 3 | import { ActivitypubHttpUnicastPayload } from '@shared/models' |
3 | import { logger } from '../../../helpers/logger' | 4 | import { logger } from '../../../helpers/logger' |
4 | import { doRequest } from '../../../helpers/requests' | 5 | import { doRequest } from '../../../helpers/requests' |
5 | import { ActorFollowHealthCache } from '../../actor-follow-health-cache' | 6 | import { ActorFollowHealthCache } from '../../actor-follow-health-cache' |
6 | import { buildGlobalHeaders, buildSignedRequestOptions, computeBody } from './utils/activitypub-http-utils' | ||
7 | 7 | ||
8 | async function processActivityPubHttpUnicast (job: Job) { | 8 | async function processActivityPubHttpUnicast (job: Job) { |
9 | logger.info('Processing ActivityPub unicast in job %d.', job.id) | 9 | logger.info('Processing ActivityPub unicast in job %d.', job.id) |
diff --git a/server/lib/job-queue/handlers/utils/activitypub-http-utils.ts b/server/lib/job-queue/handlers/utils/activitypub-http-utils.ts deleted file mode 100644 index 2a03325b7..000000000 --- a/server/lib/job-queue/handlers/utils/activitypub-http-utils.ts +++ /dev/null | |||
@@ -1,59 +0,0 @@ | |||
1 | import { buildDigest } from '@server/helpers/peertube-crypto' | ||
2 | import { buildSignedActivity } from '@server/lib/activitypub/activity' | ||
3 | import { getServerActor } from '@server/models/application/application' | ||
4 | import { ContextType } from '@shared/models/activitypub/context' | ||
5 | import { ACTIVITY_PUB, HTTP_SIGNATURE } from '../../../../initializers/constants' | ||
6 | import { ActorModel } from '../../../../models/actor/actor' | ||
7 | import { MActor } from '../../../../types/models' | ||
8 | |||
9 | type Payload <T> = { body: T, contextType?: ContextType, signatureActorId?: number } | ||
10 | |||
11 | async function computeBody <T> ( | ||
12 | payload: Payload<T> | ||
13 | ): Promise<T | T & { type: 'RsaSignature2017', creator: string, created: string }> { | ||
14 | let body = payload.body | ||
15 | |||
16 | if (payload.signatureActorId) { | ||
17 | const actorSignature = await ActorModel.load(payload.signatureActorId) | ||
18 | if (!actorSignature) throw new Error('Unknown signature actor id.') | ||
19 | |||
20 | body = await buildSignedActivity(actorSignature, payload.body, payload.contextType) | ||
21 | } | ||
22 | |||
23 | return body | ||
24 | } | ||
25 | |||
26 | async function buildSignedRequestOptions (payload: Payload<any>) { | ||
27 | let actor: MActor | null | ||
28 | |||
29 | if (payload.signatureActorId) { | ||
30 | actor = await ActorModel.load(payload.signatureActorId) | ||
31 | if (!actor) throw new Error('Unknown signature actor id.') | ||
32 | } else { | ||
33 | // We need to sign the request, so use the server | ||
34 | actor = await getServerActor() | ||
35 | } | ||
36 | |||
37 | const keyId = actor.url | ||
38 | return { | ||
39 | algorithm: HTTP_SIGNATURE.ALGORITHM, | ||
40 | authorizationHeaderName: HTTP_SIGNATURE.HEADER_NAME, | ||
41 | keyId, | ||
42 | key: actor.privateKey, | ||
43 | headers: HTTP_SIGNATURE.HEADERS_TO_SIGN | ||
44 | } | ||
45 | } | ||
46 | |||
47 | function buildGlobalHeaders (body: any) { | ||
48 | return { | ||
49 | 'digest': buildDigest(body), | ||
50 | 'content-type': 'application/activity+json', | ||
51 | 'accept': ACTIVITY_PUB.ACCEPT_HEADER | ||
52 | } | ||
53 | } | ||
54 | |||
55 | export { | ||
56 | buildGlobalHeaders, | ||
57 | computeBody, | ||
58 | buildSignedRequestOptions | ||
59 | } | ||