From 0bec84b137c6d4f7780676dc86b959b03f98ecd6 Mon Sep 17 00:00:00 2001
From: Justin Reichardt <mrjars@gmail.com>
Date: Fri, 19 Aug 2022 19:10:11 -0500
Subject: [PATCH] Added webserver flag for starting one

Later when the testing is done, the webserver will be enabled by default in the
config. This will allow the user to decide which mode they want to use.
---
 src/rhosts.go      | 9 ++++++---
 src/serve/serve.go | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/rhosts.go b/src/rhosts.go
index aadc842..9399c4b 100644
--- a/src/rhosts.go
+++ b/src/rhosts.go
@@ -51,12 +51,14 @@ const GPL = `
 
 func main() {
 	exit := make(chan bool)
+	var webserver bool = false
 	var daemon bool = false
 	var interval int = 1440
 	var versionflag bool = false
 	var removetimestamp bool = false
 
 	// Parsing Flags
+	flag.BoolVar(&webserver, "s", false, "Turn on the Webserver")
 	flag.BoolVar(&daemon, "d", false, "Should this be run in daemon mode")
 	flag.IntVar(&interval, "t", 1440, "Minutes until next run of daemon")
 	flag.BoolVar(&versionflag, "version", false, "show version information")
@@ -83,8 +85,10 @@ func main() {
 	}
 
 	// Starting web server
-	go serve.Start(exit)
-
+	if (webserver == true){
+		go serve.Start(exit)
+		defer func(){<-exit}()
+	}
 	// Update the hosts file
 	if daemon == false {
 		err := hosts.Update()
@@ -112,5 +116,4 @@ func main() {
 			}
 		}
 	}
-	<-exit
 }
diff --git a/src/serve/serve.go b/src/serve/serve.go
index d20c9a7..e74750e 100644
--- a/src/serve/serve.go
+++ b/src/serve/serve.go
@@ -10,7 +10,7 @@ import (
 func Start(exit chan bool) {
 	config := cfg.Create()
 	if config.WebServer.Enabled == false {
-		log.Print("Made it this far")
+		log.Print("Webserver was disabled in the config file")
 		exit <- true
 		return
 	}