grim/src/gui/views/wallets/wallet/message.rs

// Copyright 2026 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, RichText, ScrollArea};

use crate::gui::Colors;
use crate::gui::icons::{BROOM, CLIPBOARD_TEXT, SCAN, SEAL_CHECK};
use crate::gui::platform::PlatformCallbacks;
use crate::gui::views::wallets::wallet::proof::PaymentProofContent;
use crate::gui::views::{CameraContent, FilePickContent, FilePickContentType, Modal, View};
use crate::wallet::Wallet;
use crate::wallet::types::WalletTask;

pub struct MessageInputContent {
	/// Slatepack input text.
	message_edit: String,
	/// Flag to check if error happened at Slatepack message parsing.
	parse_error: bool,
	/// Button to parse picked file content.
	file_pick_button: FilePickContent,

	/// QR code scanner content.
	scan_qr_content: Option<CameraContent>,

	/// Payment proof input content.
	pub proof_content: Option<PaymentProofContent>,
}

/// Hint for Slatepack message input.
const SLATEPACK_MESSAGE_HINT: &'static str = "BEGINSLATEPACK.\n...\n...\n...\nENDSLATEPACK.";

impl Default for MessageInputContent {
	fn default() -> Self {
		Self {
			message_edit: "".to_string(),
			parse_error: false,
			file_pick_button: FilePickContent::new(FilePickContentType::Button(
				t!("choose_file").into(),
			)),
			scan_qr_content: None,
			proof_content: None,
		}
	}
}

impl MessageInputContent {
	/// Identifier for [`Modal`].
	pub const MODAL_ID: &'static str = "input_message_modal";

	/// Draw [`Modal`] content.
	pub fn ui(
		&mut self,
		ui: &mut egui::Ui,
		wallet: &Wallet,
		modal: &Modal,
		cb: &dyn PlatformCallbacks,
	) {
		if let Some(scan_content) = self.scan_qr_content.as_mut() {
			if let Some(result) = scan_content.qr_scan_result() {
				cb.stop_camera();
				modal.enable_closing();
				self.scan_qr_content = None;
				// Parse scan result.
				self.on_message_input(result.text(), wallet);
			} else {
				scan_content.ui(ui, cb);
			}
			ui.add_space(8.0);

			// Setup spacing between buttons.
			ui.spacing_mut().item_spacing = egui::Vec2::new(8.0, 0.0);

			// Show buttons to close modal or scanner.
			ui.columns(2, |cols| {
				cols[0].vertical_centered_justified(|ui| {
					View::button(ui, t!("close"), Colors::white_or_black(false), || {
						cb.stop_camera();
						self.scan_qr_content = None;
						Modal::close();
					});
				});
				cols[1].vertical_centered_justified(|ui| {
					View::button(ui, t!("back"), Colors::white_or_black(false), || {
						cb.stop_camera();
						self.scan_qr_content = None;
						modal.enable_closing();
					});
				});
			});
		} else if let Some(proof_content) = self.proof_content.as_mut() {
			proof_content.input_ui(ui, wallet, cb);
			ui.add_space(8.0);
			View::horizontal_line(ui, Colors::item_stroke());
			ui.add_space(8.0);

			// Show button to close modal.
			ui.vertical_centered_justified(|ui| {
				View::button(ui, t!("close"), Colors::white_or_black(false), || {
					self.message_edit = "".to_string();
					Modal::close();
				});
			});
		} else {
			ui.add_space(6.0);
			ui.vertical_centered(|ui| {
				let (text, color) = if self.parse_error {
					(t!("wallets.parse_slatepack_err"), Colors::red())
				} else {
					(t!("wallets.input_slatepack_desc"), Colors::gray())
				};
				ui.label(RichText::new(text).size(16.0).color(color));
			});
			ui.add_space(6.0);

			// Draw slatepack message content.
			let message_before = self.message_edit.clone();
			ui.vertical_centered(|ui| {
				let scroll_id = Id::from("message_input").with(wallet.identifier());
				View::horizontal_line(ui, Colors::item_stroke());
				ui.add_space(3.0);
				ScrollArea::vertical()
					.id_salt(scroll_id)
					.scroll_bar_visibility(ScrollBarVisibility::AlwaysHidden)
					.max_height(128.0)
					.auto_shrink([false; 2])
					.show(ui, |ui| {
						ui.add_space(7.0);
						let input_id = scroll_id.with("_input");
						let resp = egui::TextEdit::multiline(&mut self.message_edit)
							.id(input_id)
							.font(egui::TextStyle::Small)
							.desired_rows(5)
							.interactive(true)
							.hint_text(SLATEPACK_MESSAGE_HINT)
							.desired_width(f32::INFINITY)
							.show(ui)
							.response;
						if View::is_desktop() {
							resp.request_focus();
						}
						ui.add_space(6.0);
					});
			});
			// Parse message on input change.
			if message_before != self.message_edit {
				self.on_message_input(self.message_edit.clone(), wallet);
			}

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

			// 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| {
					if self.parse_error {
						// Draw button to clear message input.
						let clear_text = format!("{} {}", BROOM, t!("clear"));
						View::button(ui, clear_text, Colors::white_or_black(false), || {
							self.message_edit = "".to_string();
							self.parse_error = false;
						});
					} else {
						// Draw button to scan Slatepack message QR code.
						let scan_text = format!("{} {}", SCAN, t!("scan"));
						View::button(ui, scan_text, Colors::white_or_black(false), || {
							modal.disable_closing();
							self.scan_qr_content = Some(CameraContent::default());
							cb.start_camera();
						});
					}
				});
				columns[1].vertical_centered_justified(|ui| {
					// Draw paste button.
					let paste_text = format!("{} {}", CLIPBOARD_TEXT, t!("paste"));
					View::button(ui, paste_text, Colors::white_or_black(false), || {
						self.on_message_input(cb.get_string_from_buffer(), wallet);
					});
				});
			});

			// Draw button to pick Slatepack message file.
			ui.add_space(8.0);
			ui.vertical_centered(|ui| {
				let mut picked_data = None;
				self.file_pick_button.ui(ui, cb, |data| {
					picked_data = Some(data);
				});
				if let Some(data) = picked_data {
					self.on_message_input(data, wallet);
				}
			});

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

			ui.vertical_centered(|ui| {
				let proof_label = format!("{} {}", SEAL_CHECK, t!("wallets.payment_proof"));
				View::colored_text_button(
					ui,
					proof_label,
					Colors::gold_dark(),
					Colors::white_or_black(false),
					|| {
						Modal::set_title(t!("wallets.payment_proof"));
						self.proof_content = Some(PaymentProofContent::new(None));
					},
				);
			});
			ui.add_space(8.0);
			View::horizontal_line(ui, Colors::item_stroke());
			ui.add_space(8.0);

			// Show button to close modal.
			ui.vertical_centered_justified(|ui| {
				View::button(ui, t!("close"), Colors::white_or_black(false), || {
					self.message_edit = "".to_string();
					Modal::close();
				});
			});
		}
		ui.add_space(6.0);
	}

	/// Parse Slatepack message on input change.
	fn on_message_input(&mut self, text: String, wallet: &Wallet) {
		self.parse_error = false;
		self.message_edit = text;
		if self.message_edit.is_empty() {
			return;
		}
		if self.message_edit.starts_with("BEGINSLATEPACK.")
			&& self.message_edit.ends_with("ENDSLATEPACK.")
		{
			wallet.task(WalletTask::OpenMessage(self.message_edit.to_string()));
			self.message_edit = "".to_string();
			Modal::close();
		} else {
			self.parse_error = true;
		}
	}
}