server/lib/activitypub/crawl.ts

   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 }