Skip to content

Commit

Permalink
chore: Update prodHandler to use clienturl flag for Access-Control-Al…
Browse files Browse the repository at this point in the history
…low-Origin
  • Loading branch information
f-fsantos committed Jun 28, 2024
1 parent 4cac6b6 commit 0f86ae9
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions rogueserver.go
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -70,7 +72,7 @@ func main() {
}

// start web server
handler := prodHandler(mux)
handler := prodHandler(mux, clienturl)
if *debug {
handler = debugHandler(mux)
}
Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit 0f86ae9

Please sign in to comment.