feature: Support to change port

commands/run.go

@@ -6,6 +6,6 @@ import (
 )
 
 func Run(config *proxy.Config) {
-	fmt.Println("Ergo Proxy listening on port: 2000")
+	fmt.Println("Ergo Proxy listening on port: " + config.Port)
 	proxy.ServeProxy(config)
 }

main.go

@@ -24,18 +24,14 @@ Usage:
 Options:
   -h      Shows this message.
   -v      Shows ergo's version.
+
+  Run:
+	-p      Set ports to proxy
 `
 
 func main() {
-	var command string = "run"
-
-	if len(os.Args) > 1 {
-		command = os.Args[1]
-	}
-
 	help := flag.Bool("h", false, "Shows ergo's help.")
 	version := flag.Bool("v", false, "Shows ergo's version.")
-
 	flag.Parse()
 
 	if *help {
@@ -48,9 +44,15 @@ func main() {
 		os.Exit(0)
 	}
 
-	config := proxy.LoadConfig("./.ergo")
+	config := proxy.NewConfig()
+
+	command := flag.NewFlagSet(os.Args[1], flag.ExitOnError)
+
+	command.StringVar(&config.Port, "p", "2000", "Set port to the proxy")
+	command.StringVar(&config.Port, "p", "2000", "Set port to the proxy")
 
-	switch command {
+	config.Services = proxy.LoadConfig("./.ergo")
+	switch os.Args[1] {
 	case "list":
 		commands.List(config)
 

proxy/config.go

@@ -36,7 +36,16 @@ func (c *Config) GetService(host string) *Service {
 	return nil
 }
 
-func LoadConfig(filepath string) *Config {
+func NewConfig() *Config {
+	return &Config{
+		Port:       "2000",
+		Domain:     ".dev",
+		UrlPattern: `.*\.dev$`,
+		Services:   nil,
+	}
+}
+
+func LoadConfig(filepath string) []Service {
 	file, e := os.Open(filepath)
 	defer file.Close()
 	if e != nil {
@@ -58,11 +67,5 @@ func LoadConfig(filepath string) *Config {
 		}
 	}
 
-	return &Config{
-		Port:       "2000",
-		Domain:     ".dev",
-		UrlPattern: `.*\.dev$`,
-		Services:   services,
-	}
-
+	return services
 }

proxy/config_test.go

@@ -5,7 +5,8 @@ import (
 )
 
 func TestWhenHasErgoFile(t *testing.T) {
-	config := LoadConfig("../.ergo")
+	config := NewConfig()
+	config.Services = LoadConfig("../.ergo")
 
 	t.Run("It loads the services redirections", func(t *testing.T) {
 		expected := 3

proxy/proxy_test.go

@@ -10,7 +10,8 @@ import (
 )
 
 func TestWhenHasCollectionFile(t *testing.T) {
-	config := LoadConfig("../.ergo")
+	config := NewConfig()
+	config.Services = LoadConfig("../.ergo")
 	proxy := NewErgoProxy(config)
 
 	t.Run("it redirects foo.dev to localhost 3000", func(t *testing.T) {