use client_network::{ ChunkNode, MerkleNode, MerkleTree, NetworkCommand, NetworkEvent, NodeHash, big_or_chunk_to_file, filename_to_string, generate_base_tree, node_hash_to_hex_string, remove_null_bytes, }; use crossbeam_channel::{Receiver, Sender}; use egui::{ Align, CentralPanel, CollapsingHeader, Color32, Context, CornerRadius, Frame, Layout, Response, ScrollArea, SidePanel, Stroke, TopBottomPanel, Ui, ViewportCommand, }; use std::collections::HashSet; use std::{collections::HashMap, fmt::format, io::Seek}; use std::fs::{File, OpenOptions, create_dir}; enum ServerStatus { Loading, NotConnected, Connected, ConnectedHandshake, } // --- Main Application Struct --- pub struct P2PClientApp { remaining: std::time::Duration, // temps restant last_update: std::time::Instant, // pour calculer delta timer_started: bool, // Communication channels network_cmd_tx: Sender, network_event_rx: Receiver, // GUI State status_message: String, known_peers: Vec<(String, bool)>, loading_peers: Vec, connect_address_input: String, connected_address: String, connect_name_input: String, // Key: Parent Directory Hash (String), Value: List of children FileNode loaded_fs: HashMap, shared_tree: MerkleTree, // Current peer tree displayed active_peer: Option, server_status: ServerStatus, show_network_popup: bool, // gΓ©rer selon besoin error_message: Option<(String, String)>, // Some(message) -> afficher, None -> rien success_message: Option<(String, String)>, // Some(message) -> afficher, None -> rien active_server: String, current_downloading_file_map: MerkleTree, remaining_chunks: HashSet<[u8; 32]>, root_downloading_file: String, } impl P2PClientApp { pub fn new(cmd_tx: Sender, event_rx: Receiver) -> Self { //let (root_hash, tree_content) = MerkleNode::generate_base_tree(); let mut loaded_fs = HashMap::new(); let mut current_downloading_file_map = MerkleTree::new(HashMap::new(), [0; 32]); //let tree = MerkleTree::new(tree_content, root_hash); //loaded_fs.insert("bob".to_string(), tree); Self { remaining: std::time::Duration::from_secs(0), timer_started: false, last_update: std::time::Instant::now(), network_cmd_tx: cmd_tx, network_event_rx: event_rx, status_message: "Client Initialized. Awaiting network status...".to_string(), known_peers: Vec::new(), loading_peers: Vec::new(), connect_address_input: "https://jch.irif.fr:8443".to_string(), connected_address: "".to_string(), loaded_fs, active_peer: None, server_status: ServerStatus::NotConnected, show_network_popup: false, error_message: None, success_message: None, connect_name_input: "bob".to_string(), active_server: "".to_string(), shared_tree: generate_base_tree(), current_downloading_file_map: current_downloading_file_map, root_downloading_file: "".to_string(), remaining_chunks: HashSet::new(), } } pub fn show_error(&mut self, msg: impl Into, peer_username: impl Into) { self.error_message = Some((msg.into(), peer_username.into())); } pub fn show_success(&mut self, msg: impl Into, peer_username: impl Into) { self.success_message = Some((msg.into(), peer_username.into())); } pub fn clear_error(&mut self) { self.error_message = None; } pub fn clear_success(&mut self) { self.success_message = None; } } // --- eframe::App Trait Implementation --- impl eframe::App for P2PClientApp { fn update(&mut self, ctx: &Context, _frame: &mut eframe::Frame) { if matches!(self.server_status, ServerStatus::Connected) && !self.timer_started { self.remaining = std::time::Duration::from_secs(30 * 60); self.last_update = std::time::Instant::now(); self.timer_started = true; } // in update (every frame) let now = std::time::Instant::now(); let delta = now.saturating_duration_since(self.last_update); self.last_update = now; if matches!(self.server_status, ServerStatus::Connected) && self.remaining > std::time::Duration::ZERO { self.remaining = self.remaining.saturating_sub(delta); } // 1. Process incoming Network Events // We poll the channel and update the GUI state for every event received. while let Ok(event) = self.network_event_rx.try_recv() { match event { NetworkEvent::PeerConnected(addr) => { todo!(); self.status_message = format!("βœ… Peer connected: {}", addr); if !self.known_peers.contains(&(addr, true)) { self.known_peers.push((addr, true)); } } NetworkEvent::RootRequest(addr) => { let root = self.shared_tree.root; let _ = self .network_cmd_tx .send(NetworkCommand::SendRootReply(root.to_vec(), addr)); } NetworkEvent::DatumRequest(node_hash, addr) => { let hash: NodeHash = node_hash.try_into().expect("incorrect size"); let asked_datum = self.shared_tree.data.get(&hash); match asked_datum { Some(datum_found) => { let _ = self.network_cmd_tx.send(NetworkCommand::SendDatum( datum_found.clone(), node_hash, addr, )); } None => { let _ = self .network_cmd_tx .send(NetworkCommand::SendNoDatum(node_hash.to_vec(), addr)); } } } NetworkEvent::PeerListUpdated(peers) => { //todo!(); self.known_peers = peers; } NetworkEvent::FileTreeReceived(node_hash, merklenode, ip) => { match &self.active_peer { Some(active_peer) => { if let Some(maptree) = self.loaded_fs.get_mut(active_peer) { maptree.data.insert(node_hash, merklenode.clone()); match merklenode { MerkleNode::Directory(d) => { for entry in d.entries { let _ = self.network_cmd_tx.send( NetworkCommand::GetChildren( entry.content_hash, ip.clone(), false, ), ); } } MerkleNode::BigDirectory(bigd) => { for entry in bigd.children_hashes { let _ = self.network_cmd_tx.send( NetworkCommand::GetChildren( entry, ip.clone(), false, ), ); } } _ => {} } } } None => {} } } NetworkEvent::FileTreeRootReceived(peer_id, root_hash) => { // todo!(); /*self.status_message = format!( "πŸ”„ Received Merkle Root from {}: {}", peer_id, &root_hash[..8] );*/ if let Ok(chunknode) = ChunkNode::new(Vec::new()) { let data_map: HashMap = HashMap::new(); //data_map.insert(root_hash, MerkleNode::Chunk(chunknode)); println!("len root: {}", data_map.len()); println!("node hash: {:?}", root_hash.to_vec()); let tree = MerkleTree { data: data_map, root: root_hash, }; match &self.active_peer { Some(activepeer) => { self.loaded_fs.insert(activepeer.clone(), tree); } None => {} } println!("tree created"); } //self.active_peer_id = Some(peer_id.clone()); // Request the content of the root directory immediately /*let _ = self .network_cmd_tx .send(NetworkCommand::RequestDirectoryContent(peer_id, root_hash));*/ } NetworkEvent::Connected(ip) => { self.server_status = ServerStatus::Connected; self.connected_address = ip.clone(); let _ = self.network_cmd_tx.send(NetworkCommand::FetchPeerList( self.connected_address.clone(), )); } NetworkEvent::ConnectedHandshake() => { self.server_status = ServerStatus::ConnectedHandshake; } NetworkEvent::Disconnected() => { self.active_server = "".to_string(); self.connected_address = "".to_string(); self.known_peers.clear(); self.server_status = ServerStatus::NotConnected; } NetworkEvent::Error(err, peer_username) => { self.loading_peers.retain(|s| s != peer_username.as_str()); self.show_error(err, peer_username); } NetworkEvent::InitDownload(hash, ip) => { if let Some(addr) = &self.active_peer { if let Some(roottree) = self.loaded_fs.get(addr) { if let Some(root) = roottree.data.get(&hash) { let _ = self .current_downloading_file_map .data .insert(hash, root.clone()); let _ = self .network_cmd_tx .send(NetworkCommand::GetChildren(hash, ip, true)); } } } } NetworkEvent::DataReceived(hash, merkle_node, ip) => { let _ = self .current_downloading_file_map .data .insert(hash, merkle_node.clone()); println!("merkle:{}", merkle_node.get_type_byte()); match merkle_node { MerkleNode::Big(bigfile) => { for entry in bigfile.children_hashes { println!("entry: {:?}", entry); let _ = self.network_cmd_tx.send(NetworkCommand::GetChildren( entry, ip.clone(), true, )); self.remaining_chunks.insert(entry); } self.remaining_chunks.remove(&hash); } MerkleNode::Chunk(chunk) => { self.remaining_chunks.remove(&hash); } _ => {} } if self.remaining_chunks.is_empty() { println!("bigfile tΓ©lΓ©chargΓ©"); } } NetworkEvent::Success(msg, peer_username) => { self.loading_peers.retain(|s| s != peer_username.as_str()); self.show_success(msg, peer_username); } NetworkEvent::HandshakeFailed() => {} NetworkEvent::ServerHandshakeFailed(err) => { self.active_server = "".to_string(); self.server_status = ServerStatus::NotConnected; let err_msg = format!("Failed to connect to the server: {}", err); self.show_error(err_msg, ""); let res = self.network_cmd_tx.send(NetworkCommand::ResetServerPeer()); } } } // 2. Menu Bar TopBottomPanel::top("top_panel").show(ctx, |ui| { egui::MenuBar::new().ui(ui, |ui| { ui.menu_button("File", |ui| { if ui.button("Settings").clicked() { //show settings } if ui.button("Quit").clicked() { // Use ViewportCommand to request a close ctx.send_viewport_cmd(ViewportCommand::Close); } }); ui.menu_button("Network", |ui| { match self.server_status { ServerStatus::Connected | ServerStatus::ConnectedHandshake => { let desired = egui::vec2(300.0, 0.0); // width 300, auto-height if 0 ui.set_min_size(desired); ui.vertical(|ui| { if ui.button("Disconnect").clicked() { println!("Disconnecting..."); let _ = self.network_cmd_tx.send(NetworkCommand::Disconnect()); self.server_status = ServerStatus::NotConnected; self.remaining = std::time::Duration::from_secs(0); self.timer_started = false; ui.close(); } }); } ServerStatus::NotConnected => { let desired = egui::vec2(0.0, 0.0); // width 300, auto-height if 0 ui.set_min_size(desired); ui.vertical(|ui| { ui.horizontal(|ui| { ui.label("Server IP:"); ui.text_edit_singleline(&mut self.connect_address_input); }); ui.horizontal(|ui| { ui.label("Name:"); ui.text_edit_singleline(&mut self.connect_name_input); }); if ui.button("Connect").clicked() { let addr = self.connect_address_input.clone(); let name = self.connect_name_input.clone(); let _ = self .network_cmd_tx .send(NetworkCommand::ConnectToServerPut(addr, name)); self.server_status = ServerStatus::Loading; ui.close(); } }); } _ => {} } }); }); }); TopBottomPanel::bottom("bottom_panel").show(ctx, |ui| { ui.horizontal(|ui| { match self.server_status { ServerStatus::Loading => { ui.spinner(); } ServerStatus::Connected => { ui.label("Registered but no server peer chosen..."); } ServerStatus::NotConnected => { ui.label("No connection.."); } ServerStatus::ConnectedHandshake => { let str = format!("πŸ“‘"); ui.label(str); } } ui.add_space(ui.available_width() - 30.0); // formater mm:ss let secs = self.remaining.as_secs(); let minutes = secs / 60; let seconds = secs % 60; ui.label(format!("{:02}:{:02}", minutes, seconds)); }); }); SidePanel::right("right_panel") .resizable(false) .min_width(180.0) .show(ctx, |ui| { ui.horizontal(|ui| { ui.heading("🌐 Known Peers"); ui.add_space(20.0); if ui.button("πŸ”„").clicked() { println!("addr:{}", self.connected_address.clone()); let res = self.network_cmd_tx.send(NetworkCommand::FetchPeerList( self.connected_address.clone(), )); if let Some(error) = res.err() { println!( "[GUI] Error while sending crossbeam message to Network: {}", error.to_string() ); } if let Some(active_peer) = &self.active_peer { if let Some(tree) = self.loaded_fs.get(active_peer) { println!("{}", tree.data.len()); } } } }); ui.separator(); ScrollArea::vertical().show(ui, |ui| { if self.known_peers.is_empty() { ui.add_space(10.0); ui.label("No active peers."); } else { for peer in &self.known_peers { let is_active = self.active_peer.as_ref().map_or(false, |id| id == &peer.0); // if peer.id == self.active_peer_id let selectable: Response; // if &self.active_server == &peer.0 { // // Create a frame with green background and render the selectable inside it. // // Adjust rounding, padding and stroke as desired. // let frame = Frame { // fill: Color32::DARK_BLUE, // stroke: Stroke::default(), // corner_radius: CornerRadius::from(0.5), // ..Default::default() // }; // let internal = frame.show(ui, |ui| { // // horizontal row: label on the left, spinner on the right // ui.horizontal(|ui| { // // let selectable label take remaining space // ui.with_layout(Layout::left_to_right(Align::Center), |ui| { // ui.add_space(0.0); // ensure layout established // return ui.selectable_label( // is_active, // format!("{}", peer.0), // ); // }) // .inner // return Response from the inner closure if your egui version does so // }) // }); // selectable = internal.inner.inner; // } else { // selectable = ui.selectable_label(is_active, format!("{}", peer.0)); // } // place spinner to the right of the label ui.horizontal(|ui| { // Use same width for the label widget as the selectable we already created: // Recreate selectable inline so both label and spinner share the same row. let resp = if &self.active_server == &peer.0 { // draw with frame inline let frame = Frame { fill: Color32::DARK_BLUE, stroke: Stroke::default(), corner_radius: CornerRadius::from(0.5), ..Default::default() }; frame .show(ui, |ui| { ui.selectable_label(is_active, format!("{}", peer.0)) }) .inner } else { ui.selectable_label(is_active, format!("{}", peer.0)) }; ui.add_space(4.0); // small gap if self.loading_peers.contains(&peer.0) { // push spinner to right by expanding a spacer before it ui.with_layout(Layout::right_to_left(Align::Center), |ui| { ui.spinner(); }); } // use resp (click handling etc.) if resp.clicked() { // switch to displaying this peer's tree self.active_peer = Some(peer.0.clone()); // Request root content if not loaded if !self .loaded_fs .contains_key(self.active_peer.as_ref().unwrap()) { //todo!(); let _ = self.network_cmd_tx.send(NetworkCommand::Discover( peer.0.clone(), "root".to_string(), self.connected_address.clone(), )); } } resp.context_menu(|ui| { // ... action match self.server_status { ServerStatus::Connected => { if ui .button("Utiliser le peer en tant que serveur") .clicked() { self.active_server = peer.0.to_string(); let res = self.network_cmd_tx.send( NetworkCommand::ServerHandshake( peer.0.to_string(), self.connected_address.clone(), ), ); } } _ => {} } if ui.button("Send Ping").clicked() { let res = self.network_cmd_tx.send(NetworkCommand::Ping( peer.0.to_string(), self.connected_address.clone(), )); self.loading_peers.push(peer.0.to_owned()); } if ui.button("Send Nat Traversal Request").clicked() { match self.network_cmd_tx.send( NetworkCommand::NatTraversal( peer.0.to_string(), self.connected_address.clone(), ), ) { Ok(_) => { print!( "[+] successfully sent nat traversal request" ) } Err(_) => { print!("[-] failed to send nat traversal request") } } } if ui.button("Infos").clicked() { // action 3 ui.close(); } // ... autres boutons }); }); // if self.loading_peers.contains(&peer.0) { // ui.spinner(); // } //if selectable.clicked() { // switch to displaying this peer's tree //self.active_peer = Some(peer.0.clone()); //// Request root content if not loaded //if !self // .loaded_fs // .contains_key(self.active_peer.as_ref().unwrap()) //{ // //todo!(); // let _ = self.network_cmd_tx.send(NetworkCommand::Discover( // peer.0.clone(), // "root".to_string(), // self.connected_address.clone(), // )); //} //} //selectable.context_menu(|ui| { // // ... action // match self.server_status { // ServerStatus::Connected => { // if ui // .button("Utiliser le peer en tant que serveur") // .clicked() // { // self.active_server = peer.0.to_string(); // let res = self.network_cmd_tx.send( // NetworkCommand::ServerHandshake( // peer.0.to_string(), // self.connected_address.clone(), // ), // ); // } // } // _ => {} // } // if ui.button("Send Ping").clicked() { // let res = self.network_cmd_tx.send(NetworkCommand::Ping( // peer.0.to_string(), // self.connected_address.clone(), // )); // self.loading_peers.push(peer.0.to_owned()); // } // if ui.button("Send Nat Traversal Request").clicked() { // match self.network_cmd_tx.send(NetworkCommand::NatTraversal( // peer.0.to_string(), // self.connected_address.clone(), // )) { // Ok(_) => { // print!("[+] successfully sent nat traversal request") // } // Err(_) => { // print!("[-] failed to send nat traversal request") // } // } // } // if ui.button("Infos").clicked() { // // action 3 // ui.close(); // } // // ... autres boutons //}); } } }); }); CentralPanel::default().show(ctx, |ui| { ui.heading({ if let Some(peer) = &self.active_peer { format!("πŸ“‚ {}'s tree", peer) } else { "πŸ“‚ p2p-merkel client".to_string() } }); ui.separator(); if let Some(active_peer) = &self.active_peer { if let Some(tree) = self.loaded_fs.get(active_peer) { ScrollArea::vertical().show(ui, |ui| { // Start drawing the tree from the root hash self.draw_file_tree(ui, tree); }); } else { ui.label(format!("Loading root for peer: {}", active_peer)); } } else { ui.label("Connect to a peer to view a file tree."); } ui.separator(); ui.add_space(5.0); // This is now safe because draw_file_tree only takes an immutable borrow // ui.label(format!("Status: {}", self.status_message)); }); if let Some(msg) = &self.error_message { let msg = msg.clone(); egui::Window::new("Error") .collapsible(false) .resizable(false) .anchor(egui::Align2::CENTER_CENTER, [0.0, 0.0]) .show(ctx, |ui| { ui.label(&msg.1); ui.label(&msg.0); if ui.button("OK").clicked() { self.clear_error(); } }); ctx.request_repaint(); } if let Some(msg) = &self.success_message { let msg = msg.clone(); egui::Window::new("Success") .collapsible(false) .resizable(false) .anchor(egui::Align2::CENTER_CENTER, [0.0, 0.0]) .show(ctx, |ui| { ui.label(&msg.1); ui.label(&msg.0); if ui.button("OK").clicked() { self.clear_success(); } }); ctx.request_repaint(); } ctx.request_repaint_after(std::time::Duration::from_millis(10)); } } // --- Helper for Drawing the Recursive File Tree --- impl P2PClientApp { fn draw_file_tree(&self, ui: &mut Ui, tree: &MerkleTree) { assert!(self.active_peer.is_some()); assert!( self.loaded_fs .get(&self.active_peer.clone().unwrap()) .is_some() ); let root = tree.root; CollapsingHeader::new(format!("πŸ“ root")) .default_open(true) .enabled(true) .show(ui, |ui| { self.draw_file_node(ui, root, tree, 0, None); }); } fn draw_file_node( &self, ui: &mut Ui, to_draw: NodeHash, tree: &MerkleTree, depth: usize, filename: Option<[u8; 32]>, ) { if depth >= 32 { return; } if let Some(current) = tree.data.get(&to_draw) { let name = { if filename.is_some() { String::from_utf8(filename.unwrap().to_vec()).expect("err") } else { node_hash_to_hex_string(&to_draw) } }; match current { MerkleNode::Chunk(node) => { if ui .selectable_label(false, format!("πŸ“„ (C) {}...", name)) .on_hover_text("Click to request file chunks...") .clicked() { match create_dir("./Download/") { Ok(_) => println!("Directory created successfully!"), Err(e) => println!("Failed to create directory: {}", e), } let new_name = format!("./Download/{}", name); let sani = remove_null_bytes(&new_name); println!("sani:{}", sani); let mut file = OpenOptions::new() .append(true) .create(true) .open(sani) .unwrap(); big_or_chunk_to_file(tree, &MerkleNode::Chunk(node.clone()), &mut file); // if let Some(peer_id) = active_peer_id.clone() { // let _ = self.network_cmd_tx.send(NetworkCommand::RequestChunk(peer_id, entry_hash.clone())); // // self.status_message = format!("Requested file chunks for: {}...", &entry_hash[..8]); // } // todo!(); } } MerkleNode::Directory(node) => { CollapsingHeader::new(format!("πŸ“ (D) {}", name)) .default_open(false) .enabled(true) .show(ui, |ui| { for entry in &node.entries { self.draw_file_node( ui, entry.content_hash, tree, depth + 1, Some(entry.filename.try_into().expect("incorrect size")), ); } }); } MerkleNode::Big(node) => { if ui .selectable_label(false, format!("πŸ“„ (B) {}", name)) .on_hover_text("Click to request file chunks...") .clicked() { if let Some(addr) = &self.active_peer { let _ = self .network_cmd_tx .send(NetworkCommand::InitDownload(to_draw, addr.clone())); } } } MerkleNode::BigDirectory(node) => { CollapsingHeader::new(format!("πŸ“ (BD) {}", name)) .default_open(false) .enabled(true) .show(ui, |ui| { for child in &node.children_hashes { self.draw_file_node(ui, child.clone(), tree, depth + 1, None); } }); } } } } }