// 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::time::Duration; use egui::{Align, Layout, Margin, RichText, Rounding}; use grin_chain::SyncStatus; use grin_core::core::amount_to_hr_string; use crate::AppConfig; use crate::gui::Colors; use crate::gui::icons::{DOWNLOAD, FILE_ARCHIVE, GEAR_FINE, LIST, PACKAGE, PLUS, POWER, REPEAT, UPLOAD, WALLET}; use crate::gui::platform::PlatformCallbacks; use crate::gui::views::{Root, View}; use crate::gui::views::wallets::{WalletInfo, WalletReceive, WalletSend, WalletSettings}; use crate::gui::views::wallets::types::{WalletTab, WalletTabType}; use crate::node::Node; use crate::wallet::types::WalletData; use crate::wallet::Wallet; /// Selected and opened wallet content. pub struct WalletContent { /// Current tab content to show. pub current_tab: Box, } impl Default for WalletContent { fn default() -> Self { Self { current_tab: Box::new(WalletInfo::default()) } } } impl WalletContent { pub fn ui(&mut self, ui: &mut egui::Ui, frame: &mut eframe::Frame, wallet: &mut Wallet, cb: &dyn PlatformCallbacks) { let data = wallet.get_data(); let data_empty = data.is_none(); // Show wallet balance panel when data is not empty and current tab is not Settings. let show_balance = self.current_tab.get_type() != WalletTabType::Settings && !data_empty; egui::TopBottomPanel::top("wallet_balance") .frame(egui::Frame { fill: Colors::FILL, stroke: View::DEFAULT_STROKE, inner_margin: Margin { left: View::far_left_inset_margin(ui) + 4.0, right: View::get_right_inset() + 4.0, top: 4.0, bottom: 0.0, }, outer_margin: Margin { left: 0.0, right: 0.0, top: 0.0, bottom: -1.0, }, ..Default::default() }) .show_animated_inside(ui, show_balance, |ui| { ui.vertical_centered(|ui| { // Draw wallet tabs. View::max_width_ui(ui, Root::SIDE_PANEL_WIDTH * 1.35, |ui| { Self::account_ui(ui, data.as_ref().unwrap(), &wallet.config.account); }); }); }); // Show wallet tabs panel. egui::TopBottomPanel::bottom("wallet_tabs") .frame(egui::Frame { stroke: View::DEFAULT_STROKE, fill: Colors::FILL, inner_margin: Margin { left: View::far_left_inset_margin(ui) + 4.0, right: View::get_right_inset() + 4.0, top: 4.0, bottom: View::get_bottom_inset() + 4.0, }, ..Default::default() }) .show_animated_inside(ui, !Self::block_navigation_on_sync(wallet), |ui| { ui.vertical_centered(|ui| { // Draw wallet tabs. View::max_width_ui(ui, Root::SIDE_PANEL_WIDTH * 1.3, |ui| { self.tabs_ui(ui); }); }); }); // Show tab content panel. egui::CentralPanel::default() .frame(egui::Frame { fill: Colors::WHITE, stroke: View::DEFAULT_STROKE, ..Default::default() }) .show_inside(ui, |ui| { self.current_tab.ui(ui, frame, wallet, cb); }); // Refresh content after 1 second for synced wallet. if !data_empty { ui.ctx().request_repaint_after(Duration::from_millis(1000)); } else { ui.ctx().request_repaint(); } } /// Draw wallet account content. fn account_ui(ui: &mut egui::Ui, data: &WalletData, account: &Option) { let mut rect = ui.available_rect_before_wrap(); rect.set_height(75.0); // Draw round background. let rounding = View::item_rounding(0, 2, false); ui.painter().rect(rect, rounding, Colors::BUTTON, View::ITEM_STROKE); ui.allocate_ui_with_layout(rect.size(), Layout::right_to_left(Align::Center), |ui| { // Setup padding for item buttons. ui.style_mut().spacing.button_padding = egui::vec2(14.0, 0.0); // Draw button to add new account. View::item_button(ui, View::item_rounding(0, 2, true), PLUS, None, || { //TODO add account modal. }); // Draw button to show list of accounts. View::item_button(ui, Rounding::none(), LIST, None, || { //TODO: accounts list modal }); let layout_size = ui.available_size(); ui.allocate_ui_with_layout(layout_size, Layout::left_to_right(Align::Center), |ui| { ui.add_space(6.0); ui.vertical(|ui| { ui.add_space(3.0); // Show spendable amount. let amount = amount_to_hr_string(data.info.amount_currently_spendable, false); let amount_text = format!("{} ツ", amount); ui.label(RichText::new(amount_text).size(18.0).color(Colors::BLACK)); // Show account name. let account_name = match account { None => t!("wallets.default_account"), Some(name) => name.to_owned() }; let account_text = format!("{} {}", FILE_ARCHIVE, account_name); ui.add_space(-2.0); View::ellipsize_text(ui, account_text, 15.0, Colors::TEXT); // Show confirmed height. let height_text = format!("{} {}", PACKAGE, data.info.last_confirmed_height); ui.label(RichText::new(height_text).size(15.0).color(Colors::GRAY)); }) }); }); } /// Draw tab buttons in the bottom of the screen. fn tabs_ui(&mut self, ui: &mut egui::Ui) { ui.scope(|ui| { // Setup spacing between tabs. ui.style_mut().spacing.item_spacing = egui::vec2(4.0, 0.0); // Setup vertical padding inside tab button. ui.style_mut().spacing.button_padding = egui::vec2(0.0, 8.0); // Draw tab buttons. let current_type = self.current_tab.get_type(); ui.columns(4, |columns| { columns[0].vertical_centered_justified(|ui| { View::tab_button(ui, WALLET, current_type == WalletTabType::Info, || { self.current_tab = Box::new(WalletInfo::default()); }); }); columns[1].vertical_centered_justified(|ui| { View::tab_button(ui, DOWNLOAD, current_type == WalletTabType::Receive, || { self.current_tab = Box::new(WalletReceive::default()); }); }); columns[2].vertical_centered_justified(|ui| { View::tab_button(ui, UPLOAD, current_type == WalletTabType::Send, || { self.current_tab = Box::new(WalletSend::default()); }); }); columns[3].vertical_centered_justified(|ui| { View::tab_button(ui, GEAR_FINE, current_type == WalletTabType::Settings, || { self.current_tab = Box::new(WalletSettings::default()); }); }); }); }); } /// Draw content when wallet is syncing and not ready to use, returns `true` at this case. pub fn sync_ui(ui: &mut egui::Ui, frame: &mut eframe::Frame, wallet: &Wallet) -> bool { if wallet.is_repairing() && !wallet.sync_error() { Self::sync_progress_ui(ui, wallet); return true; } else if wallet.is_closing() { Self::sync_progress_ui(ui, wallet); return true; } else if wallet.get_current_ext_conn_id().is_none() { if !Node::is_running() || Node::is_stopping() { let dual_panel_root = Root::is_dual_panel_mode(frame); View::center_content(ui, 108.0, |ui| { let text = t!("wallets.enable_node", "settings" => GEAR_FINE); ui.label(RichText::new(text).size(16.0).color(Colors::INACTIVE_TEXT)); ui.add_space(8.0); // Show button to enable integrated node at non-dual root panel mode // or when network connections are not showing and node is not stopping if (!dual_panel_root || AppConfig::show_connections_network_panel()) && !Node::is_stopping() { let enable_node_text = format!("{} {}", POWER, t!("network.enable_node")); View::button(ui, enable_node_text, Colors::GOLD, || { Node::start(); }); } }); return true } else if wallet.sync_error() && Node::get_sync_status() == Some(SyncStatus::NoSync) { Self::sync_error_ui(ui, wallet); return true; } else if wallet.get_data().is_none() { Self::sync_progress_ui(ui, wallet); return true; } } else if wallet.sync_error() { Self::sync_error_ui(ui, wallet); return true; } else if wallet.get_data().is_none() { Self::sync_progress_ui(ui, wallet); return true; } false } /// Draw wallet sync error content. fn sync_error_ui(ui: &mut egui::Ui, wallet: &Wallet) { View::center_content(ui, 108.0, |ui| { let text = t!("wallets.wallet_loading_err", "settings" => GEAR_FINE); ui.label(RichText::new(text).size(16.0).color(Colors::INACTIVE_TEXT)); ui.add_space(8.0); let retry_text = format!("{} {}", REPEAT, t!("retry")); View::button(ui, retry_text, Colors::GOLD, || { wallet.retry_sync(); }); }); } /// Check when to block tabs navigation on sync progress. pub fn block_navigation_on_sync(wallet: &Wallet) -> bool { let sync_error = wallet.sync_error(); let integrated_node = wallet.get_current_ext_conn_id().is_none(); let integrated_node_ready = Node::get_sync_status() == Some(SyncStatus::NoSync); let sync_after_opening = wallet.get_data().is_none() && !wallet.sync_error(); // Block navigation if wallet is repairing and integrated node is not launching // and if wallet is closing or syncing after opening when there is no data to show. (wallet.is_repairing() && (integrated_node_ready || !integrated_node) && !sync_error) || wallet.is_closing() || (sync_after_opening && !integrated_node) } /// Draw wallet sync progress content. pub fn sync_progress_ui(ui: &mut egui::Ui, wallet: &Wallet) { View::center_content(ui, 162.0, |ui| { View::big_loading_spinner(ui); ui.add_space(18.0); // Setup sync progress text. let text = { let integrated_node = wallet.get_current_ext_conn_id().is_none(); let integrated_node_ready = Node::get_sync_status() == Some(SyncStatus::NoSync); let info_progress = wallet.info_sync_progress(); if wallet.is_closing() { t!("wallets.wallet_closing") } else if integrated_node && !integrated_node_ready { t!("wallets.node_loading", "settings" => GEAR_FINE) } else if wallet.is_repairing() { let repair_progress = wallet.repairing_progress(); if repair_progress == 0 { t!("wallets.wallet_checking") } else { format!("{}: {}%", t!("wallets.wallet_checking"), repair_progress) } } else if info_progress != 100 { if info_progress == 0 { t!("wallets.wallet_loading") } else { format!("{}: {}%", t!("wallets.wallet_loading"), info_progress) } } else { let tx_progress = wallet.txs_sync_progress(); if tx_progress == 0 { t!("wallets.tx_loading") } else { format!("{}: {}%", t!("wallets.tx_loading"), tx_progress) } } }; ui.label(RichText::new(text).size(16.0).color(Colors::INACTIVE_TEXT)); }); } }