From 0aab8b16c7341d33587e05cacac2f52aa308aed1 Mon Sep 17 00:00:00 2001
From: soruh <mail@soruh.de>
Date: Tue, 28 Mar 2023 14:21:47 +0200
Subject: [PATCH] register name changes as updates

---
 src/client.rs | 4 +++-
 src/main.rs   | 5 -----
 src/ports.rs  | 3 +++
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/client.rs b/src/client.rs
index 1dd8e8f..ce405de 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -333,7 +333,9 @@ pub async fn handler(
     {
         if let Some(name) = peer_query(&config.dyn_ip_server, number).await? {
             info!(%name, "found client name");
-            port_handler.lock().await.names.insert(number, name);
+            let mut port_handler = port_handler.lock().await;
+            port_handler.names.insert(number, name);
+            port_handler.register_update();
         }
     }
 
diff --git a/src/main.rs b/src/main.rs
index da0bb99..211741d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -272,11 +272,6 @@ async fn connection_handler(
     if let Some(port) = handler_metadata.port {
         let mut port_handler = port_handler.lock().await;
 
-        #[cfg(feature = "debug_server")]
-        if let Some(number) = handler_metadata.number {
-            port_handler.names.remove(&number);
-        }
-
         if let Some(port_state) = port_handler.port_state.get_mut(&port) {
             port_state.new_state(PortStatus::Disconnected);
             port_handler.register_update();
diff --git a/src/ports.rs b/src/ports.rs
index 671571f..dd0220a 100644
--- a/src/ports.rs
+++ b/src/ports.rs
@@ -533,6 +533,9 @@ impl PortHandler {
             self.register_update();
             info!(port, old_number, "reused port");
             assert!(self.allocated_ports.remove(&old_number).is_some());
+            #[cfg(feature = "debug_server")]
+            self.names.remove(&old_number);
+
             return Some(port);
         }