rnk 0.17.3

//! Terminal handling with ANSI escape codes (ink-style)
//!
//! This module provides terminal abstraction supporting both inline and fullscreen modes,
//! following patterns from Ink (JavaScript) and Bubbletea (Go).
//!
//! ## Modes
//!
//! - **Inline mode** (default): Renders in the current terminal position, output persists
//!   in terminal history. Like Ink and Bubbletea's default behavior.
//!
//! - **Fullscreen mode**: Uses alternate screen buffer, content is cleared on exit.
//!   Like vim, less, or Bubbletea's `WithAltScreen()`.

use crossterm::{
    cursor::{Hide, MoveTo, Show},
    event::{self, DisableMouseCapture, EnableMouseCapture, Event, KeyCode, KeyModifiers},
    execute,
    terminal::{EnterAlternateScreen, LeaveAlternateScreen, disable_raw_mode, enable_raw_mode},
};
use std::io::{Write, stdout};
use std::time::Duration;

/// ANSI escape codes for terminal control
mod ansi {
    /// Move cursor to specific position (1-indexed)
    pub fn cursor_to(row: u16, col: u16) -> String {
        format!("\x1b[{};{}H", row + 1, col + 1)
    }

    /// Move cursor to home position (0, 0)
    pub fn cursor_home() -> &'static str {
        "\x1b[H"
    }

    /// Move cursor to column (0-indexed)
    pub fn cursor_to_column(col: u16) -> String {
        format!("\x1b[{}G", col + 1)
    }

    /// Move cursor up n lines
    pub fn cursor_up(n: u16) -> String {
        if n == 0 {
            String::new()
        } else {
            format!("\x1b[{}A", n)
        }
    }

    /// Erase from cursor to end of line
    pub fn erase_end_of_line() -> &'static str {
        "\x1b[K"
    }

    /// Erase entire line
    pub fn erase_line() -> &'static str {
        "\x1b[2K"
    }

    /// Erase entire screen
    pub fn erase_screen() -> &'static str {
        "\x1b[2J"
    }

    /// Hide cursor
    pub fn hide_cursor() -> &'static str {
        "\x1b[?25l"
    }

    /// Show cursor
    pub fn show_cursor() -> &'static str {
        "\x1b[?25h"
    }

    /// Enter alternate screen buffer (like vim, less)
    pub fn enter_alt_screen() -> &'static str {
        "\x1b[?1049h"
    }

    /// Leave alternate screen buffer
    pub fn leave_alt_screen() -> &'static str {
        "\x1b[?1049l"
    }
}

/// Terminal abstraction with ink-style rendering
///
/// Supports both inline and fullscreen (alternate screen) modes:
///
/// - **Inline mode**: Output appears at current cursor position, persists in terminal history
/// - **Fullscreen mode**: Uses alternate screen buffer, cleared on exit
///
/// Runtime mode switching is supported via `switch_to_alt_screen()` and `switch_to_inline()`.
///
/// ## Performance Optimizations
///
/// The terminal uses a two-level diff strategy (inspired by Bubbletea):
///
/// 1. **Fast path**: If the entire output is identical to the last frame, skip rendering entirely
/// 2. **Line-level diff**: Only update lines that have changed
///
/// This significantly reduces terminal I/O and improves perceived performance.
pub struct Terminal {
    /// Previous frame's lines for incremental rendering (line-level diff)
    previous_lines: Vec<String>,
    /// Last complete output string (for fast-path identical frame detection)
    last_output: String,
    /// Whether we're in alternate screen mode
    alternate_screen: bool,
    /// Whether cursor is hidden
    cursor_hidden: bool,
    /// Whether raw mode is enabled
    raw_mode: bool,
    /// Whether mouse mode is enabled
    mouse_enabled: bool,
    /// Number of lines rendered in inline mode (for cursor positioning)
    inline_lines_rendered: usize,
}

impl Terminal {
    /// Create a new terminal instance
    pub fn new() -> Self {
        Self {
            previous_lines: Vec::new(),
            last_output: String::new(),
            alternate_screen: false,
            cursor_hidden: false,
            raw_mode: false,
            mouse_enabled: false,
            inline_lines_rendered: 0,
        }
    }

    /// Check if currently in alternate screen mode
    pub fn is_alt_screen(&self) -> bool {
        self.alternate_screen
    }

    /// Enter raw mode and alternate screen (fullscreen mode)
    pub fn enter(&mut self) -> std::io::Result<()> {
        enable_raw_mode()?;
        self.raw_mode = true;
        execute!(stdout(), EnterAlternateScreen, Hide)?;
        self.alternate_screen = true;
        self.cursor_hidden = true;
        Ok(())
    }

    /// Exit raw mode and alternate screen
    pub fn exit(&mut self) -> std::io::Result<()> {
        // Disable mouse capture first
        if self.mouse_enabled {
            execute!(stdout(), DisableMouseCapture)?;
            self.mouse_enabled = false;
        }
        if self.alternate_screen {
            execute!(stdout(), Show, LeaveAlternateScreen)?;
            self.alternate_screen = false;
            self.cursor_hidden = false;
        }
        if self.raw_mode {
            disable_raw_mode()?;
            self.raw_mode = false;
        }
        Ok(())
    }

    /// Enter inline mode (renders in current terminal position)
    pub fn enter_inline(&mut self) -> std::io::Result<()> {
        enable_raw_mode()?;
        self.raw_mode = true;

        // Hide cursor during rendering
        let mut stdout = stdout();
        write!(stdout, "{}", ansi::hide_cursor())?;
        stdout.flush()?;
        self.cursor_hidden = true;
        self.inline_lines_rendered = 0;

        Ok(())
    }

    /// Exit inline mode
    pub fn exit_inline(&mut self) -> std::io::Result<()> {
        let mut stdout = stdout();

        // Disable mouse capture first
        if self.mouse_enabled {
            execute!(stdout, DisableMouseCapture)?;
            self.mouse_enabled = false;
        }

        // Show cursor
        if self.cursor_hidden {
            write!(stdout, "{}", ansi::show_cursor())?;
            self.cursor_hidden = false;
        }

        // Move to the end of output and add newline
        let line_count = self.previous_lines.len();
        if line_count > 0 {
            // We're at the last line, just add a newline
            writeln!(stdout)?;
        }

        stdout.flush()?;

        if self.raw_mode {
            disable_raw_mode()?;
            self.raw_mode = false;
        }

        Ok(())
    }

    /// Switch to alternate screen mode at runtime
    ///
    /// This clears the current inline output and enters fullscreen mode.
    /// Like Bubbletea's `EnterAltScreen` command.
    pub fn switch_to_alt_screen(&mut self) -> std::io::Result<()> {
        if self.alternate_screen {
            return Ok(());
        }

        let mut stdout = stdout();

        // First, clear any inline content we've rendered
        self.clear_inline_content()?;

        // Enter alternate screen using raw ANSI (more reliable for runtime switch)
        write!(stdout, "{}", ansi::enter_alt_screen())?;
        write!(stdout, "{}", ansi::erase_screen())?;
        write!(stdout, "{}", ansi::cursor_home())?;

        if !self.cursor_hidden {
            write!(stdout, "{}", ansi::hide_cursor())?;
            self.cursor_hidden = true;
        }

        stdout.flush()?;

        self.alternate_screen = true;
        self.previous_lines.clear();
        self.inline_lines_rendered = 0;

        Ok(())
    }

    /// Switch to inline mode at runtime
    ///
    /// This exits fullscreen mode and returns to normal terminal output.
    /// Like Bubbletea's `ExitAltScreen` command.
    pub fn switch_to_inline(&mut self) -> std::io::Result<()> {
        if !self.alternate_screen {
            return Ok(());
        }

        let mut stdout = stdout();

        // Leave alternate screen using raw ANSI
        write!(stdout, "{}", ansi::leave_alt_screen())?;

        // Show cursor temporarily (we'll hide it again on next render)
        if self.cursor_hidden {
            write!(stdout, "{}", ansi::show_cursor())?;
            self.cursor_hidden = false;
        }

        stdout.flush()?;

        self.alternate_screen = false;
        self.previous_lines.clear();
        self.inline_lines_rendered = 0;

        // Re-hide cursor for rendering
        write!(stdout, "{}", ansi::hide_cursor())?;
        stdout.flush()?;
        self.cursor_hidden = true;

        Ok(())
    }

    /// Clear inline content (for mode switching or println)
    fn clear_inline_content(&mut self) -> std::io::Result<()> {
        if self.previous_lines.is_empty() {
            return Ok(());
        }

        let mut stdout = stdout();
        let line_count = self.previous_lines.len();

        // Move up to the start of our content
        if line_count > 1 {
            write!(stdout, "{}", ansi::cursor_up(line_count as u16 - 1))?;
        }

        // Clear each line
        write!(stdout, "{}", ansi::cursor_to_column(0))?;
        for i in 0..line_count {
            write!(stdout, "{}", ansi::erase_line())?;
            if i < line_count - 1 {
                write!(stdout, "\r\n")?;
            }
        }

        // Move back up to where we started
        if line_count > 1 {
            write!(stdout, "{}", ansi::cursor_up(line_count as u16 - 1))?;
        }
        write!(stdout, "{}", ansi::cursor_to_column(0))?;

        stdout.flush()?;

        self.previous_lines.clear();
        self.inline_lines_rendered = 0;

        Ok(())
    }

    /// Write persistent output above the UI (like Bubbletea's Println)
    ///
    /// In inline mode, this clears the current UI, writes the message,
    /// and the UI will be re-rendered below it.
    ///
    /// In fullscreen mode, this is a no-op (messages are ignored).
    pub fn println(&mut self, message: &str) -> std::io::Result<()> {
        // Println only works in inline mode (like Bubbletea)
        if self.alternate_screen {
            return Ok(());
        }

        let mut stdout = stdout();

        // Clear current UI content
        self.clear_inline_content()?;

        // Write the message with proper line endings
        for line in message.lines() {
            write!(stdout, "{}{}\r\n", line, ansi::erase_end_of_line())?;
        }

        stdout.flush()?;

        // Mark for repaint to ensure clean render after println
        self.repaint();

        Ok(())
    }

    /// Render output to terminal (ink-style incremental rendering)
    ///
    /// Uses a two-level optimization strategy:
    /// 1. Fast path: Skip entirely if output is identical to last frame
    /// 2. Line-level diff: Only update changed lines
    pub fn render(&mut self, output: &str) -> std::io::Result<()> {
        // Fast path: if output is identical to last frame, skip rendering entirely
        // This is a key optimization from Bubbletea that significantly reduces I/O
        if output == self.last_output && !self.previous_lines.is_empty() {
            return Ok(());
        }

        let result = if self.alternate_screen {
            self.render_fullscreen(output)
        } else {
            self.render_inline(output)
        };

        // Store the complete output for fast-path comparison
        if result.is_ok() {
            self.last_output = output.to_string();
        }

        result
    }

    /// Render in fullscreen/alternate screen mode
    fn render_fullscreen(&mut self, output: &str) -> std::io::Result<()> {
        let mut stdout = stdout();

        // Move to top-left
        execute!(stdout, MoveTo(0, 0))?;

        let new_lines: Vec<&str> = output.lines().collect();

        // Incremental update - only redraw changed lines
        for (i, new_line) in new_lines.iter().enumerate() {
            let old_line = self.previous_lines.get(i).map(|s| s.as_str());

            if old_line != Some(*new_line) {
                // Move to line and clear it, then write new content
                write!(
                    stdout,
                    "{}{}{}",
                    ansi::cursor_to(i as u16, 0),
                    ansi::erase_line(),
                    new_line
                )?;
            }
        }

        // Clear any extra lines from previous render
        if self.previous_lines.len() > new_lines.len() {
            for i in new_lines.len()..self.previous_lines.len() {
                write!(
                    stdout,
                    "{}{}",
                    ansi::cursor_to(i as u16, 0),
                    ansi::erase_line()
                )?;
            }
        }

        stdout.flush()?;

        // Store current lines for next comparison
        self.previous_lines = new_lines.iter().map(|s| s.to_string()).collect();

        Ok(())
    }

    /// Render in inline mode (like ink's default behavior)
    ///
    /// This renders at the current cursor position, using cursor movement
    /// to update in place. Content persists in terminal history.
    ///
    /// Key insight from Ink/Bubbletea: use `inline_lines_rendered` to track
    /// how many lines are actually on screen, separate from `previous_lines`
    /// which is used for diff optimization. After repaint(), previous_lines
    /// is cleared but inline_lines_rendered still reflects screen state.
    fn render_inline(&mut self, output: &str) -> std::io::Result<()> {
        let mut stdout = stdout();
        let new_lines: Vec<&str> = output.lines().collect();
        let new_count = new_lines.len();

        // Use inline_lines_rendered to know how many lines are on screen
        // This is separate from previous_lines which may be cleared by repaint()
        let lines_on_screen = self.inline_lines_rendered;

        // Move cursor to the start of our output area if we have content on screen
        if lines_on_screen > 0 {
            if lines_on_screen > 1 {
                write!(stdout, "{}", ansi::cursor_up(lines_on_screen as u16 - 1))?;
            }
            write!(stdout, "{}", ansi::cursor_to_column(0))?;
        }

        // Calculate max lines to handle (max of screen content and new content)
        let max_lines = lines_on_screen.max(new_count);

        // Render each line
        for (i, new_line) in new_lines.iter().enumerate() {
            let old_line = self.previous_lines.get(i).map(|s| s.as_str());

            // Only rewrite if content changed or we don't have previous content
            if old_line != Some(*new_line) {
                write!(stdout, "{}{}", ansi::erase_line(), new_line)?;
            }

            // Move to next line if not the last
            if i < max_lines - 1 {
                write!(stdout, "\r\n")?;
            }
        }

        // Clear extra lines from previous render
        for i in new_count..max_lines {
            write!(stdout, "{}", ansi::erase_line())?;

            // Move to next line if not the last
            if i < max_lines - 1 {
                write!(stdout, "\r\n")?;
            }
        }

        // Position cursor correctly at the end
        // If new content is shorter, we need to move cursor back up
        if new_count < lines_on_screen {
            let lines_to_go_up = lines_on_screen - new_count;
            write!(stdout, "{}", ansi::cursor_up(lines_to_go_up as u16))?;
        }
        write!(stdout, "{}", ansi::cursor_to_column(0))?;

        stdout.flush()?;

        // Store current lines for next comparison
        self.previous_lines = new_lines.iter().map(|s| s.to_string()).collect();
        self.inline_lines_rendered = new_count;

        Ok(())
    }

    /// Clear the current output
    pub fn clear(&mut self) -> std::io::Result<()> {
        if self.previous_lines.is_empty() {
            return Ok(());
        }

        let mut stdout = stdout();
        let line_count = self.previous_lines.len();

        if self.alternate_screen {
            execute!(stdout, MoveTo(0, 0))?;
            for i in 0..line_count {
                write!(
                    stdout,
                    "{}{}",
                    ansi::cursor_to(i as u16, 0),
                    ansi::erase_line()
                )?;
            }
        } else {
            // Move up and clear each line
            if line_count > 1 {
                write!(stdout, "{}", ansi::cursor_up(line_count as u16 - 1))?;
            }
            for _ in 0..line_count {
                writeln!(
                    stdout,
                    "{}{}",
                    ansi::cursor_to_column(0),
                    ansi::erase_line()
                )?;
            }
            // Move back up
            write!(stdout, "{}", ansi::cursor_up(line_count as u16))?;
        }

        stdout.flush()?;
        self.previous_lines.clear();
        self.inline_lines_rendered = 0;

        Ok(())
    }

    /// Force a full repaint on next render
    pub fn repaint(&mut self) {
        self.previous_lines.clear();
        self.last_output.clear();
    }

    /// Get terminal size
    pub fn size() -> std::io::Result<(u16, u16)> {
        crossterm::terminal::size()
    }

    /// Poll for input event
    pub fn poll_event(timeout: Duration) -> std::io::Result<Option<Event>> {
        if event::poll(timeout)? {
            Ok(Some(event::read()?))
        } else {
            Ok(None)
        }
    }

    /// Read input event (blocking)
    pub fn read_event() -> std::io::Result<Event> {
        event::read()
    }

    /// Check if Ctrl+C was pressed
    pub fn is_ctrl_c(event: &Event) -> bool {
        matches!(
            event,
            Event::Key(crossterm::event::KeyEvent {
                code: KeyCode::Char('c'),
                modifiers,
                ..
            }) if modifiers.contains(KeyModifiers::CONTROL)
        )
    }

    /// Enable mouse capture
    pub fn enable_mouse(&mut self) -> std::io::Result<()> {
        if !self.mouse_enabled {
            execute!(stdout(), EnableMouseCapture)?;
            self.mouse_enabled = true;
        }
        Ok(())
    }

    /// Disable mouse capture
    pub fn disable_mouse(&mut self) -> std::io::Result<()> {
        if self.mouse_enabled {
            execute!(stdout(), DisableMouseCapture)?;
            self.mouse_enabled = false;
        }
        Ok(())
    }

    /// Check if mouse is enabled
    pub fn is_mouse_enabled(&self) -> bool {
        self.mouse_enabled
    }

    /// Suspend the terminal for external process execution
    ///
    /// This restores the terminal to a normal state so that an external
    /// interactive process (like vim, less, etc.) can take over.
    ///
    /// Call `resume()` after the external process exits to restore TUI state.
    pub fn suspend(&mut self) -> std::io::Result<()> {
        let mut stdout = stdout();

        // Disable mouse capture
        if self.mouse_enabled {
            execute!(stdout, DisableMouseCapture)?;
            // Note: we keep mouse_enabled = true so resume() knows to re-enable it
        }

        // Show cursor
        if self.cursor_hidden {
            write!(stdout, "{}", ansi::show_cursor())?;
        }

        // Leave alternate screen if in fullscreen mode
        if self.alternate_screen {
            write!(stdout, "{}", ansi::leave_alt_screen())?;
        }

        // Disable raw mode
        if self.raw_mode {
            disable_raw_mode()?;
        }

        stdout.flush()?;
        Ok(())
    }

    /// Resume the terminal after external process execution
    ///
    /// This restores the TUI state after an external process has finished.
    /// Should be called after `suspend()` and the external process has exited.
    pub fn resume(&mut self) -> std::io::Result<()> {
        let mut stdout = stdout();

        // Re-enable raw mode
        if self.raw_mode {
            enable_raw_mode()?;
        }

        // Re-enter alternate screen if we were in fullscreen mode
        if self.alternate_screen {
            write!(stdout, "{}", ansi::enter_alt_screen())?;
            write!(stdout, "{}", ansi::erase_screen())?;
            write!(stdout, "{}", ansi::cursor_home())?;
        }

        // Hide cursor again
        if self.cursor_hidden {
            write!(stdout, "{}", ansi::hide_cursor())?;
        }

        // Re-enable mouse capture if it was enabled
        if self.mouse_enabled {
            execute!(stdout, EnableMouseCapture)?;
        }

        stdout.flush()?;

        // Force full repaint
        self.repaint();

        Ok(())
    }
}

impl Default for Terminal {
    fn default() -> Self {
        Self::new()
    }
}

impl Drop for Terminal {
    fn drop(&mut self) {
        // Ensure we clean up on drop
        let _ = self.exit();
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_terminal_size() {
        // This test may fail in CI environments without a terminal
        if let Ok((width, height)) = Terminal::size() {
            assert!(width > 0);
            assert!(height > 0);
        }
    }

    #[test]
    fn test_ansi_codes() {
        assert_eq!(ansi::cursor_to(0, 0), "\x1b[1;1H");
        assert_eq!(ansi::cursor_to(5, 10), "\x1b[6;11H");
        assert_eq!(ansi::cursor_up(3), "\x1b[3A");
        assert_eq!(ansi::erase_line(), "\x1b[2K");
        assert_eq!(ansi::cursor_home(), "\x1b[H");
        assert_eq!(ansi::erase_screen(), "\x1b[2J");
        assert_eq!(ansi::enter_alt_screen(), "\x1b[?1049h");
        assert_eq!(ansi::leave_alt_screen(), "\x1b[?1049l");
    }

    #[test]
    fn test_terminal_new() {
        let terminal = Terminal::new();
        assert!(!terminal.is_alt_screen());
        assert!(terminal.previous_lines.is_empty());
        assert!(terminal.last_output.is_empty());
        assert_eq!(terminal.inline_lines_rendered, 0);
    }

    #[test]
    fn test_repaint_clears_previous_lines() {
        let mut terminal = Terminal::new();
        terminal.previous_lines = vec!["line1".to_string(), "line2".to_string()];
        terminal.last_output = "line1\nline2".to_string();
        terminal.repaint();
        assert!(terminal.previous_lines.is_empty());
        assert!(terminal.last_output.is_empty());
    }

    #[test]
    fn test_fast_path_identical_output() {
        let mut terminal = Terminal::new();
        // Simulate a previous render
        terminal.previous_lines = vec!["line1".to_string(), "line2".to_string()];
        terminal.last_output = "line1\nline2".to_string();
        terminal.inline_lines_rendered = 2;

        // The fast path should detect identical output
        // We can't easily test the actual render without a terminal,
        // but we can verify the state is set up correctly for the optimization
        assert_eq!(terminal.last_output, "line1\nline2");
        assert!(!terminal.previous_lines.is_empty());
    }
}