Add support for modifying data with a callback function

Author: Michael A. Plikk

index.js

@@ -9,6 +9,8 @@ module.exports = {
 		const data = {version: options.version};
 		options.pre = options.pre || [];
 
+		const cb = typeof options.callback === 'function' ? options.callback : (data) => data;
+
 		if (!options.revisionFile) {
 			return done(data);
 		}
@@ -30,7 +32,7 @@ module.exports = {
 				config: {
 					auth: false,
 					handler() {
-						return data;
+						return cb(data);
 					},
 					pre: options.pre,
 					description: 'Show status',

test/index.test.js

@@ -31,4 +31,34 @@ describe('Status Route Plugin', () => {
 		res.statusCode.should.equal(200);
 		res.result.should.have.property('version', '1.0.0');
 	});
+
+	it('should support callback to modify data', async () => {
+		const srv = new Hapi.Server({debug: {request: '*'}});
+		await srv.register({
+			plugin,
+			options: {
+				version: '1.0.0',
+				callback: (data) => {
+					data.count = n;
+					return data;
+				}
+			}
+		});
+		let n = 1;
+		const options = {
+			method: 'GET',
+			url: '/status'
+		};
+
+		let res = await srv.inject(options);
+		res.statusCode.should.equal(200);
+		res.result.should.have.property('version', '1.0.0');
+		res.result.should.have.property('count', 1);
+
+		n = 2;
+		res = await srv.inject(options);
+		res.statusCode.should.equal(200);
+		res.result.should.have.property('version', '1.0.0');
+		res.result.should.have.property('count', 2);
+	});
 });