Merge branch 'release/2.1.0' into develop
[oweals/peertube.git] / 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 { removeRedundanciesOfServer } 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     actorType: req.query.actorType,
105     state: req.query.state
106   })
107
108   return res.json(getFormattedObjects(resultList.data, resultList.total))
109 }
110
111 async function listFollowers (req: express.Request, res: express.Response) {
112   const serverActor = await getServerActor()
113   const resultList = await ActorFollowModel.listFollowersForApi({
114     actorId: serverActor.id,
115     start: req.query.start,
116     count: req.query.count,
117     sort: req.query.sort,
118     search: req.query.search,
119     actorType: req.query.actorType,
120     state: req.query.state
121   })
122
123   return res.json(getFormattedObjects(resultList.data, resultList.total))
124 }
125
126 async function followInstance (req: express.Request, res: express.Response) {
127   const hosts = req.body.hosts as string[]
128   const follower = await getServerActor()
129
130   for (const host of hosts) {
131     const payload = {
132       host,
133       name: SERVER_ACTOR_NAME,
134       followerActorId: follower.id
135     }
136
137     JobQueue.Instance.createJob({ type: 'activitypub-follow', payload })
138   }
139
140   return res.status(204).end()
141 }
142
143 async function removeFollowing (req: express.Request, res: express.Response) {
144   const follow = res.locals.follow
145
146   await sequelizeTypescript.transaction(async t => {
147     if (follow.state === 'accepted') await sendUndoFollow(follow, t)
148
149     // Disable redundancy on unfollowed instances
150     const server = follow.ActorFollowing.Server
151     server.redundancyAllowed = false
152     await server.save({ transaction: t })
153
154     // Async, could be long
155     removeRedundanciesOfServer(server.id)
156       .catch(err => logger.error('Cannot remove redundancy of %s.', server.host, err))
157
158     await follow.destroy({ transaction: t })
159   })
160
161   return res.status(204).end()
162 }
163
164 async function removeOrRejectFollower (req: express.Request, res: express.Response) {
165   const follow = res.locals.follow
166
167   await sendReject(follow.ActorFollower, follow.ActorFollowing)
168
169   await follow.destroy()
170
171   return res.status(204).end()
172 }
173
174 async function acceptFollower (req: express.Request, res: express.Response) {
175   const follow = res.locals.follow
176
177   await sendAccept(follow)
178
179   follow.state = 'accepted'
180   await follow.save()
181
182   await autoFollowBackIfNeeded(follow)
183
184   return res.status(204).end()
185 }