// 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, Margin, RichText, ScrollArea}; use egui::scroll_area::ScrollBarVisibility; use grin_core::core::{amount_from_hr_string, amount_to_hr_string}; use grin_wallet_libwallet::{Slate, SlateState}; use log::error; use crate::gui::Colors; use crate::gui::icons::{BROOM, CLIPBOARD_TEXT, COPY, DOWNLOAD_SIMPLE, PROHIBIT, UPLOAD_SIMPLE}; use crate::gui::platform::PlatformCallbacks; use crate::gui::views::{Modal, Root, View}; use crate::gui::views::types::{ModalPosition, TextEditOptions}; use crate::gui::views::wallets::wallet::types::{SLATEPACK_MESSAGE_HINT, WalletTab, WalletTabType}; use crate::gui::views::wallets::wallet::WalletContent; use crate::wallet::types::WalletTransaction; use crate::wallet::Wallet; #[derive(Clone, Eq, PartialEq, Debug, thiserror::Error)] enum MessageError { #[error("{0}")] Response(String), #[error("{0}")] Parse(String), #[error("{0}")] Finalize(String), #[error("{0}")] Other(String), } impl MessageError { pub fn text(&self) -> &String { match self { MessageError::Response(text) => text, MessageError::Parse(text) => text, MessageError::Finalize(text) => text, MessageError::Other(text) => text } } } /// Slatepacks messages interaction tab content. pub struct WalletMessages { /// Slatepack message to create response message. message_edit: String, /// Parsed Slatepack message. message_slate: Option, /// Slatepack error on finalization, parse and response creation. message_error: Option, /// Generated Slatepack response message. response_edit: String, /// Flag to check if Dandelion is needed to finalize transaction. dandelion: bool, /// Flag to check if send or invoice request was opened for [`Modal`]. send_request: bool, /// Amount to send or receive at [`Modal`]. amount_edit: String, /// Generated Slatepack message as request to send or receive funds at [`Modal`]. request_edit: String, /// Flag to check if there is an error happened on request creation at [`Modal`]. request_error: Option, } /// Identifier for amount input [`Modal`]. const AMOUNT_MODAL: &'static str = "amount_modal"; impl WalletTab for WalletMessages { fn get_type(&self) -> WalletTabType { WalletTabType::Messages } fn ui(&mut self, ui: &mut egui::Ui, _: &mut eframe::Frame, wallet: &mut Wallet, cb: &dyn PlatformCallbacks) { if WalletContent::sync_ui(ui, wallet) { return; } // Show modal content for this ui container. self.modal_content_ui(ui, wallet, cb); // Show manual wallet content panel. egui::CentralPanel::default() .frame(egui::Frame { stroke: View::ITEM_STROKE, fill: Colors::WHITE, inner_margin: Margin { left: View::far_left_inset_margin(ui) + 4.0, right: View::get_right_inset() + 4.0, top: 3.0, bottom: 4.0, }, ..Default::default() }) .show_inside(ui, |ui| { ScrollArea::vertical() .scroll_bar_visibility(ScrollBarVisibility::AlwaysVisible) .id_source(Id::from("wallet_messages").with(wallet.get_config().id)) .auto_shrink([false; 2]) .show(ui, |ui| { ui.vertical_centered(|ui| { View::max_width_ui(ui, Root::SIDE_PANEL_WIDTH * 1.3, |ui| { self.ui(ui, wallet, cb); }); }); }); }); } } impl WalletMessages { /// Create new content instance, put message into input if provided. pub fn new(dandelion: bool, message: Option) -> Self { Self { send_request: false, message_edit: message.unwrap_or("".to_string()), message_slate: None, message_error: None, response_edit: "".to_string(), dandelion, amount_edit: "".to_string(), request_edit: "".to_string(), request_error: None, } } /// Draw manual wallet transaction interaction content. pub fn ui(&mut self, ui: &mut egui::Ui, wallet: &mut Wallet, cb: &dyn PlatformCallbacks) { ui.add_space(3.0); // Show creation of request to send or receive funds. self.request_ui(ui, wallet, cb); ui.add_space(12.0); View::horizontal_line(ui, Colors::ITEM_STROKE); ui.add_space(6.0); // Show Slatepack message input field. self.input_slatepack_ui(ui, wallet, cb); ui.add_space(6.0); } /// Draw [`Modal`] content for this ui container. fn modal_content_ui(&mut self, ui: &mut egui::Ui, wallet: &mut Wallet, cb: &dyn PlatformCallbacks) { match Modal::opened() { None => {} Some(id) => { match id { AMOUNT_MODAL => { Modal::ui(ui.ctx(), |ui, modal| { self.amount_modal_ui(ui, wallet, modal, cb); }); } _ => {} } } } } /// Draw creation of request to send or receive funds. fn request_ui(&mut self, ui: &mut egui::Ui, wallet: &Wallet, cb: &dyn PlatformCallbacks) { ui.label(RichText::new(t!("wallets.create_request_desc")) .size(16.0) .color(Colors::INACTIVE_TEXT)); ui.add_space(7.0); // Show send button only if balance is not empty. let data = wallet.get_data().unwrap(); if data.info.amount_currently_spendable > 0 { // Setup spacing between buttons. ui.spacing_mut().item_spacing = egui::Vec2::new(6.0, 0.0); ui.columns(2, |columns| { columns[0].vertical_centered_justified(|ui| { // Draw send request creation button. let send_text = format!("{} {}", UPLOAD_SIMPLE, t!("wallets.send")); View::button(ui, send_text, Colors::BUTTON, || { // Setup modal values. self.send_request = true; self.amount_edit = "".to_string(); self.request_error = None; // Show send amount modal. Modal::new(AMOUNT_MODAL) .position(ModalPosition::CenterTop) .title(t!("wallets.send")) .show(); cb.show_keyboard(); }); }); columns[1].vertical_centered_justified(|ui| { // Draw invoice request creation button. self.receive_button_ui(ui, cb); }); }); } else { // Draw invoice creation button. self.receive_button_ui(ui, cb); } } /// Draw invoice request creation button. fn receive_button_ui(&mut self, ui: &mut egui::Ui, cb: &dyn PlatformCallbacks) { let receive_text = format!("{} {}", DOWNLOAD_SIMPLE, t!("wallets.receive")); View::button(ui, receive_text, Colors::BUTTON, || { // Setup modal values. self.send_request = false; self.amount_edit = "".to_string(); self.request_error = None; // Show receive amount modal. Modal::new(AMOUNT_MODAL) .position(ModalPosition::CenterTop) .title(t!("wallets.receive")) .show(); cb.show_keyboard(); }); } /// Draw Slatepack message input content. fn input_slatepack_ui(&mut self, ui: &mut egui::Ui, wallet: &mut Wallet, cb: &dyn PlatformCallbacks) { // Setup description. let response_empty = self.response_edit.is_empty(); if let Some(err) = &self.message_error { ui.label(RichText::new(err.text()).size(16.0).color(Colors::RED)); } else { let desc_text = if self.message_slate.is_none() { t!("wallets.input_slatepack_desc") } else { let slate = self.message_slate.clone().unwrap(); let amount = amount_to_hr_string(slate.amount, true); match slate.state { SlateState::Standard1 => { t!("wallets.parse_s1_slatepack_desc","amount" => amount) } SlateState::Standard2 => { t!("wallets.parse_s2_slatepack_desc","amount" => amount) } SlateState::Standard3 => { t!("wallets.parse_s3_slatepack_desc","amount" => amount) } SlateState::Invoice1 => { t!("wallets.parse_i1_slatepack_desc","amount" => amount) } SlateState::Invoice2 => { t!("wallets.parse_i2_slatepack_desc","amount" => amount) } SlateState::Invoice3 => { t!("wallets.parse_i3_slatepack_desc","amount" => amount) } _ => { t!("wallets.input_slatepack_desc") } } }; ui.label(RichText::new(desc_text).size(16.0).color(Colors::INACTIVE_TEXT)); } ui.add_space(6.0); // Setup Slatepack message text input. let message = if response_empty { &mut self.message_edit } else { &mut self.response_edit }; // Save message to check for changes. let message_before = message.clone(); View::horizontal_line(ui, Colors::ITEM_STROKE); ui.add_space(3.0); ScrollArea::vertical() .max_height(128.0) .id_source(Id::from( if response_empty { "message_input" } else { "response_input" }).with(wallet.get_config().id)) .auto_shrink([false; 2]) .show(ui, |ui| { ui.add_space(7.0); egui::TextEdit::multiline(message) .font(egui::TextStyle::Small) .desired_rows(5) .interactive(response_empty) .hint_text(SLATEPACK_MESSAGE_HINT) .desired_width(f32::INFINITY) .show(ui); ui.add_space(6.0); }); ui.add_space(2.0); View::horizontal_line(ui, Colors::ITEM_STROKE); ui.add_space(10.0); // Parse Slatepack message if input field was changed, resetting message error. if &message_before != message { self.parse_message(wallet); } // Draw buttons to clear/copy/paste. let fields_empty = self.message_edit.is_empty() && self.response_edit.is_empty(); let columns_num = if fields_empty { 1 } else { 2 }; let mut show_dandelion = false; ui.scope(|ui| { // Setup spacing between buttons. ui.spacing_mut().item_spacing = egui::Vec2::new(6.0, 0.0); ui.columns(columns_num, |columns| { let first_column_content = |ui: &mut egui::Ui| { if self.message_slate.is_some() { if self.response_edit.is_empty() { let clear_text = format!("{} {}", BROOM, t!("clear")); View::button(ui, clear_text, Colors::BUTTON, || { self.message_edit.clear(); self.response_edit.clear(); self.message_error = None; self.message_slate = None; }); } else { let cancel = format!("{} {}", PROHIBIT, t!("modal.cancel")); View::colored_text_button(ui, cancel, Colors::RED, Colors::BUTTON, || { let slate = self.message_slate.clone().unwrap(); if let Some(tx) = wallet.tx_by_slate(&slate) { wallet.cancel(tx.data.id); self.message_edit.clear(); self.response_edit.clear(); self.message_slate = None; } }); } } else { let paste = format!("{} {}", CLIPBOARD_TEXT, t!("paste")); View::button(ui, paste, Colors::BUTTON, || { let buf = cb.get_string_from_buffer(); let previous = self.message_edit.clone(); self.message_edit = buf.clone(); // Parse Slatepack message resetting message error. if buf != previous { self.parse_message(wallet); } }); } }; if columns_num == 1 { columns[0].vertical_centered(first_column_content); } else { columns[0].vertical_centered_justified(first_column_content); columns[1].vertical_centered_justified(|ui| { if self.message_slate.is_some() { if !self.response_edit.is_empty() { let copy_text = format!("{} {}", COPY, t!("copy")); View::button(ui, copy_text, Colors::BUTTON, || { cb.copy_string_to_buffer(self.response_edit.clone()); self.message_edit.clear(); self.response_edit.clear(); self.message_slate = None; }); } else { show_dandelion = true; View::button(ui, t!("wallets.finalize"), Colors::GOLD, || { let slate = self.message_slate.clone().unwrap(); if slate.state == SlateState::Invoice3 || slate.state == SlateState::Standard3 { if wallet.post(&slate, self.dandelion).is_ok() { self.message_edit.clear(); self.message_slate = None; } else { self.message_error = Some( MessageError::Finalize( t!("wallets.finalize_slatepack_err") ) ); } } else { let r = wallet.finalize(&self.message_edit, self.dandelion); if r.is_ok() { self.message_edit.clear(); self.message_slate = None; } else { self.message_error = Some( MessageError::Finalize( t!("wallets.finalize_slatepack_err") ) ); } } }); } } else { let clear_text = format!("{} {}", BROOM, t!("clear")); View::button(ui, clear_text, Colors::BUTTON, || { self.message_error = None; self.message_edit.clear(); self.response_edit.clear(); self.message_slate = None; }); } }); } }); ui.add_space(12.0); // Draw clear button on response. if !self.response_edit.is_empty() && self.message_slate.is_some() { let clear_text = format!("{} {}", BROOM, t!("clear")); View::button(ui, clear_text, Colors::BUTTON, || { self.message_error = None; self.message_edit.clear(); self.response_edit.clear(); self.message_slate = None; }); } }); // Draw setup of ability to post transaction with Dandelion. if show_dandelion { let dandelion_before = self.dandelion; View::checkbox(ui, dandelion_before, t!("wallets.use_dandelion"), || { self.dandelion = !dandelion_before; wallet.update_use_dandelion(self.dandelion); }); } } /// Parse message input into [`Slate`] updating slate and response input. pub fn parse_message(&mut self, wallet: &mut Wallet) { self.message_error = None; if self.message_edit.is_empty() { return; } if let Ok(mut slate) = wallet.parse_slatepack(&self.message_edit) { println!("parse_message: {}", slate); // Try to setup empty amount from transaction by id. if slate.amount == 0 { let _ = wallet.get_data().unwrap().txs.clone().iter().map(|tx| { if tx.data.tx_slate_id == Some(slate.id) { if slate.amount == 0 { slate.amount = tx.amount; } } tx }).collect::>(); } if slate.amount == 0 { self.message_error = Some( MessageError::Response(t!("wallets.resp_slatepack_err")) ); return; } // Make operation based on incoming state status. match slate.state { SlateState::Standard1 | SlateState::Invoice1 => { let resp = if slate.state == SlateState::Standard1 { wallet.receive(&self.message_edit) } else { wallet.pay(&self.message_edit) }; if resp.is_ok() { self.response_edit = resp.unwrap(); } else { match resp.err().unwrap() { grin_wallet_libwallet::Error::TransactionWasCancelled {..} => { // Set already canceled transaction error message. self.message_error = Some( MessageError::Response(t!("wallets.resp_canceled_err")) ); return; } _ => {} } // Check if tx with same slate id already exists. let exists_tx = wallet.tx_by_slate(&slate).is_some(); if exists_tx { let mut sl = slate.clone(); sl.state = if sl.state == SlateState::Standard1 { SlateState::Standard2 } else { SlateState::Invoice2 }; match wallet.read_slatepack(&sl) { None => { self.message_error = Some( MessageError::Response(t!("wallets.resp_slatepack_err")) ); } Some(sp) => { self.message_slate = Some(slate); self.response_edit = sp; } } return; } // Set default response error message. self.message_error = Some( MessageError::Response(t!("wallets.resp_slatepack_err")) ); } } SlateState::Standard2 | SlateState::Invoice2 => { // Check if slatepack with same id and state already exists. let mut sl = slate.clone(); sl.state = if sl.state == SlateState::Standard2 { SlateState::Standard1 } else { SlateState::Invoice1 }; match wallet.read_slatepack(&sl) { None => { match wallet.read_slatepack(&slate) { None => { self.message_error = Some( MessageError::Response(t!("wallets.resp_slatepack_err")) ); } Some(sp) => { self.message_slate = Some(sl); self.response_edit = sp; return; } } } Some(_) => { self.message_slate = Some(slate.clone()); return; } } } _ => { self.response_edit = "".to_string(); } } self.message_slate = Some(slate); } else { self.message_slate = None; self.message_error = Some(MessageError::Parse(t!("wallets.resp_slatepack_err"))); } } /// Draw amount input [`Modal`] content to create invoice or request to send funds. fn amount_modal_ui(&mut self, ui: &mut egui::Ui, wallet: &mut Wallet, modal: &Modal, cb: &dyn PlatformCallbacks) { ui.add_space(6.0); if self.request_edit.is_empty() { ui.vertical_centered(|ui| { let enter_text = if self.send_request { let data = wallet.get_data().unwrap(); let amount = amount_to_hr_string(data.info.amount_currently_spendable, true); t!("wallets.enter_amount_send","amount" => amount) } else { t!("wallets.enter_amount_receive") }; ui.label(RichText::new(enter_text) .size(17.0) .color(Colors::GRAY)); }); ui.add_space(8.0); // Draw invoice amount text edit. let amount_edit_id = Id::from(modal.id).with(wallet.get_config().id); let amount_edit_opts = TextEditOptions::new(amount_edit_id).h_center(); let amount_edit_before = self.amount_edit.clone(); View::text_edit(ui, cb, &mut self.amount_edit, amount_edit_opts); // Check value if input was changed. if amount_edit_before != self.amount_edit { self.request_error = None; if !self.amount_edit.is_empty() { match amount_from_hr_string(self.amount_edit.as_str()) { Ok(a) => { if !self.amount_edit.contains(".") { // To avoid input of several "0". if a == 0 { self.amount_edit = "0".to_string(); return; } } else { // Check input after ".". let parts = self.amount_edit.split(".").collect::>(); if parts.len() == 2 && parts[1].len() > 9 { self.amount_edit = amount_edit_before; return; } } // Do not input amount more than balance in sending. if self.send_request { let b = wallet.get_data().unwrap().info.amount_currently_spendable; if b < a { self.amount_edit = amount_edit_before; } } } Err(_) => { self.amount_edit = amount_edit_before; } } } } // Show request creation error. if self.request_error.is_some() { ui.add_space(12.0); ui.vertical_centered(|ui| { ui.label(RichText::new(self.request_error.clone().unwrap().text()) .size(17.0) .color(Colors::RED)); }); } ui.add_space(12.0); // Setup spacing between buttons. ui.spacing_mut().item_spacing = egui::Vec2::new(6.0, 0.0); ui.columns(2, |columns| { columns[0].vertical_centered_justified(|ui| { View::button(ui, t!("modal.cancel"), Colors::WHITE, || { self.amount_edit = "".to_string(); self.request_error = None; cb.hide_keyboard(); modal.close(); }); }); columns[1].vertical_centered_justified(|ui| { // Button to create Slatepack message for request. View::button(ui, t!("continue"), Colors::WHITE, || { if self.amount_edit.is_empty() { return; } if let Ok(a) = amount_from_hr_string(self.amount_edit.as_str()) { let message = if self.send_request { wallet.send(a) } else { wallet.issue_invoice(a) }; match message { Ok((_, message)) => { self.request_edit = message; cb.hide_keyboard(); } Err(err) => { match err { grin_wallet_libwallet::Error::NotEnoughFunds { .. } => { let m = t!( "wallets.pay_balance_error", "amount" => self.amount_edit ); self.request_error = Some(MessageError::Other(m)); } _ => { let m = t!("wallets.invoice_slatepack_err"); self.request_error = Some(MessageError::Other(m)); } } } } } else { self.request_error = Some( MessageError::Other(t!("wallets.invoice_slatepack_err")) ); } }); }); }); ui.add_space(6.0); } else { ui.vertical_centered(|ui| { let amount = amount_from_hr_string(self.amount_edit.as_str()).unwrap(); let amount_format = amount_to_hr_string(amount, true); let desc_text = if self.send_request { t!("wallets.send_request_desc","amount" => amount_format) } else { t!("wallets.invoice_desc","amount" => amount_format) }; ui.label(RichText::new(desc_text).size(16.0).color(Colors::GRAY)); ui.add_space(6.0); View::horizontal_line(ui, Colors::ITEM_STROKE); ui.add_space(3.0); // Draw output Slatepack message text. let input_id = if self.send_request { Id::from("send_request_output").with(wallet.get_config().id) } else { Id::from("receive_request_output").with(wallet.get_config().id) }; ScrollArea::vertical() .max_height(128.0) .id_source(input_id) .auto_shrink([false; 2]) .show(ui, |ui| { ui.add_space(7.0); egui::TextEdit::multiline(&mut self.request_edit) .font(egui::TextStyle::Small) .desired_rows(5) .interactive(false) .hint_text(SLATEPACK_MESSAGE_HINT) .desired_width(f32::INFINITY) .show(ui); ui.add_space(6.0); }); ui.add_space(2.0); View::horizontal_line(ui, Colors::ITEM_STROKE); }); ui.add_space(12.0); // Setup spacing between buttons. ui.spacing_mut().item_spacing = egui::Vec2::new(6.0, 0.0); ui.columns(2, |columns| { columns[0].vertical_centered_justified(|ui| { // Button to cancel transaction. let cancel = format!("{} {}", PROHIBIT, t!("modal.cancel")); View::colored_text_button(ui, cancel, Colors::RED, Colors::BUTTON, || { if let Ok(slate) = wallet.parse_slatepack(&self.request_edit) { if let Some(tx) = wallet.tx_by_slate(&slate) { wallet.cancel(tx.data.id); } } self.amount_edit = "".to_string(); self.request_edit = "".to_string(); cb.hide_keyboard(); modal.close(); }); }); columns[1].vertical_centered_justified(|ui| { // Draw copy button. let copy_text = format!("{} {}", COPY, t!("copy")); View::button(ui, copy_text, Colors::BUTTON, || { cb.copy_string_to_buffer(self.request_edit.clone()); self.amount_edit = "".to_string(); self.request_edit = "".to_string(); modal.close(); }); }); }); // Draw button to close modal. ui.add_space(12.0); ui.vertical_centered_justified(|ui| { View::button(ui, t!("close"), Colors::WHITE, || { self.amount_edit = "".to_string(); self.request_edit = "".to_string(); modal.close(); }); }); ui.add_space(6.0); } } }