implement dns indicator (#119)

src/server/frontend_task.rs

@@ -151,6 +151,16 @@ async fn handle_frontend_event(
             update_pos(server, handle, capture, emulate, pos).await;
             broadcast_client_update(server, frontend, handle).await;
         }
+        FrontendRequest::ResolveDns(handle) => {
+            let hostname = server
+                .client_manager
+                .borrow()
+                .get(handle)
+                .and_then(|(c, _)| c.hostname.clone());
+            if let Some(hostname) = hostname {
+                let _ = resolve_tx.send(DnsRequest { hostname, handle }).await;
+            }
+        }
     };
     false
 }

src/server/resolver_task.rs

@@ -2,7 +2,7 @@ use std::collections::HashSet;
 
 use tokio::{sync::mpsc::Sender, task::JoinHandle};
 
-use crate::{client::ClientHandle, dns::DnsResolver};
+use crate::{client::ClientHandle, dns::DnsResolver, frontend::FrontendEvent};
 
 use super::Server;
 
@@ -12,7 +12,11 @@ pub struct DnsRequest {
     pub handle: ClientHandle,
 }
 
-pub fn new(resolver: DnsResolver, server: Server) -> (JoinHandle<()>, Sender<DnsRequest>) {
+pub fn new(
+    resolver: DnsResolver,
+    mut server: Server,
+    mut frontend: Sender<FrontendEvent>,
+) -> (JoinHandle<()>, Sender<DnsRequest>) {
     let (dns_tx, mut dns_rx) = tokio::sync::mpsc::channel::<DnsRequest>(32);
     let resolver_task = tokio::task::spawn_local(async move {
         loop {
@@ -20,6 +24,13 @@ pub fn new(resolver: DnsResolver, server: Server) -> (JoinHandle<()>, Sender<Dns
                 Some(r) => (r.hostname, r.handle),
                 None => break,
             };
+
+            /* update resolving status */
+            if let Some((_, s)) = server.client_manager.borrow_mut().get_mut(handle) {
+                s.resolving = true;
+            }
+            notify_state_change(&mut frontend, &mut server, handle).await;
+
             let ips = match resolver.resolve(&host).await {
                 Ok(ips) => ips,
                 Err(e) => {
@@ -27,14 +38,35 @@ pub fn new(resolver: DnsResolver, server: Server) -> (JoinHandle<()>, Sender<Dns
                     continue;
                 }
             };
+
+            /* update ips and resolving state */
             if let Some((c, s)) = server.client_manager.borrow_mut().get_mut(handle) {
                 let mut addrs = HashSet::from_iter(c.fix_ips.iter().cloned());
                 for ip in ips {
                     addrs.insert(ip);
                 }
                 s.ips = addrs;
+                s.resolving = false;
             }
+            notify_state_change(&mut frontend, &mut server, handle).await;
         }
     });
     (resolver_task, dns_tx)
 }
+
+async fn notify_state_change(
+    frontend: &mut Sender<FrontendEvent>,
+    server: &mut Server,
+    handle: ClientHandle,
+) {
+    let state = server
+        .client_manager
+        .borrow_mut()
+        .get_mut(handle)
+        .map(|(_, s)| s.clone());
+    if let Some(state) = state {
+        let _ = frontend
+            .send(FrontendEvent::StateChange(handle, state))
+            .await;
+    }
+}