mirror/rogueserver
1
0
Fork 0
mirror of https://github.com/pagefaultgames/rogueserver.git synced 2025-04-02 02:57:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 0f86ae9b339df6c5a754bd2693ea9bc0137e580b into 4cac6b6ce8fbdcaf6ef640b8c0d80a8afe2c9a37

Browse Source
This commit is contained in:
Frederico Santos 2024-06-28 01:58:22 +01:00 committed by GitHub
commit 96a1e7fff8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
rogueserver.go
View File

@ -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)