From fe05331edb0f180373721ecb73870f42bad61377 Mon Sep 17 00:00:00 2001
From: joshua bauer <josh@wurrly.com>
Date: Wed, 5 Dec 2018 21:45:15 -0800
Subject: [PATCH] Added redoc support for OpenAPI.

---
 .../proteus/services/OpenAPIService.java      | 51 ++++++++++++++++---
 .../proteus/services/SwaggerService.java      |  2 +-
 .../proteus/server/tools/openapi/redoc.html   | 25 +++++++++
 .../proteus/server/tools/swagger/redoc.html   | 10 ++--
 src/main/resources/reference.conf             |  2 +
 5 files changed, 79 insertions(+), 11 deletions(-)
 create mode 100644 src/main/resources/io/sinistral/proteus/server/tools/openapi/redoc.html

diff --git a/src/main/java/io/sinistral/proteus/services/OpenAPIService.java b/src/main/java/io/sinistral/proteus/services/OpenAPIService.java
index ad223a7..d0720e1 100644
--- a/src/main/java/io/sinistral/proteus/services/OpenAPIService.java
+++ b/src/main/java/io/sinistral/proteus/services/OpenAPIService.java
@@ -83,6 +83,8 @@ public class OpenAPIService extends BaseService implements Supplier<RoutingHandl
 	protected OpenAPI openApi = null;
 	protected String spec = null;
 	protected String indexHTML = null;
+	protected String redocHTML = null;
+	
 	@Inject
 	@Named("openapi.resourcePrefix")
 	protected String resourcePrefix;
@@ -101,9 +103,15 @@ public class OpenAPIService extends BaseService implements Supplier<RoutingHandl
 	@Inject
 	@Named("openapi.port")
 	protected Integer port;
+	 
+	@Inject
+	@Named("openapi.redocPath")
+	protected String redocPath;
+	
 	@Inject
 	@Named("application.path")
 	protected String applicationPath;
+	 
 	@Inject
 	protected RoutingHandler router;
 	@Inject
@@ -141,6 +149,13 @@ public void generateHTML()
 				templateString = templateString.replaceAll("\\{\\{ title \\}\\}", applicationName + " Swagger UI");
 				this.indexHTML = templateString;
 			}
+			
+			try (InputStream templateInputStream = getClass().getClassLoader().getResourceAsStream(resourcePrefix + "/redoc.html"))
+			{
+				byte[] templateBytes = IOUtils.toByteArray(templateInputStream);
+ 
+				this.redocHTML = new String(templateBytes, Charset.defaultCharset());
+			}
 
 			URL url = this.getClass().getClassLoader().getResource(resourcePrefix);
 
@@ -234,9 +249,7 @@ public void generateSpec() throws Exception
 
 		openApi.getComponents().setSecuritySchemes(securitySchemes);
 
-		List<Server> servers = mapper.convertValue(openAPIConfig.getValue("servers").unwrapped(), new TypeReference<List<Server>>()
-		{
-		});
+		List<Server> servers = mapper.convertValue(openAPIConfig.getValue("servers").unwrapped(), new TypeReference<List<Server>>(){});
 
 		openApi.setServers(servers);
 
@@ -317,11 +330,10 @@ public void handleRequest(HttpServerExchange exchange) throws Exception
 				.withMethod(Methods.GET)
 				.withProduces(io.sinistral.proteus.server.MediaType.TEXT_YAML.contentType())
 				.build());
-
-		pathTemplate = this.basePath;
+ 
 
 		router.add(	HttpMethod.GET,
-					pathTemplate,
+		           	basePath,
 					new HttpHandler()
 					{
 						@Override
@@ -339,6 +351,33 @@ public void handleRequest(HttpServerExchange exchange) throws Exception
 				.withControllerName(this.getClass().getSimpleName())
 				.withMethod(Methods.GET)
 				.build());
+		 
+		final String specPath = pathTemplate;
+		
+		router.add(	HttpMethod.GET,
+		           	this.basePath + "/" + this.redocPath,
+					new HttpHandler()
+					{
+						@Override
+						public void handleRequest(HttpServerExchange exchange) throws Exception
+						{
+							exchange.getResponseHeaders().put(Headers.CONTENT_TYPE, MediaType.TEXT_HTML);
+							
+							final String fullPath = String.format("%s://%s%s",exchange.getRequestScheme(), exchange.getHostAndPort(), specPath);
+							
+							final String html = redocHTML.replaceAll("\\{\\{ specPath \\}\\}",  fullPath);
+ 
+							exchange.getResponseSender().send(html);
+						}
+					});
+
+		this.registeredEndpoints.add(EndpointInfo.builder()
+				.withConsumes(MediaType.WILDCARD)
+				.withProduces(MediaType.TEXT_HTML)
+				.withPathTemplate(this.basePath + "/" + this.redocPath)
+				.withControllerName(this.getClass().getSimpleName())
+				.withMethod(Methods.GET)
+				.build());
 
 		try
 		{
diff --git a/src/main/java/io/sinistral/proteus/services/SwaggerService.java b/src/main/java/io/sinistral/proteus/services/SwaggerService.java
index ae53a8d..7e79eee 100644
--- a/src/main/java/io/sinistral/proteus/services/SwaggerService.java
+++ b/src/main/java/io/sinistral/proteus/services/SwaggerService.java
@@ -366,7 +366,7 @@ public void generateSwaggerHTML()
 
 				String templateString = new String(templateBytes, Charset.defaultCharset());
 
-				templateString = templateString.replaceAll("\\{\\{ swaggerSpecPath \\}\\}", this.basePath + ".json");
+				templateString = templateString.replaceAll("\\{\\{ specPath \\}\\}", this.basePath + ".json");
 				templateString = templateString.replaceAll("\\{\\{ applicationName \\}\\}", applicationName);
 
 				this.redocHTML = templateString;
diff --git a/src/main/resources/io/sinistral/proteus/server/tools/openapi/redoc.html b/src/main/resources/io/sinistral/proteus/server/tools/openapi/redoc.html
new file mode 100644
index 0000000..b99afb5
--- /dev/null
+++ b/src/main/resources/io/sinistral/proteus/server/tools/openapi/redoc.html
@@ -0,0 +1,25 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <title>ReDoc</title>
+    <!-- needed for adaptive design -->
+    <meta charset="utf-8"/>
+    <meta name="viewport" content="width=device-width, initial-scale=1">
+    <link href="https://fonts.googleapis.com/css?family=Montserrat:300,400,700|Roboto:300,400,700" rel="stylesheet">
+
+    <!--
+    ReDoc doesn't change outer page styles
+    -->
+    <style>
+      body {
+        margin: 0;
+        padding: 0;
+      }
+    </style>
+  </head> 
+  <body>
+    <redoc spec-url='{{ specPath }}'></redoc>
+    <script src="https://cdn.jsdelivr.net/npm/redoc@next/bundles/redoc.standalone.js"> </script>
+
+  </body>
+</html>
\ No newline at end of file
diff --git a/src/main/resources/io/sinistral/proteus/server/tools/swagger/redoc.html b/src/main/resources/io/sinistral/proteus/server/tools/swagger/redoc.html
index c3eb485..b99afb5 100644
--- a/src/main/resources/io/sinistral/proteus/server/tools/swagger/redoc.html
+++ b/src/main/resources/io/sinistral/proteus/server/tools/swagger/redoc.html
@@ -1,10 +1,11 @@
 <!DOCTYPE html>
 <html>
   <head>
-    <title>{{ applicationName }} ReDoc</title>
+    <title>ReDoc</title>
     <!-- needed for adaptive design -->
     <meta charset="utf-8"/>
     <meta name="viewport" content="width=device-width, initial-scale=1">
+    <link href="https://fonts.googleapis.com/css?family=Montserrat:300,400,700|Roboto:300,400,700" rel="stylesheet">
 
     <!--
     ReDoc doesn't change outer page styles
@@ -15,9 +16,10 @@
         padding: 0;
       }
     </style>
-  </head>
+  </head> 
   <body>
-    <redoc spec-url='{{ swaggerSpecPath }}'></redoc>
-    <script src="https://cdn.jsdelivr.net/npm/redoc/bundles/redoc.standalone.js"> </script>
+    <redoc spec-url='{{ specPath }}'></redoc>
+    <script src="https://cdn.jsdelivr.net/npm/redoc@next/bundles/redoc.standalone.js"> </script>
+
   </body>
 </html>
\ No newline at end of file
diff --git a/src/main/resources/reference.conf b/src/main/resources/reference.conf
index 8f088f1..32c068d 100644
--- a/src/main/resources/reference.conf
+++ b/src/main/resources/reference.conf
@@ -65,6 +65,8 @@ openapi {
 
   basePath= ${application.path}"/openapi"
 
+  redocPath= "redoc"
+
   port = ${application.ports.http}
 
   specFilename="openapi.yaml"