feat(http-api): add joi validation to bootstrap

ipfs · Sep 30, 2016 · 028a98c · 028a98c
1 parent 8509199
commit 028a98c
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/src/http-api/routes/bootstrap.js b/src/http-api/routes/bootstrap.js
@@ -1,5 +1,6 @@
 'use strict'
 
+const Joi = require('joi')
 const resources = require('./../resources')
 
 module.exports = (server) => {
@@ -20,7 +21,14 @@ module.exports = (server) => {
       pre: [
         { method: resources.bootstrap.add.parseArgs, assign: 'args' }
       ],
-      handler: resources.bootstrap.add.handler
+      handler: resources.bootstrap.add.handler,
+      validate: {
+        query: {
+          arg: Joi.string().required(),
+          default: Joi.boolean(),
+          'stream-channels': Joi.boolean()
+        }
+      }
     }
   })
 
@@ -39,7 +47,14 @@ module.exports = (server) => {
       pre: [
         { method: resources.bootstrap.rm.parseArgs, assign: 'args' }
       ],
-      handler: resources.bootstrap.rm.handler
+      handler: resources.bootstrap.rm.handler,
+      validate: {
+        query: {
+          arg: Joi.string().required(),
+          default: Joi.boolean(),
+          'stream-channels': Joi.boolean()
+        }
+      }
     }
   })
 }