serverless-simple-middleware 0.0.58 → 0.0.60

package/dist/middleware/base.d.ts

@@ -22,14 +22,15 @@ export declare class HandlerResponse {
     callback: any;
     completed: boolean;
     result: any | Promise<any> | undefined;
-    private corsHeaders;
     private cookies;
     private crossOrigin?;
+    private customHeaders;
     constructor(callback: any);
     ok(body?: {}, code?: number): any;
     fail(body?: {}, code?: number): any;
     addCookie(key: string, value: string, domain?: string, specifyCrossOrigin?: true, path?: string): void;
     setCrossOrigin: (origin?: string | undefined) => void;
+    addHeader: (header: string, value: string) => void;
 }
 export interface HandlerAuxBase {
     [key: string]: any;

package/dist/middleware/base.js

@@ -58,26 +58,30 @@ var HandlerRequest = /** @class */ (function () {
     return HandlerRequest;
 }());
 exports.HandlerRequest = HandlerRequest;
+var CORS_HEADERS = {
+    'Access-Control-Allow-Origin': '*',
+    'Access-Control-Allow-Headers': 'X-Version',
+    'Access-Control-Allow-Credentials': true,
+};
 var HandlerResponse = /** @class */ (function () {
     function HandlerResponse(callback) {
         var _this = this;
         this.setCrossOrigin = function (origin) {
             _this.crossOrigin = origin;
         };
+        this.addHeader = function (header, value) {
+            _this.customHeaders[header] = value;
+        };
         this.callback = callback;
         this.completed = false;
-        this.corsHeaders = {
-            'Access-Control-Allow-Origin': '*',
-            'Access-Control-Allow-Headers': 'X-Version',
-            'Access-Control-Allow-Credentials': true,
-        };
         this.cookies = [];
+        this.customHeaders = {};
     }
     HandlerResponse.prototype.ok = function (body, code) {
         if (body === void 0) { body = {}; }
         if (code === void 0) { code = 200; }
         logger.stupid("ok", body);
-        var headers = __assign({}, this.corsHeaders);
+        var headers = __assign({}, CORS_HEADERS, this.customHeaders);
         if (this.crossOrigin) {
             headers['Access-Control-Allow-Origin'] = this.crossOrigin;
         }
@@ -100,7 +104,7 @@ var HandlerResponse = /** @class */ (function () {
         logger.stupid("fail", body);
         var result = this.callback(null, {
             statusCode: code,
-            headers: this.corsHeaders,
+            headers: CORS_HEADERS,
             body: JSON.stringify(body),
         });
         this.completed = true;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "serverless-simple-middleware",
   "description": "Simple middleware to translate the interface of lambda's handler to request => response",
-  "version": "0.0.58",
+  "version": "0.0.60",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "author": "VoyagerX",

package/src/middleware/base.ts

@@ -50,30 +50,33 @@ export class HandlerRequest {
   }
 }
 
+const CORS_HEADERS = {
+  'Access-Control-Allow-Origin': '*',
+  'Access-Control-Allow-Headers': 'X-Version',
+  'Access-Control-Allow-Credentials': true,
+};
+
 export class HandlerResponse {
   public callback: any;
   public completed: boolean;
   public result: any | Promise<any> | undefined;
 
-  private corsHeaders: { [header: string]: any };
   private cookies: string[];
   private crossOrigin?: string;
+  private customHeaders: { [header: string]: any };
 
   constructor(callback: any) {
     this.callback = callback;
     this.completed = false;
-    this.corsHeaders = {
-      'Access-Control-Allow-Origin': '*',
-      'Access-Control-Allow-Headers': 'X-Version',
-      'Access-Control-Allow-Credentials': true,
-    };
     this.cookies = [];
+    this.customHeaders = {};
   }
 
   public ok(body = {}, code = 200) {
     logger.stupid(`ok`, body);
     const headers = {
-      ...this.corsHeaders,
+      ...CORS_HEADERS,
+      ...this.customHeaders,
     };
     if (this.crossOrigin) {
       headers['Access-Control-Allow-Origin'] = this.crossOrigin;
@@ -96,7 +99,7 @@ export class HandlerResponse {
     logger.stupid(`fail`, body);
     const result = this.callback(null, {
       statusCode: code,
-      headers: this.corsHeaders,
+      headers: CORS_HEADERS,
       body: JSON.stringify(body),
     });
     this.completed = true;
@@ -124,6 +127,10 @@ export class HandlerResponse {
   public setCrossOrigin = (origin?: string) => {
     this.crossOrigin = origin;
   };
+
+  public addHeader = (header: string, value: string) => {
+    this.customHeaders[header] = value;
+  };
 }
 
 export interface HandlerAuxBase {