server/controllers/api/server/follows.ts

   1 import * as express from 'express'
   2 import { UserRight } from '../../../../shared/models/users'
   3 import { getFormattedObjects, getServerActor, loadActorUrlOrGetFromWebfinger, logger, retryTransactionWrapper } from '../../../helpers'
   4 import { sequelizeTypescript, SERVER_ACTOR_NAME } from '../../../initializers'
   5 import { getOrCreateActorAndServerAndModel } from '../../../lib/activitypub'
   6 import { sendFollow, sendUndoFollow } from '../../../lib/activitypub/send'
   7 import {
   8   asyncMiddleware,
   9   authenticate,
  10   ensureUserHasRight,
  11   paginationValidator,
  12   removeFollowingValidator,
  13   setBodyHostsPort,
  14   setFollowersSort,
  15   setFollowingSort,
  16   setPagination
  17 } from '../../../middlewares'
  18 import { followersSortValidator, followingSortValidator, followValidator } from '../../../middlewares/validators'
  19 import { ActorModel } from '../../../models/activitypub/actor'
  20 import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
  21
  22 const serverFollowsRouter = express.Router()
  23
  24 serverFollowsRouter.get('/following',
  25   paginationValidator,
  26   followingSortValidator,
  27   setFollowingSort,
  28   setPagination,
  29   asyncMiddleware(listFollowing)
  30 )
  31
  32 serverFollowsRouter.post('/following',
  33   authenticate,
  34   ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
  35   followValidator,
  36   setBodyHostsPort,
  37   asyncMiddleware(followRetry)
  38 )
  39
  40 serverFollowsRouter.delete('/following/:host',
  41   authenticate,
  42   ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
  43   asyncMiddleware(removeFollowingValidator),
  44   asyncMiddleware(removeFollow)
  45 )
  46
  47 serverFollowsRouter.get('/followers',
  48   paginationValidator,
  49   followersSortValidator,
  50   setFollowersSort,
  51   setPagination,
  52   asyncMiddleware(listFollowers)
  53 )
  54
  55 // ---------------------------------------------------------------------------
  56
  57 export {
  58   serverFollowsRouter
  59 }
  60
  61 // ---------------------------------------------------------------------------
  62
  63 async function listFollowing (req: express.Request, res: express.Response, next: express.NextFunction) {
  64   const serverActor = await getServerActor()
  65   const resultList = await ActorFollowModel.listFollowingForApi(serverActor.id, req.query.start, req.query.count, req.query.sort)
  66
  67   return res.json(getFormattedObjects(resultList.data, resultList.total))
  68 }
  69
  70 async function listFollowers (req: express.Request, res: express.Response, next: express.NextFunction) {
  71   const serverActor = await getServerActor()
  72   const resultList = await ActorFollowModel.listFollowersForApi(serverActor.id, req.query.start, req.query.count, req.query.sort)
  73
  74   return res.json(getFormattedObjects(resultList.data, resultList.total))
  75 }
  76
  77 async function followRetry (req: express.Request, res: express.Response, next: express.NextFunction) {
  78   const hosts = req.body.hosts as string[]
  79   const fromActor = await getServerActor()
  80
  81   const tasks: Promise<any>[] = []
  82   const actorName = SERVER_ACTOR_NAME
  83
  84   for (const host of hosts) {
  85     // We process each host in a specific transaction
  86     // First, we add the follow request in the database
  87     // Then we send the follow request to other actor
  88     const p = loadActorUrlOrGetFromWebfinger(actorName, host)
  89       .then(actorUrl => getOrCreateActorAndServerAndModel(actorUrl))
  90       .then(targetActor => {
  91         const options = {
  92           arguments: [ fromActor, targetActor ],
  93           errorMessage: 'Cannot follow with many retries.'
  94         }
  95
  96         return retryTransactionWrapper(follow, options)
  97       })
  98       .catch(err => logger.warn('Cannot follow server %s.', host, err))
  99
 100     tasks.push(p)
 101   }
 102
 103   // Don't make the client wait the tasks
 104   Promise.all(tasks)
 105     .catch(err => logger.error('Error in follow.', err))
 106
 107   return res.status(204).end()
 108 }
 109
 110 function follow (fromActor: ActorModel, targetActor: ActorModel) {
 111   return sequelizeTypescript.transaction(async t => {
 112     const [ actorFollow ] = await ActorFollowModel.findOrCreate({
 113       where: {
 114         actorId: fromActor.id,
 115         targetActorId: targetActor.id
 116       },
 117       defaults: {
 118         state: 'pending',
 119         actorId: fromActor.id,
 120         targetActorId: targetActor.id
 121       },
 122       transaction: t
 123     })
 124     actorFollow.ActorFollowing = targetActor
 125     actorFollow.ActorFollower = fromActor
 126
 127     // Send a notification to remote server
 128     if (actorFollow.state === 'pending') {
 129       await sendFollow(actorFollow, t)
 130     }
 131   })
 132 }
 133
 134 async function removeFollow (req: express.Request, res: express.Response, next: express.NextFunction) {
 135   const follow: ActorFollowModel = res.locals.follow
 136
 137   await sequelizeTypescript.transaction(async t => {
 138     if (follow.state === 'accepted') await sendUndoFollow(follow, t)
 139
 140     await follow.destroy({ transaction: t })
 141   })
 142
 143   // Destroy the actor that will destroy video channels, videos and video files too
 144   // This could be long so don't wait this task
 145   const following = follow.ActorFollowing
 146   following.destroy()
 147     .catch(err => logger.error('Cannot destroy actor that we do not follow anymore %s.', following.url, err))
 148
 149   return res.status(204).end()
 150 }