From 6fa0df139582972ad328e1c346c2027c22038cef Mon Sep 17 00:00:00 2001
From: Andrea Zagli <azagli@libero.it>
Date: Tue, 12 Apr 2022 10:48:17 +0200
Subject: [PATCH] Config.

---
 app.toml.dist |  9 ++++++
 go.mod        |  2 ++
 main.go       | 80 ++++++++++++++++++++++++++++++++++++++++++---------
 3 files changed, 77 insertions(+), 14 deletions(-)
 create mode 100644 app.toml.dist

diff --git a/app.toml.dist b/app.toml.dist
new file mode 100644
index 0000000..119ac7f
--- /dev/null
+++ b/app.toml.dist
@@ -0,0 +1,9 @@
+[Proxy]
+Port=9081
+Headers=[{Name="First", Value="One"}, {Name="Second", Value="Val"}]
+
+[Service]
+Address="localhost"
+Port=9090
+Path="/some/app/path"
+Executable="executable_file_name"
diff --git a/go.mod b/go.mod
index 01ef2b4..9b58b14 100644
--- a/go.mod
+++ b/go.mod
@@ -1,3 +1,5 @@
 module example.com/devrevproxy
 
 go 1.15
+
+require github.com/spf13/viper v1.11.0
diff --git a/main.go b/main.go
index bbeeac4..dd2132a 100644
--- a/main.go
+++ b/main.go
@@ -12,11 +12,36 @@ import (
 	"os/exec"
 	"time"
 	"strings"
+	"flag"
+
+	"github.com/spf13/viper"
 )
 
-var pid int
+type Header struct {
+	Name string
+	Value string
+}
 
-var path = "/home/tux/Documenti/sviluppo/centriestivi"
+type ConfigProxy struct {
+	Port int
+	Headers []Header
+}
+
+type ConfigService struct {
+	Address string
+	Port int
+	Path string
+	Executable string
+}
+
+type Config struct {
+	Proxy ConfigProxy `mapstructure:"Proxy"`
+	Service ConfigService `mapstructure:"Service"`
+}
+
+var config Config
+
+var pid int
 
 var watched = make(map[string]time.Time)
 
@@ -42,12 +67,15 @@ func NewProxy(targetHost string) (*httputil.ReverseProxy, error) {
 
 func modifyRequest(req *http.Request) {
 	//req.Header.Set("X-Proxy", "Simple-Reverse-Proxy")
-	req.Header.Set("ZType", "be")
+	for _, h := range config.Proxy.Headers {
+		fmt.Println(h.Name, h.Value)
+		req.Header.Set(h.Name, h.Value)
+	}
 	fmt.Println(req.URL)
 
 	/* controllo se i file *.go sono cambiati */
 	rebuild := false
-	dir, _ := os.Open(path)
+	dir, _ := os.Open(config.Service.Path)
 	dirEntries, _ := dir.ReadDir(0)
 	for _, entry := range dirEntries {
 		if !entry.IsDir() && strings.HasSuffix(entry.Name(), ".go") {
@@ -59,6 +87,7 @@ func modifyRequest(req *http.Request) {
 				break
 			} else {
 				if watched[entry.Name()].Before(info.ModTime()) {
+					fmt.Println("File modificato", entry.Name())
 					watched[entry.Name()] = info.ModTime()
 					rebuild = true
 					break
@@ -110,9 +139,9 @@ func startApp() {
 	}
 
 	cmd := &exec.Cmd {
-		Dir: path,
+		Dir: config.Service.Path,
 		Path: goExecPath,
-		Args: []string{goExecPath, "build", "."},
+		Args: []string{goExecPath, "build", "-o", config.Service.Executable, "."},
 		Stdout: os.Stdout,
 		Stderr: os.Stdout,
 	}
@@ -124,9 +153,9 @@ func startApp() {
 
 	fmt.Println("ESEGUO RUN")
 	cmd = &exec.Cmd {
-		Dir: path,
-		Path: path + "/centriestivi",
-		Args: []string{path + "/centriestivi"},
+		Dir: config.Service.Path,
+		Path: config.Service.Path + "/" + config.Service.Executable,
+		Args: []string{config.Service.Path + "/" + config.Service.Executable},
 		Stdout: os.Stdout,
 		Stderr: os.Stdout,
 	}
@@ -143,14 +172,14 @@ func startApp() {
 	/* per adesso testo se la porta tcp è aperta */
 	for {
 		timeout := time.Second
-		conn, err := net.DialTimeout("tcp", net.JoinHostPort("10.101.101.103", "9083"), timeout)
+		conn, err := net.DialTimeout("tcp", net.JoinHostPort(config.Service.Address, fmt.Sprintf("%d", config.Service.Port)), timeout)
 		if err != nil {
 			fmt.Println("Connecting error:", err)
 			time.Sleep(5 * time.Second)
 		} else {
 			if conn != nil {
 				defer conn.Close()
-				fmt.Println("Opened", net.JoinHostPort("10.101.101.103", "9083"))
+				fmt.Println("Opened", net.JoinHostPort(config.Service.Address, fmt.Sprintf("%d", config.Service.Port)))
 				break
 			}
 		}
@@ -158,7 +187,7 @@ func startApp() {
 }
 
 func watchedInit() {
-	dir, _ := os.Open(path)
+	dir, _ := os.Open(config.Service.Path)
 	dirEntries, _ := dir.ReadDir(0)
 	for _, entry := range dirEntries {
 		fmt.Println(entry.Name())
@@ -170,17 +199,40 @@ func watchedInit() {
 }
 
 func main() {
+	var config_file string
+
+	flag.StringVar(&config_file, "c", "./app.toml", "Config file")
+
+	flag.Parse()
+
+	viper.SetConfigFile(config_file)
+
+	viper.AutomaticEnv()
+
+	err := viper.ReadInConfig()
+	if err != nil {
+	 	fmt.Printf("error %v", err)
+		return
+	}
+	err = viper.Unmarshal(&config)
+	if err != nil {
+	 	fmt.Printf("error %v", err)
+		return
+	}
+
 	watchedInit()
 
 	startApp()
 
 	// initialize a reverse proxy and pass the actual backend server url here
-	proxy, err := NewProxy("http://localhost:9083")
+	proxy, err := NewProxy(fmt.Sprintf("http://%s:%d", config.Service.Address, config.Service.Port))
 	if err != nil {
 		panic(err)
 	}
 
 	// handle all requests to your server using the proxy
 	http.HandleFunc("/", ProxyRequestHandler(proxy))
-	log.Fatal(http.ListenAndServe(":9080", nil))
+	log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", config.Proxy.Port), nil))
+
+	fmt.Printf("Listening and serving")
 }
-- 
2.49.0