1 import * as express from 'express'
2 import { UserRight } from '../../../../shared/models/users'
3 import { sanitizeHost } from '../../../helpers/core-utils'
4 import { retryTransactionWrapper } from '../../../helpers/database-utils'
5 import { logger } from '../../../helpers/logger'
6 import { getFormattedObjects, getServerActor } from '../../../helpers/utils'
7 import { loadActorUrlOrGetFromWebfinger } from '../../../helpers/webfinger'
8 import { REMOTE_SCHEME, sequelizeTypescript, SERVER_ACTOR_NAME } from '../../../initializers'
9 import { getOrCreateActorAndServerAndModel } from '../../../lib/activitypub/actor'
10 import { sendFollow, sendUndoFollow } from '../../../lib/activitypub/send'
12 asyncMiddleware, authenticate, ensureUserHasRight, paginationValidator, removeFollowingValidator, setBodyHostsPort, setDefaultSort,
14 } from '../../../middlewares'
15 import { followersSortValidator, followingSortValidator, followValidator } from '../../../middlewares/validators'
16 import { ActorModel } from '../../../models/activitypub/actor'
17 import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
19 const serverFollowsRouter = express.Router()
21 serverFollowsRouter.get('/following',
23 followingSortValidator,
26 asyncMiddleware(listFollowing)
29 serverFollowsRouter.post('/following',
31 ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
34 asyncMiddleware(followRetry)
37 serverFollowsRouter.delete('/following/:host',
39 ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW),
40 asyncMiddleware(removeFollowingValidator),
41 asyncMiddleware(removeFollow)
44 serverFollowsRouter.get('/followers',
46 followersSortValidator,
49 asyncMiddleware(listFollowers)
52 // ---------------------------------------------------------------------------
58 // ---------------------------------------------------------------------------
60 async function listFollowing (req: express.Request, res: express.Response, next: express.NextFunction) {
61 const serverActor = await getServerActor()
62 const resultList = await ActorFollowModel.listFollowingForApi(serverActor.id, req.query.start, req.query.count, req.query.sort)
64 return res.json(getFormattedObjects(resultList.data, resultList.total))
67 async function listFollowers (req: express.Request, res: express.Response, next: express.NextFunction) {
68 const serverActor = await getServerActor()
69 const resultList = await ActorFollowModel.listFollowersForApi(serverActor.id, req.query.start, req.query.count, req.query.sort)
71 return res.json(getFormattedObjects(resultList.data, resultList.total))
74 async function followRetry (req: express.Request, res: express.Response, next: express.NextFunction) {
75 const hosts = req.body.hosts as string[]
76 const fromActor = await getServerActor()
78 const tasks: Promise<any>[] = []
79 const actorName = SERVER_ACTOR_NAME
81 for (const host of hosts) {
82 const sanitizedHost = sanitizeHost(host, REMOTE_SCHEME.HTTP)
84 // We process each host in a specific transaction
85 // First, we add the follow request in the database
86 // Then we send the follow request to other actor
87 const p = loadActorUrlOrGetFromWebfinger(actorName, sanitizedHost)
88 .then(actorUrl => getOrCreateActorAndServerAndModel(actorUrl))
89 .then(targetActor => {
91 arguments: [ fromActor, targetActor ],
92 errorMessage: 'Cannot follow with many retries.'
95 return retryTransactionWrapper(follow, options)
97 .catch(err => logger.warn('Cannot follow server %s.', sanitizedHost, err))
102 // Don't make the client wait the tasks
104 .catch(err => logger.error('Error in follow.', err))
106 return res.status(204).end()
109 function follow (fromActor: ActorModel, targetActor: ActorModel) {
110 return sequelizeTypescript.transaction(async t => {
111 const [ actorFollow ] = await ActorFollowModel.findOrCreate({
113 actorId: fromActor.id,
114 targetActorId: targetActor.id
118 actorId: fromActor.id,
119 targetActorId: targetActor.id
123 actorFollow.ActorFollowing = targetActor
124 actorFollow.ActorFollower = fromActor
126 // Send a notification to remote server
127 await sendFollow(actorFollow, t)
131 async function removeFollow (req: express.Request, res: express.Response, next: express.NextFunction) {
132 const follow: ActorFollowModel = res.locals.follow
134 await sequelizeTypescript.transaction(async t => {
135 if (follow.state === 'accepted') await sendUndoFollow(follow, t)
137 await follow.destroy({ transaction: t })
140 // Destroy the actor that will destroy video channels, videos and video files too
141 // This could be long so don't wait this task
142 const following = follow.ActorFollowing
144 .catch(err => logger.error('Cannot destroy actor that we do not follow anymore %s.', following.url, err))
146 return res.status(204).end()