Merge pull request #12 from Gandem/oxlay/config-as-flag

Load config file using CLI argument
Gandem · Feb 12, 2018 · 620c456 · 620c456
2 parents 4c0cc7b + d07b70c
commit 620c456
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/main.go b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"log"
 	"net"
@@ -10,11 +11,11 @@ import (
 	"github.com/google/gopacket/pcap"
 )
 
-const configPath = "./config.toml"
-
 func main() {
 	// Read config file and generate mDNS forwarding maps
-	cfg, err := readConfig(configPath)
+	configPath := flag.String("config", "", "Config file in TOML format")
+	flag.Parse()
+	cfg, err := readConfig(*configPath)
 	if err != nil {
 		log.Fatalf("Could not read configuration: %v", err)
 	}