server/middlewares/validators/user-subscriptions.ts

   1 import express from 'express'
   2 import { body, param, query } from 'express-validator'
   3 import { HttpStatusCode } from '../../../shared/models/http/http-error-codes'
   4 import { areValidActorHandles, isValidActorHandle } from '../../helpers/custom-validators/activitypub/actor'
   5 import { toArray } from '../../helpers/custom-validators/misc'
   6 import { WEBSERVER } from '../../initializers/constants'
   7 import { ActorFollowModel } from '../../models/actor/actor-follow'
   8 import { areValidationErrors } from './shared'
   9
  10 const userSubscriptionListValidator = [
  11   query('search')
  12     .optional()
  13     .not().isEmpty(),
  14
  15   (req: express.Request, res: express.Response, next: express.NextFunction) => {
  16     if (areValidationErrors(req, res)) return
  17
  18     return next()
  19   }
  20 ]
  21
  22 const userSubscriptionAddValidator = [
  23   body('uri')
  24     .custom(isValidActorHandle).withMessage('Should have a valid URI to follow (username@domain)'),
  25
  26   (req: express.Request, res: express.Response, next: express.NextFunction) => {
  27     if (areValidationErrors(req, res)) return
  28
  29     return next()
  30   }
  31 ]
  32
  33 const areSubscriptionsExistValidator = [
  34   query('uris')
  35     .customSanitizer(toArray)
  36     .custom(areValidActorHandles).withMessage('Should have a valid array of URIs'),
  37
  38   (req: express.Request, res: express.Response, next: express.NextFunction) => {
  39     if (areValidationErrors(req, res)) return
  40
  41     return next()
  42   }
  43 ]
  44
  45 const userSubscriptionGetValidator = [
  46   param('uri')
  47     .custom(isValidActorHandle),
  48
  49   async (req: express.Request, res: express.Response, next: express.NextFunction) => {
  50     if (areValidationErrors(req, res)) return
  51
  52     let [ name, host ] = req.params.uri.split('@')
  53     if (host === WEBSERVER.HOST) host = null
  54
  55     const user = res.locals.oauth.token.User
  56     const subscription = await ActorFollowModel.loadByActorAndTargetNameAndHostForAPI({
  57       actorId: user.Account.Actor.id,
  58       targetName: name,
  59       targetHost: host,
  60       state: 'accepted'
  61     })
  62
  63     if (!subscription || !subscription.ActorFollowing.VideoChannel) {
  64       return res.fail({
  65         status: HttpStatusCode.NOT_FOUND_404,
  66         message: `Subscription ${req.params.uri} not found.`
  67       })
  68     }
  69
  70     res.locals.subscription = subscription
  71     return next()
  72   }
  73 ]
  74
  75 // ---------------------------------------------------------------------------
  76
  77 export {
  78   areSubscriptionsExistValidator,
  79   userSubscriptionListValidator,
  80   userSubscriptionAddValidator,
  81   userSubscriptionGetValidator
  82 }