From 9ee7c22969bb00f3eb90f83eebb8e647058062c5 Mon Sep 17 00:00:00 2001
From: oleg <zelhat@gmail.com>
Date: Sat, 6 Jan 2024 13:14:48 +0300
Subject: [PATCH] add env configuration

---
 cmd/gophermart/main.go | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/cmd/gophermart/main.go b/cmd/gophermart/main.go
index c638f17..0f893fe 100644
--- a/cmd/gophermart/main.go
+++ b/cmd/gophermart/main.go
@@ -1,8 +1,10 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"net/http"
+	"os"
 	"time"
 
 	"github.com/go-chi/chi/v5"
@@ -13,6 +15,23 @@ import (
 
 func main() {
 
+	var (
+		serverAddr  string
+		accrualAddr string
+	)
+
+	flag.StringVar(&serverAddr, "a", "localhost:8080", "адрес и порт запуска сервиса")
+	flag.StringVar(&accrualAddr, "r", "localhost:34567", "адрес системы расчёта начислений")
+
+	flag.Parse()
+
+	if envRunAddr := os.Getenv("RUN_ADDRESS"); envRunAddr != "" {
+		serverAddr = envRunAddr
+	}
+	if envAccrualAddr := os.Getenv("ACCRUAL_SYSTEM_ADDRESS"); envAccrualAddr != "" {
+		accrualAddr = envAccrualAddr
+	}
+
 	router := chi.NewRouter()
 
 	router.Route("/api/user", func(r chi.Router) {
@@ -33,7 +52,7 @@ func main() {
 
 	go func() {
 		client := resty.New()
-		url := "http://localhost:34567/api/orders/"
+		url := accrualAddr + "/api/orders/"
 		ball := struct {
 			Order   string  `json:"order"`
 			Status  string  `json:"status"`
@@ -69,6 +88,6 @@ func main() {
 		}
 	}()
 
-	fmt.Println("start server port 8080")
-	http.ListenAndServe(":8080", router)
+	fmt.Println("start server:", serverAddr)
+	http.ListenAndServe(serverAddr, router)
 }