1 import * as Bull from 'bull'
2 import * as Bluebird from 'bluebird'
3 import { logger } from '../../../helpers/logger'
4 import { processActivities } from '../../activitypub/process'
5 import { addVideoComments } from '../../activitypub/video-comments'
6 import { crawlCollectionPage } from '../../activitypub/crawl'
7 import { VideoModel } from '../../../models/video/video'
8 import { addVideoShares, createRates } from '../../activitypub'
9 import { createAccountPlaylists } from '../../activitypub/playlist'
10 import { AccountModel } from '../../../models/account/account'
11 import { AccountVideoRateModel } from '../../../models/account/account-video-rate'
12 import { VideoShareModel } from '../../../models/video/video-share'
13 import { VideoCommentModel } from '../../../models/video/video-comment'
14 import { MAccountDefault, MVideoFullLight } from '../../../typings/models'
16 type FetchType = 'activity' | 'video-likes' | 'video-dislikes' | 'video-shares' | 'video-comments' | 'account-playlists'
18 export type ActivitypubHttpFetcherPayload = {
25 async function processActivityPubHttpFetcher (job: Bull.Job) {
26 logger.info('Processing ActivityPub fetcher in job %d.', job.id)
28 const payload = job.data as ActivitypubHttpFetcherPayload
30 let video: MVideoFullLight
31 if (payload.videoId) video = await VideoModel.loadAndPopulateAccountAndServerAndTags(payload.videoId)
33 let account: MAccountDefault
34 if (payload.accountId) account = await AccountModel.load(payload.accountId)
36 const fetcherType: { [ id in FetchType ]: (items: any[]) => Promise<any> } = {
37 'activity': items => processActivities(items, { outboxUrl: payload.uri, fromFetch: true }),
38 'video-likes': items => createRates(items, video, 'like'),
39 'video-dislikes': items => createRates(items, video, 'dislike'),
40 'video-shares': items => addVideoShares(items, video),
41 'video-comments': items => addVideoComments(items),
42 'account-playlists': items => createAccountPlaylists(items, account)
45 const cleanerType: { [ id in FetchType ]?: (crawlStartDate: Date) => Bluebird<any> } = {
46 'video-likes': crawlStartDate => AccountVideoRateModel.cleanOldRatesOf(video.id, 'like' as 'like', crawlStartDate),
47 'video-dislikes': crawlStartDate => AccountVideoRateModel.cleanOldRatesOf(video.id, 'dislike' as 'dislike', crawlStartDate),
48 'video-shares': crawlStartDate => VideoShareModel.cleanOldSharesOf(video.id, crawlStartDate),
49 'video-comments': crawlStartDate => VideoCommentModel.cleanOldCommentsOf(video.id, crawlStartDate)
52 return crawlCollectionPage(payload.uri, fetcherType[payload.type], cleanerType[payload.type])
55 // ---------------------------------------------------------------------------
58 processActivityPubHttpFetcher