From 920477bbb2d9fde761fc1c1321ea19bca7d58912 Mon Sep 17 00:00:00 2001
From: 21pages <pages21@163.com>
Date: Fri, 24 Feb 2023 13:13:51 +0800
Subject: [PATCH] delete discovery from RustDesk_lan_peers.toml

Signed-off-by: 21pages <pages21@163.com>
---
 flutter/lib/common/widgets/peer_card.dart | 8 ++++++--
 src/flutter_ffi.rs                        | 4 ++++
 src/ui.rs                                 | 4 +---
 src/ui_interface.rs                       | 7 +++++++
 4 files changed, 18 insertions(+), 5 deletions(-)

diff --git a/flutter/lib/common/widgets/peer_card.dart b/flutter/lib/common/widgets/peer_card.dart
index 8d4d58772..470b631ce 100644
--- a/flutter/lib/common/widgets/peer_card.dart
+++ b/flutter/lib/common/widgets/peer_card.dart
@@ -534,7 +534,7 @@ abstract class BasePeerCard extends StatelessWidget {
       proc: () {
         () async {
           if (isLan) {
-            // TODO
+            bind.mainRemoveDiscovered(id: id);
           } else {
             final favs = (await bind.mainGetFav()).toList();
             if (favs.remove(id)) {
@@ -859,7 +859,11 @@ class DiscoveredPeerCard extends BasePeerCard {
     }
 
     menuItems.add(MenuEntryDivider());
-    menuItems.add(_removeAction(peer.id, () async {}));
+    menuItems.add(
+      _removeAction(peer.id, () async {
+        await bind.mainLoadLanPeers();
+      }, isLan: true),
+    );
     return menuItems;
   }
 
diff --git a/src/flutter_ffi.rs b/src/flutter_ffi.rs
index 8a8bf4de4..23a65c2da 100644
--- a/src/flutter_ffi.rs
+++ b/src/flutter_ffi.rs
@@ -796,6 +796,10 @@ pub fn main_load_lan_peers() {
     };
 }
 
+pub fn main_remove_discovered(id: String) {
+    remove_discovered(id);
+}
+
 fn main_broadcast_message(data: &HashMap<&str, &str>) {
     let apps = vec![
         flutter::APP_TYPE_DESKTOP_REMOTE,
diff --git a/src/ui.rs b/src/ui.rs
index 1b6838e46..a197cb257 100644
--- a/src/ui.rs
+++ b/src/ui.rs
@@ -413,9 +413,7 @@ impl UI {
     }
 
     fn remove_discovered(&mut self, id: String) {
-        let mut peers = config::LanPeers::load().peers;
-        peers.retain(|x| x.id != id);
-        config::LanPeers::store(&peers);
+        remove_discovered(id);
     }
 
     fn send_wol(&mut self, id: String) {
diff --git a/src/ui_interface.rs b/src/ui_interface.rs
index dd111f86e..3b2ba0897 100644
--- a/src/ui_interface.rs
+++ b/src/ui_interface.rs
@@ -596,6 +596,13 @@ pub fn get_lan_peers() -> Vec<HashMap<&'static str, String>> {
         .collect()
 }
 
+#[inline]
+pub fn remove_discovered(id: String) {
+    let mut peers = config::LanPeers::load().peers;
+    peers.retain(|x| x.id != id);
+    config::LanPeers::store(&peers);
+}
+
 #[inline]
 pub fn get_uuid() -> String {
     base64::encode(hbb_common::get_uuid())