jiq 3.21.0

use ratatui::crossterm::event::{KeyCode, KeyEvent, KeyModifiers};

use super::super::app_state::{App, Focus, OutputMode};
use crate::help::HelpTab;

fn accept_autocomplete_suggestion(app: &mut App) -> bool {
    if app.focus == Focus::InputField && app.autocomplete.is_visible() {
        if let Some(suggestion) = app.autocomplete.selected() {
            let suggestion_clone = suggestion.clone();
            app.insert_autocomplete_suggestion(&suggestion_clone);
            app.debouncer.mark_executed();
            app.update_tooltip();
        }
        return true;
    }
    false
}

pub fn handle_global_keys(app: &mut App, key: KeyEvent) -> bool {
    if let Some(query) = &mut app.query
        && crate::ai::ai_events::handle_suggestion_selection(
            key,
            &mut app.ai,
            &mut app.input,
            query,
            &mut app.autocomplete,
        )
    {
        return true;
    }

    match key.code {
        KeyCode::Char('q') if !key.modifiers.contains(KeyModifiers::CONTROL) => match app.focus {
            Focus::ResultsPane => {
                app.should_quit = true;
                true
            }
            Focus::InputField => {
                if app.input.editor_mode == crate::editor::EditorMode::Normal {
                    app.should_quit = true;
                    true
                } else {
                    false
                }
            }
        },

        KeyCode::Char('q') if key.modifiers.contains(KeyModifiers::CONTROL) => {
            if app.debouncer.has_pending() {
                crate::editor::editor_events::execute_query(app);
                app.debouncer.mark_executed();
            }
            if let Some(query) = &app.query
                && query.result.is_ok()
                && !app.query().is_empty()
            {
                let query_str = app.query().to_string();
                app.history.add_entry(&query_str);
            }
            app.output_mode = Some(OutputMode::Query);
            app.should_quit = true;
            true
        }
        KeyCode::Enter if key.modifiers.contains(KeyModifiers::SHIFT) => {
            if app.debouncer.has_pending() {
                crate::editor::editor_events::execute_query(app);
                app.debouncer.mark_executed();
            }
            if let Some(query) = &app.query
                && query.result.is_ok()
                && !app.query().is_empty()
            {
                let query_str = app.query().to_string();
                app.history.add_entry(&query_str);
            }
            app.output_mode = Some(OutputMode::Query);
            app.should_quit = true;
            true
        }
        KeyCode::Enter if key.modifiers.contains(KeyModifiers::ALT) => {
            if app.debouncer.has_pending() {
                crate::editor::editor_events::execute_query(app);
                app.debouncer.mark_executed();
            }
            if let Some(query) = &app.query
                && query.result.is_ok()
                && !app.query().is_empty()
            {
                let query_str = app.query().to_string();
                app.history.add_entry(&query_str);
            }
            app.output_mode = Some(OutputMode::Query);
            app.should_quit = true;
            true
        }
        KeyCode::Enter => {
            if accept_autocomplete_suggestion(app) {
                return true;
            }

            if app.debouncer.has_pending() {
                crate::editor::editor_events::execute_query(app);
                app.debouncer.mark_executed();
            }
            if let Some(query) = &app.query
                && query.result.is_ok()
                && !app.query().is_empty()
            {
                let query_str = app.query().to_string();
                app.history.add_entry(&query_str);
            }
            app.output_mode = Some(OutputMode::Results);
            app.should_quit = true;
            true
        }

        KeyCode::Tab if !key.modifiers.contains(KeyModifiers::CONTROL) => {
            accept_autocomplete_suggestion(app)
        }

        KeyCode::BackTab => {
            if app.history.is_visible() {
                app.history.close();
            }

            match app.focus {
                Focus::InputField => app.focus_results_pane(),
                Focus::ResultsPane => app.focus_input_field(),
            };
            true
        }

        KeyCode::Char('?') => {
            let snippets_allows = app.snippets.is_visible() && !app.snippets.is_editing();
            if snippets_allows
                || app.input.editor_mode == crate::editor::EditorMode::Normal
                || app.focus == Focus::ResultsPane
            {
                if app.help.visible {
                    app.help.reset();
                } else {
                    // Context-aware tab selection
                    // Note: History and AI tabs never auto-focus
                    let tab = if app.snippets.is_visible() {
                        HelpTab::Snippet
                    } else if app.search.is_visible() {
                        HelpTab::Search
                    } else if app.focus == Focus::ResultsPane {
                        HelpTab::Result
                    } else {
                        HelpTab::Input
                    };
                    app.help.active_tab = tab;
                    app.help.visible = true;
                }
                true
            } else {
                false
            }
        }

        KeyCode::Char('e') if key.modifiers.contains(KeyModifiers::CONTROL) => {
            if let Some(query) = &app.query
                && query.result.is_err()
            {
                app.error_overlay_visible = !app.error_overlay_visible;
            }
            true
        }

        KeyCode::Char('t') if key.modifiers.contains(KeyModifiers::CONTROL) => {
            crate::tooltip::tooltip_events::handle_tooltip_toggle(&mut app.tooltip);
            true
        }

        KeyCode::Char('f') if key.modifiers.contains(KeyModifiers::CONTROL) => {
            crate::search::search_events::open_search(app);
            true
        }

        KeyCode::Char('a') if key.modifiers.contains(KeyModifiers::CONTROL) => {
            let was_visible = app.ai.visible;
            app.ai.toggle();

            if !was_visible && app.ai.visible {
                // AI popup just became visible - hide tooltip
                app.saved_tooltip_visibility = app.tooltip.enabled;
                app.tooltip.enabled = false;

                // Trigger AI request for current context
                app.trigger_ai_request();
            } else if was_visible && !app.ai.visible {
                // AI popup just became hidden - restore tooltip
                app.tooltip.enabled = app.saved_tooltip_visibility;
            }

            true
        }

        KeyCode::Char('s') if key.modifiers.contains(KeyModifiers::CONTROL) => {
            app.snippets.open();
            app.autocomplete.hide();
            app.history.close();
            true
        }

        _ => false,
    }
}

#[cfg(test)]
#[path = "global_tests.rs"]
mod global_tests;