Add config for default value

- Create a new method to set the default value for `headers`, `body` and
`statusCodes`.
- Also it should not break the old implementation.
- Move to ES6 codestyle.

Fix #2

Author: themyth92

LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2016 
+Copyright (c) 2017
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

README.md

@@ -8,13 +8,13 @@ A wrap around response for aws lambda integration proxy
 
 ## Response creation
 1. Response can be created using `response` method
-  ```
-  var lambdaProxy = require('lambda-proxy-reponse');
+  ```javascript
+  const lambdaProxy = require('lambda-proxy-reponse');
 
   exports.handler = (event, context, callback) => {
     // response method needs at least 3 parameters
     // statusCode, headers, body
-    var response = lambdaProxy.response(200, { 'X-header': 'Your headers value' }, { 'bodyData': 'yourBodyData' });
+    const response = lambdaProxy.response(200, { 'X-header': 'Your headers value' }, { 'bodyData': 'yourBodyData' });
 
     // -> response = {
     //     statusCodes: 200,
@@ -39,6 +39,82 @@ A wrap around response for aws lambda integration proxy
   - `lambdaProxy.notFound = lambdaProxy.response(404, ...)`
   - `lambdaProxy.serverError = lambdaProxy.response(500, ...)`
 
+## Options
+From version 2.0, new config option has been introduced to set default values for response.
+
+### body
+Default body for every response. Will be overriden if you support new body in the call
+```javascript
+// ....
+lambdaProxy.config({ body: { bodyData: 'yourDefaultBodyData' } });
+
+// ......
+  let response = lambdaProxy.response(200, { 'X-header': 'Your headers value' });
+
+  // -> response = {
+  //     statusCodes: 200,
+  //     headers: {'X-header': 'Your headers value'},
+  //     body: '{ "bodyData": "yourDefaultBodyData" }' 
+  //   }
+
+  response = lambdaProxy.response(200, { 'X-header': 'Your headers value' }, { 'bodyData': 'yourBodyData' });
+  // -> response = {
+  //     statusCodes: 200,
+  //     headers: {'X-header': 'Your headers value'},
+  //     body: '{ "bodyData": "yourBodyData" }' 
+  //   }
+```
+
+### headers
+Default headers for every response. By default, it will extend whatever headers you provide inside
+your response. Can be turned off by setting `extendHeader` in options.
+```javascript
+// ....
+lambdaProxy.config({ header: { 'X-Default-Header': 'Your default header' } });
+
+// ......
+  let response = lambdaProxy.response(200, null, { bodyData: 'yourBodyData' });
+
+  // -> response = {
+  //     statusCodes: 200,
+  //     headers: { 'X-Default-Header': 'Your default header' },
+  //     body: '{ "bodyData": "yourBodyData" }' 
+  //   }
+
+  // by default it will extend your header
+  response = lambdaProxy.response(200, { 'X-header': 'Your headers value' }, { 'bodyData': 'yourBodyData' });
+  // -> response = {
+  //     statusCodes: 200,
+  //     headers: { 'X-header': 'Your headers value', 'X-Default-Header': 'Your default header' },
+  //     body: '{ "bodyData": "yourBodyData" }' 
+  //   }
+
+  // you can turn it off by setting config
+  lambdaProxy.config({ header: { 'X-Default-Header': 'Your default header' }, extendHeader: false });
+
+  // .......
+  response = lambdaProxy.response(200, { 'X-header': 'Your headers value' }, { 'bodyData': 'yourBodyData' });
+  // -> response = {
+  //     statusCodes: 200,
+  //     headers: { 'X-header': 'Your headers value' },
+  //     body: '{ "bodyData": "yourBodyData" }' 
+  //   }
+```
+### status
+Default status code.
+```javascript
+// ....
+lambdaProxy.config({ status: 200 });
+
+// ....
+  response = lambdaProxy.response();
+  // -> response = {
+  //     statusCodes: 200,
+  //     headers: {},
+  //     body: '{}' 
+  //   }
+```
+
 ## Note
 This response template is only for used with newly aws lambda proxy integration.
 

index.js

@@ -1,57 +1,93 @@
 'use strict';
-exports.response = function response(status, headers, body, cb) {
-  // hearders must be an object, if not APIGateway will throw error
-  if (typeof headers !== 'object') {
-    headers = {};
+class LambdaProxyResponse {
+  constructor(options) {
+    this.options = options || { headers: {}, body: {}, status: null, extendHeader: true };
   }
 
-  if (typeof body === 'undefined' || typeof body === 'null') {
-    body = {};
+  config(options) {
+    if (typeof options === 'object' && options !== null) {
+      Object.assign(this.options, options);
+    }
   }
 
-  status = status || 400;
-  var responseTemplate = createResponseTemplate(status, headers, body);
+  response(status, headers, body, cb, options) {
+    let responseStatus = {};
+    let responseHeader = {};
+    let responseBody = {};
 
-  // call cb
-  if (cb) {
-    cb(null, responseTemplate);
+    responseStatus = status || this.options.status || 400;
+
+    // set headers
+    if (typeof headers !== 'object' || headers === null ) {
+      if (this.options && this.options.headers) {
+        responseHeader = this.options.headers;
+      } else {
+        responseHeader = {};
+      }
+    } else {
+      if (this.options.headers && this.options.extendHeader) {
+        responseHeader = Object.assign(responseHeader, headers, this.options.headers);
+      } else {
+        responseHeader = headers;
+      }
+    }
+
+    // set body
+    if (typeof body === 'undefined' || body === null) {
+      if (this.options && this.options.body) {
+        responseBody = this.options.body;
+      } else {
+        responseBody = {};
+      }
+    } else {
+      responseBody = body;
+    }
+
+    const responseTemplate = this._createResponseTemplate(responseStatus, responseHeader, responseBody);
+
+    // call cb
+    if (typeof cb === 'function') {
+      cb(null, responseTemplate);
+    }
+
+    return responseTemplate;
   }
 
-  return responseTemplate;
-};
+  ok(headers, body, cb, options) {
+    return this.response(200, headers, body, cb, options || {});
+  }
 
-exports.ok = function ok(headers, body, cb) {
-  return exports.response(200, headers, body, cb);
-}
+  created(headers, body, cb, options) {
+    return this.response(201, headers, body, cb, options || {});
+  }
 
-exports.created = function created(headers, body, cb) {
-  return exports.response(201, headers, body, cb);
-}
+  badRequest(headers, body, cb, options) {
+    return this.response(400, headers, body, cb, options || {});
+  }
 
-exports.badRequest = function badRequest(headers, body, cb) {
-  return exports.response(400, headers, body, cb);
-}
+  notAuthorized(headers, body, cb, options) {
+    return this.response(401, headers, body, cb, options || {});
+  }
 
-exports.notAuthorized = function notAuthorized(headers, body, cb) {
-  return exports.response(401, headers, body, cb);
-}
+  forbidden(headers, body, cb, options) {
+    return this.response(403, headers, body, cb, options || {});
+  }
 
-exports.forbidden = function forbidden(headers, body, cb) {
-  return exports.response(403, headers, body, cb);
-}
+  notFound(headers, body, cb, options) {
+    return this.response(404, headers, body, cb, options || {});
+  }
 
-exports.notFound = function notFound(headers, body, cb) {
-  return exports.response(404, headers, body, cb);
-}
+  serverError(headers, body, cb, options) {
+    return this.response(500, headers, body, cb, options || {});
+  }
 
-exports.serverError = function serverError(headers, body, cb) {
-  return exports.response(500, headers, body, cb);
+  _createResponseTemplate(status, headers, body) {
+    return {
+      statusCode: status,
+      body: JSON.stringify(body),
+      headers: headers
+    }; 
+  }
 }
 
-function createResponseTemplate(status, headers, body) {
-  return {
-    statusCode: status,
-    body: JSON.stringify(body),
-    headers: headers
-  };
-}
+module.exports = new LambdaProxyResponse();