use anyhow::Result;
use crossterm::{
execute,
terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
};
use ratatui::backend::CrosstermBackend;
use std::io::{self, IsTerminal, Stdout, Write};
pub type Terminal = ratatui::Terminal<CrosstermBackend<Stdout>>;
pub type Tui = Terminal;
pub fn init() -> Result<Tui> {
let mut stdout = io::stdout();
if !stdout.is_terminal() {
return Err(anyhow::anyhow!(
"Fido TUI must run in an interactive terminal (TTY). Stdout is not a TTY."
));
}
enable_raw_mode()?;
execute!(stdout, EnterAlternateScreen)?;
print!("\x1b[?1000l"); print!("\x1b[?1002l"); print!("\x1b[?1003l"); print!("\x1b[?1006l"); stdout.flush()?;
#[cfg(windows)]
disable_windows_mouse_input()?;
let backend = CrosstermBackend::new(stdout);
let terminal = ratatui::Terminal::new(backend)?;
Ok(terminal)
}
#[cfg(windows)]
fn disable_windows_mouse_input() -> Result<()> {
use windows::Win32::System::Console::{
GetConsoleMode, GetStdHandle, SetConsoleMode, CONSOLE_MODE, ENABLE_MOUSE_INPUT,
STD_INPUT_HANDLE,
};
unsafe {
let handle = GetStdHandle(STD_INPUT_HANDLE)
.map_err(|e| anyhow::anyhow!("Failed to get console handle: {}", e))?;
let mut mode = CONSOLE_MODE(0);
GetConsoleMode(handle, &mut mode)
.map_err(|e| anyhow::anyhow!("Failed to get console mode: {}", e))?;
mode &= !ENABLE_MOUSE_INPUT;
SetConsoleMode(handle, mode)
.map_err(|e| anyhow::anyhow!("Failed to set console mode: {}", e))?;
}
Ok(())
}
pub fn restore() -> Result<()> {
disable_raw_mode()?;
execute!(io::stdout(), LeaveAlternateScreen)?;
#[cfg(windows)]
enable_windows_mouse_input()?;
Ok(())
}
#[cfg(windows)]
fn enable_windows_mouse_input() -> Result<()> {
use windows::Win32::System::Console::{
GetConsoleMode, GetStdHandle, SetConsoleMode, CONSOLE_MODE, ENABLE_MOUSE_INPUT,
STD_INPUT_HANDLE,
};
unsafe {
let handle = GetStdHandle(STD_INPUT_HANDLE)
.map_err(|e| anyhow::anyhow!("Failed to get console handle: {}", e))?;
let mut mode = CONSOLE_MODE(0);
GetConsoleMode(handle, &mut mode)
.map_err(|e| anyhow::anyhow!("Failed to get console mode: {}", e))?;
mode |= ENABLE_MOUSE_INPUT;
SetConsoleMode(handle, mode)
.map_err(|e| anyhow::anyhow!("Failed to set console mode: {}", e))?;
}
Ok(())
}