feat: add validation for pagination params

Chia-Network · Jan 5, 2022 · a4e428d · a4e428d
1 parent c9b9ceb
commit a4e428d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/src/routes/v1/resources/projects.js b/src/routes/v1/resources/projects.js
@@ -8,7 +8,15 @@ import joiExpress from 'express-joi-validation';
 const validator = joiExpress.createValidator({});
 const ProjectRouter = express.Router();
 
-ProjectRouter.get('/', (req, res) => {
+const querySchema = Joi.object()
+  .keys({
+    page: Joi.number(),
+    limit: Joi.number(),
+    search: Joi.string(),
+  })
+  .with('page', 'limit');
+
+ProjectRouter.get('/', validator.query(querySchema), (req, res) => {
   return req.query.warehouseProjectId
     ? ProjectController.findOne(req, res)
     : ProjectController.findAll(req, res);

diff --git a/src/routes/v1/resources/units.js b/src/routes/v1/resources/units.js
@@ -8,7 +8,15 @@ import joiExpress from 'express-joi-validation';
 const validator = joiExpress.createValidator({});
 const UnitRouter = express.Router();
 
-UnitRouter.get('/', (req, res) => {
+const querySchema = Joi.object()
+  .keys({
+    page: Joi.number(),
+    limit: Joi.number(),
+    search: Joi.string(),
+  })
+  .with('page', 'limit');
+
+UnitRouter.get('/', validator.query(querySchema), (req, res) => {
   return req.query.id
     ? UnitController.findOne(req, res)
     : UnitController.findAll(req, res);