feat: cleanly replace an existing vhost

.gitignore

@@ -29,4 +29,4 @@ _testmain.go
 /vproxy*
 dist/
 .vproxy.conf
-
+t/

daemon.go

@@ -218,17 +218,9 @@ func (d *Daemon) relayLogsUntilClose(vhost *Vhost, w http.ResponseWriter, reqCtx
 	logChan := vhost.NewLogListener()
 
 	// read existing logs first
-	if vhost.logRing.Len() > 0 {
-		buff := ""
-		for i := 0; i < vhost.logRing.Len(); i++ {
-			s := vhost.logRing.At(i).(string)
-			if s != "" {
-				buff += s + "\n"
-			}
-		}
-		if buff != "" {
-			fmt.Fprint(w, buff)
-		}
+	buff := vhost.BufferAsString()
+	if buff != "" {
+		fmt.Fprint(w, buff)
 	}
 
 	// Listen to connection close and un-register logChan
@@ -254,6 +246,13 @@ func (d *Daemon) addVhost(binding string, w http.ResponseWriter) *Vhost {
 		return nil
 	}
 
+	// remove any existing vhost
+	if v := d.loggedHandler.GetVhost(vhost.Host); v != nil {
+		fmt.Printf("[*] removing existing vhost: %s -> %d\n", v.Host, v.Port)
+		v.Close()
+		d.loggedHandler.RemoveVhost(vhost.Host)
+	}
+
 	fmt.Printf("[*] registering new vhost: %s -> %d\n", vhost.Host, vhost.Port)
 
 	// Set the headers related to event streaming.

vhost.go

@@ -146,8 +146,23 @@ func (v *Vhost) RemoveLogListener(logChan LogListener) {
 	v.listeners = v.listeners[:index]
 }
 
-func (v *Vhost) Close() {
+func (v *Vhost) BufferAsString() string {
+	if v.logRing.Len() == 0 {
+		return ""
+	}
+	buff := ""
+	for i := 0; i < v.logRing.Len(); i++ {
+		s := v.logRing.At(i).(string)
+		if s != "" {
+			buff += s + "\n"
+		}
+	}
+	return buff
+}
 
+func (v *Vhost) Close() {
+	close(v.logChan)
+	v.logRing.Clear()
 }
 
 func (v *Vhost) populateLogBuffer() {