import express from 'express' import { body, param, query, ValidationChain } from 'express-validator' import { HttpStatusCode } from '../../../shared/models/http/http-error-codes' import { PluginType } from '../../../shared/models/plugins/plugin.type' import { InstallOrUpdatePlugin } from '../../../shared/models/plugins/server/api/install-plugin.model' import { exists, isBooleanValid, isSafePath, toBooleanOrNull, toIntOrNull } from '../../helpers/custom-validators/misc' import { isNpmPluginNameValid, isPluginNameValid, isPluginTypeValid, isPluginVersionValid } from '../../helpers/custom-validators/plugins' import { CONFIG } from '../../initializers/config' import { PluginManager } from '../../lib/plugins/plugin-manager' import { PluginModel } from '../../models/server/plugin' import { areValidationErrors } from './shared' const getPluginValidator = (pluginType: PluginType, withVersion = true) => { const validators: (ValidationChain | express.Handler)[] = [ param('pluginName') .custom(isPluginNameValid) ] if (withVersion) { validators.push( param('pluginVersion') .custom(isPluginVersionValid) ) } return validators.concat([ (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return const npmName = PluginModel.buildNpmName(req.params.pluginName, pluginType) const plugin = PluginManager.Instance.getRegisteredPluginOrTheme(npmName) if (!plugin) { return res.fail({ status: HttpStatusCode.NOT_FOUND_404, message: 'No plugin found named ' + npmName }) } if (withVersion && plugin.version !== req.params.pluginVersion) { return res.fail({ status: HttpStatusCode.NOT_FOUND_404, message: 'No plugin found named ' + npmName + ' with version ' + req.params.pluginVersion }) } res.locals.registeredPlugin = plugin return next() } ]) } const getExternalAuthValidator = [ param('authName') .custom(exists), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return const plugin = res.locals.registeredPlugin if (!plugin.registerHelpers) { return res.fail({ status: HttpStatusCode.NOT_FOUND_404, message: 'No registered helpers were found for this plugin' }) } const externalAuth = plugin.registerHelpers.getExternalAuths().find(a => a.authName === req.params.authName) if (!externalAuth) { return res.fail({ status: HttpStatusCode.NOT_FOUND_404, message: 'No external auths were found for this plugin' }) } res.locals.externalAuth = externalAuth return next() } ] const pluginStaticDirectoryValidator = [ param('staticEndpoint') .custom(isSafePath), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return return next() } ] const listPluginsValidator = [ query('pluginType') .optional() .customSanitizer(toIntOrNull) .custom(isPluginTypeValid), query('uninstalled') .optional() .customSanitizer(toBooleanOrNull) .custom(isBooleanValid), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return return next() } ] const installOrUpdatePluginValidator = [ body('npmName') .optional() .custom(isNpmPluginNameValid), body('pluginVersion') .optional() .custom(isPluginVersionValid), body('path') .optional() .custom(isSafePath), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return const body: InstallOrUpdatePlugin = req.body if (!body.path && !body.npmName) { return res.fail({ message: 'Should have either a npmName or a path' }) } if (body.pluginVersion && !body.npmName) { return res.fail({ message: 'Should have a npmName when specifying a pluginVersion' }) } return next() } ] const uninstallPluginValidator = [ body('npmName') .custom(isNpmPluginNameValid), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return return next() } ] const existingPluginValidator = [ param('npmName') .custom(isNpmPluginNameValid), async (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return const plugin = await PluginModel.loadByNpmName(req.params.npmName) if (!plugin) { return res.fail({ status: HttpStatusCode.NOT_FOUND_404, message: 'Plugin not found' }) } res.locals.plugin = plugin return next() } ] const updatePluginSettingsValidator = [ body('settings') .exists(), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return return next() } ] const listAvailablePluginsValidator = [ query('search') .optional() .exists(), query('pluginType') .optional() .customSanitizer(toIntOrNull) .custom(isPluginTypeValid), query('currentPeerTubeEngine') .optional() .custom(isPluginVersionValid), (req: express.Request, res: express.Response, next: express.NextFunction) => { if (areValidationErrors(req, res)) return if (CONFIG.PLUGINS.INDEX.ENABLED === false) { return res.fail({ message: 'Plugin index is not enabled' }) } return next() } ] // --------------------------------------------------------------------------- export { pluginStaticDirectoryValidator, getPluginValidator, updatePluginSettingsValidator, uninstallPluginValidator, listAvailablePluginsValidator, existingPluginValidator, installOrUpdatePluginValidator, listPluginsValidator, getExternalAuthValidator }