add --daemon

1 files changed, 15 insertions, 11 deletions

diff --git a/main.go b/main.go
index 2eae239..c74409b 100644
--- a/main.go
+++ b/main.go
@@ -73,18 +73,22 @@ func main() {
 		okExit("")
 	}
 
-	//	if argv.Daemon == true {
-	http.HandleFunc("/", okHandler)
-	// go https() // use caddy instead
-	p := fmt.Sprintf(":%d", argv.Port)
-	log.Println(argv.Version(), "HOSTNAME set to:", HOSTNAME)
-	log.Println("Running on port", "http://localhost"+p)
-	log.Println("Running on port", "http://localhost"+p+"/ipv6.png")
-	err := http.ListenAndServe(p, nil)
-	if err != nil {
-		log.Println("Error starting server:", err)
+	if argv.Daemon == true {
+		log.Info("Running in --daemon mode")
+		http.HandleFunc("/", okHandler)
+		// go https() // use caddy instead
+		p := fmt.Sprintf(":%d", argv.Port)
+		log.Println(argv.Version(), "HOSTNAME set to:", HOSTNAME)
+		log.Println("Running on port", "http://localhost"+p)
+		log.Println("Running on port", "http://localhost"+p+"/ipv6.png")
+		err := http.ListenAndServe(p, nil)
+		if err != nil {
+			log.Println("Error starting server:", err)
+		}
+	} else {
+		log.Info("--daemon was not set. Just list the patches.")
+		doList()
 	}
-	// }
 }
 
 func formatDuration(d time.Duration) string {