feat(swagger): use response body as sample for definitions

nazieb · Apr 21, 2018 · 1b5296b · 1b5296b
1 parent 51a6a3b
commit 1b5296b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/formatters/swagger.js b/src/formatters/swagger.js
@@ -4,6 +4,7 @@ import Regex from "re.js"
 import moment from "moment"
 
 let securityDefinitions = {};
+let structureExamples = {};
 
 /**
  *  @param blueprint
@@ -100,6 +101,17 @@ function getActions(actions) {
                     pathResponse.schema["type"] = "string";
                 } else {
                     pathResponse.schema = getResponseSchema(response.content);
+
+                    if (pathResponse.schema.hasOwnProperty("$ref")) {
+                        const structureName = pathResponse.schema["$ref"].replace("#/definitions/", "");
+
+                        try {
+                            const responseBody = response.body.replace("\\n", "");
+                            structureExamples[structureName] = JSON.parse(responseBody);
+                        } catch (e) {
+                            // do nothing
+                        }
+                    }
                 }
 
                 path.responses[response.name] = pathResponse;
@@ -429,6 +441,10 @@ function getDefinitions(dataStructures) {
             definition["properties"] = properties;
         }
 
+        if (structureExamples.hasOwnProperty(definitionName)) {
+            definition["example"] = structureExamples[definitionName];
+        }
+
         result[definitionName] = definition;
     }