grim/src/gui/views/network.rs

// Copyright 2023 The Grim Developers
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use std::borrow::Cow;
use std::collections::hash_map::DefaultHasher;
use std::sync::atomic::Ordering;
use std::time::Duration;
use chrono::Utc;
use eframe::epaint::{Color32, FontId, Stroke};
use eframe::epaint::text::{LayoutJob, TextFormat, TextWrapping};
use egui::{Response, RichText, Sense, Spinner, Widget};
use egui::style::Margin;
use egui_extras::{Size, StripBuilder};
use grin_chain::SyncStatus;

use grin_core::global::ChainTypes;
use grin_servers::ServerStats;
use crate::gui::app::is_dual_panel_mode;
use crate::gui::platform::PlatformCallbacks;
use crate::gui::screens::Navigator;
use crate::gui::{COLOR_DARK, COLOR_LIGHT, COLOR_YELLOW, SYM_ACCOUNTS, SYM_METRICS, SYM_NETWORK};

use crate::gui::views::{NetworkTab, TitlePanel, TitlePanelAction};
use crate::gui::views::network_node::NetworkNode;
use crate::node;
use crate::node::Node;

enum Mode {
    Node,
    // Miner,
    Metrics,
    Tuning
}

pub struct Network {
    current_mode: Mode,
    node: Node,
    node_view: NetworkNode,
}

impl Default for Network {
    fn default() -> Self {
        let node = Node::new(ChainTypes::Mainnet, true);
        Self {
            node,
            current_mode: Mode::Node,
            node_view: NetworkNode::default()
        }
    }
}

impl Network {
    pub fn ui(&mut self,
              ui: &mut egui::Ui,
              frame: &mut eframe::Frame,
              nav: &mut Navigator,
              cb: &dyn PlatformCallbacks) {

        egui::TopBottomPanel::top("network_title")
            .resizable(false)
            .frame(egui::Frame {
                fill: COLOR_YELLOW,
                inner_margin: Margin::same(0.0),
                outer_margin: Margin::same(0.0),
                stroke: Stroke::NONE,
                ..Default::default()
            })
            .show_inside(ui, |ui| {
                self.draw_title(ui, frame, nav);
            });

        egui::CentralPanel::default().frame(egui::Frame {
            stroke: Stroke::new(1.0, Color32::from_gray(190)),
            fill: Color32::WHITE,
            .. Default::default()
        }).show_inside(ui, |ui| {
            self.draw_tab_content(ui);
        });

        egui::TopBottomPanel::bottom("network_tabs")
            .frame(egui::Frame {
                stroke: Stroke::new(1.0, Color32::from_gray(190)),
                .. Default::default()
            })
            .resizable(false)
            .show_inside(ui, |ui| {
                self.draw_tabs(ui);
            });

        ui.ctx().request_repaint_after(Duration::from_millis(500));
    }

    fn draw_tabs(&self, ui: &mut egui::Ui) {
        ui.vertical_centered(|ui| {
            ui.columns(3, |columns| {
                columns[0].horizontal_wrapped(|ui| {
                });
                columns[1].vertical_centered(|ui| {
                });
                columns[2].horizontal_wrapped(|ui| {
                });
            });
        });
    }

    fn draw_tab_content(&mut self, ui: &mut egui::Ui) {
        match self.current_mode {
            Mode::Node => {
                self.node_view.ui(ui, &mut self.node);
            }
            Mode::Metrics => {}
            Mode::Tuning => {}
        }
    }

    fn draw_title(&mut self, ui: &mut egui::Ui, frame: &mut eframe::Frame, nav: &mut Navigator) {
        // Disable stroke around title buttons on hover
        ui.style_mut().visuals.widgets.active.bg_stroke = Stroke::NONE;

        StripBuilder::new(ui)
            .size(Size::exact(52.0))
            .vertical(|mut strip| {
                strip.strip(|builder| {
                    builder
                        .size(Size::exact(52.0))
                        .size(Size::remainder())
                        .size(Size::exact(52.0))
                        .horizontal(|mut strip| {
                            strip.empty();
                            strip.strip(|builder| {
                                self.draw_title_text(builder);
                            });
                            strip.cell(|ui| {
                                if !is_dual_panel_mode(frame) {
                                    ui.centered_and_justified(|ui| {
                                        let b = egui::widgets::Button::new(
                                            RichText::new(SYM_ACCOUNTS)
                                                .size(24.0)
                                                .color(COLOR_DARK)
                                        ).fill(Color32::TRANSPARENT)
                                            .ui(ui).interact(Sense::click_and_drag());
                                        if b.drag_released() || b.clicked() {
                                            nav.toggle_left_panel();
                                        };
                                    });
                                }
                            });
                        });
                });
            });
    }

    fn draw_title_text(&self, mut builder: StripBuilder) {
        let Self { node, ..} = self;

        let title_text = match &self.current_mode {
            Mode::Node => {
                self.node_view.title()
            }
            Mode::Metrics => {
                self.node_view.title()
            }
            Mode::Tuning => {
                self.node_view.title()
            }
        };

        let r_stats = node.state.read_stats();
        let syncing = r_stats.is_some() &&
            r_stats.as_ref().unwrap().sync_status != SyncStatus::NoSync;

        let mut b = builder.size(Size::remainder());
        if syncing {
            b = b.size(Size::remainder());
        }
        b.vertical(|mut strip| {
            strip.cell(|ui| {
                ui.centered_and_justified(|ui| {
                    ui.heading(title_text.to_uppercase());
                });
            });
            if syncing {
                strip.cell(|ui| {
                    ui.centered_and_justified(|ui| {
                        let status_text = if node.state.is_stopping() {
                            get_sync_status(SyncStatus::Shutdown).to_string()
                        } else if node.state.is_restarting() {
                            "Restarting".to_string()
                        } else {
                            get_sync_status(r_stats.as_ref().unwrap().sync_status).to_string()
                        };
                        let mut job = LayoutJob::single_section(status_text, TextFormat {
                            font_id: FontId::proportional(15.0),
                            color: COLOR_DARK,
                            .. Default::default()
                        });
                        job.wrap = TextWrapping {
                            max_rows: 1,
                            break_anywhere: false,
                            overflow_character: Option::from('…'),
                            ..Default::default()
                        };
                        ui.label(job);
                    });
                });
            }
        });
    }
}

fn get_sync_status(sync_status: SyncStatus) -> Cow<'static, str> {
    match sync_status {
        SyncStatus::Initial => Cow::Borrowed("Initializing"),
        SyncStatus::NoSync => Cow::Borrowed("Running"),
        SyncStatus::AwaitingPeers(_) => Cow::Borrowed("Waiting for peers"),
        SyncStatus::HeaderSync {
            sync_head,
            highest_height,
            ..
        } => {
            if highest_height == 0 {
                Cow::Borrowed("Downloading headers")
            } else {
                let percent = sync_head.height * 100 / highest_height;
                Cow::Owned(format!("Downloading headers: {}%", percent))
            }
        }
        SyncStatus::TxHashsetDownload(stat) => {
            if stat.total_size > 0 {
                let percent = stat.downloaded_size * 100 / stat.total_size;
                Cow::Owned(format!("Downloading chain state: {}%", percent))
            } else {
                Cow::Borrowed("Downloading chain state")
            }
        }
        SyncStatus::TxHashsetSetup => {
            Cow::Borrowed("Preparing state for validation")
        }
        SyncStatus::TxHashsetRangeProofsValidation {
            rproofs,
            rproofs_total,
        } => {
            let r_percent = if rproofs_total > 0 {
                (rproofs * 100) / rproofs_total
            } else {
                0
            };
            Cow::Owned(format!("Validating state - range proofs: {}%", r_percent))
        }
        SyncStatus::TxHashsetKernelsValidation {
            kernels,
            kernels_total,
        } => {
            let k_percent = if kernels_total > 0 {
                (kernels * 100) / kernels_total
            } else {
                0
            };
            Cow::Owned(format!("Validating state - kernels: {}%", k_percent))
        }
        SyncStatus::TxHashsetSave => {
            Cow::Borrowed("Finalizing chain state")
        }
        SyncStatus::TxHashsetDone => {
            Cow::Borrowed("Finalized chain state")
        }
        SyncStatus::BodySync {
            current_height,
            highest_height,
        } => {
            if highest_height == 0 {
                Cow::Borrowed("Downloading blocks data")
            } else {
                Cow::Owned(format!(
                    "Downloading blocks: {}%",
                    current_height * 100 / highest_height
                ))
            }
        }
        SyncStatus::Shutdown => Cow::Borrowed("Shutting down"),
    }
}