X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fcontrollers%2Fapi%2Fserver%2Ffollows.ts;h=517d1897ecd0ee531e50f654f537adf576e16ba0;hb=de94ac86a211dec657332d964693857ec235ce40;hp=e7d81f7c3e97f48599373726411fb18f298f0611;hpb=50d6de9c286abcb34ff4234d56d9cbb803db7665;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/controllers/api/server/follows.ts b/server/controllers/api/server/follows.ts index e7d81f7c3..517d1897e 100644 --- a/server/controllers/api/server/follows.ts +++ b/server/controllers/api/server/follows.ts @@ -1,31 +1,41 @@ import * as express from 'express' import { UserRight } from '../../../../shared/models/users' -import { getFormattedObjects, getServerActor, loadActorUrlOrGetFromWebfinger, logger, retryTransactionWrapper } from '../../../helpers' -import { sequelizeTypescript, SERVER_ACTOR_NAME } from '../../../initializers' -import { getOrCreateActorAndServerAndModel } from '../../../lib/activitypub' -import { sendFollow, sendUndoFollow } from '../../../lib/activitypub/send' +import { logger } from '../../../helpers/logger' +import { getFormattedObjects } from '../../../helpers/utils' +import { SERVER_ACTOR_NAME } from '../../../initializers/constants' +import { sendAccept, sendReject, sendUndoFollow } from '../../../lib/activitypub/send' import { asyncMiddleware, authenticate, ensureUserHasRight, paginationValidator, - removeFollowingValidator, setBodyHostsPort, - setFollowersSort, - setFollowingSort, - setPagination + setDefaultPagination, + setDefaultSort } from '../../../middlewares' -import { followersSortValidator, followingSortValidator, followValidator } from '../../../middlewares/validators' -import { ActorModel } from '../../../models/activitypub/actor' +import { + acceptOrRejectFollowerValidator, + followersSortValidator, + followingSortValidator, + followValidator, + getFollowerValidator, + removeFollowingValidator, + listFollowsValidator +} from '../../../middlewares/validators' import { ActorFollowModel } from '../../../models/activitypub/actor-follow' +import { JobQueue } from '../../../lib/job-queue' +import { removeRedundanciesOfServer } from '../../../lib/redundancy' +import { sequelizeTypescript } from '../../../initializers/database' +import { autoFollowBackIfNeeded } from '../../../lib/activitypub/follow' +import { getServerActor } from '@server/models/application/application' const serverFollowsRouter = express.Router() - serverFollowsRouter.get('/following', + listFollowsValidator, paginationValidator, followingSortValidator, - setFollowingSort, - setPagination, + setDefaultSort, + setDefaultPagination, asyncMiddleware(listFollowing) ) @@ -34,24 +44,48 @@ serverFollowsRouter.post('/following', ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW), followValidator, setBodyHostsPort, - asyncMiddleware(followRetry) + asyncMiddleware(followInstance) ) serverFollowsRouter.delete('/following/:host', authenticate, ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW), asyncMiddleware(removeFollowingValidator), - asyncMiddleware(removeFollow) + asyncMiddleware(removeFollowing) ) serverFollowsRouter.get('/followers', + listFollowsValidator, paginationValidator, followersSortValidator, - setFollowersSort, - setPagination, + setDefaultSort, + setDefaultPagination, asyncMiddleware(listFollowers) ) +serverFollowsRouter.delete('/followers/:nameWithHost', + authenticate, + ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW), + asyncMiddleware(getFollowerValidator), + asyncMiddleware(removeOrRejectFollower) +) + +serverFollowsRouter.post('/followers/:nameWithHost/reject', + authenticate, + ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW), + asyncMiddleware(getFollowerValidator), + acceptOrRejectFollowerValidator, + asyncMiddleware(removeOrRejectFollower) +) + +serverFollowsRouter.post('/followers/:nameWithHost/accept', + authenticate, + ensureUserHasRight(UserRight.MANAGE_SERVER_FOLLOW), + asyncMiddleware(getFollowerValidator), + acceptOrRejectFollowerValidator, + asyncMiddleware(acceptFollower) +) + // --------------------------------------------------------------------------- export { @@ -60,91 +94,93 @@ export { // --------------------------------------------------------------------------- -async function listFollowing (req: express.Request, res: express.Response, next: express.NextFunction) { +async function listFollowing (req: express.Request, res: express.Response) { const serverActor = await getServerActor() - const resultList = await ActorFollowModel.listFollowingForApi(serverActor.id, req.query.start, req.query.count, req.query.sort) + const resultList = await ActorFollowModel.listFollowingForApi({ + id: serverActor.id, + start: req.query.start, + count: req.query.count, + sort: req.query.sort, + search: req.query.search, + actorType: req.query.actorType, + state: req.query.state + }) return res.json(getFormattedObjects(resultList.data, resultList.total)) } -async function listFollowers (req: express.Request, res: express.Response, next: express.NextFunction) { +async function listFollowers (req: express.Request, res: express.Response) { const serverActor = await getServerActor() - const resultList = await ActorFollowModel.listFollowersForApi(serverActor.id, req.query.start, req.query.count, req.query.sort) + const resultList = await ActorFollowModel.listFollowersForApi({ + actorId: serverActor.id, + start: req.query.start, + count: req.query.count, + sort: req.query.sort, + search: req.query.search, + actorType: req.query.actorType, + state: req.query.state + }) return res.json(getFormattedObjects(resultList.data, resultList.total)) } -async function followRetry (req: express.Request, res: express.Response, next: express.NextFunction) { +async function followInstance (req: express.Request, res: express.Response) { const hosts = req.body.hosts as string[] - const fromActor = await getServerActor() - - const tasks: Promise[] = [] - const actorName = SERVER_ACTOR_NAME + const follower = await getServerActor() for (const host of hosts) { - // We process each host in a specific transaction - // First, we add the follow request in the database - // Then we send the follow request to other actor - const p = loadActorUrlOrGetFromWebfinger(actorName, host) - .then(actorUrl => getOrCreateActorAndServerAndModel(actorUrl)) - .then(targetActor => { - const options = { - arguments: [ fromActor, targetActor ], - errorMessage: 'Cannot follow with many retries.' - } - - return retryTransactionWrapper(follow, options) - }) - .catch(err => logger.warn('Cannot follow server %s.', host, err)) - - tasks.push(p) - } + const payload = { + host, + name: SERVER_ACTOR_NAME, + followerActorId: follower.id + } - // Don't make the client wait the tasks - Promise.all(tasks) - .catch(err => logger.error('Error in follow.', err)) + JobQueue.Instance.createJob({ type: 'activitypub-follow', payload }) + } return res.status(204).end() } -function follow (fromActor: ActorModel, targetActor: ActorModel) { - return sequelizeTypescript.transaction(async t => { - const [ actorFollow ] = await ActorFollowModel.findOrCreate({ - where: { - actorId: fromActor.id, - targetActorId: targetActor.id - }, - defaults: { - state: 'pending', - actorId: fromActor.id, - targetActorId: targetActor.id - }, - transaction: t - }) - actorFollow.ActorFollowing = targetActor - actorFollow.ActorFollower = fromActor - - // Send a notification to remote server - if (actorFollow.state === 'pending') { - await sendFollow(actorFollow, t) - } - }) -} - -async function removeFollow (req: express.Request, res: express.Response, next: express.NextFunction) { - const follow: ActorFollowModel = res.locals.follow +async function removeFollowing (req: express.Request, res: express.Response) { + const follow = res.locals.follow await sequelizeTypescript.transaction(async t => { if (follow.state === 'accepted') await sendUndoFollow(follow, t) + // Disable redundancy on unfollowed instances + const server = follow.ActorFollowing.Server + server.redundancyAllowed = false + await server.save({ transaction: t }) + + // Async, could be long + removeRedundanciesOfServer(server.id) + .catch(err => logger.error('Cannot remove redundancy of %s.', server.host, err)) + await follow.destroy({ transaction: t }) }) - // Destroy the actor that will destroy video channels, videos and video files too - // This could be long so don't wait this task - const following = follow.ActorFollowing - following.destroy() - .catch(err => logger.error('Cannot destroy actor that we do not follow anymore %s.', following.url, err)) + return res.status(204).end() +} + +async function removeOrRejectFollower (req: express.Request, res: express.Response) { + const follow = res.locals.follow + + await sendReject(follow.url, follow.ActorFollower, follow.ActorFollowing) + + await follow.destroy() + + return res.status(204).end() +} + +async function acceptFollower (req: express.Request, res: express.Response) { + const follow = res.locals.follow + + await sendAccept(follow) + + follow.state = 'accepted' + await follow.save() + + await autoFollowBackIfNeeded(follow) return res.status(204).end() }