support for header parameter

nazieb · Apr 11, 2017 · c9acbc4 · c9acbc4
1 parent b346625
commit c9acbc4
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/formatters/swagger.js b/src/formatters/swagger.js
@@ -235,6 +235,18 @@ function getActionParams(action) {
 
             params.push(bodyParam);
         }
+
+        for (let header of request.headers) {
+            if (/\{.*}/g.test(header.value)) {
+                const headerParam = {
+                    "name": header.name,
+                    "in": "header",
+                    "type": "string",
+                };
+
+                params.push(headerParam);
+            }
+        }
     }
 
     return params;
@@ -245,7 +257,7 @@ function mergeResourceAndActionParams(resourceParams, actionParams) {
 
     if (actionParams.length > 0) {
         for (let actionParam of actionParams) {
-            if (actionParam.hasOwnProperty("in") && actionParam.in == "body") {
+            if (actionParam.hasOwnProperty("in") && (actionParam.in == "body" || actionParam.in == "header")) {
                 result.push(actionParam);
                 continue;
             }