use client_network::{
    ChunkNode, MerkleNode, MerkleTree, NetworkCommand, NetworkEvent, NodeHash,
    big_or_chunk_to_file, 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, Id, Layout,
    ProgressBar, ScrollArea, SidePanel, Stroke, TopBottomPanel, Ui, ViewportCommand,
};
use std::collections::HashMap;
use std::collections::HashSet;

use std::fs::{OpenOptions, create_dir};

enum ServerStatus {
    Loading,
    NotConnected,
    Connected,
    ConnectedHandshake,
}

pub struct P2PClientApp {
    remaining: std::time::Duration,  // temps restant
    last_update: std::time::Instant, // pour calculer delta
    timer_started: bool,
    network_cmd_tx: Sender<NetworkCommand>,
    network_event_rx: Receiver<NetworkEvent>,

    known_peers: Vec<(String, bool)>,
    loading_peers: Vec<String>,
    connect_address_input: String,
    connected_address: String,
    connect_name_input: String,

    loaded_fs: HashMap<String, MerkleTree>,
    shared_tree: MerkleTree,

    // Current peer tree displayed
    active_peer: Option<String>,

    server_status: ServerStatus,

    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]>,

    // total number of chunks expected for the current download (set when download starts)
    current_total_chunks: Option<usize>,

    // number of chunks received so far (count of removed remaining_chunks)
    current_received_chunks: usize,

    root_downloading_file: String,
    show_network_window: bool,
    show_choose_server_window: bool,
}

impl P2PClientApp {
    pub fn new(cmd_tx: Sender<NetworkCommand>, event_rx: Receiver<NetworkEvent>) -> Self {
        let loaded_fs = HashMap::new();
        let current_downloading_file_map = MerkleTree::new(HashMap::new(), [0; 32]);

        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,
            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,
            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,
            current_total_chunks: None,
            current_received_chunks: 0,
            root_downloading_file: "".to_string(),
            remaining_chunks: HashSet::new(),
            show_network_window: false,
            show_choose_server_window: false,
        }
    }
    pub fn show_error(&mut self, msg: impl Into<String>, peer_username: impl Into<String>) {
        self.error_message = Some((msg.into(), peer_username.into()));
    }
    pub fn show_success(&mut self, msg: impl Into<String>, peer_username: impl Into<String>) {
        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;
    }

    fn set_current_total_chunks(&mut self, len: Option<usize>) {
        self.current_total_chunks = len
    }

    fn set_current_received_chunks(&mut self, arg: usize) {
        self.current_received_chunks = arg
    }
}

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;
        }

        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);
        }

        while let Ok(event) = self.network_event_rx.try_recv() {
            match event {
                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) => {
                    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) => {
                    if let Ok(chunknode) = ChunkNode::new(Vec::new()) {
                        let data_map: HashMap<NodeHash, MerkleNode> = 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");
                    }
                }
                NetworkEvent::Connected(ip) => {
                    self.server_status = ServerStatus::Connected;
                    self.connected_address = ip.clone();
                    self.show_choose_server_window = true;
                    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, name) => {
                    if let Some(addr) = &self.active_peer {
                        if let Some(roottree) = self.loaded_fs.get(addr) {
                            if let Some(root) = roottree.data.get(&hash) {
                                self.root_downloading_file = name;
                                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.current_total_chunks = Some(self.remaining_chunks.len());
                            }
                            self.remaining_chunks.remove(&hash);
                        }
                        MerkleNode::Chunk(_) => {
                            self.remaining_chunks.remove(&hash);
                        }
                        _ => {}
                    }

                    if let Some(total) = self.current_total_chunks {
                        // recompute received (safer than incrementing)
                        let received = total.saturating_sub(self.remaining_chunks.len());
                        self.current_received_chunks = received;
                    }

                    if self.remaining_chunks.is_empty() {
                        self.current_total_chunks = None;
                        self.current_received_chunks = 0;
                        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, "");
                    match self.network_cmd_tx.send(NetworkCommand::ResetServerPeer()) {
                        Ok(_) => {}
                        Err(err) => {
                            println!("GUI Error : {}", err.to_string());
                        }
                    };
                }
            }
        }

        if self.show_choose_server_window {
            let full_rect = ctx.input(|i| i.content_rect());
            let modal_size = egui::vec2(400.0, 160.0);

            egui::Area::new(Id::new("modal_blocker_bg"))
                .order(egui::Order::Background)
                .show(ctx, |ui| {
                    let painter = ui.painter();
                    painter.rect_filled(full_rect, 0.0, egui::Color32::from_black_alpha(160));

                    let sense = egui::Sense::click_and_drag();
                    ui.allocate_exact_size(full_rect.size(), sense);
                });
            egui::Window::new("Choose the server")
                .resizable(false)
                .collapsible(false)
                .title_bar(true)
                .anchor(egui::Align2::CENTER_CENTER, egui::Vec2::ZERO)
                .fixed_size(modal_size)
                .show(ctx, |ui| {
                    ScrollArea::vertical()
                        .auto_shrink([false; 2])
                        .show(ui, |ui| {
                            ui.style_mut().visuals.widgets.inactive.bg_fill =
                                ui.style().visuals.widgets.inactive.bg_fill; // no-op to get mutable borrow
                            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

                                    // 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

                                        // use resp (click handling etc.)
                                        if resp.clicked() {
                                            self.active_server = peer.0.to_string();
                                            match self.network_cmd_tx.send(
                                                NetworkCommand::ServerHandshake(
                                                    peer.0.to_string(),
                                                    self.connected_address.clone(),
                                                ),
                                            ) {
                                                Ok(_) => {}
                                                Err(e) => {
                                                    println!("GUI Error : {}", e.to_string());
                                                }
                                            };
                                            ui.close();
                                            self.show_choose_server_window = false;
                                        }
                                    });
                                }
                            }
                        });
                });
        }

        // 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() {}

                    if ui.button("Quit").clicked() {
                        ctx.send_viewport_cmd(ViewportCommand::Close);
                    }
                });

                if ui.button("Network").clicked() {
                    self.show_network_window = true;
                }

                if self.show_network_window {
                    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;
                                    self.show_network_window = false;
                                }
                            });
                        }
                        ServerStatus::NotConnected => {
                            egui::Window::new("Network")
                                .resizable(false)
                                .show(ctx, |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.to_string(),
                                            ),
                                        );
                                        self.server_status = ServerStatus::Loading;
                                        ui.close();
                                        self.show_network_window = false;
                                        self.loaded_fs
                                            .insert(name.to_string(), self.shared_tree.clone());
                                    }
                                });
                        }
                        _ => {}
                    }
                }
            });
        });

        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 => ui.label("📡"),
                };

                ui.add_space(8.0); // small gap

                // desired progress bar width
                let bar_width = 220.0f32;
                // push it to the right by adding space equal to remaining width minus bar width
                let push = (ui.available_width() - bar_width).max(0.0);
                ui.add_space(push);

                if let Some(total) = self.current_total_chunks {
                    let received = self.current_received_chunks;
                    let frac = if total == 0 {
                        1.0
                    } else {
                        received as f32 / total as f32
                    };

                    ui.add(
                        ProgressBar::new(frac)
                            .show_percentage()
                            .animate(true)
                            .desired_height(10.0),
                    );
                }
            });
        });

        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()
                    .auto_shrink([false; 2])
                    .show(ui, |ui| {
                        ui.style_mut().visuals.widgets.inactive.bg_fill =
                            ui.style().visuals.widgets.inactive.bg_fill; // no-op to get mutable borrow
                        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
                                //
                                if peer.0.eq(&self.connect_name_input) {
                                    ui.horizontal(|ui| {
                                        let resp = ui.selectable_label(
                                            is_active,
                                            format!("{} (you)", peer.0),
                                        );

                                        if resp.clicked() {
                                            self.active_peer = Some(peer.0.clone());
                                        }
                                    });
                                } else {
                                    // place spinner to the right of the label
                                    ui.horizontal(|ui| {
                                        let resp = if &self.active_server == &peer.0 {
                                            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) {
                                            ui.with_layout(
                                                Layout::right_to_left(Align::Center),
                                                |ui| {
                                                    ui.spinner();
                                                },
                                            );
                                        }

                                        if resp.clicked() {
                                            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| {
                                    match self.server_status {
                                        ServerStatus::Connected => {
                                            if ui
                                                .button("Utiliser le peer en tant que serveur")
                                                .clicked()
                                            {
                                                self.active_server = peer.0.to_string();
                                                match self.network_cmd_tx.send(
                                                    NetworkCommand::ServerHandshake(
                                                        peer.0.to_string(),
                                                        self.connected_address.clone(),
                                                    ),
                                                ) {
                                                Ok(_) => {}
                                                Err(e) => {
                                                    println!("GUI Error : {}", e.to_string());
                                                },
                                                };
                                            }
                                        }
                                        _ => {}
                                    }
                                    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.clone().get(active_peer) {
                    ScrollArea::vertical().show(ui, |ui| {
                        // Start drawing the tree from the root hash
                        self.draw_file_tree(ui, tree);
                    });
                } else {
                    ui.horizontal(|ui| {
                        ui.label(format!("Loading root for peer: {}", active_peer));
                        ui.add_space(8.0);
                        ui.spinner(); // or conditional: if is_loading { ui.spinner(); }
                    });
                }
            } 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(&mut 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(
        &mut 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(name) = filename {
                            if let Ok(nameb) = String::from_utf8(name.to_vec()) {
                                if let Some(addr) = &self.active_peer {
                                    let _ = self.network_cmd_tx.send(NetworkCommand::InitDownload(
                                        to_draw,
                                        addr.clone(),
                                        nameb,
                                    ));
                                }
                            }
                        }
                    }
                }
                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);
                            }
                        });
                }
            }
        }
    }
}