1 files changed, 75 insertions, 0 deletions

diff --git a/echo/middleware/https_redirect.go b/echo/middleware/https_redirect.go
new file mode 100644
index 0000000..311f1c9
--- /dev/null
+++ b/echo/middleware/https_redirect.go
@@ -0,0 +1,75 @@
+package middleware
+
+/*
+HTTP to HTTPS Redirect Middleware
+
+This is a duplicate of existing functionality in Echo because the Echo default
+middleware doesn't support redirecting to a different HTTPS port which is
+needed in dev environments and some prod environments where the server runs on
+an off-port.
+*/
+
+import (
+        "fmt"
+        "net"
+        "net/http"
+
+        "github.com/labstack/echo/v4"
+        "github.com/labstack/echo/v4/middleware"
+)
+
+type HTTPSRedirectConfig struct {
+        Skipper middleware.Skipper
+        Port    int
+        Code    int
+}
+
+var DefaultHTTPSRedirectConfig = HTTPSRedirectConfig{
+        Skipper: middleware.DefaultSkipper,
+        Port:    443,
+        Code:    http.StatusMovedPermanently,
+}
+
+func HTTPSRedirect() echo.MiddlewareFunc {
+        return HTTPSRedirectWithConfig(DefaultHTTPSRedirectConfig)
+}
+
+func HTTPSRedirectWithConfig(config HTTPSRedirectConfig) echo.MiddlewareFunc {
+        if config.Skipper == nil {
+                config.Skipper = DefaultHTTPSRedirectConfig.Skipper
+        }
+        if config.Code == 0 {
+                config.Code = DefaultHTTPSRedirectConfig.Code
+        }
+        if config.Port == 0 {
+                config.Port = DefaultHTTPSRedirectConfig.Port
+        }
+
+        return func(next echo.HandlerFunc) echo.HandlerFunc {
+                return func(c echo.Context) error {
+                        if config.Skipper(c) || c.Scheme() == "https" {
+                                return next(c)
+                        }
+
+                        var err error
+                        req := c.Request()
+
+                        host := req.URL.Host
+                        if host == "" {
+                                host, _, err = net.SplitHostPort(req.Host)
+                                if err != nil {
+                                        return echo.ErrBadRequest
+                                }
+                        }
+
+                        // Browers assume 443 if it's an https request, otherwise the port
+                        // needs to be specified in the URL
+                        redir := fmt.Sprintf("https://%s%s", host, req.RequestURI)
+                        if config.Port != 443 {
+                                redir = fmt.Sprintf("https://%s:%d%s", host, config.Port, req.RequestURI)
+                        }
+
+                        return c.Redirect(config.Code, redir)
+                }
+        }
+}