use a flag for config file

1 files changed, 6 insertions, 3 deletions

diff --git a/main.go b/main.go
index d7ab887..d46e53e 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"sync"
 	"time"
 
@@ -26,9 +27,11 @@ func main() {
 
 	// Config
 	var cfg config.Config
-	//TODO use a flag here
-	//TODO ensure that handler names only contain AZa-z and '-'
-	_, err := toml.DecodeFile("sample-config.toml", &cfg)
+	var cFlag = flag.String("c", "sample-config.toml", "path to config file")
+
+	flag.Parse()
+
+	_, err := toml.DecodeFile(*cFlag, &cfg)
 	if err != nil {
 		log.Fatal(err)
 	}