grim/src/gui/views/wallets/wallet/settings/recovery.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::{Id, RichText};
use grin_chain::SyncStatus;
use grin_util::ZeroingString;

use crate::gui::Colors;
use crate::gui::icons::{EYE, KEY, LIFEBUOY, STETHOSCOPE, TRASH};
use crate::gui::platform::PlatformCallbacks;
use crate::gui::views::types::ModalPosition;
use crate::gui::views::wallets::wallet::types::WalletContentContainer;
use crate::gui::views::{Modal, TextEdit, View};
use crate::node::Node;
use crate::wallet::Wallet;
use crate::wallet::types::ConnectionMethod;

/// Wallet recovery settings content.
pub struct RecoverySettings {
	/// Wallet password [`Modal`] value.
	pass_edit: String,
	/// Flag to check if wrong password was entered.
	wrong_pass: bool,

	/// Recovery phrase value.
	recovery_phrase: Option<ZeroingString>,
}

/// Identifier for recovery phrase [`Modal`].
const RECOVERY_PHRASE_MODAL: &'static str = "recovery_phrase_modal";
/// Identifier to confirm wallet deletion [`Modal`].
const DELETE_CONFIRMATION_MODAL: &'static str = "delete_wallet_confirmation_modal";

impl WalletContentContainer for RecoverySettings {
	fn modal_ids(&self) -> Vec<&'static str> {
		vec![RECOVERY_PHRASE_MODAL, DELETE_CONFIRMATION_MODAL]
	}

	fn modal_ui(
		&mut self,
		ui: &mut egui::Ui,
		wallet: &Wallet,
		modal: &Modal,
		cb: &dyn PlatformCallbacks,
	) {
		match modal.id {
			RECOVERY_PHRASE_MODAL => {
				self.recovery_phrase_modal_ui(ui, wallet, modal, cb);
			}
			DELETE_CONFIRMATION_MODAL => {
				Self::deletion_modal_ui(ui, wallet);
			}
			_ => {}
		}
	}

	fn container_ui(&mut self, ui: &mut egui::Ui, wallet: &Wallet, _: &dyn PlatformCallbacks) {
		ui.add_space(10.0);
		View::horizontal_line(ui, Colors::stroke());
		ui.add_space(6.0);
		View::sub_title(ui, format!("{} {}", KEY, t!("wallets.recovery")));
		View::horizontal_line(ui, Colors::stroke());
		ui.add_space(4.0);

		ui.vertical_centered(|ui| {
			let integrated_node = wallet.get_current_connection() == ConnectionMethod::Integrated;
			let integrated_node_ready = Node::get_sync_status() == Some(SyncStatus::NoSync);
			if wallet.sync_error() || (integrated_node && !integrated_node_ready) {
				ui.add_space(2.0);
				ui.label(
					RichText::new(t!("wallets.repair_unavailable"))
						.size(16.0)
						.color(Colors::red()),
				);
				ui.add_space(2.0);
			} else if !wallet.is_repairing() {
				ui.add_space(6.0);

				// Draw button to repair the wallet.
				let repair_text = format!("{} {}", STETHOSCOPE, t!("wallets.repair_wallet"));
				View::action_button(ui, repair_text, || {
					wallet.repair();
				});

				ui.add_space(6.0);
				ui.label(
					RichText::new(t!("wallets.repair_desc"))
						.size(16.0)
						.color(Colors::inactive_text()),
				);
			}

			ui.add_space(6.0);
			View::horizontal_line(ui, Colors::item_stroke());
			ui.add_space(6.0);

			// Draw button to restore the wallet.
			ui.add_space(4.0);
			View::colored_text_button(
				ui,
				format!("{} {}", LIFEBUOY, t!("wallets.recover")),
				Colors::green(),
				Colors::white_or_black(false),
				|| {
					wallet.delete_db();
				},
			);
			ui.add_space(6.0);
			ui.label(
				RichText::new(t!("wallets.restore_wallet_desc"))
					.size(16.0)
					.color(Colors::inactive_text()),
			);

			ui.add_space(6.0);
			View::horizontal_line(ui, Colors::item_stroke());
			ui.add_space(6.0);

			let recovery_text = format!("{}:", t!("wallets.recovery_phrase"));
			ui.label(
				RichText::new(recovery_text)
					.size(16.0)
					.color(Colors::gray()),
			);
			ui.add_space(6.0);

			// Draw button to show recovery phrase.
			let show_text = format!("{} {}", EYE, t!("show"));
			View::button(ui, show_text, Colors::white_or_black(false), || {
				self.show_recovery_phrase_modal();
			});

			ui.add_space(12.0);
			View::horizontal_line(ui, Colors::item_stroke());
			ui.add_space(6.0);
			ui.label(
				RichText::new(t!("wallets.delete_desc"))
					.size(16.0)
					.color(Colors::red()),
			);
			ui.add_space(6.0);

			// Draw button to delete the wallet.
			View::colored_text_button(
				ui,
				format!("{} {}", TRASH, t!("wallets.delete")),
				Colors::red(),
				Colors::white_or_black(false),
				|| {
					Modal::new(DELETE_CONFIRMATION_MODAL)
						.position(ModalPosition::Center)
						.title(t!("confirmation"))
						.show();
				},
			);
			ui.add_space(8.0);
		});
	}
}

impl Default for RecoverySettings {
	fn default() -> Self {
		Self {
			wrong_pass: false,
			pass_edit: "".to_string(),
			recovery_phrase: None,
		}
	}
}

impl RecoverySettings {
	/// Show recovery phrase [`Modal`].
	fn show_recovery_phrase_modal(&mut self) {
		// Setup modal values.
		self.pass_edit = "".to_string();
		self.wrong_pass = false;
		self.recovery_phrase = None;
		// Show recovery phrase modal.
		Modal::new(RECOVERY_PHRASE_MODAL)
			.position(ModalPosition::CenterTop)
			.title(t!("wallets.recovery_phrase"))
			.show();
	}

	/// Draw recovery phrase [`Modal`] content.
	fn recovery_phrase_modal_ui(
		&mut self,
		ui: &mut egui::Ui,
		wallet: &Wallet,
		modal: &Modal,
		cb: &dyn PlatformCallbacks,
	) {
		let on_next = |c: &mut RecoverySettings| match wallet.get_recovery(c.pass_edit.clone()) {
			Ok(phrase) => {
				c.wrong_pass = false;
				c.recovery_phrase = Some(phrase);
			}
			Err(_) => {
				c.wrong_pass = true;
			}
		};

		ui.add_space(6.0);
		if self.recovery_phrase.is_some() {
			ui.vertical_centered(|ui| {
				ui.label(
					RichText::new(self.recovery_phrase.clone().unwrap().to_string())
						.size(17.0)
						.color(Colors::white_or_black(true)),
				);
			});
			ui.add_space(10.0);
			ui.vertical_centered_justified(|ui| {
				View::button(ui, t!("close"), Colors::white_or_black(false), || {
					self.recovery_phrase = None;
					Modal::close();
				});
			});
		} else {
			ui.vertical_centered(|ui| {
				ui.label(
					RichText::new(t!("wallets.pass"))
						.size(17.0)
						.color(Colors::gray()),
				);
				ui.add_space(8.0);

				// Draw current wallet password text edit.
				let pass_edit_id = Id::from(modal.id).with(wallet.get_config().id);
				let mut pass_edit = TextEdit::new(pass_edit_id).password();
				pass_edit.ui(ui, &mut self.pass_edit, cb);
				if pass_edit.enter_pressed {
					on_next(self);
				}

				// Show information when password is empty or wrong.
				if self.pass_edit.is_empty() {
					ui.add_space(12.0);
					ui.label(
						RichText::new(t!("wallets.pass_empty"))
							.size(17.0)
							.color(Colors::inactive_text()),
					);
				} else if self.wrong_pass {
					ui.add_space(12.0);
					ui.label(
						RichText::new(t!("wallets.wrong_pass"))
							.size(17.0)
							.color(Colors::red()),
					);
				}
			});
			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);

				ui.columns(2, |columns| {
					columns[0].vertical_centered_justified(|ui| {
						View::button(
							ui,
							t!("modal.cancel"),
							Colors::white_or_black(false),
							|| {
								self.recovery_phrase = None;
								Modal::close();
							},
						);
					});
					columns[1].vertical_centered_justified(|ui| {
						View::button(ui, "OK".to_owned(), Colors::white_or_black(false), || {
							on_next(self);
						});
					});
				});
			});
		}
		ui.add_space(6.0);
	}

	/// Draw wallet deletion [`Modal`] content.
	pub fn deletion_modal_ui(ui: &mut egui::Ui, wallet: &Wallet) {
		ui.add_space(8.0);
		ui.vertical_centered(|ui| {
			ui.label(
				RichText::new(t!("wallets.delete_conf"))
					.size(17.0)
					.color(Colors::text(false)),
			);
		});
		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);

			ui.columns(2, |columns| {
				columns[0].vertical_centered_justified(|ui| {
					View::button(
						ui,
						t!("modal.cancel"),
						Colors::white_or_black(false),
						|| {
							Modal::close();
						},
					);
				});
				columns[1].vertical_centered_justified(|ui| {
					View::button(ui, t!("delete"), Colors::white_or_black(false), || {
						wallet.delete_wallet();
						Modal::close();
					});
				});
			});
			ui.add_space(6.0);
		});
	}
}