refactor: extract WithRoutes config option

1 files changed, 14 insertions, 8 deletions

diff --git a/cmd/gtwy/main.go b/cmd/gtwy/main.go
index 84c9f18..1e9d3a3 100644
--- a/cmd/gtwy/main.go
+++ b/cmd/gtwy/main.go
@@ -10,16 +10,22 @@ import (
 	"gitlab.com/mokhax/spike/pkg/srv"
 )
 
-func main() {
-	mux := http.NewServeMux()
-	mux.Handle("/", prxy.New(map[string]string{
-		"idp.example.com": "localhost:8282",
-		"ui.example.com":  "localhost:8283",
-		"api.example.com": "localhost:8284",
-	}))
+func WithRoutes() cfg.Option {
+	return func(c *cfg.Config) {
+		mux := http.NewServeMux()
+		mux.Handle("/", prxy.New(map[string]string{
+			"idp.example.com": "localhost:8282",
+			"ui.example.com":  "localhost:8283",
+			"api.example.com": "localhost:8284",
+		}))
+
+		cfg.WithMux(mux)(c)
+	}
+}
 
+func main() {
 	log.Fatal(srv.Run(cfg.New(
 		env.Fetch("BIND_ADDR", ":8080"),
-		cfg.WithMux(mux),
+		WithRoutes(),
 	)))
 }