WIP

peer/on_hub.go

@@ -38,6 +38,9 @@ func (a *App) onAddPeer(p HubPeer) {
 
 	if !peer.IsPublic {
 		a.devAddPeer(peer)
+		if a.forwarder != nil && endpoint.IsValid() {
+			a.forwarder.SetEndpoint(peer.VPNIP.As4()[3], endpoint)
+		}
 		return
 	}
 
@@ -50,6 +53,9 @@ func (a *App) onRemovePeer(key wgtypes.Key) {
 		return
 	}
 	a.devRemove(peer)
+	if a.forwarder != nil {
+		a.forwarder.RemoveEndpoint(peer.VPNIP.As4()[3])
+	}
 	delete(a.peersByKey, key)
 	delete(a.peersByIP, peer.VPNIP)