Search code examples
gogorilla

how to redirect http to https in Gorilla Mux?


I am developing a golang application and I am using Gorilla Mux and i want to redirect HTTP requests to HTTPS

here is what i have so far

package main

import (
    "net/http"

    "github.com/gorilla/mux"
    "github.com/zolamk/deviant/handlers"
    "github.com/zolamk/deviant/lib"
)

func main() {

    router := mux.NewRouter()
    // this is where i am trying to redirect
    router.PathPrefix("/").Schemes("HTTP").HandlerFunc(func(res http.ResponseWriter, req *http.Request) {

        http.Redirect(res, req, fmt.Sprintf("https://%s", req.URL), http.StatusSeeOther)

    })

    router.Handle("/", handlers.ContextHandler(handlers.Index)).Methods("GET")

    router.Handle("/register/", handlers.ContextHandler(handlers.Register)).Methods("GET")

    router.Handle("/register/", handlers.ContextHandler(handlers.RegisterPost)).Methods("POST")

    router.Handle("/login/", handlers.ContextHandler(handlers.Login)).Methods("GET")

    router.Handle("/login/", handlers.ContextHandler(handlers.LoginPost)).Methods("POST")

    router.Handle("/logout/", handlers.ContextHandler(handlers.Logout)).Methods("GET")

    if lib.Settings.ServeStatic {

        router.PathPrefix("/public/").Handler(http.FileServer(http.Dir("./")))

    }

    router.NotFoundHandler = handlers.ContextHandler(handlers.NotFound)

    log.Printf("Deviant running @ http://%s\n", lib.Settings.Address)

    loggedRouter := handlers.LoggedRouter(os.Stdout, router)

    log.Fatal(http.ListenAndServe(lib.Settings.Address, loggedRouter))

}

so like i said before how do i redirect HTTP traffic to HTTPS without affecting my other routes? Thank You.


Solution

  • What I ended up doing was that, I wrote a middleware that redirects HTTP requests to HTTPS

    func RedirectToHTTPSRouter(next http.Handler) http.Handler {
        return http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
            proto := req.Header.Get("x-forwarded-proto")
            if proto == "http" || proto == "HTTP" {
                http.Redirect(res, req, fmt.Sprintf("https://%s%s", req.Host, req.URL), http.StatusPermanentRedirect)
                return
            }
    
            next.ServeHTTP(res, req)
    
        })
    }
    
    func main() {
    
        router := mux.NewRouter()
    
        httpsRouter := RedirectToHTTPSRouter(router)
    
        log.Fatal(http.ListenAndServe(lib.Settings.Address, httpsRouter))
    
    }