//! Mapping of inputs to sequences of actions.

use super::actions::{Action, Direction};

use std::collections::HashMap;

use strum::IntoEnumIterator;
use zellij_tile::data::*;

type Keybinds = HashMap<InputMode, ModeKeybinds>;
type ModeKeybinds = HashMap<Key, Vec<Action>>;

/// Populates the default hashmap of keybinds.
/// @@@khs26 What about an input config file?
pub fn get_default_keybinds() -> Result<Keybinds, String> {
    let mut defaults = Keybinds::new();

    for mode in InputMode::iter() {
        defaults.insert(mode, get_defaults_for_mode(&mode));
    }

    Ok(defaults)
}

/// Returns the default keybinds for a givent [`InputMode`].
fn get_defaults_for_mode(mode: &InputMode) -> ModeKeybinds {
    let mut defaults = ModeKeybinds::new();

    match *mode {
        InputMode::Normal => {
            defaults.insert(
                Key::Ctrl('g'),
                vec![Action::SwitchToMode(InputMode::Locked)],
            );
            defaults.insert(Key::Ctrl('p'), vec![Action::SwitchToMode(InputMode::Pane)]);
            defaults.insert(
                Key::Ctrl('r'),
                vec![Action::SwitchToMode(InputMode::Resize)],
            );
            defaults.insert(Key::Ctrl('t'), vec![Action::SwitchToMode(InputMode::Tab)]);
            defaults.insert(
                Key::Ctrl('s'),
                vec![Action::SwitchToMode(InputMode::Scroll)],
            );
            defaults.insert(Key::Ctrl('q'), vec![Action::Quit]);
        }
        InputMode::Locked => {
            defaults.insert(
                Key::Ctrl('g'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
        }
        InputMode::Resize => {
            defaults.insert(
                Key::Ctrl('g'),
                vec![Action::SwitchToMode(InputMode::Locked)],
            );
            defaults.insert(Key::Ctrl('p'), vec![Action::SwitchToMode(InputMode::Pane)]);
            defaults.insert(
                Key::Ctrl('r'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            defaults.insert(Key::Ctrl('t'), vec![Action::SwitchToMode(InputMode::Tab)]);
            defaults.insert(
                Key::Ctrl('s'),
                vec![Action::SwitchToMode(InputMode::Scroll)],
            );
            defaults.insert(Key::Ctrl('q'), vec![Action::Quit]);
            defaults.insert(Key::Esc, vec![Action::SwitchToMode(InputMode::Normal)]);
            defaults.insert(
                Key::Char('\n'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            defaults.insert(
                Key::Char(' '),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );

            defaults.insert(Key::Char('h'), vec![Action::Resize(Direction::Left)]);
            defaults.insert(Key::Char('j'), vec![Action::Resize(Direction::Down)]);
            defaults.insert(Key::Char('k'), vec![Action::Resize(Direction::Up)]);
            defaults.insert(Key::Char('l'), vec![Action::Resize(Direction::Right)]);

            defaults.insert(Key::Left, vec![Action::Resize(Direction::Left)]);
            defaults.insert(Key::Down, vec![Action::Resize(Direction::Down)]);
            defaults.insert(Key::Up, vec![Action::Resize(Direction::Up)]);
            defaults.insert(Key::Right, vec![Action::Resize(Direction::Right)]);
        }
        InputMode::Pane => {
            defaults.insert(
                Key::Ctrl('g'),
                vec![Action::SwitchToMode(InputMode::Locked)],
            );
            defaults.insert(
                Key::Ctrl('p'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            defaults.insert(
                Key::Ctrl('r'),
                vec![Action::SwitchToMode(InputMode::Resize)],
            );
            defaults.insert(Key::Ctrl('t'), vec![Action::SwitchToMode(InputMode::Tab)]);
            defaults.insert(
                Key::Ctrl('s'),
                vec![Action::SwitchToMode(InputMode::Scroll)],
            );
            defaults.insert(Key::Ctrl('q'), vec![Action::Quit]);
            defaults.insert(Key::Esc, vec![Action::SwitchToMode(InputMode::Normal)]);
            defaults.insert(
                Key::Char('\n'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            defaults.insert(
                Key::Char(' '),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );

            defaults.insert(Key::Char('h'), vec![Action::MoveFocus(Direction::Left)]);
            defaults.insert(Key::Char('j'), vec![Action::MoveFocus(Direction::Down)]);
            defaults.insert(Key::Char('k'), vec![Action::MoveFocus(Direction::Up)]);
            defaults.insert(Key::Char('l'), vec![Action::MoveFocus(Direction::Right)]);

            defaults.insert(Key::Left, vec![Action::MoveFocus(Direction::Left)]);
            defaults.insert(Key::Down, vec![Action::MoveFocus(Direction::Down)]);
            defaults.insert(Key::Up, vec![Action::MoveFocus(Direction::Up)]);
            defaults.insert(Key::Right, vec![Action::MoveFocus(Direction::Right)]);

            defaults.insert(Key::Char('p'), vec![Action::SwitchFocus(Direction::Right)]);
            defaults.insert(Key::Char('n'), vec![Action::NewPane(None)]);
            defaults.insert(Key::Char('d'), vec![Action::NewPane(Some(Direction::Down))]);
            defaults.insert(
                Key::Char('r'),
                vec![Action::NewPane(Some(Direction::Right))],
            );
            defaults.insert(Key::Char('x'), vec![Action::CloseFocus]);
            defaults.insert(Key::Char('f'), vec![Action::ToggleFocusFullscreen]);
        }
        InputMode::Tab => {
            defaults.insert(
                Key::Ctrl('g'),
                vec![Action::SwitchToMode(InputMode::Locked)],
            );
            defaults.insert(Key::Ctrl('p'), vec![Action::SwitchToMode(InputMode::Pane)]);
            defaults.insert(
                Key::Ctrl('r'),
                vec![Action::SwitchToMode(InputMode::Resize)],
            );
            defaults.insert(
                Key::Ctrl('t'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            defaults.insert(
                Key::Ctrl('s'),
                vec![Action::SwitchToMode(InputMode::Scroll)],
            );
            defaults.insert(Key::Ctrl('q'), vec![Action::Quit]);
            defaults.insert(Key::Esc, vec![Action::SwitchToMode(InputMode::Normal)]);
            defaults.insert(
                Key::Char('\n'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            defaults.insert(
                Key::Char(' '),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );

            defaults.insert(Key::Char('h'), vec![Action::GoToPreviousTab]);
            defaults.insert(Key::Char('j'), vec![Action::GoToNextTab]);
            defaults.insert(Key::Char('k'), vec![Action::GoToPreviousTab]);
            defaults.insert(Key::Char('l'), vec![Action::GoToNextTab]);

            defaults.insert(Key::Left, vec![Action::GoToPreviousTab]);
            defaults.insert(Key::Down, vec![Action::GoToNextTab]);
            defaults.insert(Key::Up, vec![Action::GoToPreviousTab]);
            defaults.insert(Key::Right, vec![Action::GoToNextTab]);

            defaults.insert(Key::Char('n'), vec![Action::NewTab]);
            defaults.insert(Key::Char('x'), vec![Action::CloseTab]);

            defaults.insert(
                Key::Char('r'),
                vec![
                    Action::SwitchToMode(InputMode::RenameTab),
                    Action::TabNameInput(vec![0]),
                ],
            );
            defaults.insert(Key::Char('q'), vec![Action::Quit]);
            defaults.insert(
                Key::Ctrl('g'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            for i in '1'..='9' {
                defaults.insert(Key::Char(i), vec![Action::GoToTab(i.to_digit(10).unwrap())]);
            }
        }
        InputMode::Scroll => {
            defaults.insert(
                Key::Ctrl('g'),
                vec![Action::SwitchToMode(InputMode::Locked)],
            );
            defaults.insert(Key::Ctrl('p'), vec![Action::SwitchToMode(InputMode::Pane)]);
            defaults.insert(
                Key::Ctrl('r'),
                vec![Action::SwitchToMode(InputMode::Resize)],
            );
            defaults.insert(Key::Ctrl('t'), vec![Action::SwitchToMode(InputMode::Tab)]);
            defaults.insert(
                Key::Ctrl('s'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            defaults.insert(Key::Ctrl('q'), vec![Action::Quit]);
            defaults.insert(Key::Esc, vec![Action::SwitchToMode(InputMode::Normal)]);
            defaults.insert(
                Key::Char('\n'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            defaults.insert(
                Key::Char(' '),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );

            defaults.insert(Key::Char('j'), vec![Action::ScrollDown]);
            defaults.insert(Key::Char('k'), vec![Action::ScrollUp]);

            defaults.insert(Key::Down, vec![Action::ScrollDown]);
            defaults.insert(Key::Up, vec![Action::ScrollUp]);
        }
        InputMode::RenameTab => {
            defaults.insert(Key::Char('\n'), vec![Action::SwitchToMode(InputMode::Tab)]);
            defaults.insert(
                Key::Ctrl('g'),
                vec![Action::SwitchToMode(InputMode::Normal)],
            );
            defaults.insert(
                Key::Esc,
                vec![
                    Action::TabNameInput(vec![0x1b]),
                    Action::SwitchToMode(InputMode::Tab),
                ],
            );
        }
    }

    defaults
}

/// Converts a [`Key`] terminal event to a sequence of [`Action`]s according to the current
/// [`InputMode`] and [`Keybinds`].
pub fn key_to_actions(
    key: &Key,
    input: Vec<u8>,
    mode: &InputMode,
    keybinds: &Keybinds,
) -> Vec<Action> {
    let mode_keybind_or_action = |action: Action| {
        keybinds
            .get(mode)
            .unwrap_or_else(|| unreachable!("Unrecognized mode: {:?}", mode))
            .get(key)
            .cloned()
            .unwrap_or_else(|| vec![action])
    };
    match *mode {
        InputMode::Normal | InputMode::Locked => mode_keybind_or_action(Action::Write(input)),
        InputMode::RenameTab => mode_keybind_or_action(Action::TabNameInput(input)),
        _ => mode_keybind_or_action(Action::NoOp),
    }
}