Keep the original payload at the handler level

lib/index.js

@@ -3,6 +3,7 @@
 const Stream = require('stream');
 
 const Henning = require('henning');
+const Recourier = require('recourier');
 const Subtext = require('subtext');
 
 const internals = {};
@@ -36,14 +37,20 @@ internals.onPostAuth = function (options) {
 
 exports.register = function (server, options, next) {
 
-    const plugin = {
+    const plugins = [{
         register: Henning,
         options: options
-    };
+    }, {
+        register: Recourier,
+        options: {
+            namespace: 'burton'
+        }
+    }];
 
-    server.register(plugin, (err) => {
+    server.register(plugins, (err) => {
 
         server.ext('onPostAuth', internals.onPostAuth(options));
+
         next(err);
     });
 };

package.json

@@ -29,6 +29,7 @@
   "homepage": "https://github.com/ruiquelhas/burton#readme",
   "devDependencies": {
     "code": "2.x.x",
+    "content": "3.x.x",
     "coveralls": "2.x.x",
     "form-data": "0.x.x",
     "hapi": "12.x.x",
@@ -37,6 +38,7 @@
   },
   "dependencies": {
     "henning": "1.x.x",
+    "recourier": "ruiquelhas/recourier",
     "subtext": "4.x.x"
   },
   "engines": {

test/index.js

@@ -5,6 +5,7 @@ const Os = require('os');
 const Path = require('path');
 
 const Code = require('code');
+const Content = require('content');
 const FormData = require('form-data');
 const Hapi = require('hapi');
 const Lab = require('lab');
@@ -32,7 +33,7 @@ lab.experiment('burton', () => {
 
         const main = {
             config: {
-                handler: (request, reply) => reply(),
+                handler: (request, reply) => reply(request.payload),
                 payload: {
                     output: 'stream',
                     parse: false
@@ -109,4 +110,26 @@ lab.experiment('burton', () => {
             });
         });
     });
+
+    lab.test('should return control to the server if all files the in payload are allowed', (done) => {
+
+        const png = Path.join(Os.tmpdir(), 'foo.png');
+        Fs.createWriteStream(png).end(new Buffer([0x89, 0x50]));
+
+        const form = new FormData();
+        form.append('file1', Fs.createReadStream(png));
+        form.append('file2', Fs.createReadStream(png));
+        form.append('foo', 'bar');
+
+        StreamToPromise(form).then((payload) => {
+
+            server.inject({ headers: form.getHeaders(), method: 'POST', payload: payload, url: '/main' }, (response) => {
+
+                Code.expect(response.statusCode).to.equal(200);
+                Code.expect(response.headers['content-validation']).to.equal('success');
+                Code.expect(Content.type(response.headers['content-type']).mime).to.equal('multipart/form-data');
+                done();
+            });
+        });
+    });
 });