chore: Update prodHandler to use clienturl flag for Access-Control-Allow-Origin

pull/44/head
Frederico Santos 6 months ago
parent 4cac6b6ce8
commit 0f86ae9b33

@ -44,6 +44,8 @@ func main() {
dbaddr := flag.String("dbaddr", "localhost", "database address")
dbname := flag.String("dbname", "pokeroguedb", "database name")
clienturl := flag.String("clienturl", "https://pokerogue.net", "client url")
flag.Parse()
// register gob types
@ -70,7 +72,7 @@ func main() {
}
// start web server
handler := prodHandler(mux)
handler := prodHandler(mux, clienturl)
if *debug {
handler = debugHandler(mux)
}
@ -105,11 +107,11 @@ func createListener(proto, addr string) (net.Listener, error) {
return listener, nil
}
func prodHandler(router *http.ServeMux) http.Handler {
func prodHandler(router *http.ServeMux, clienturl *string) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Access-Control-Allow-Headers", "Authorization, Content-Type")
w.Header().Set("Access-Control-Allow-Methods", "OPTIONS, GET, POST")
w.Header().Set("Access-Control-Allow-Origin", "https://pokerogue.net")
w.Header().Set("Access-Control-Allow-Origin", *clienturl)
if r.Method == "OPTIONS" {
w.WriteHeader(http.StatusOK)

Loading…
Cancel
Save