From bc855bc4c55b735772b034a42ce943e280130c3a Mon Sep 17 00:00:00 2001
From: Owen <owen@txv.io>
Date: Mon, 28 Jul 2025 22:40:49 -0700
Subject: [PATCH] Fix what happens if there are no sites

---
 main.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/main.go b/main.go
index 6c17388..d6d94a2 100644
--- a/main.go
+++ b/main.go
@@ -822,6 +822,24 @@ func runOlmMainWithArgs(ctx context.Context, args []string) {
 		peerMonitor.HandleFailover(removeData.SiteId, primaryRelay)
 	})
 
+	olm.RegisterHandler("olm/register/no-sites", func(msg websocket.WSMessage) {
+		logger.Info("Received no-sites message - no sites available for connection")
+
+		// if stopRegister != nil {
+		// 	stopRegister()
+		// 	stopRegister = nil
+		// }
+
+		// select {
+		// case <-stopHolepunch:
+		// 	// Channel already closed, do nothing
+		// default:
+		// 	close(stopHolepunch)
+		// }
+
+		logger.Info("No sites available - stopped registration and holepunch processes")
+	})
+
 	olm.RegisterHandler("olm/terminate", func(msg websocket.WSMessage) {
 		logger.Info("Received terminate message")
 		olm.Close()