Merge pull request #10 from kuno1/issue/9

Support log-path
Close #9.

immortalize.go

@@ -73,22 +73,32 @@ func run(minLifetime uint, maxLifetime uint, command string) int {
 	return status
 }
 
-func configLog(level string) error {
+func configLog(level string, logPath string) error {
 	var l logrus.Level
 	switch level {
 	case "trace":
 		l = logrus.TraceLevel
 	case "debug":
 		l = logrus.DebugLevel
 	case "info":
-		l = logrus.DebugLevel
+		l = logrus.InfoLevel
 	default:
 		return errors.New("log level must be one of 'info', 'debug', or 'trace'")
 	}
 
 	log.SetFormatter(&logrus.JSONFormatter{})
 	log.SetLevel(l)
-	log.Out = os.Stderr
+
+	if logPath == "" {
+		log.Out = os.Stderr
+		return nil
+	}
+
+	f, err := os.Create(logPath)
+	if err != nil {
+		return err
+	}
+	log.Out = f
 	return nil
 }
 
@@ -100,9 +110,13 @@ func main() {
 	commandPtr := flag.String("command", "command", "command to immortalize")
 	levelPtr := flag.String(
 		"log-level", "info", "Log level: 'info', 'debug', or 'trace'")
+	logPathPtr := flag.String(
+		"log-path", "", "Log path: default to stderr")
 	flag.Parse()
 
-	configLog(*levelPtr)
+	if err := configLog(*levelPtr, *logPathPtr); err != nil {
+		panic(err)
+	}
 
 	log.Debugf("PID: %v", os.Getpid())
 

test/test.bats

@@ -161,3 +161,39 @@ assert_exit_with_0 () {
   assert_exit_with_nonzero
   [ "$(result)" == 0 ]
 }
+
+@test "Log path" {
+  log=work/immortalize.log
+  rm -f "$log"
+  ./work/immortalize \
+    -max-lifetime 2 \
+    -command test-bin/command-zero \
+    -log-path "$log" >&3 2>&1
+
+  [ 1 -lt "$(wc -l < "$log")" ]
+}
+
+@test "log level - default (info)" {
+  log=work/immortalize.log
+  rm -f "$log"
+  ./work/immortalize \
+    -max-lifetime 2 \
+    -command test-bin/command-zero \
+    -log-path "$log" >&3 2>&1
+
+  [ 1 -lt "$(grep info < "$log" | wc -l)" ]
+  [ 0 -eq "$(grep debug < "$log" | wc -l)" ]
+}
+
+@test "log level - debug" {
+  log=work/immortalize.log
+  rm -f "$log"
+  ./work/immortalize \
+    -max-lifetime 2 \
+    -command test-bin/command-zero \
+    -log-path "$log" \
+    -log-level debug >&3 2>&1
+
+  [ 1 -lt "$(grep info < "$log" | wc -l)" ]
+  [ 0 -lt "$(grep debug < "$log" | wc -l)" ]
+}