From 3027cb7c5139d80f7f225959b6e72e1fdb71f0ae Mon Sep 17 00:00:00 2001
From: Akos Marton <akos.marton@gmail.com>
Date: Tue, 20 Oct 2020 09:49:17 +0300
Subject: [PATCH] Fix exit handler

---
 main.go | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index b8f45e6..40ee415 100644
--- a/main.go
+++ b/main.go
@@ -24,8 +24,13 @@ func exit(err error) {
 		log.Error(err.Error())
 	}
 
-	streams.audio.sendDisconnect()
-	streams.control.sendDisconnect()
+	if streams.audio.common.conn != nil {
+		streams.audio.sendDisconnect()
+	}
+
+	if streams.control.common.conn != nil {
+		streams.control.sendDisconnect()
+	}
 
 	if audioPipes.source.IsOpen() {
 		if err := audioPipes.source.Close(); err != nil {
@@ -85,10 +90,10 @@ func main() {
 		exit(err)
 	}
 
-	setupCloseHandler()
-
 	streams.audio.init()
 	streams.control.init()
 
+	setupCloseHandler()
+
 	streams.control.start()
 }