sql_cli/ui/input/

action_handlers.rs

//! Action handlers using visitor pattern
//!
//! This module implements a visitor pattern for handling different groups of actions,
//! allowing us to break down the massive try_handle_action function into manageable chunks.

use crate::buffer::AppMode;
use crate::ui::input::actions::{Action, ActionContext, ActionResult, NavigateAction, YankTarget};
use anyhow::Result;

/// Trait for handling groups of related actions
pub trait ActionHandler {
    /// Handle an action if this handler is responsible for it
    fn handle_action(
        &self,
        action: &Action,
        context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>>;

    /// Get the name of this handler (for debugging/logging)
    fn name(&self) -> &'static str;
}

/// Context interface for action handlers to interact with the TUI
/// This abstracts the TUI methods that action handlers need
pub trait ActionHandlerContext {
    // Navigation methods
    fn previous_row(&mut self);
    fn next_row(&mut self);
    fn move_column_left(&mut self);
    fn move_column_right(&mut self);
    fn page_up(&mut self);
    fn page_down(&mut self);
    fn goto_first_row(&mut self);
    fn goto_last_row(&mut self);
    fn goto_first_column(&mut self);
    fn goto_last_column(&mut self);
    fn goto_row(&mut self, row: usize);
    fn goto_column(&mut self, col: usize);

    // Mode and UI state
    fn set_mode(&mut self, mode: AppMode);
    fn get_mode(&self) -> AppMode;
    fn set_status_message(&mut self, message: String);

    // Column operations
    fn toggle_column_pin(&mut self);
    fn hide_current_column(&mut self);
    fn unhide_all_columns(&mut self);
    fn clear_all_pinned_columns(&mut self);

    // Export operations
    fn export_to_csv(&mut self);
    fn export_to_json(&mut self);

    // Yank operations
    fn yank_cell(&mut self);
    fn yank_row(&mut self);
    fn yank_column(&mut self);
    fn yank_all(&mut self);
    fn yank_query(&mut self);

    // Toggle operations
    fn toggle_selection_mode(&mut self);
    fn toggle_row_numbers(&mut self);
    fn toggle_compact_mode(&mut self);
    fn toggle_case_insensitive(&mut self);
    fn toggle_key_indicator(&mut self);

    // Clear operations
    fn clear_filter(&mut self);
    fn clear_line(&mut self);

    // Mode operations
    fn start_search(&mut self);
    fn start_column_search(&mut self);
    fn start_filter(&mut self);
    fn start_fuzzy_filter(&mut self);
    fn exit_current_mode(&mut self);
    fn toggle_debug_mode(&mut self);

    // Column arrangement operations
    fn move_current_column_left(&mut self);
    fn move_current_column_right(&mut self);

    // Search navigation
    fn next_search_match(&mut self);
    fn previous_search_match(&mut self);

    // Statistics and display
    fn show_column_statistics(&mut self);
    fn cycle_column_packing(&mut self);

    // Viewport navigation
    fn navigate_to_viewport_top(&mut self);
    fn navigate_to_viewport_middle(&mut self);
    fn navigate_to_viewport_bottom(&mut self);

    // Input and text editing
    fn move_input_cursor_left(&mut self);
    fn move_input_cursor_right(&mut self);
    fn move_input_cursor_home(&mut self);
    fn move_input_cursor_end(&mut self);
    fn backspace(&mut self);
    fn delete(&mut self);
    fn undo(&mut self);
    fn redo(&mut self);

    // Jump-to-row operations
    fn start_jump_to_row(&mut self);
    fn clear_jump_to_row_input(&mut self);

    // Debug and development
    fn show_debug_info(&mut self);
    fn show_pretty_query(&mut self);
    fn show_help(&mut self);

    // Text editing
    fn kill_line(&mut self);
    fn kill_line_backward(&mut self);
    fn delete_word_backward(&mut self);
    fn delete_word_forward(&mut self);
    fn expand_asterisk(&mut self);
    fn expand_asterisk_visible(&mut self);

    // History navigation
    fn previous_history_command(&mut self);
    fn next_history_command(&mut self);

    // Viewport lock operations
    fn toggle_cursor_lock(&mut self);
    fn toggle_viewport_lock(&mut self);
}

/// Handler for navigation actions (Up, Down, Left, Right, PageUp, etc.)
pub struct NavigationActionHandler;

impl ActionHandler for NavigationActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::Navigate(nav_action) => match nav_action {
                NavigateAction::Up(count) => {
                    for _ in 0..*count {
                        tui.previous_row();
                    }
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::Down(count) => {
                    for _ in 0..*count {
                        tui.next_row();
                    }
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::Left(count) => {
                    for _ in 0..*count {
                        tui.move_column_left();
                    }
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::Right(count) => {
                    for _ in 0..*count {
                        tui.move_column_right();
                    }
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::PageUp => {
                    tui.page_up();
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::PageDown => {
                    tui.page_down();
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::Home => {
                    tui.goto_first_row();
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::End => {
                    tui.goto_last_row();
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::FirstColumn => {
                    tui.goto_first_column();
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::LastColumn => {
                    tui.goto_last_column();
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::JumpToRow(row) => {
                    tui.goto_row(*row);
                    Some(Ok(ActionResult::Handled))
                }
                NavigateAction::JumpToColumn(col) => {
                    tui.goto_column(*col);
                    Some(Ok(ActionResult::Handled))
                }
            },
            Action::NextColumn => {
                tui.move_column_right();
                Some(Ok(ActionResult::Handled))
            }
            Action::PreviousColumn => {
                tui.move_column_left();
                Some(Ok(ActionResult::Handled))
            }
            _ => None, // Not handled by this handler
        }
    }

    fn name(&self) -> &'static str {
        "Navigation"
    }
}

/// Handler for column operations (pin, hide, sort, etc.)
pub struct ColumnActionHandler;

impl ActionHandler for ColumnActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::ToggleColumnPin => {
                tui.toggle_column_pin();
                Some(Ok(ActionResult::Handled))
            }
            Action::HideColumn => {
                tui.hide_current_column();
                Some(Ok(ActionResult::Handled))
            }
            Action::UnhideAllColumns => {
                tui.unhide_all_columns();
                Some(Ok(ActionResult::Handled))
            }
            Action::ClearAllPins => {
                tui.clear_all_pinned_columns();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "Column"
    }
}

/// Handler for export operations (CSV, JSON)
pub struct ExportActionHandler;

impl ActionHandler for ExportActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::ExportToCsv => {
                tui.export_to_csv();
                Some(Ok(ActionResult::Handled))
            }
            Action::ExportToJson => {
                tui.export_to_json();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "Export"
    }
}

/// Handler for yank operations (cell, row, column, etc.)
pub struct YankActionHandler;

impl ActionHandler for YankActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::Yank(target) => {
                match target {
                    YankTarget::Cell => tui.yank_cell(),
                    YankTarget::Row => tui.yank_row(),
                    YankTarget::Column => tui.yank_column(),
                    YankTarget::All => tui.yank_all(),
                    YankTarget::Query => tui.yank_query(),
                }
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "Yank"
    }
}

/// Handler for UI mode and display operations
pub struct UIActionHandler;

impl ActionHandler for UIActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::ShowHelp => {
                tui.set_mode(AppMode::Help);
                tui.set_status_message("Help mode - Press 'q' or Escape to return".to_string());
                Some(Ok(ActionResult::Handled))
            }
            // ShowDebugInfo needs to stay in the legacy switch for now
            // because it calls toggle_debug_mode() which generates debug data
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "UI"
    }
}

/// Handler for toggle operations (selection mode, row numbers, etc.)
pub struct ToggleActionHandler;

impl ActionHandler for ToggleActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::ToggleSelectionMode => {
                tui.toggle_selection_mode();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleRowNumbers => {
                tui.toggle_row_numbers();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleCompactMode => {
                tui.toggle_compact_mode();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleCaseInsensitive => {
                tui.toggle_case_insensitive();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleKeyIndicator => {
                tui.toggle_key_indicator();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "Toggle"
    }
}

/// Handler for clear/reset operations
pub struct ClearActionHandler;

impl ActionHandler for ClearActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::ClearFilter => {
                tui.clear_filter();
                Some(Ok(ActionResult::Handled))
            }
            Action::ClearLine => {
                // Only handle in Command mode
                if context.mode == AppMode::Command {
                    tui.clear_line();
                    Some(Ok(ActionResult::Handled))
                } else {
                    None
                }
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "Clear"
    }
}

/// Handler for exit/quit actions
pub struct ExitActionHandler;

impl ActionHandler for ExitActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        _tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::Quit | Action::ForceQuit => Some(Ok(ActionResult::Exit)),
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "Exit"
    }
}

/// Handler for mode transition actions (search modes, debug, etc.)
pub struct ModeActionHandler;

impl ActionHandler for ModeActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::StartSearch => {
                tui.start_search();
                Some(Ok(ActionResult::Handled))
            }
            Action::StartColumnSearch => {
                tui.start_column_search();
                Some(Ok(ActionResult::Handled))
            }
            Action::StartFilter => {
                tui.start_filter();
                Some(Ok(ActionResult::Handled))
            }
            Action::StartFuzzyFilter => {
                tui.start_fuzzy_filter();
                Some(Ok(ActionResult::Handled))
            }
            Action::ExitCurrentMode => {
                tui.exit_current_mode();
                Some(Ok(ActionResult::Handled))
            }
            Action::ShowDebugInfo => {
                tui.toggle_debug_mode();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "Mode"
    }
}

/// Handler for viewport navigation actions (H, M, L vim commands)
pub struct ViewportNavigationHandler;

impl ActionHandler for ViewportNavigationHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::NavigateToViewportTop => {
                tui.navigate_to_viewport_top();
                Some(Ok(ActionResult::Handled))
            }
            Action::NavigateToViewportMiddle => {
                tui.navigate_to_viewport_middle();
                Some(Ok(ActionResult::Handled))
            }
            Action::NavigateToViewportBottom => {
                tui.navigate_to_viewport_bottom();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "ViewportNavigation"
    }
}

/// Handler for column arrangement actions (move left/right)
pub struct ColumnArrangementActionHandler;

impl ActionHandler for ColumnArrangementActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::MoveColumnLeft => {
                tui.move_current_column_left();
                Some(Ok(ActionResult::Handled))
            }
            Action::MoveColumnRight => {
                tui.move_current_column_right();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "ColumnArrangement"
    }
}

/// Handler for search navigation actions
pub struct SearchNavigationActionHandler;

impl ActionHandler for SearchNavigationActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::NextSearchMatch => {
                tui.next_search_match();
                Some(Ok(ActionResult::Handled))
            }
            Action::PreviousSearchMatch => {
                // When Shift+N is pressed and there's no active search,
                // toggle row numbers instead of navigating search
                if !context.has_search {
                    tui.toggle_row_numbers();
                } else {
                    tui.previous_search_match();
                }
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "SearchNavigation"
    }
}

/// Handler for statistics and display actions
pub struct StatisticsActionHandler;

impl ActionHandler for StatisticsActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::ShowColumnStatistics => {
                tui.show_column_statistics();
                Some(Ok(ActionResult::Handled))
            }
            Action::CycleColumnPacking => {
                tui.cycle_column_packing();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "Statistics"
    }
}

/// Handler for cursor movement actions in Command mode
pub struct InputCursorActionHandler;

impl ActionHandler for InputCursorActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        // Only handle in Command mode
        if context.mode != AppMode::Command {
            return None;
        }

        match action {
            Action::MoveCursorLeft => {
                tui.move_input_cursor_left();
                Some(Ok(ActionResult::Handled))
            }
            Action::MoveCursorRight => {
                tui.move_input_cursor_right();
                Some(Ok(ActionResult::Handled))
            }
            Action::MoveCursorHome => {
                tui.move_input_cursor_home();
                Some(Ok(ActionResult::Handled))
            }
            Action::MoveCursorEnd => {
                tui.move_input_cursor_end();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "InputCursor"
    }
}

/// Handler for debug and viewport control operations
pub struct DebugViewportActionHandler;

impl ActionHandler for DebugViewportActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::ShowDebugInfo => {
                tui.toggle_debug_mode();
                Some(Ok(ActionResult::Handled))
            }
            Action::StartJumpToRow => {
                tui.start_jump_to_row();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleCursorLock => {
                tui.toggle_cursor_lock();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleViewportLock => {
                tui.toggle_viewport_lock();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "DebugViewport"
    }
}

/// Handler for function key actions (F1-F12)
pub struct FunctionKeyActionHandler;

impl ActionHandler for FunctionKeyActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        _context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        match action {
            Action::ShowHelp => {
                tui.show_help();
                Some(Ok(ActionResult::Handled))
            }
            Action::ShowPrettyQuery => {
                tui.show_pretty_query();
                Some(Ok(ActionResult::Handled))
            }
            Action::ShowDebugInfo => {
                tui.toggle_debug_mode();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleRowNumbers => {
                tui.toggle_row_numbers();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleCompactMode => {
                tui.toggle_compact_mode();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleCaseInsensitive => {
                tui.toggle_case_insensitive();
                Some(Ok(ActionResult::Handled))
            }
            Action::ToggleKeyIndicator => {
                tui.toggle_key_indicator();
                Some(Ok(ActionResult::Handled))
            }
            Action::KillLine => {
                tui.kill_line();
                Some(Ok(ActionResult::Handled))
            }
            Action::KillLineBackward => {
                tui.kill_line_backward();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "FunctionKey"
    }
}

/// Handler for text editing actions in Command mode
pub struct TextEditActionHandler;

impl ActionHandler for TextEditActionHandler {
    fn handle_action(
        &self,
        action: &Action,
        context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Option<Result<ActionResult>> {
        // Only handle in Command mode
        if context.mode != AppMode::Command {
            return None;
        }

        match action {
            Action::Backspace => {
                tui.backspace();
                Some(Ok(ActionResult::Handled))
            }
            Action::Delete => {
                tui.delete();
                Some(Ok(ActionResult::Handled))
            }
            Action::Undo => {
                tui.undo();
                Some(Ok(ActionResult::Handled))
            }
            Action::Redo => {
                tui.redo();
                Some(Ok(ActionResult::Handled))
            }
            Action::DeleteWordBackward => {
                tui.delete_word_backward();
                Some(Ok(ActionResult::Handled))
            }
            Action::DeleteWordForward => {
                tui.delete_word_forward();
                Some(Ok(ActionResult::Handled))
            }
            Action::KillLine => {
                tui.kill_line();
                Some(Ok(ActionResult::Handled))
            }
            Action::KillLineBackward => {
                tui.kill_line_backward();
                Some(Ok(ActionResult::Handled))
            }
            Action::ExpandAsterisk => {
                tui.expand_asterisk();
                Some(Ok(ActionResult::Handled))
            }
            Action::ExpandAsteriskVisible => {
                tui.expand_asterisk_visible();
                Some(Ok(ActionResult::Handled))
            }
            Action::PreviousHistoryCommand => {
                tui.previous_history_command();
                Some(Ok(ActionResult::Handled))
            }
            Action::NextHistoryCommand => {
                tui.next_history_command();
                Some(Ok(ActionResult::Handled))
            }
            _ => None,
        }
    }

    fn name(&self) -> &'static str {
        "TextEdit"
    }
}

/// Main action dispatcher using visitor pattern
pub struct ActionDispatcher {
    handlers: Vec<Box<dyn ActionHandler>>,
}

impl ActionDispatcher {
    pub fn new() -> Self {
        let handlers: Vec<Box<dyn ActionHandler>> = vec![
            Box::new(NavigationActionHandler),
            Box::new(ColumnActionHandler),
            Box::new(ExportActionHandler),
            Box::new(YankActionHandler),
            Box::new(UIActionHandler),
            Box::new(ToggleActionHandler),
            Box::new(ClearActionHandler),
            Box::new(ExitActionHandler),
            Box::new(FunctionKeyActionHandler),
            Box::new(ModeActionHandler),
            Box::new(ColumnArrangementActionHandler),
            Box::new(SearchNavigationActionHandler),
            Box::new(StatisticsActionHandler),
            Box::new(InputCursorActionHandler),
            Box::new(TextEditActionHandler),
            Box::new(ViewportNavigationHandler),
            Box::new(DebugViewportActionHandler),
        ];

        Self { handlers }
    }

    /// Dispatch an action to the appropriate handler
    pub fn dispatch(
        &self,
        action: &Action,
        context: &ActionContext,
        tui: &mut dyn ActionHandlerContext,
    ) -> Result<ActionResult> {
        for handler in &self.handlers {
            if let Some(result) = handler.handle_action(action, context, tui) {
                return result;
            }
        }

        // No handler found for this action
        Ok(ActionResult::NotHandled)
    }
}

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

#[cfg(test)]
mod tests {
    use super::*;
    use crate::ui::input::actions::{Action, NavigateAction};

    // Mock implementation for testing
    struct MockTui {
        pub last_action: String,
        pub mode: AppMode,
        pub status_message: String,
    }

    impl MockTui {
        fn new() -> Self {
            Self {
                last_action: String::new(),
                mode: AppMode::Results,
                status_message: String::new(),
            }
        }
    }

    impl ActionHandlerContext for MockTui {
        fn previous_row(&mut self) {
            self.last_action = "previous_row".to_string();
        }
        fn next_row(&mut self) {
            self.last_action = "next_row".to_string();
        }
        fn move_column_left(&mut self) {
            self.last_action = "move_column_left".to_string();
        }
        fn move_column_right(&mut self) {
            self.last_action = "move_column_right".to_string();
        }
        fn page_up(&mut self) {
            self.last_action = "page_up".to_string();
        }
        fn page_down(&mut self) {
            self.last_action = "page_down".to_string();
        }
        fn goto_first_row(&mut self) {
            self.last_action = "goto_first_row".to_string();
        }
        fn goto_last_row(&mut self) {
            self.last_action = "goto_last_row".to_string();
        }
        fn goto_first_column(&mut self) {
            self.last_action = "goto_first_column".to_string();
        }
        fn goto_last_column(&mut self) {
            self.last_action = "goto_last_column".to_string();
        }
        fn goto_row(&mut self, row: usize) {
            self.last_action = format!("goto_row_{}", row);
        }
        fn goto_column(&mut self, col: usize) {
            self.last_action = format!("goto_column_{}", col);
        }

        fn set_mode(&mut self, mode: AppMode) {
            self.mode = mode;
        }
        fn get_mode(&self) -> AppMode {
            self.mode.clone()
        }
        fn set_status_message(&mut self, message: String) {
            self.status_message = message;
        }

        fn toggle_column_pin(&mut self) {
            self.last_action = "toggle_column_pin".to_string();
        }
        fn hide_current_column(&mut self) {
            self.last_action = "hide_current_column".to_string();
        }
        fn unhide_all_columns(&mut self) {
            self.last_action = "unhide_all_columns".to_string();
        }
        fn clear_all_pinned_columns(&mut self) {
            self.last_action = "clear_all_pinned_columns".to_string();
        }

        fn export_to_csv(&mut self) {
            self.last_action = "export_to_csv".to_string();
        }
        fn export_to_json(&mut self) {
            self.last_action = "export_to_json".to_string();
        }

        fn yank_cell(&mut self) {
            self.last_action = "yank_cell".to_string();
        }
        fn yank_row(&mut self) {
            self.last_action = "yank_row".to_string();
        }
        fn yank_column(&mut self) {
            self.last_action = "yank_column".to_string();
        }
        fn yank_all(&mut self) {
            self.last_action = "yank_all".to_string();
        }
        fn yank_query(&mut self) {
            self.last_action = "yank_query".to_string();
        }

        // Toggle operations
        fn toggle_selection_mode(&mut self) {
            self.last_action = "toggle_selection_mode".to_string();
        }
        fn toggle_row_numbers(&mut self) {
            self.last_action = "toggle_row_numbers".to_string();
        }
        fn toggle_compact_mode(&mut self) {
            self.last_action = "toggle_compact_mode".to_string();
        }
        fn toggle_case_insensitive(&mut self) {
            self.last_action = "toggle_case_insensitive".to_string();
        }
        fn toggle_key_indicator(&mut self) {
            self.last_action = "toggle_key_indicator".to_string();
        }

        // Clear operations
        fn clear_filter(&mut self) {
            self.last_action = "clear_filter".to_string();
        }
        fn clear_line(&mut self) {
            self.last_action = "clear_line".to_string();
        }

        // Mode operations
        fn start_search(&mut self) {
            self.last_action = "start_search".to_string();
        }
        fn start_column_search(&mut self) {
            self.last_action = "start_column_search".to_string();
        }
        fn start_filter(&mut self) {
            self.last_action = "start_filter".to_string();
        }
        fn start_fuzzy_filter(&mut self) {
            self.last_action = "start_fuzzy_filter".to_string();
        }
        fn exit_current_mode(&mut self) {
            self.last_action = "exit_current_mode".to_string();
        }
        fn toggle_debug_mode(&mut self) {
            self.last_action = "toggle_debug_mode".to_string();
        }

        // Column arrangement operations
        fn move_current_column_left(&mut self) {
            self.last_action = "move_current_column_left".to_string();
        }
        fn move_current_column_right(&mut self) {
            self.last_action = "move_current_column_right".to_string();
        }

        // Search navigation
        fn next_search_match(&mut self) {
            self.last_action = "next_search_match".to_string();
        }
        fn previous_search_match(&mut self) {
            self.last_action = "previous_search_match".to_string();
        }

        // Statistics and display
        fn show_column_statistics(&mut self) {
            self.last_action = "show_column_statistics".to_string();
        }
        fn cycle_column_packing(&mut self) {
            self.last_action = "cycle_column_packing".to_string();
        }

        fn navigate_to_viewport_top(&mut self) {
            self.last_action = "navigate_to_viewport_top".to_string();
        }

        fn navigate_to_viewport_middle(&mut self) {
            self.last_action = "navigate_to_viewport_middle".to_string();
        }

        fn navigate_to_viewport_bottom(&mut self) {
            self.last_action = "navigate_to_viewport_bottom".to_string();
        }

        // Jump-to-row operations
        fn start_jump_to_row(&mut self) {
            self.last_action = "start_jump_to_row".to_string();
        }
        fn clear_jump_to_row_input(&mut self) {
            self.last_action = "clear_jump_to_row_input".to_string();
        }

        // Viewport lock operations
        fn toggle_cursor_lock(&mut self) {
            self.last_action = "toggle_cursor_lock".to_string();
        }
        fn toggle_viewport_lock(&mut self) {
            self.last_action = "toggle_viewport_lock".to_string();
        }

        // Input and text editing
        fn move_input_cursor_left(&mut self) {
            self.last_action = "move_input_cursor_left".to_string();
        }
        fn move_input_cursor_right(&mut self) {
            self.last_action = "move_input_cursor_right".to_string();
        }
        fn move_input_cursor_home(&mut self) {
            self.last_action = "move_input_cursor_home".to_string();
        }
        fn move_input_cursor_end(&mut self) {
            self.last_action = "move_input_cursor_end".to_string();
        }
        fn backspace(&mut self) {
            self.last_action = "backspace".to_string();
        }
        fn delete(&mut self) {
            self.last_action = "delete".to_string();
        }
        fn undo(&mut self) {
            self.last_action = "undo".to_string();
        }
        fn redo(&mut self) {
            self.last_action = "redo".to_string();
        }

        // Debug and development methods
        fn show_debug_info(&mut self) {
            self.last_action = "show_debug_info".to_string();
        }
        fn show_pretty_query(&mut self) {
            self.last_action = "show_pretty_query".to_string();
        }
        fn show_help(&mut self) {
            self.last_action = "show_help".to_string();
        }

        // Text editing operations
        fn kill_line(&mut self) {
            self.last_action = "kill_line".to_string();
        }
        fn kill_line_backward(&mut self) {
            self.last_action = "kill_line_backward".to_string();
        }
        fn delete_word_backward(&mut self) {
            self.last_action = "delete_word_backward".to_string();
        }
        fn delete_word_forward(&mut self) {
            self.last_action = "delete_word_forward".to_string();
        }
        fn expand_asterisk(&mut self) {
            self.last_action = "expand_asterisk".to_string();
        }
        fn expand_asterisk_visible(&mut self) {
            self.last_action = "expand_asterisk_visible".to_string();
        }

        // History navigation
        fn previous_history_command(&mut self) {
            self.last_action = "previous_history_command".to_string();
        }
        fn next_history_command(&mut self) {
            self.last_action = "next_history_command".to_string();
        }
    }

    #[test]
    fn test_debug_viewport_handler() {
        let mut tui = MockTui::new();
        let handler = DebugViewportActionHandler;
        let context = ActionContext {
            mode: AppMode::Results,
            selection_mode: crate::app_state_container::SelectionMode::Cell,
            has_results: true,
            has_filter: false,
            has_search: false,
            row_count: 100,
            column_count: 10,
            current_row: 0,
            current_column: 0,
        };

        // Test ShowDebugInfo
        let result = handler
            .handle_action(&Action::ShowDebugInfo, &context, &mut tui)
            .unwrap();
        assert!(matches!(result, Ok(ActionResult::Handled)));
        assert_eq!(tui.last_action, "toggle_debug_mode");

        // Test StartJumpToRow
        let result = handler
            .handle_action(&Action::StartJumpToRow, &context, &mut tui)
            .unwrap();
        assert!(matches!(result, Ok(ActionResult::Handled)));
        assert_eq!(tui.last_action, "start_jump_to_row");

        // Test ToggleCursorLock
        let result = handler
            .handle_action(&Action::ToggleCursorLock, &context, &mut tui)
            .unwrap();
        assert!(matches!(result, Ok(ActionResult::Handled)));
        assert_eq!(tui.last_action, "toggle_cursor_lock");

        // Test ToggleViewportLock
        let result = handler
            .handle_action(&Action::ToggleViewportLock, &context, &mut tui)
            .unwrap();
        assert!(matches!(result, Ok(ActionResult::Handled)));
        assert_eq!(tui.last_action, "toggle_viewport_lock");

        // Test unhandled action
        let result = handler.handle_action(&Action::Quit, &context, &mut tui);
        assert!(result.is_none());
    }

    #[test]
    fn test_navigation_handler() {
        let handler = NavigationActionHandler;
        let mut mock_tui = MockTui::new();
        let context = ActionContext {
            mode: AppMode::Results,
            selection_mode: crate::app_state_container::SelectionMode::Row,
            has_results: true,
            has_filter: false,
            has_search: false,
            row_count: 10,
            column_count: 5,
            current_row: 0,
            current_column: 0,
        };

        let action = Action::Navigate(NavigateAction::Up(1));
        let result = handler.handle_action(&action, &context, &mut mock_tui);

        assert!(result.is_some());
        assert_eq!(mock_tui.last_action, "previous_row");

        match result.unwrap() {
            Ok(ActionResult::Handled) => {}
            _ => panic!("Expected Handled result"),
        }
    }

    #[test]
    fn test_action_dispatcher() {
        let dispatcher = ActionDispatcher::new();
        let mut mock_tui = MockTui::new();
        let context = ActionContext {
            mode: AppMode::Results,
            selection_mode: crate::app_state_container::SelectionMode::Row,
            has_results: true,
            has_filter: false,
            has_search: false,
            row_count: 10,
            column_count: 5,
            current_row: 0,
            current_column: 0,
        };

        // Test navigation action
        let action = Action::Navigate(NavigateAction::Down(2));
        let result = dispatcher
            .dispatch(&action, &context, &mut mock_tui)
            .unwrap();

        assert_eq!(result, ActionResult::Handled);
        assert_eq!(mock_tui.last_action, "next_row"); // Called twice for count=2

        // Test export action
        let action = Action::ExportToCsv;
        let result = dispatcher
            .dispatch(&action, &context, &mut mock_tui)
            .unwrap();

        assert_eq!(result, ActionResult::Handled);
        assert_eq!(mock_tui.last_action, "export_to_csv");
    }

    #[test]
    fn test_toggle_handler() {
        let handler = ToggleActionHandler;
        let mut mock_tui = MockTui::new();
        let context = ActionContext {
            mode: AppMode::Results,
            selection_mode: crate::app_state_container::SelectionMode::Row,
            has_results: true,
            has_filter: false,
            has_search: false,
            row_count: 10,
            column_count: 5,
            current_row: 0,
            current_column: 0,
        };

        // Test toggle selection mode
        let action = Action::ToggleSelectionMode;
        let result = handler
            .handle_action(&action, &context, &mut mock_tui)
            .unwrap()
            .unwrap();
        assert_eq!(result, ActionResult::Handled);
        assert_eq!(mock_tui.last_action, "toggle_selection_mode");

        // Test toggle row numbers
        let action = Action::ToggleRowNumbers;
        let result = handler
            .handle_action(&action, &context, &mut mock_tui)
            .unwrap()
            .unwrap();
        assert_eq!(result, ActionResult::Handled);
        assert_eq!(mock_tui.last_action, "toggle_row_numbers");
    }

    #[test]
    fn test_clear_handler() {
        let handler = ClearActionHandler;
        let mut mock_tui = MockTui::new();

        // Test clear filter
        let context = ActionContext {
            mode: AppMode::Results,
            selection_mode: crate::app_state_container::SelectionMode::Row,
            has_results: true,
            has_filter: true,
            has_search: false,
            row_count: 10,
            column_count: 5,
            current_row: 0,
            current_column: 0,
        };

        let action = Action::ClearFilter;
        let result = handler
            .handle_action(&action, &context, &mut mock_tui)
            .unwrap()
            .unwrap();
        assert_eq!(result, ActionResult::Handled);
        assert_eq!(mock_tui.last_action, "clear_filter");

        // Test clear line in Command mode
        let context = ActionContext {
            mode: AppMode::Command,
            selection_mode: crate::app_state_container::SelectionMode::Row,
            has_results: true,
            has_filter: false,
            has_search: false,
            row_count: 10,
            column_count: 5,
            current_row: 0,
            current_column: 0,
        };

        let action = Action::ClearLine;
        let result = handler
            .handle_action(&action, &context, &mut mock_tui)
            .unwrap()
            .unwrap();
        assert_eq!(result, ActionResult::Handled);
        assert_eq!(mock_tui.last_action, "clear_line");

        // Test clear line not handled in Results mode
        let context = ActionContext {
            mode: AppMode::Results,
            selection_mode: crate::app_state_container::SelectionMode::Row,
            has_results: true,
            has_filter: false,
            has_search: false,
            row_count: 10,
            column_count: 5,
            current_row: 0,
            current_column: 0,
        };

        let action = Action::ClearLine;
        let result = handler.handle_action(&action, &context, &mut mock_tui);
        assert!(result.is_none());
    }

    #[test]
    fn test_exit_handler() {
        let handler = ExitActionHandler;
        let mut mock_tui = MockTui::new();
        let context = ActionContext {
            mode: AppMode::Results,
            selection_mode: crate::app_state_container::SelectionMode::Row,
            has_results: true,
            has_filter: false,
            has_search: false,
            row_count: 10,
            column_count: 5,
            current_row: 0,
            current_column: 0,
        };

        // Test quit
        let action = Action::Quit;
        let result = handler
            .handle_action(&action, &context, &mut mock_tui)
            .unwrap()
            .unwrap();
        assert_eq!(result, ActionResult::Exit);

        // Test force quit
        let action = Action::ForceQuit;
        let result = handler
            .handle_action(&action, &context, &mut mock_tui)
            .unwrap()
            .unwrap();
        assert_eq!(result, ActionResult::Exit);
    }
}