Clients list

src/server.rs

@@ -1,16 +1,30 @@
-use std::sync::Arc;
+use std::{
+    collections::HashMap,
+    sync::{Arc, atomic::AtomicU16},
+};
 
 use axum::{
     extract::{WebSocketUpgrade, ws::WebSocket},
     response::Response,
 };
-use ed25519_dalek::SigningKey;
+use ed25519_dalek::{SigningKey, VerifyingKey};
+use tokio::sync::Mutex;
 
-use crate::{config::Config, protocol::Client};
+use crate::{
+    config::Config,
+    protocol::{Client, ClientMeta},
+};
+
+pub struct OnlineClientMeta {
+    pub meta: ClientMeta,
+    pub counter: AtomicU16,
+    pub connections: HashMap<u16, Client>,
+}
 
 pub struct Server {
     pub key: SigningKey,
     pub config: Config,
+    pub clients: Mutex<HashMap<VerifyingKey, OnlineClientMeta>>,
 }
 
 impl Server {
@@ -18,6 +32,7 @@ impl Server {
         Ok(Arc::new(Self {
             key: crate::signature::get().await?,
             config: Config::get().await?,
+            clients: Mutex::new(HashMap::new()),
         }))
     }
 }
@@ -28,7 +43,25 @@ impl Server {
 
         ws.on_upgrade(move |socket: WebSocket| async move {
             match Client::initialize(&s, socket).await {
-                Ok(mut client) => {
+                Ok((mut client, meta)) => {
+                    let mut clients_meta = s.clients.lock().await;
+
+                    let client_meta =
+                        clients_meta
+                            .entry(client.public_key)
+                            .or_insert_with(|| OnlineClientMeta {
+                                meta,
+                                counter: AtomicU16::new(0),
+                                connections: HashMap::new(),
+                            });
+
+                    client_meta.connections.insert(
+                        client_meta
+                            .counter
+                            .fetch_add(1, std::sync::atomic::Ordering::Relaxed),
+                        client.clone(),
+                    );
+
                     if let Err(e) = client.read_loop().await {
                         eprintln!("Failed to handle client: {e}");
                     } else {