Refractor and optimize AP collections
[oweals/peertube.git] / server / controllers / activitypub / outbox.ts
1 import * as express from 'express'
2 import { Activity } from '../../../shared/models/activitypub/activity'
3 import { VideoPrivacy } from '../../../shared/models/videos'
4 import { activityPubCollectionPagination, activityPubContextify } from '../../helpers/activitypub'
5 import { logger } from '../../helpers/logger'
6 import { announceActivityData, createActivityData } from '../../lib/activitypub/send'
7 import { buildAudience } from '../../lib/activitypub/audience'
8 import { asyncMiddleware, localAccountValidator } from '../../middlewares'
9 import { AccountModel } from '../../models/account/account'
10 import { ActorModel } from '../../models/activitypub/actor'
11 import { VideoModel } from '../../models/video/video'
12 import { activityPubResponse } from './utils'
13
14 const outboxRouter = express.Router()
15
16 outboxRouter.get('/accounts/:name/outbox',
17   localAccountValidator,
18   asyncMiddleware(outboxController)
19 )
20
21 // ---------------------------------------------------------------------------
22
23 export {
24   outboxRouter
25 }
26
27 // ---------------------------------------------------------------------------
28
29 async function outboxController (req: express.Request, res: express.Response, next: express.NextFunction) {
30   const account: AccountModel = res.locals.account
31   const actor = account.Actor
32   const actorOutboxUrl = account.Actor.url + '/outbox'
33
34   logger.info('Receiving outbox request for %s.', actorOutboxUrl)
35
36   const handler = (start: number, count: number) => buildActivities(actor, start, count)
37   const json = await activityPubCollectionPagination(actorOutboxUrl, handler, req.query.page)
38
39   return activityPubResponse(activityPubContextify(json), res)
40 }
41
42 async function buildActivities (actor: ActorModel, start: number, count: number) {
43   const data = await VideoModel.listAllAndSharedByActorForOutbox(actor.id, start, count)
44   const activities: Activity[] = []
45
46   // Avoid too many SQL requests
47   const actors = data.data.map(v => v.VideoChannel.Account.Actor)
48   actors.push(actor)
49
50   const followersMatrix = await ActorModel.getActorsFollowerSharedInboxUrls(actors, undefined)
51
52   for (const video of data.data) {
53     const byActor = video.VideoChannel.Account.Actor
54     const createActivityAudience = buildAudience(followersMatrix[byActor.id], video.privacy === VideoPrivacy.PUBLIC)
55
56     // This is a shared video
57     if (video.VideoShares !== undefined && video.VideoShares.length !== 0) {
58       const videoShare = video.VideoShares[0]
59       const announceAudience = buildAudience(followersMatrix[actor.id], video.privacy === VideoPrivacy.PUBLIC)
60       const announceActivity = await announceActivityData(videoShare.url, actor, video.url, undefined, announceAudience)
61
62       activities.push(announceActivity)
63     } else {
64       const videoObject = video.toActivityPubObject()
65       const createActivity = await createActivityData(video.url, byActor, videoObject, undefined, createActivityAudience)
66
67       activities.push(createActivity)
68     }
69   }
70
71   return {
72     data: activities,
73     total: data.total
74   }
75 }