PeerTube/server/controllers/activitypub/client.ts

// Intercept ActivityPub client requests
import * as express from 'express'
import { pageToStartAndCount } from '../../helpers'
import { activityPubCollectionPagination } from '../../helpers/activitypub'

import { database as db } from '../../initializers'
import { ACTIVITY_PUB, CONFIG } from '../../initializers/constants'
import { buildVideoChannelAnnounceToFollowers } from '../../lib/activitypub/send/send-announce'
import { buildVideoAnnounceToFollowers } from '../../lib/index'
import { executeIfActivityPub, localAccountValidator } from '../../middlewares'
import { asyncMiddleware } from '../../middlewares/async'
import { videoChannelsGetValidator, videoChannelsShareValidator } from '../../middlewares/validators/video-channels'
import { videosGetValidator, videosShareValidator } from '../../middlewares/validators/videos'
import { AccountInstance, VideoChannelInstance } from '../../models'
import { VideoChannelShareInstance } from '../../models/video/video-channel-share-interface'
import { VideoInstance } from '../../models/video/video-interface'
import { VideoShareInstance } from '../../models/video/video-share-interface'

const activityPubClientRouter = express.Router()

activityPubClientRouter.get('/account/:name',
  executeIfActivityPub(localAccountValidator),
  executeIfActivityPub(accountController)
)

activityPubClientRouter.get('/account/:name/followers',
  executeIfActivityPub(localAccountValidator),
  executeIfActivityPub(asyncMiddleware(accountFollowersController))
)

activityPubClientRouter.get('/account/:name/following',
  executeIfActivityPub(localAccountValidator),
  executeIfActivityPub(asyncMiddleware(accountFollowingController))
)

activityPubClientRouter.get('/videos/watch/:id',
  executeIfActivityPub(videosGetValidator),
  executeIfActivityPub(videoController)
)

activityPubClientRouter.get('/videos/watch/:id/announces/:accountId',
  executeIfActivityPub(asyncMiddleware(videosShareValidator)),
  executeIfActivityPub(asyncMiddleware(videoAnnounceController))
)

activityPubClientRouter.get('/video-channels/:id',
  executeIfActivityPub(videoChannelsGetValidator),
  executeIfActivityPub(asyncMiddleware(videoChannelController))
)

activityPubClientRouter.get('/video-channels/:id/announces/:accountId',
  executeIfActivityPub(asyncMiddleware(videoChannelsShareValidator)),
  executeIfActivityPub(asyncMiddleware(videoChannelAnnounceController))
)

// ---------------------------------------------------------------------------

export {
  activityPubClientRouter
}

// ---------------------------------------------------------------------------

function accountController (req: express.Request, res: express.Response, next: express.NextFunction) {
  const account: AccountInstance = res.locals.account

  return res.json(account.toActivityPubObject()).end()
}

async function accountFollowersController (req: express.Request, res: express.Response, next: express.NextFunction) {
  const account: AccountInstance = res.locals.account

  const page = req.query.page || 1
  const { start, count } = pageToStartAndCount(page, ACTIVITY_PUB.COLLECTION_ITEMS_PER_PAGE)

  const result = await db.AccountFollow.listAcceptedFollowerUrlsForApi([ account.id ], start, count)
  const activityPubResult = activityPubCollectionPagination(CONFIG.WEBSERVER.URL + req.url, page, result)

  return res.json(activityPubResult)
}

async function accountFollowingController (req: express.Request, res: express.Response, next: express.NextFunction) {
  const account: AccountInstance = res.locals.account

  const page = req.query.page || 1
  const { start, count } = pageToStartAndCount(page, ACTIVITY_PUB.COLLECTION_ITEMS_PER_PAGE)

  const result = await db.AccountFollow.listAcceptedFollowingUrlsForApi([ account.id ], start, count)
  const activityPubResult = activityPubCollectionPagination(CONFIG.WEBSERVER.URL + req.url, page, result)

  return res.json(activityPubResult)
}

function videoController (req: express.Request, res: express.Response, next: express.NextFunction) {
  const video: VideoInstance = res.locals.video

  return res.json(video.toActivityPubObject())
}

async function videoAnnounceController (req: express.Request, res: express.Response, next: express.NextFunction) {
  const share = res.locals.videoShare as VideoShareInstance
  const object = await buildVideoAnnounceToFollowers(share.Account, res.locals.video, undefined)

  return res.json(object)
}

async function videoChannelAnnounceController (req: express.Request, res: express.Response, next: express.NextFunction) {
  const share = res.locals.videoChannelShare as VideoChannelShareInstance
  const object = await buildVideoChannelAnnounceToFollowers(share.Account, share.VideoChannel, undefined)

  return res.json(object)
}

async function videoChannelController (req: express.Request, res: express.Response, next: express.NextFunction) {
  const videoChannel: VideoChannelInstance = res.locals.videoChannel

  return res.json(videoChannel.toActivityPubObject())
}