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