import express from 'express' import { body, param, query } from 'express-validator' import { arrayify } from '@peertube/peertube-core-utils' import { FollowState, HttpStatusCode } from '@peertube/peertube-models' import { areValidActorHandles, isValidActorHandle } from '../../helpers/custom-validators/activitypub/actor.js' import { WEBSERVER } from '../../initializers/constants.js' import { ActorFollowModel } from '../../models/actor/actor-follow.js' import { areValidationErrors } from './shared/index.js' const userSubscriptionListValidator = [ query('search') .optional() .not().isEmpty(), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return return next() } ] const userSubscriptionAddValidator = [ body('uri') .custom(isValidActorHandle).withMessage('Should have a valid URI to follow (username@domain)'), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return return next() } ] const areSubscriptionsExistValidator = [ query('uris') .customSanitizer(arrayify) .custom(areValidActorHandles).withMessage('Should have a valid array of URIs'), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return return next() } ] const userSubscriptionGetValidator = [ param('uri') .custom(isValidActorHandle), async (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return if (!await doesSubscriptionExist({ uri: req.params.uri, res, state: 'accepted' })) return return next() } ] const userSubscriptionDeleteValidator = [ param('uri') .custom(isValidActorHandle), async (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return if (!await doesSubscriptionExist({ uri: req.params.uri, res })) return return next() } ] // --------------------------------------------------------------------------- export { areSubscriptionsExistValidator, userSubscriptionListValidator, userSubscriptionAddValidator, userSubscriptionGetValidator, userSubscriptionDeleteValidator } // --------------------------------------------------------------------------- async function doesSubscriptionExist (options: { uri: string res: express.Response state?: FollowState }) { const { uri, res, state } = options let [ name, host ] = uri.split('@') if (host === WEBSERVER.HOST) host = null const user = res.locals.oauth.token.User const subscription = await ActorFollowModel.loadByActorAndTargetNameAndHostForAPI({ actorId: user.Account.Actor.id, targetName: name, targetHost: host, state }) if (!subscription?.ActorFollowing.VideoChannel) { res.fail({ status: HttpStatusCode.NOT_FOUND_404, message: `Subscription ${uri} not found.` }) return false } res.locals.subscription = subscription return true }