37 lines
957 B
Go
37 lines
957 B
Go
package cookies
|
|
|
|
import (
|
|
"net/http"
|
|
"net/url"
|
|
)
|
|
|
|
// Check the value of "pagefrom" cookie, delete the cookie, and return the value
|
|
func CheckPageFrom(w http.ResponseWriter, r *http.Request) string {
|
|
pageFromCookie, err := r.Cookie("pagefrom")
|
|
if err != nil {
|
|
return "/"
|
|
}
|
|
pageFrom := pageFromCookie.Value
|
|
DeleteCookie(w, pageFromCookie.Name, pageFromCookie.Path)
|
|
return pageFrom
|
|
}
|
|
|
|
// Check the referer of the request, and if it matches the trustedHost, set
|
|
// the "pagefrom" cookie as the Path of the referer
|
|
func SetPageFrom(w http.ResponseWriter, r *http.Request, trustedHost string) {
|
|
referer := r.Referer()
|
|
parsedURL, err := url.Parse(referer)
|
|
if err != nil {
|
|
return
|
|
}
|
|
var pageFrom string
|
|
if parsedURL.Path == "" || parsedURL.Host != trustedHost {
|
|
pageFrom = "/"
|
|
} else if parsedURL.Path == "/login" || parsedURL.Path == "/register" {
|
|
return
|
|
} else {
|
|
pageFrom = parsedURL.Path
|
|
}
|
|
SetCookie(w, "pagefrom", "/", pageFrom, 0)
|
|
}
|