server/middlewares/validators/user-subscriptions.ts

   1 import * as express from 'express'
   2 import 'express-validator'
   3 import { body, param, query } from 'express-validator/check'
   4 import { logger } from '../../helpers/logger'
   5 import { areValidationErrors } from './utils'
   6 import { ActorFollowModel } from '../../models/activitypub/actor-follow'
   7 import { areValidActorHandles, isValidActorHandle } from '../../helpers/custom-validators/activitypub/actor'
   8 import { UserModel } from '../../models/account/user'
   9 import { CONFIG } from '../../initializers'
  10 import { toArray } from '../../helpers/custom-validators/misc'
  11
  12 const userSubscriptionAddValidator = [
  13   body('uri').custom(isValidActorHandle).withMessage('Should have a valid URI to follow (username@domain)'),
  14
  15   (req: express.Request, res: express.Response, next: express.NextFunction) => {
  16     logger.debug('Checking userSubscriptionAddValidator parameters', { parameters: req.body })
  17
  18     if (areValidationErrors(req, res)) return
  19
  20     return next()
  21   }
  22 ]
  23
  24 const areSubscriptionsExistValidator = [
  25   query('uris')
  26     .customSanitizer(toArray)
  27     .custom(areValidActorHandles).withMessage('Should have a valid uri array'),
  28
  29   (req: express.Request, res: express.Response, next: express.NextFunction) => {
  30     logger.debug('Checking areSubscriptionsExistValidator parameters', { parameters: req.query })
  31
  32     if (areValidationErrors(req, res)) return
  33
  34     return next()
  35   }
  36 ]
  37
  38 const userSubscriptionGetValidator = [
  39   param('uri').custom(isValidActorHandle).withMessage('Should have a valid URI to unfollow'),
  40
  41   async (req: express.Request, res: express.Response, next: express.NextFunction) => {
  42     logger.debug('Checking userSubscriptionGetValidator parameters', { parameters: req.params })
  43
  44     if (areValidationErrors(req, res)) return
  45
  46     let [ name, host ] = req.params.uri.split('@')
  47     if (host === CONFIG.WEBSERVER.HOST) host = null
  48
  49     const user: UserModel = res.locals.oauth.token.User
  50     const subscription = await ActorFollowModel.loadByActorAndTargetNameAndHostForAPI(user.Account.Actor.id, name, host)
  51
  52     if (!subscription || !subscription.ActorFollowing.VideoChannel) {
  53       return res
  54         .status(404)
  55         .json({
  56           error: `Subscription ${req.params.uri} not found.`
  57         })
  58         .end()
  59     }
  60
  61     res.locals.subscription = subscription
  62     return next()
  63   }
  64 ]
  65
  66 // ---------------------------------------------------------------------------
  67
  68 export {
  69   areSubscriptionsExistValidator,
  70   userSubscriptionAddValidator,
  71   userSubscriptionGetValidator
  72 }