From a864e689bed5561c5d844e0b18a18101a088e0e2 Mon Sep 17 00:00:00 2001 From: trafficlunar Date: Sun, 1 Dec 2024 20:33:56 +0000 Subject: [PATCH] fix: add development mode environment variable --- .env.example | 2 ++ internal/server/router.go | 12 +++++++++++- 2 files changed, 13 insertions(+), 1 deletion(-) 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,