set scheme order based on X-Forwarded-Proto header

tgunsch · Sep 13, 2023 · 297ec21 · 297ec21
1 parent 7b0e1f5
commit 297ec21
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 6 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -18,7 +18,7 @@ import (
 )
 
 // @title httPod
-// @version 0.0.8
+// @version 1.0.0
 // @description A simple HTTP Request & HTTPResponse Service, shamelessly stolen from httpbin.org.
 // @tag.name HTTP Methods
 // @tag.description Testing different HTTP methods
@@ -90,9 +90,14 @@ func swaggerMiddleware(path string) echo.MiddlewareFunc {
 		return func(c echo.Context) error {
 			if strings.HasSuffix(c.Request().URL.Path, "index.html") {
 
-				_, host := util.GetSchemeHost(c.Request())
+				scheme, host := util.GetSchemeHost(c.Request())
 				docs.SwaggerInfo.Host = host
-				docs.SwaggerInfo.Schemes = []string{"http", "https"}
+				switch scheme {
+				case "https", "HTTPS":
+					docs.SwaggerInfo.Schemes = []string{"https", "http"}
+				default:
+					docs.SwaggerInfo.Schemes = []string{"http", "https"}
+				}
 
 				docs.SwaggerInfo.BasePath = util.GetPath(path, c.Request())
 			}

diff --git a/internal/docs/docs.go b/internal/docs/docs.go
diff --git a/internal/docs/swagger.json b/internal/docs/swagger.json
@@ -4,7 +4,7 @@
         "description": "A simple HTTP Request \u0026 HTTPResponse Service, shamelessly stolen from httpbin.org.",
         "title": "httPod",
         "contact": {},
-        "version": "0.0.8"
+        "version": "1.0.0"
     },
     "paths": {
         "/cookies": {

diff --git a/internal/docs/swagger.yaml b/internal/docs/swagger.yaml
@@ -88,7 +88,7 @@ info:
   description: A simple HTTP Request & HTTPResponse Service, shamelessly stolen from
     httpbin.org.
   title: httPod
-  version: 0.0.8
+  version: 1.0.0
 paths:
   /cookies:
     get: