Remove duplicated videos on unfollow/delete redundancy
[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 { sequelizeTypescript, SERVER_ACTOR_NAME } from '../../../initializers'
6 import { sendUndoFollow } from '../../../lib/activitypub/send'
7 import {
8   asyncMiddleware,
9   authenticate,
10   ensureUserHasRight,
11   paginationValidator,
12   removeFollowingValidator,
13   setBodyHostsPort,
14   setDefaultPagination,
15   setDefaultSort
16 } from '../../../middlewares'
17 import { followersSortValidator, followingSortValidator, followValidator } from '../../../middlewares/validators'
18 import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
19 import { JobQueue } from '../../../lib/job-queue'
20 import { removeRedundancyOf } from '../../../lib/redundancy'
21
22 const serverFollowsRouter = express.Router()
23 serverFollowsRouter.get('/following',
24   paginationValidator,
25   followingSortValidator,
26   setDefaultSort,
27   setDefaultPagination,
28   asyncMiddleware(listFollowing)
29 )
30
31 serverFollowsRouter.post('/following',
32   authenticate,
33   ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
34   followValidator,
35   setBodyHostsPort,
36   asyncMiddleware(followInstance)
37 )
38
39 serverFollowsRouter.delete('/following/:host',
40   authenticate,
41   ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
42   asyncMiddleware(removeFollowingValidator),
43   asyncMiddleware(removeFollow)
44 )
45
46 serverFollowsRouter.get('/followers',
47   paginationValidator,
48   followersSortValidator,
49   setDefaultSort,
50   setDefaultPagination,
51   asyncMiddleware(listFollowers)
52 )
53
54 // ---------------------------------------------------------------------------
55
56 export {
57   serverFollowsRouter
58 }
59
60 // ---------------------------------------------------------------------------
61
62 async function listFollowing (req: express.Request, res: express.Response, next: express.NextFunction) {
63   const serverActor = await getServerActor()
64   const resultList = await ActorFollowModel.listFollowingForApi(serverActor.id, req.query.start, req.query.count, req.query.sort)
65
66   return res.json(getFormattedObjects(resultList.data, resultList.total))
67 }
68
69 async function listFollowers (req: express.Request, res: express.Response, next: express.NextFunction) {
70   const serverActor = await getServerActor()
71   const resultList = await ActorFollowModel.listFollowersForApi(serverActor.id, req.query.start, req.query.count, req.query.sort)
72
73   return res.json(getFormattedObjects(resultList.data, resultList.total))
74 }
75
76 async function followInstance (req: express.Request, res: express.Response, next: express.NextFunction) {
77   const hosts = req.body.hosts as string[]
78   const follower = await getServerActor()
79
80   for (const host of hosts) {
81     const payload = {
82       host,
83       name: SERVER_ACTOR_NAME,
84       followerActorId: follower.id
85     }
86
87     JobQueue.Instance.createJob({ type: 'activitypub-follow', payload })
88       .catch(err => logger.error('Cannot create follow job for %s.', host, err))
89   }
90
91   return res.status(204).end()
92 }
93
94 async function removeFollow (req: express.Request, res: express.Response, next: express.NextFunction) {
95   const follow: ActorFollowModel = res.locals.follow
96
97   await sequelizeTypescript.transaction(async t => {
98     if (follow.state === 'accepted') await sendUndoFollow(follow, t)
99
100     // Disable redundancy on unfollowed instances
101     const server = follow.ActorFollowing.Server
102     server.redundancyAllowed = false
103     await server.save({ transaction: t })
104
105     // Async, could be long
106     removeRedundancyOf(server.id)
107       .catch(err => logger.error('Cannot remove redundancy of %s.', server.host, err))
108
109     await follow.destroy({ transaction: t })
110   })
111
112   return res.status(204).end()
113 }