server/controllers/api/server/follows.ts

   1 import * as express from 'express'
   2 import { UserRight } from '../../../../shared/models/users'
   3 import { logger } from '../../../helpers/logger'
   4 import { getFormattedObjects, getServerActor } from '../../../helpers/utils'
   5 import { SERVER_ACTOR_NAME } from '../../../initializers/constants'
   6 import { sendAccept, sendReject, sendUndoFollow } from '../../../lib/activitypub/send'
   7 import {
   8   asyncMiddleware,
   9   authenticate,
  10   ensureUserHasRight,
  11   paginationValidator,
  12   setBodyHostsPort,
  13   setDefaultPagination,
  14   setDefaultSort
  15 } from '../../../middlewares'
  16 import {
  17   acceptOrRejectFollowerValidator,
  18   followersSortValidator,
  19   followingSortValidator,
  20   followValidator,
  21   getFollowerValidator,
  22   removeFollowingValidator
  23 } from '../../../middlewares/validators'
  24 import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
  25 import { JobQueue } from '../../../lib/job-queue'
  26 import { removeRedundancyOf } from '../../../lib/redundancy'
  27 import { sequelizeTypescript } from '../../../initializers/database'
  28
  29 const serverFollowsRouter = express.Router()
  30 serverFollowsRouter.get('/following',
  31   paginationValidator,
  32   followingSortValidator,
  33   setDefaultSort,
  34   setDefaultPagination,
  35   asyncMiddleware(listFollowing)
  36 )
  37
  38 serverFollowsRouter.post('/following',
  39   authenticate,
  40   ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
  41   followValidator,
  42   setBodyHostsPort,
  43   asyncMiddleware(followInstance)
  44 )
  45
  46 serverFollowsRouter.delete('/following/:host',
  47   authenticate,
  48   ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
  49   asyncMiddleware(removeFollowingValidator),
  50   asyncMiddleware(removeFollowing)
  51 )
  52
  53 serverFollowsRouter.get('/followers',
  54   paginationValidator,
  55   followersSortValidator,
  56   setDefaultSort,
  57   setDefaultPagination,
  58   asyncMiddleware(listFollowers)
  59 )
  60
  61 serverFollowsRouter.delete('/followers/:nameWithHost',
  62   authenticate,
  63   ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
  64   asyncMiddleware(getFollowerValidator),
  65   asyncMiddleware(removeOrRejectFollower)
  66 )
  67
  68 serverFollowsRouter.post('/followers/:nameWithHost/reject',
  69   authenticate,
  70   ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
  71   asyncMiddleware(getFollowerValidator),
  72   acceptOrRejectFollowerValidator,
  73   asyncMiddleware(removeOrRejectFollower)
  74 )
  75
  76 serverFollowsRouter.post('/followers/:nameWithHost/accept',
  77   authenticate,
  78   ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
  79   asyncMiddleware(getFollowerValidator),
  80   acceptOrRejectFollowerValidator,
  81   asyncMiddleware(acceptFollower)
  82 )
  83
  84 // ---------------------------------------------------------------------------
  85
  86 export {
  87   serverFollowsRouter
  88 }
  89
  90 // ---------------------------------------------------------------------------
  91
  92 async function listFollowing (req: express.Request, res: express.Response) {
  93   const serverActor = await getServerActor()
  94   const resultList = await ActorFollowModel.listFollowingForApi(
  95     serverActor.id,
  96     req.query.start,
  97     req.query.count,
  98     req.query.sort,
  99     req.query.search
 100   )
 101
 102   return res.json(getFormattedObjects(resultList.data, resultList.total))
 103 }
 104
 105 async function listFollowers (req: express.Request, res: express.Response) {
 106   const serverActor = await getServerActor()
 107   const resultList = await ActorFollowModel.listFollowersForApi(
 108     serverActor.id,
 109     req.query.start,
 110     req.query.count,
 111     req.query.sort,
 112     req.query.search
 113   )
 114
 115   return res.json(getFormattedObjects(resultList.data, resultList.total))
 116 }
 117
 118 async function followInstance (req: express.Request, res: express.Response) {
 119   const hosts = req.body.hosts as string[]
 120   const follower = await getServerActor()
 121
 122   for (const host of hosts) {
 123     const payload = {
 124       host,
 125       name: SERVER_ACTOR_NAME,
 126       followerActorId: follower.id
 127     }
 128
 129     JobQueue.Instance.createJob({ type: 'activitypub-follow', payload })
 130       .catch(err => logger.error('Cannot create follow job for %s.', host, err))
 131   }
 132
 133   return res.status(204).end()
 134 }
 135
 136 async function removeFollowing (req: express.Request, res: express.Response) {
 137   const follow = res.locals.follow
 138
 139   await sequelizeTypescript.transaction(async t => {
 140     if (follow.state === 'accepted') await sendUndoFollow(follow, t)
 141
 142     // Disable redundancy on unfollowed instances
 143     const server = follow.ActorFollowing.Server
 144     server.redundancyAllowed = false
 145     await server.save({ transaction: t })
 146
 147     // Async, could be long
 148     removeRedundancyOf(server.id)
 149       .catch(err => logger.error('Cannot remove redundancy of %s.', server.host, err))
 150
 151     await follow.destroy({ transaction: t })
 152   })
 153
 154   return res.status(204).end()
 155 }
 156
 157 async function removeOrRejectFollower (req: express.Request, res: express.Response) {
 158   const follow = res.locals.follow
 159
 160   await sendReject(follow.ActorFollower, follow.ActorFollowing)
 161
 162   await follow.destroy()
 163
 164   return res.status(204).end()
 165 }
 166
 167 async function acceptFollower (req: express.Request, res: express.Response) {
 168   const follow = res.locals.follow
 169
 170   await sendAccept(follow)
 171
 172   follow.state = 'accepted'
 173   await follow.save()
 174
 175   return res.status(204).end()
 176 }