grim/src/gui/views/network/content.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 egui::scroll_area::ScrollBarVisibility;
use egui::{Id, Margin, RichText, ScrollArea};

use crate::gui::icons::{ARROWS_COUNTER_CLOCKWISE, ARROW_LEFT, BRIEFCASE, DATABASE, DOTS_THREE_OUTLINE_VERTICAL, FACTORY, FADERS, GAUGE, GEAR, GLOBE, POWER};
use crate::gui::platform::PlatformCallbacks;
use crate::gui::views::network::types::{NodeTab, NodeTabType};
use crate::gui::views::network::{ConnectionsContent, NetworkMetrics, NetworkMining, NetworkNode, NetworkSettings};
use crate::gui::views::settings::SettingsContent;
use crate::gui::views::types::{ContentContainer, LinePosition, TitleContentType, TitleType};
use crate::gui::views::{Content, TitlePanel, View};
use crate::gui::Colors;
use crate::node::{Node, NodeConfig, NodeError};
use crate::AppConfig;

/// Network content.
pub struct NetworkContent {
    /// Current integrated node tab content.
    node_tab_content: Box<dyn NodeTab>,
    /// Connections content.
    connections: ConnectionsContent,

    /// Application settings content.
    settings_content: Option<SettingsContent>,
}

impl Default for NetworkContent {
    fn default() -> Self {
        Self {
            node_tab_content: Box::new(NetworkNode::default()),
            connections: ConnectionsContent::default(),
            settings_content: None,
        }
    }
}

impl NetworkContent {
    pub fn ui(&mut self, ui: &mut egui::Ui, cb: &dyn PlatformCallbacks) {
        let show_settings = self.showing_settings();
        let show_connections = AppConfig::show_connections_network_panel();
        let dual_panel = Content::is_dual_panel_mode(ui.ctx());

        // Show title panel.
        self.title_ui(ui, dual_panel, show_connections);

        // Show integrated node tabs content.
        if !show_connections && !show_settings {
            let side_padding = View::TAB_ITEMS_PADDING + if View::is_desktop() {
                0.0
            } else {
                4.0
            };
            let tabs_margin = Margin {
                left: (View::get_left_inset() + side_padding) as i8,
                right: (View::far_right_inset_margin(ui) + side_padding) as i8,
                top: View::TAB_ITEMS_PADDING as i8,
                bottom: (View::get_bottom_inset() + View::TAB_ITEMS_PADDING) as i8,
            };
            egui::TopBottomPanel::bottom("network_tabs_content")
                .min_height(0.5)
                .resizable(false)
                .frame(egui::Frame {
                    inner_margin: tabs_margin,
                    fill: Colors::fill(),
                    ..Default::default()
                })
                .show_inside(ui, |ui| {
                    let rect = ui.available_rect_before_wrap();
                    View::max_width_ui(ui, Content::SIDE_PANEL_WIDTH * 1.3, |ui| {
                        self.tabs_ui(ui);
                    });
                    // Draw content divider line.
                    let r = {
                        let mut r = rect.clone();
                        r.min.x -= tabs_margin.left as f32;
                        r.min.y -= tabs_margin.top as f32;
                        r.max.x += tabs_margin.right as f32;
                        r
                    };
                    View::line(ui, LinePosition::TOP, &r, Colors::stroke());
                });
        }

        // Show settings or integrated node content.
        egui::SidePanel::right("network_side_content")
            .resizable(false)
            .exact_width(ui.available_width())
            .frame(egui::Frame {
                ..Default::default()
            })
            .show_animated_inside(ui, show_settings || !show_connections, |ui| {
                egui::CentralPanel::default()
                    .frame(egui::Frame {
                        inner_margin: Margin {
                            left: (View::get_left_inset() + View::content_padding()) as i8,
                            right: (View::far_right_inset_margin(ui) +
                                View::content_padding()) as i8,
                            top: 3.0 as i8,
                            bottom: 4.0 as i8,
                        },
                        ..Default::default()
                    })
                    .show_inside(ui, |ui| {
                        let rect = ui.available_rect_before_wrap();
                        if let Some(c) = &mut self.settings_content {
                            ScrollArea::vertical()
                                .id_salt("app_settings_network")
                                .scroll_bar_visibility(ScrollBarVisibility::AlwaysHidden)
                                .auto_shrink([false; 2])
                                .show(ui, |ui| {
                                    ui.add_space(1.0);
                                    ui.vertical_centered(|ui| {
                                        // Show application settings content.
                                        View::max_width_ui(ui,
                                                           Content::SIDE_PANEL_WIDTH * 1.3,
                                                           |ui| {
                                                               c.ui(ui, cb);
                                                           });
                                    });
                                });
                        } else if self.node_tab_content.get_type() != NodeTabType::Settings {
                            View::max_width_ui(ui, Content::SIDE_PANEL_WIDTH * 1.3, |ui| {
                                let node_err = Node::get_error();
                                if let Some(err) = node_err {
                                    node_error_ui(ui, err);
                                } else if !Node::is_running() {
                                    disabled_node_ui(ui);
                                } else if Node::get_stats().is_none() || Node::is_restarting() ||
                                    Node::is_stopping() {
                                    NetworkContent::loading_ui(ui, None);
                                } else {
                                    self.node_tab_content.tab_ui(ui, cb);
                                }
                            });
                        } else {
                            self.node_tab_content.tab_ui(ui, cb);
                        }
                        // Draw content divider line.
                        let r = {
                            let mut r = rect.clone();
                            r.min.y -= 3.0;
                            r.max.x += View::content_padding();
                            r.max.y += View::content_padding();
                            r
                        };
                        if dual_panel {
                            View::line(ui, LinePosition::RIGHT, &r, Colors::item_stroke());
                        }
                    });
            });

        // Show connections content.
        egui::CentralPanel::default()
            .frame(egui::Frame {
                inner_margin: Margin {
                    left: if show_connections {
                        View::get_left_inset() + View::content_padding()
                    } else {
                        0.0
                    } as i8,
                    right: if show_connections {
                        View::far_right_inset_margin(ui) + View::content_padding()
                    } else {
                        0.0
                    } as i8,
                    top: 3.0 as i8,
                    bottom: 0.0 as i8,
                },
                ..Default::default()
            })
            .show_inside(ui, |ui| {
                let rect = ui.available_rect_before_wrap();
                ScrollArea::vertical()
                    .id_salt("connections_scroll")
                    .scroll_bar_visibility(ScrollBarVisibility::AlwaysHidden)
                    .auto_shrink([false; 2])
                    .show(ui, |ui| {
                        ui.add_space(1.0);
                        ui.vertical_centered(|ui| {
                            let max_width = if !dual_panel {
                                Content::SIDE_PANEL_WIDTH * 1.3
                            } else {
                                ui.available_width()
                            };
                            View::max_width_ui(ui, max_width, |ui| {
                                self.connections.ui(ui, cb);
                            });
                        });
                        ui.add_space(32.0);
                    });
                // Draw content divider line.
                let r = {
                    let mut r = rect.clone();
                    r.min.y -= 3.0;
                    r.max.x += View::content_padding();
                    r.max.y += View::content_padding() + View::get_bottom_inset();
                    r
                };
                if show_connections && dual_panel {
                    View::line(ui, LinePosition::RIGHT, &r, Colors::item_stroke());
                }
            });

        // Redraw after delay if node is running at non-dual-panel mode.
        if ((!dual_panel && Content::is_network_panel_open()) || dual_panel) && Node::is_running() {
            ui.ctx().request_repaint_after(Node::STATS_UPDATE_DELAY);
        }
    }

    /// Navigate back, return `true` if action was not consumed.
    pub fn on_back(&mut self) -> bool {
        if self.showing_settings() {
            // Close settings.
            self.settings_content = None;
            return false;
        }
        true
    }

    /// Check if application settings content is showing.
    pub fn showing_settings(&self) -> bool {
        self.settings_content.is_some()
    }

    /// Draw tab buttons at bottom of the screen.
    fn tabs_ui(&mut self, ui: &mut egui::Ui) {
        ui.vertical_centered(|ui| {
            // Setup spacing between tabs.
            ui.style_mut().spacing.item_spacing = egui::vec2(View::TAB_ITEMS_PADDING, 0.0);

            // Draw tab buttons.
            let current_type = self.node_tab_content.get_type();
            ui.columns(4, |columns| {
                columns[0].vertical_centered_justified(|ui| {
                    let active = Some(current_type == NodeTabType::Info);
                    View::tab_button(ui, DATABASE, None, active, |_| {
                        self.node_tab_content = Box::new(NetworkNode::default());
                    });
                });
                columns[1].vertical_centered_justified(|ui| {
                    let active = Some(current_type == NodeTabType::Metrics);
                    View::tab_button(ui, GAUGE, None, active, |_| {
                        self.node_tab_content = Box::new(NetworkMetrics::default());
                    });
                });
                columns[2].vertical_centered_justified(|ui| {
                    let active = Some(current_type == NodeTabType::Mining);
                    View::tab_button(ui, FACTORY, None, active, |_| {
                        self.node_tab_content = Box::new(NetworkMining::default());
                    });
                });
                columns[3].vertical_centered_justified(|ui| {
                    let active = Some(current_type == NodeTabType::Settings);
                    View::tab_button(ui, FADERS, None, active, |_| {
                        self.node_tab_content = Box::new(NetworkSettings::default());
                    });
                });
            });
        });
    }

    /// Draw title content.
    fn title_ui(&mut self, ui: &mut egui::Ui, dual_panel: bool, show_connections: bool) {
        let show_settings = self.showing_settings();

        // Setup values for title panel.
        let title_text = self.node_tab_content.get_type().title();
        let subtitle_text = Node::get_sync_status_text();
        let not_syncing = Node::not_syncing() && !Node::data_dir_changing();
        let title_content = if show_settings {
            TitleContentType::Title(t!("settings"))
        } else if !show_connections {
            TitleContentType::WithSubTitle(title_text, subtitle_text, !not_syncing)
        } else {
            TitleContentType::Title(t!("network.connections"))
        };

        // Draw title panel.
        TitlePanel::new(Id::from("network_title_panel")).ui(TitleType::Single(title_content), |ui| {
            if show_settings {
                View::title_button_big(ui, ARROW_LEFT, |_| {
                    self.settings_content = None;
                });
            } else if !show_connections {
                View::title_button_big(ui, GLOBE, |_| {
                    AppConfig::toggle_show_connections_network_panel();
                });
            } else if !dual_panel {
                View::title_button_big(ui, GEAR, |_| {
                    self.settings_content = Some(SettingsContent::default());
                });
            }
        }, |ui| {
            if !dual_panel && !show_settings {
                View::title_button_big(ui, BRIEFCASE, |_| {
                    Content::toggle_network_panel();
                });
            }
        }, ui);
    }

    /// Content to draw on loading.
    pub fn loading_ui(ui: &mut egui::Ui, text: Option<String>) {
        match text {
            None => {
                ui.centered_and_justified(|ui| {
                    View::big_loading_spinner(ui);
                });
            }
            Some(t) => {
                View::center_content(ui, 162.0, |ui| {
                    View::big_loading_spinner(ui);
                    ui.add_space(18.0);
                    ui.label(RichText::new(t)
                        .size(16.0)
                        .color(Colors::inactive_text())
                    );
                });
            }
        }
    }

    /// Draw checkbox to run integrated node on application launch.
    pub fn autorun_node_ui(ui: &mut egui::Ui) {
        let autostart = AppConfig::autostart_node();
        View::checkbox(ui, autostart, t!("network.autorun"), || {
            AppConfig::toggle_node_autostart();
        });
    }
}

/// Content to draw when node is disabled.
fn disabled_node_ui(ui: &mut egui::Ui) {
    View::center_content(ui, 156.0, |ui| {
        let text = t!("network.disabled_server", "dots" => DOTS_THREE_OUTLINE_VERTICAL);
        ui.label(RichText::new(text)
            .size(16.0)
            .color(Colors::inactive_text())
        );
        ui.add_space(8.0);
        View::action_button(ui, format!("{} {}", POWER, t!("network.enable_node")), || {
            Node::start();
        });
        ui.add_space(2.0);
        NetworkContent::autorun_node_ui(ui);
    });
}

/// Draw integrated node error content.
pub fn node_error_ui(ui: &mut egui::Ui, e: NodeError) {
    match e {
        NodeError::Storage => {
            View::center_content(ui, 156.0, |ui| {
                ui.label(RichText::new(t!("network_node.error_clean"))
                    .size(16.0)
                    .color(Colors::red())
                );
                ui.add_space(8.0);
                let btn_txt = format!("{} {}",
                                      ARROWS_COUNTER_CLOCKWISE,
                                      t!("network_node.resync"));
                View::action_button(ui, btn_txt, || {
                    Node::clean_up_data();
                    Node::start();
                });
                ui.add_space(2.0);
            });
            return;
        }
        NodeError::P2P | NodeError::API => {
            let msg_type = match e {
                NodeError::API => "API",
                _ => "P2P"
            };
            View::center_content(ui, 106.0, |ui| {
                let text = t!(
                        "network_node.error_p2p_api",
                        "p2p_api" => msg_type,
                        "settings" => FADERS
                    );
                ui.label(RichText::new(text)
                    .size(16.0)
                    .color(Colors::red())
                );
                ui.add_space(2.0);
            });
            return;
        }
        NodeError::Configuration => {
            View::center_content(ui, 106.0, |ui| {
                ui.label(RichText::new(t!("network_node.error_config", "settings" => FADERS))
                    .size(16.0)
                    .color(Colors::red())
                );
                ui.add_space(8.0);
                let btn_txt = format!("{} {}",
                                      ARROWS_COUNTER_CLOCKWISE,
                                      t!("network_settings.reset"));
                View::action_button(ui, btn_txt, || {
                    NodeConfig::reset_to_default();
                    Node::start();
                });
                ui.add_space(2.0);
            });
        }
        NodeError::Unknown => {
            View::center_content(ui, 156.0, |ui| {
                ui.label(RichText::new(t!("network_node.error_unknown", "settings" => FADERS))
                    .size(16.0)
                    .color(Colors::red())
                );
                ui.add_space(8.0);
                let btn_txt = format!("{} {}",
                                      ARROWS_COUNTER_CLOCKWISE,
                                      t!("network_node.resync"));
                View::action_button(ui, btn_txt, || {
                    Node::clean_up_data();
                    Node::start();
                });
                ui.add_space(2.0);
            });
        }
    }
}