diff --git a/.env.example b/.env.example index 1fc6a48..1a514a9 100644 --- a/.env.example +++ b/.env.example @@ -1,3 +1,5 @@ +# Adds localhost to allowed origins +DEVELOPMENT_MODE=false # Port for the server PORT=8888 diff --git a/internal/server/router.go b/internal/server/router.go index 06dbfb7..3255405 100644 --- a/internal/server/router.go +++ b/internal/server/router.go @@ -14,6 +14,16 @@ import ( "backend/internal/handler" ) +func getAllowedOrigins() []string { + allowedOrigins := []string{"https://axolotlmaid.com"} + + if os.Getenv("DEVELOPMENT_MODE") == "true" { + allowedOrigins = append(allowedOrigins, "http://localhost:4321") + } + + return allowedOrigins +} + func NewRouter() { r := chi.NewRouter() @@ -26,7 +36,7 @@ func NewRouter() { r.Use(middleware.Timeout(60 * time.Second)) r.Use(httprate.LimitByRealIP(32, time.Minute)) r.Use(cors.Handler(cors.Options{ - AllowedOrigins: []string{"https://axolotlmaid.com"}, + AllowedOrigins: getAllowedOrigins(), AllowedMethods: []string{"GET", "PATCH"}, AllowedHeaders: []string{"Content-Type"}, MaxAge: 300,