Add ref support inside fromProperty (OpenAPITools#1235)

ihsmarkitoss · Oct 20, 2018 · aad23e6 · aad23e6
1 parent 343e120
commit aad23e6
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 1 deletion.
diff --git a/...i-generator/src/main/java/org/openapitools/codegen/languages/CppRestSdkClientCodegen.java b/...i-generator/src/main/java/org/openapitools/codegen/languages/CppRestSdkClientCodegen.java
@@ -38,6 +38,7 @@
 import java.util.Collection;
 import java.util.HashMap;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Locale;
 import java.util.Map;
 import java.util.Set;
@@ -154,7 +155,7 @@ public CppRestSdkClientCodegen() {
         typeMapping.put("map", "std::map");
         typeMapping.put("file", "HttpContent");
         typeMapping.put("object", "Object");
-        typeMapping.put("binary", "std::string");
+        typeMapping.put("binary", "utility::string_t");
         typeMapping.put("number", "double");
         typeMapping.put("UUID", "utility::string_t");
         typeMapping.put("ByteArray", "utility::string_t");
@@ -249,6 +250,7 @@ public CodegenOperation fromOperation(String path, String httpMethod, Operation
 
             if (methodResponse != null) {
                 Schema response = ModelUtils.getSchemaFromResponse(methodResponse);
+                response = ModelUtils.unaliasSchema(openAPI.getComponents().getSchemas(), response);
                 if (response != null) {
                     CodegenProperty cm = fromProperty("response", response);
                     op.vendorExtensions.put("x-codegen-response", cm);
@@ -276,6 +278,24 @@ public void postProcessModelProperty(CodegenModel model, CodegenProperty propert
         }
     }
 
+    // override with any special post-processing
+    @SuppressWarnings("unchecked")
+    @Override
+    public Map<String, Object> postProcessOperationsWithModels(Map<String, Object> objs, List<Object> allModels) {
+        Map<String, Object> operations = (Map<String, Object>) objs.get("operations");
+        List<CodegenOperation> operationList = (List<CodegenOperation>) operations.get("operation");
+        for (CodegenOperation op : operationList) {
+            for(String hdr : op.imports) {
+                if(importMapping.containsKey(hdr)) {
+                    continue;
+                }
+                operations.put("hasModelImport", true);
+                break;
+            }
+        }
+        return objs;
+    }
+
     protected boolean isFileSchema(CodegenProperty property) {
         return property.baseType.equals("HttpContent");
     }

diff --git a/modules/openapi-generator/src/main/resources/cpp-rest-sdk-client/api-header.mustache b/modules/openapi-generator/src/main/resources/cpp-rest-sdk-client/api-header.mustache
@@ -13,6 +13,7 @@
 
 {{#imports}}{{{import}}}
 {{/imports}}
+{{^hasModelImport}}#include "../ModelBase.h"{{/hasModelImport}}
 
 #include <boost/optional.hpp>
 

diff --git a/samples/client/petstore/cpp-restsdk/api/PetApi.h b/samples/client/petstore/cpp-restsdk/api/PetApi.h
@@ -26,6 +26,7 @@
 #include "Pet.h"
 #include <cpprest/details/basic_types.h>
 
+
 #include <boost/optional.hpp>
 
 namespace org {

diff --git a/samples/client/petstore/cpp-restsdk/api/StoreApi.h b/samples/client/petstore/cpp-restsdk/api/StoreApi.h
@@ -25,6 +25,7 @@
 #include <map>
 #include <cpprest/details/basic_types.h>
 
+
 #include <boost/optional.hpp>
 
 namespace org {

diff --git a/samples/client/petstore/cpp-restsdk/api/UserApi.h b/samples/client/petstore/cpp-restsdk/api/UserApi.h
@@ -25,6 +25,7 @@
 #include <vector>
 #include <cpprest/details/basic_types.h>
 
+
 #include <boost/optional.hpp>
 
 namespace org {