From 2e1e0be2d5f240597dd666c065105cf03462b4b8 Mon Sep 17 00:00:00 2001 From: unknown Date: Tue, 9 Feb 2016 13:45:12 +0100 Subject: [PATCH] Add VendorExtensions to root swagger --- .../src/main/java/io/swagger/codegen/CodegenConfig.java | 2 ++ .../src/main/java/io/swagger/codegen/DefaultCodegen.java | 5 +++++ .../main/java/io/swagger/codegen/DefaultGenerator.java | 8 ++++++++ 3 files changed, 15 insertions(+) diff --git a/modules/swagger-codegen/src/main/java/io/swagger/codegen/CodegenConfig.java b/modules/swagger-codegen/src/main/java/io/swagger/codegen/CodegenConfig.java index 720f2f8aa67..9bb5b89089a 100644 --- a/modules/swagger-codegen/src/main/java/io/swagger/codegen/CodegenConfig.java +++ b/modules/swagger-codegen/src/main/java/io/swagger/codegen/CodegenConfig.java @@ -18,6 +18,8 @@ public interface CodegenConfig { String getHelp(); Map additionalProperties(); + + Map vendorExtensions(); String testPackage(); diff --git a/modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultCodegen.java b/modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultCodegen.java index dd1faa82ed4..0ea82226cef 100644 --- a/modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultCodegen.java +++ b/modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultCodegen.java @@ -71,6 +71,7 @@ public class DefaultCodegen { protected String templateDir; protected String embeddedTemplateDir; protected Map additionalProperties = new HashMap(); + protected Map vendorExtensions = new HashMap(); protected List supportingFiles = new ArrayList(); protected List cliOptions = new ArrayList(); protected boolean skipOverwrite; @@ -247,6 +248,10 @@ public class DefaultCodegen { return additionalProperties; } + public Map vendorExtensions() { + return vendorExtensions; + } + public List supportingFiles() { return supportingFiles; } diff --git a/modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultGenerator.java b/modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultGenerator.java index eda6c2a456b..94716051020 100644 --- a/modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultGenerator.java +++ b/modules/swagger-codegen/src/main/java/io/swagger/codegen/DefaultGenerator.java @@ -133,6 +133,10 @@ public class DefaultGenerator extends AbstractGenerator implements Generator { config.additionalProperties().put("termsOfService", info.getTermsOfService()); } } + + if(swagger.getVendorExtensions() != null) { + config.vendorExtensions().putAll(swagger.getVendorExtensions()); + } StringBuilder hostBuilder = new StringBuilder(); String scheme; @@ -273,6 +277,10 @@ public class DefaultGenerator extends AbstractGenerator implements Generator { operation.put("classname", config.toApiName(tag)); operation.put("classVarName", config.toApiVarName(tag)); operation.put("importPath", config.toApiImport(tag)); + + if(!config.vendorExtensions().isEmpty()) { + operation.put("vendorExtensions", config.vendorExtensions()); + } // Pass sortParamsByRequiredFlag through to the Mustache template... boolean sortParamsByRequiredFlag = true;