]>
Commit | Line | Data |
---|---|---|
1 | import { ACTIVITY_PUB, JOB_REQUEST_TIMEOUT } from '../../initializers' | |
2 | import { doRequest } from '../../helpers/requests' | |
3 | import { logger } from '../../helpers/logger' | |
4 | import * as Bluebird from 'bluebird' | |
5 | import { ActivityPubOrderedCollection } from '../../../shared/models/activitypub' | |
6 | ||
7 | async function crawlCollectionPage <T> (uri: string, handler: (items: T[]) => Promise<any> | Bluebird<any>) { | |
8 | logger.info('Crawling ActivityPub data on %s.', uri) | |
9 | ||
10 | const options = { | |
11 | method: 'GET', | |
12 | uri, | |
13 | json: true, | |
14 | activityPub: true, | |
15 | timeout: JOB_REQUEST_TIMEOUT | |
16 | } | |
17 | ||
18 | const response = await doRequest<ActivityPubOrderedCollection<T>>(options) | |
19 | const firstBody = response.body | |
20 | ||
21 | let limit = ACTIVITY_PUB.FETCH_PAGE_LIMIT | |
22 | let i = 0 | |
23 | let nextLink = firstBody.first | |
24 | while (nextLink && i < limit) { | |
25 | options.uri = nextLink | |
26 | ||
27 | const { body } = await doRequest<ActivityPubOrderedCollection<T>>(options) | |
28 | nextLink = body.next | |
29 | i++ | |
30 | ||
31 | if (Array.isArray(body.orderedItems)) { | |
32 | const items = body.orderedItems | |
33 | logger.info('Processing %i ActivityPub items for %s.', items.length, options.uri) | |
34 | ||
35 | await handler(items) | |
36 | } | |
37 | } | |
38 | } | |
39 | ||
40 | export { | |
41 | crawlCollectionPage | |
42 | } |