[aspnetcore] Add processing of Port defined in spec (#368)

OpenAPITools · Jul 3, 2018 · bece8d2 · bece8d2
1 parent 83e14a7
commit bece8d2
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/...i-generator/src/main/java/org/openapitools/codegen/languages/AspNetCoreServerCodegen.java b/...i-generator/src/main/java/org/openapitools/codegen/languages/AspNetCoreServerCodegen.java
@@ -21,13 +21,16 @@
 
 import com.samskivert.mustache.Mustache;
 
+import io.swagger.v3.oas.models.OpenAPI;
+
 import org.openapitools.codegen.*;
 import org.openapitools.codegen.utils.ModelUtils;
-
+import org.openapitools.codegen.utils.URLPathUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import java.io.File;
+import java.net.URL;
 import java.util.*;
 
 import static java.util.UUID.randomUUID;
@@ -42,6 +45,9 @@ public class AspNetCoreServerCodegen extends AbstractCSharpCodegen {
     protected Logger LOGGER = LoggerFactory.getLogger(AspNetCoreServerCodegen.class);
 
     private boolean useSwashbuckle = true;
+    protected int serverPort = 8080;
+    protected String serverHost = "0.0.0.0";
+
 
     public AspNetCoreServerCodegen() {
         super();
@@ -112,6 +118,13 @@ public String getName() {
     public String getHelp() {
         return "Generates an ASP.NET Core Web API server.";
     }
+    @Override
+    public void preprocessOpenAPI(OpenAPI openAPI) {
+        super.preprocessOpenAPI(openAPI);
+        URL url = URLPathUtils.getServerURL(openAPI);
+        additionalProperties.put("serverHost", url.getHost());
+        additionalProperties.put("serverPort", URLPathUtils.getPort(url, 8080));
+    }
 
     @Override
     public void processOpts() {

diff --git a/modules/openapi-generator/src/main/resources/aspnetcore/Program.mustache b/modules/openapi-generator/src/main/resources/aspnetcore/Program.mustache
@@ -24,7 +24,8 @@ namespace {{packageName}}
         /// <returns>Webhost</returns>
         public static IWebHost BuildWebHost(string[] args) =>
             WebHost.CreateDefaultBuilder(args)
-                .UseStartup<Startup>()
+                .UseStartup<Startup>()                
+                .UseUrls("http://0.0.0.0:{{#serverPort}}{{serverPort}}{{/serverPort}}{{^serverPort}}8080{{/serverPort}}/")
                 .Build();
     }
 }
diff --git a/samples/server/petstore/aspnetcore/src/Org.OpenAPITools/Program.cs b/samples/server/petstore/aspnetcore/src/Org.OpenAPITools/Program.cs
@@ -24,7 +24,8 @@ public static void Main(string[] args)
         /// <returns>Webhost</returns>
         public static IWebHost BuildWebHost(string[] args) =>
             WebHost.CreateDefaultBuilder(args)
-                .UseStartup<Startup>()
+                .UseStartup<Startup>()                
+                .UseUrls("http://0.0.0.0:8080/")
                 .Build();
     }
 }