// 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::{Id, RichText, TextStyle, Widget}; use crate::gui::Colors; use crate::gui::icons::{BARBELL, HARD_DRIVES, PLUG, POWER, TIMER}; use crate::gui::platform::PlatformCallbacks; use crate::gui::views::{Modal, View}; use crate::gui::views::network::settings::NetworkSettings; use crate::gui::views::types::{ModalContainer, ModalPosition}; use crate::node::{Node, NodeConfig}; /// Stratum server setup section content. pub struct StratumSetup { /// IP Addresses available at system. available_ips: Vec, /// Stratum port value. stratum_port_edit: String, /// Flag to check if stratum port is available. stratum_port_available_edit: bool, /// Flag to check if stratum port from saved config value is available. is_port_available: bool, /// Attempt time value in seconds to mine on a particular header. attempt_time_edit: String, /// Minimum share difficulty value to request from miners. min_share_diff_edit: String, /// [`Modal`] identifiers allowed at this ui container. modal_ids: Vec<&'static str> } /// Identifier for stratum port [`Modal`]. const STRATUM_PORT_MODAL: &'static str = "stratum_port"; /// Identifier for attempt time [`Modal`]. const ATTEMPT_TIME_MODAL: &'static str = "stratum_attempt_time"; /// Identifier for minimum share difficulty [`Modal`]. const MIN_SHARE_DIFF_MODAL: &'static str = "stratum_min_share_diff"; impl Default for StratumSetup { fn default() -> Self { let (ip, port) = NodeConfig::get_stratum_address(); let is_port_available = NodeConfig::is_stratum_port_available(&ip, &port); Self { available_ips: NodeConfig::get_ip_addrs(), stratum_port_edit: port, stratum_port_available_edit: is_port_available, is_port_available, attempt_time_edit: NodeConfig::get_stratum_attempt_time(), min_share_diff_edit: NodeConfig::get_stratum_min_share_diff(), modal_ids: vec![ STRATUM_PORT_MODAL, ATTEMPT_TIME_MODAL, MIN_SHARE_DIFF_MODAL ] } } } impl ModalContainer for StratumSetup { fn modal_ids(&self) -> &Vec<&'static str> { &self.modal_ids } fn modal_ui(&mut self, ui: &mut egui::Ui, _: &mut eframe::Frame, modal: &Modal, cb: &dyn PlatformCallbacks) { match modal.id { STRATUM_PORT_MODAL => self.port_modal(ui, modal, cb), ATTEMPT_TIME_MODAL => self.attempt_modal(ui, modal, cb), MIN_SHARE_DIFF_MODAL => self.min_diff_modal(ui, modal, cb), _ => {} } } } impl StratumSetup { pub fn ui(&mut self, ui: &mut egui::Ui, frame: &mut eframe::Frame, cb: &dyn PlatformCallbacks) { // Draw modal content for current ui container. self.current_modal_ui(ui, frame, cb); View::sub_title(ui, format!("{} {}", HARD_DRIVES, t!("network_mining.server"))); View::horizontal_line(ui, Colors::STROKE); ui.add_space(6.0); ui.vertical_centered(|ui| { // Show loading indicator or controls to start/stop stratum server if port is available. if self.is_port_available { if Node::is_stratum_starting() || Node::is_stratum_stopping() { ui.vertical_centered(|ui| { ui.add_space(8.0); View::small_loading_spinner(ui); ui.add_space(8.0); }); } else if Node::get_stratum_stats().is_running { ui.add_space(6.0); let disable_text = format!("{} {}", POWER, t!("network_settings.disable")); View::button(ui, disable_text, Colors::GOLD, || { Node::stop_stratum(); }); ui.add_space(6.0); } else { ui.add_space(6.0); let enable_text = format!("{} {}", POWER, t!("network_settings.enable")); View::button(ui, enable_text, Colors::GOLD, || { Node::start_stratum(); }); ui.add_space(6.0); } } // Show stratum server autorun checkbox. let stratum_enabled = NodeConfig::is_stratum_autorun_enabled(); View::checkbox(ui, stratum_enabled, t!("network.autorun"), || { NodeConfig::toggle_stratum_autorun(); }); // Show reminder to restart running server. if Node::get_stratum_stats().is_running { ui.add_space(2.0); ui.label(RichText::new(t!("network_mining.restart_server_required")) .size(16.0) .color(Colors::INACTIVE_TEXT) ); } ui.add_space(8.0); }); View::horizontal_line(ui, Colors::ITEM_STROKE); ui.add_space(6.0); // Show message when IP addresses are not available on the system. if self.available_ips.is_empty() { NetworkSettings::no_ip_address_ui(ui); return; } ui.vertical_centered(|ui| { ui.label(RichText::new(t!("network_settings.stratum_ip")) .size(16.0) .color(Colors::GRAY) ); ui.add_space(6.0); // Show stratum IP addresses to select. let (ip, port) = NodeConfig::get_stratum_address(); NetworkSettings::ip_addrs_ui(ui, &ip, &self.available_ips, |selected_ip| { NodeConfig::save_stratum_address(selected_ip, &port); self.is_port_available = NodeConfig::is_stratum_port_available(selected_ip, &port); }); // Show stratum port setup. self.port_setup_ui(ui, cb); View::horizontal_line(ui, Colors::ITEM_STROKE); ui.add_space(6.0); // Show attempt time setup. self.attempt_time_ui(ui, cb); View::horizontal_line(ui, Colors::ITEM_STROKE); ui.add_space(6.0); // Show minimum acceptable share difficulty setup. self.min_diff_ui(ui, cb); }); } /// Draw stratum port value setup content. fn port_setup_ui(&mut self, ui: &mut egui::Ui, cb: &dyn PlatformCallbacks) { ui.label(RichText::new(t!("network_settings.stratum_port")) .size(16.0) .color(Colors::GRAY) ); ui.add_space(6.0); let (_, port) = NodeConfig::get_stratum_address(); View::button(ui, format!("{} {}", PLUG, port.clone()), Colors::BUTTON, || { // Setup values for modal. self.stratum_port_edit = port; self.stratum_port_available_edit = self.is_port_available; // Show stratum port modal. Modal::new(STRATUM_PORT_MODAL) .position(ModalPosition::CenterTop) .title(t!("network_settings.change_value")) .show(); cb.show_keyboard(); }); ui.add_space(12.0); // Show error when stratum server port is unavailable. if !self.is_port_available { ui.add_space(6.0); ui.label(RichText::new(t!("network_settings.port_unavailable")) .size(16.0) .color(Colors::RED)); ui.add_space(12.0); } } /// Draw stratum port [`Modal`] content. fn port_modal(&mut self, ui: &mut egui::Ui, modal: &Modal, cb: &dyn PlatformCallbacks) { ui.add_space(6.0); ui.vertical_centered(|ui| { ui.label(RichText::new(t!("network_settings.stratum_port")) .size(17.0) .color(Colors::GRAY)); ui.add_space(8.0); // Draw stratum port text edit. let text_edit_resp = egui::TextEdit::singleline(&mut self.stratum_port_edit) .id(Id::from(modal.id)) .font(TextStyle::Heading) .desired_width(64.0) .cursor_at_end(true) .ui(ui); text_edit_resp.request_focus(); if text_edit_resp.clicked() { cb.show_keyboard(); } // Show error when specified port is unavailable. if !self.stratum_port_available_edit { ui.add_space(12.0); ui.label(RichText::new(t!("network_settings.port_unavailable")) .size(17.0) .color(Colors::RED)); } else { server_restart_required_ui(ui); } ui.add_space(12.0); // Show modal buttons. ui.scope(|ui| { // Setup spacing between buttons. ui.spacing_mut().item_spacing = egui::Vec2::new(8.0, 0.0); // Save button callback. let on_save = || { // Check if port is available. let (stratum_ip, _) = NodeConfig::get_stratum_address(); let available = NodeConfig::is_stratum_port_available( &stratum_ip, &self.stratum_port_edit ); self.stratum_port_available_edit = available; // Save port at config if it's available. if available { NodeConfig::save_stratum_address(&stratum_ip, &self.stratum_port_edit); self.is_port_available = true; cb.hide_keyboard(); modal.close(); } }; ui.columns(2, |columns| { columns[0].vertical_centered_justified(|ui| { View::button(ui, t!("modal.cancel"), Colors::WHITE, || { // Close modal. cb.hide_keyboard(); modal.close(); }); }); columns[1].vertical_centered_justified(|ui| { View::button(ui, t!("modal.save"), Colors::WHITE, on_save); }); }); ui.add_space(6.0); }); }); } /// Draw attempt time value setup content. fn attempt_time_ui(&mut self, ui: &mut egui::Ui, cb: &dyn PlatformCallbacks) { ui.label(RichText::new(t!("network_settings.attempt_time")) .size(16.0) .color(Colors::GRAY) ); ui.add_space(6.0); let time = NodeConfig::get_stratum_attempt_time(); View::button(ui, format!("{} {}", TIMER, time.clone()), Colors::BUTTON, || { // Setup values for modal. self.attempt_time_edit = time; // Show attempt time modal. Modal::new(ATTEMPT_TIME_MODAL) .position(ModalPosition::CenterTop) .title(t!("network_settings.change_value")) .show(); cb.show_keyboard(); }); ui.add_space(6.0); ui.label(RichText::new(t!("network_settings.attempt_time_desc")) .size(16.0) .color(Colors::INACTIVE_TEXT) ); ui.add_space(6.0); } /// Draw attempt time [`Modal`] content. fn attempt_modal(&mut self, ui: &mut egui::Ui, modal: &Modal, cb: &dyn PlatformCallbacks) { ui.add_space(6.0); ui.vertical_centered(|ui| { ui.label(RichText::new(t!("network_settings.attempt_time")) .size(17.0) .color(Colors::GRAY)); ui.add_space(8.0); // Draw attempt time text edit. let text_edit_resp = egui::TextEdit::singleline(&mut self.attempt_time_edit) .id(Id::from(modal.id)) .font(TextStyle::Heading) .desired_width(42.0) .cursor_at_end(true) .ui(ui); text_edit_resp.request_focus(); if text_edit_resp.clicked() { cb.show_keyboard(); } // Show error when specified value is not valid or reminder to restart enabled node. if self.attempt_time_edit.parse::().is_err() { ui.add_space(12.0); ui.label(RichText::new(t!("network_settings.not_valid_value")) .size(17.0) .color(Colors::RED)); } else { server_restart_required_ui(ui); } ui.add_space(12.0); }); // Show modal buttons. ui.scope(|ui| { // Setup spacing between buttons. ui.spacing_mut().item_spacing = egui::Vec2::new(8.0, 0.0); // Save button callback. let on_save = || { if let Ok(time) = self.attempt_time_edit.parse::() { NodeConfig::save_stratum_attempt_time(time); cb.hide_keyboard(); modal.close(); } }; ui.columns(2, |columns| { columns[0].vertical_centered_justified(|ui| { View::button(ui, t!("modal.cancel"), Colors::WHITE, || { // Close modal. cb.hide_keyboard(); modal.close(); }); }); columns[1].vertical_centered_justified(|ui| { View::button(ui, t!("modal.save"), Colors::WHITE, on_save); }); }); ui.add_space(6.0); }); } /// Draw minimum share difficulty value setup content. fn min_diff_ui(&mut self, ui: &mut egui::Ui, cb: &dyn PlatformCallbacks) { ui.label(RichText::new(t!("network_settings.min_share_diff")) .size(16.0) .color(Colors::GRAY) ); ui.add_space(6.0); let diff = NodeConfig::get_stratum_min_share_diff(); View::button(ui, format!("{} {}", BARBELL, diff.clone()), Colors::BUTTON, || { // Setup values for modal. self.min_share_diff_edit = diff; // Show share difficulty setup modal. Modal::new(MIN_SHARE_DIFF_MODAL) .position(ModalPosition::CenterTop) .title(t!("network_settings.change_value")) .show(); cb.show_keyboard(); }); ui.add_space(6.0); } /// Draw minimum acceptable share difficulty [`Modal`] content. fn min_diff_modal(&mut self, ui: &mut egui::Ui, modal: &Modal, cb: &dyn PlatformCallbacks) { ui.add_space(6.0); ui.vertical_centered(|ui| { ui.label(RichText::new(t!("network_settings.min_share_diff")) .size(17.0) .color(Colors::GRAY)); ui.add_space(8.0); // Draw share difficulty text edit. let text_edit_resp = egui::TextEdit::singleline(&mut self.min_share_diff_edit) .id(Id::from(modal.id)) .font(TextStyle::Heading) .desired_width(42.0) .cursor_at_end(true) .ui(ui); text_edit_resp.request_focus(); if text_edit_resp.clicked() { cb.show_keyboard(); } // Show error when specified value is not valid or reminder to restart enabled node. if self.min_share_diff_edit.parse::().is_err() { ui.add_space(12.0); ui.label(RichText::new(t!("network_settings.not_valid_value")) .size(17.0) .color(Colors::RED)); } else { server_restart_required_ui(ui); } ui.add_space(12.0); }); // Show modal buttons. ui.scope(|ui| { // Setup spacing between buttons. ui.spacing_mut().item_spacing = egui::Vec2::new(8.0, 0.0); // Save button callback. let on_save = || { if let Ok(diff) = self.min_share_diff_edit.parse::() { NodeConfig::save_stratum_min_share_diff(diff); cb.hide_keyboard(); modal.close(); } }; ui.columns(2, |columns| { columns[0].vertical_centered_justified(|ui| { View::button(ui, t!("modal.cancel"), Colors::WHITE, || { // Close modal. cb.hide_keyboard(); modal.close(); }); }); columns[1].vertical_centered_justified(|ui| { View::button(ui, t!("modal.save"), Colors::WHITE, on_save); }); }); ui.add_space(6.0); }); } } /// Reminder to restart enabled node to show on edit setting at [`Modal`]. pub fn server_restart_required_ui(ui: &mut egui::Ui) { if Node::get_stratum_stats().is_running { ui.add_space(12.0); ui.label(RichText::new(t!("network_mining.restart_server_required")) .size(16.0) .color(Colors::GREEN) ); } }