Add config path flag

1 files changed, 23 insertions, 4 deletions

diff --git a/statusbar.go b/statusbar.go
index 93e6c22..5db93d4 100644
--- a/statusbar.go
+++ b/statusbar.go
@@ -10,6 +10,7 @@ import (
 	"os/exec"
 	"io/ioutil"
 	"time"
+	"flag"
 
 	"github.com/BurntSushi/toml"
 )
@@ -300,15 +301,33 @@ func buildBattery(config BatteryConfig) string {
 	return b.String()
 }
 
-func main() {
-	var configPath = "/etc/statusbar.conf"
+func getDefaultConfigPath() string {
+	return "/etc/statusbar.conf"
+}
+
+func readConfig(configPath *string) ([]byte, error) {
 	var content []byte
 	var err error
-	if _, err := os.Stat(configPath); err != nil {
+	if configPath == nil {
 		content, err = ioutil.ReadAll(os.Stdin)
 	} else {
-		content, err = ioutil.ReadFile(configPath)
+		if _, err := os.Stat(*configPath); err != nil {
+			fmt.Fprintf(os.Stderr, "Config not found")
+			content, err = ioutil.ReadAll(os.Stdin)
+		} else {
+			content, err = ioutil.ReadFile(*configPath)
+		}
+	}
+	return content, err
+}
+
+func main() {
+	configPath := flag.String("c", getDefaultConfigPath(), "The config path")
+	flag.Parse()
+	if *configPath == "-" {
+		configPath = nil
 	}
+	content, err := readConfig(configPath)
 	if err != nil {
 		panic(err)
 	}