From 0f1e51f391de1c9fdca7e5fb710693b1fbee4452 Mon Sep 17 00:00:00 2001
From: Owen <owen@txv.io>
Date: Sun, 23 Nov 2025 16:00:29 -0500
Subject: [PATCH] Add callback functions

Former-commit-id: 1aecf6208a38c90e3016053e0e96014870579996
---
 olm/olm.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/olm/olm.go b/olm/olm.go
index 9803516..70ecc7c 100644
--- a/olm/olm.go
+++ b/olm/olm.go
@@ -31,6 +31,10 @@ type GlobalConfig struct {
 	SocketPath string
 	Version    string
 
+	// Callbacks
+	OnRegistered func()
+	OnConnected  func()
+
 	// Source tracking (not in JSON)
 	sources map[string]string
 }
@@ -525,6 +529,11 @@ func StartTunnel(config TunnelConfig) {
 
 		connected = true
 
+		// Invoke onConnected callback if configured
+		if globalConfig.OnConnected != nil {
+			go globalConfig.OnConnected()
+		}
+
 		logger.Info("WireGuard device created.")
 	})
 
@@ -987,6 +996,11 @@ func StartTunnel(config TunnelConfig) {
 				"orgId":      config.OrgID,
 				// "doNotCreateNewClient": config.DoNotCreateNewClient,
 			}, 1*time.Second)
+
+			// Invoke onRegistered callback if configured
+			if globalConfig.OnRegistered != nil {
+				go globalConfig.OnRegistered()
+			}
 		}
 
 		go keepSendingPing(olm)