path: root/src/handler.rs

use crate::browser::FileExtension;
use crate::{
    app::{App, AppResult, SelectedTab},
    ui::InputMode,
};
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
use rust_fuzzy_search::{self, fuzzy_search_sorted};
use std::{path::Path, time::Duration};

pub fn handle_key_events(key_event: KeyEvent, app: &mut App) -> AppResult<()> {
    if app.inputmode == InputMode::Editing {
        // Live search update
        match app.selected_tab {
            SelectedTab::DirectoryBrowser => {
                let list: Vec<&str> = app
                    .browser
                    .filetree
                    .iter()
                    .map(|(_, f)| f.as_str())
                    .collect::<Vec<&str>>();

                let res: Vec<(&str, f32)> = fuzzy_search_sorted(&app.search_input, &list);
                let res = res.iter().map(|(x, _)| *x).collect::<Vec<&str>>();

                for (i, (_, item)) in app.browser.filetree.iter().enumerate() {
                    if item.contains(res.get(0).unwrap()) {
                        app.browser.selected = i;
                    }
                }
            }

            SelectedTab::Queue => {
                let list: Vec<&str> = app
                    .queue_list
                    .list
                    .iter()
                    .map(|f| f.file.as_str())
                    .collect::<Vec<&str>>();
                let res: Vec<(&str, f32)> = fuzzy_search_sorted(&app.search_input, &list);
                let res = res.iter().map(|(x, _)| *x).collect::<Vec<&str>>();

                for (i, item) in app.queue_list.list.iter().enumerate() {
                    if item.file.contains(res.get(0).unwrap()) {
                        app.queue_list.index = i;
                    }
                }
            }

            SelectedTab::Playlists => {
                let list: Vec<&str> = app
                    .pl_list
                    .list
                    .iter()
                    .map(|f| f.as_str())
                    .collect::<Vec<&str>>();
                let res: Vec<(&str, f32)> = fuzzy_search_sorted(&app.search_input, &list);
                let res = res.iter().map(|(x, _)| *x).collect::<Vec<&str>>();

                for (i, item) in app.pl_list.list.iter().enumerate() {
                    if item.contains(res.get(0).unwrap()) {
                        app.pl_list.index = i;
                    }
                }
            }
        }

        // Keybind for searching
        //
        // Keybinds for when the search prompt is visible
        match key_event.code {
            KeyCode::Esc => {
                app.inputmode = InputMode::Normal;
            }
            KeyCode::Char(to_insert) => {
                app.enter_char(to_insert);
            }
            KeyCode::Enter => {
                let list: Vec<&str> = app
                    .browser
                    .filetree
                    .iter()
                    .map(|(_, f)| f.as_str())
                    .collect::<Vec<&str>>();

                let res: Vec<(&str, f32)> = fuzzy_search_sorted(&app.search_input, &list);
                let (res, _) = res.get(0).unwrap();

                for (i, (_, item)) in app.browser.filetree.iter().enumerate() {
                    if item.contains(res) {
                        app.browser.selected = i;
                    }
                }

                app.search_input.clear();
                app.reset_cursor();
                app.inputmode = InputMode::Normal;
            }

            KeyCode::Backspace => {
                app.delete_char();
            }

            KeyCode::Left => {
                app.move_cursor_left();
            }

            KeyCode::Right => {
                app.move_cursor_right();
            }

            _ => {}
        }
    } else if app.inputmode == InputMode::PlaylistRename {
        match key_event.code {
            KeyCode::Esc => {
                app.pl_newname_input.clear();
                app.reset_cursor();
                app.inputmode = InputMode::Normal;
            }
            KeyCode::Char(to_insert) => {
                app.enter_char(to_insert);
            }
            KeyCode::Enter => {
                app.conn.conn.pl_rename(app.pl_list.list.get(app.pl_list.index).unwrap(), &app.pl_newname_input)?;
                app.pl_list.list = App::get_playlist(&mut app.conn.conn)?;
                app.pl_newname_input.clear();
                app.reset_cursor();
                app.inputmode = InputMode::Normal;
            }

            KeyCode::Backspace => {
                app.delete_char();
            }

            KeyCode::Left => {
                app.move_cursor_left();
            }

            KeyCode::Right => {
                app.move_cursor_right();
            }

            _ => {}
        }
        // Playlist popup keybinds
        //
        // Keybind for when the "append to playlist" popup is visible
    } else if app.playlist_popup {
        match key_event.code {
            KeyCode::Char('q') | KeyCode::Esc => {
                app.playlist_popup = false;
            }

            KeyCode::Char('j') | KeyCode::Down => app.append_list.next(),
            KeyCode::Char('k') | KeyCode::Up => app.append_list.prev(),

            KeyCode::Enter => {
                let pl_index = app.append_list.index;
                let pl_name = &app.append_list.list.get(pl_index).unwrap();

                let s_index: usize;
                match app.selected_tab {
                    SelectedTab::Queue => {
                        s_index = app.queue_list.index;
                        let short_path = &app.queue_list.list.get(s_index).unwrap().file;

                        if let Some(full_path) = app.conn.get_full_path(&short_path) {
                            let song = app.conn.get_song_with_only_filename(&full_path);

                            if pl_name.to_string() == "Current Playlist" {
                                app.conn.conn.push(&song)?;
                                app.update_queue();
                            } else {
                                app.conn.add_to_playlist(pl_name, &song)?;
                            }
                        }
                    }

                    SelectedTab::DirectoryBrowser => {
                        let (t, f) = app.browser.filetree.get(app.browser.selected).unwrap();
                        if t == "file" {
                            let short_path = f;
                            if let Some(full_path) = app.conn.get_full_path(short_path) {
                                let song = app.conn.get_song_with_only_filename(&full_path);

                                if pl_name.to_string() == "Current Playlist" {
                                    app.conn.conn.push(&song)?;
                                    app.update_queue();
                                } else {
                                    app.conn.add_to_playlist(pl_name, &song)?;
                                }
                            }
                        } else if t == "directory" {
                            for (t, f) in app.conn.conn.listfiles(f)?.iter() {
                                // dir_vec.push((t, f));
                                if t == "file"
                                    && Path::new(&f).has_extension(&[
                                        "mp3", "ogg", "flac", "m4a", "wav", "aac", "opus", "ape",
                                        "wma", "mpc", "aiff", "dff", "mp2", "mka",
                                    ])
                                {
                                    let full_path = app.conn.get_full_path(&f).unwrap_or_default();
                                    let song = app.conn.get_song_with_only_filename(&full_path);
                                    if pl_name.to_string() == "Current Playlist" {
                                        app.conn.conn.push(&song)?;
                                    } else {
                                        app.conn.add_to_playlist(pl_name, &song)?;
                                    }
                                }
                            }
                        }
                    }
                    _ => {}
                }

                // hide the playlist popup
                app.playlist_popup = false;
                app.append_list.index = 0;
            }
            _ => {}
        }
    } else {
        // Global keymaps
        //
        // Keymaps related to all the tabs
        match key_event.code {
            // Quit
            KeyCode::Char('q') => app.quit(),
            KeyCode::Char('c') | KeyCode::Char('C') => {
                if key_event.modifiers == KeyModifiers::CONTROL {
                    app.quit();
                } else {
                    app.conn.conn.clear()?;
                    app.conn.update_status();
                    app.queue_list.list.clear();
                    app.queue_list.reset_index();
                }
            }

            // Go Up
            KeyCode::Char('j') | KeyCode::Down => match app.selected_tab {
                SelectedTab::DirectoryBrowser => app.browser.next(),
                SelectedTab::Queue => app.queue_list.next(),
                SelectedTab::Playlists => app.pl_list.next(),
            },

            // Go down
            KeyCode::Char('k') | KeyCode::Up => match app.selected_tab {
                SelectedTab::DirectoryBrowser => app.browser.prev(),
                SelectedTab::Queue => app.queue_list.prev(),
                SelectedTab::Playlists => app.pl_list.prev(),
            },

            // Next directory
            KeyCode::Enter | KeyCode::Char('l') | KeyCode::Right => {
                // app.update_queue();

                match app.selected_tab {
                    SelectedTab::DirectoryBrowser => {
                        app.handle_enter()?;
                    }

                    SelectedTab::Queue => {
                        app.conn.conn.switch(app.queue_list.index as u32)?;
                    }

                    SelectedTab::Playlists => {
                        app.conn
                            .load_playlist(app.pl_list.list.get(app.pl_list.index).unwrap())?;
                    }
                }
                app.conn.update_status();
            }

            // head back to previous directory
            KeyCode::Char('h') | KeyCode::Left => match app.selected_tab {
                SelectedTab::DirectoryBrowser => {
                    app.browser.handle_go_back(&mut app.conn)?;
                }
                SelectedTab::Queue => {}
                SelectedTab::Playlists => {}
            },

            // Playback controls
            // Toggle Pause
            KeyCode::Char('p') => app.conn.toggle_pause(),

            // Pause
            KeyCode::Char('s') => app.conn.pause(),

            // Toggle rpeat
            KeyCode::Char('r') => {
                app.conn.toggle_repeat();
                app.conn.update_status();
            }

            // Toggle random
            KeyCode::Char('z') => {
                app.conn.toggle_random();
                app.conn.update_status();
            }

            // Dmenu prompt
            KeyCode::Char('D') => app.conn.play_dmenu()?,

            // add to queue
            KeyCode::Char('a') => app.playlist_popup = true,

            // Fast forward
            KeyCode::Char('f') => {
                let place = app.conn.conn.status().unwrap().song.unwrap().pos;
                let (pos, _) = app.conn.conn.status().unwrap().time.unwrap();
                let pos = Duration::from_secs(pos.as_secs().wrapping_add(2));
                app.conn.conn.seek(place, pos)?;
            }

            // backward
            KeyCode::Char('b') => {
                let place = app.conn.conn.status().unwrap().song.unwrap().pos;
                let (pos, _) = app.conn.conn.status().unwrap().time.unwrap();
                let pos = Duration::from_secs(pos.as_secs().wrapping_add(2));
                app.conn.conn.seek(place, pos)?;
            }

            // Cycle through tabs
            KeyCode::Tab => {
                app.cycle_tabls();
            }

            // Directory browser tab
            KeyCode::Char('1') => {
                app.selected_tab = SelectedTab::Queue;
            }

            // Playing queue tab
            KeyCode::Char('2') => {
                app.selected_tab = SelectedTab::DirectoryBrowser;
            }

            // Playlists tab
            KeyCode::Char('3') => {
                app.selected_tab = SelectedTab::Playlists;
            }

            // Play next song
            KeyCode::Char('>') => {
                if !app.queue_list.list.is_empty() {
                    app.conn.conn.next()?;
                    app.update_queue();
                }
            }

            // Play previous song
            KeyCode::Char('<') => {
                if !app.queue_list.list.is_empty() {
                    app.conn.conn.prev()?;
                    app.update_queue();
                }
            }

            // Volume controls
            KeyCode::Char('=') => {
                app.conn.inc_volume(2);
                app.conn.update_status();
            }

            KeyCode::Char('-') => {
                app.conn.dec_volume(2);
                app.conn.update_status();
            }

            // Delete highlighted song from the queue
            KeyCode::Char('d') => {
                if app.queue_list.index >= app.queue_list.list.len() - 1 {
                    if app.queue_list.index != 0 {
                        app.queue_list.index -= 1;
                    }
                }

                app.conn.conn.delete(app.queue_list.index as u32)?;
                app.update_queue();
            }

            // Update MPD database
            KeyCode::Char('U') => {
                app.conn.conn.rescan()?;
                app.browser.update_directory(&mut app.conn)?;
            }

            // Search for songs
            KeyCode::Char('/') => {
                if app.inputmode == InputMode::Normal {
                    app.inputmode = InputMode::Editing;
                } else {
                    app.inputmode = InputMode::Normal;
                }
            }

            // Remove from Current Playlsit
            KeyCode::Char(' ') | KeyCode::Backspace => {
                app.handle_add_or_remove_from_current_playlist()?;
            }

            // go to top of list
            KeyCode::Char('g') => match app.selected_tab {
                SelectedTab::Queue => app.queue_list.index = 0,
                SelectedTab::DirectoryBrowser => app.browser.selected = 0,
                SelectedTab::Playlists => app.pl_list.index = 0,
            },

            // go to bottom of list
            KeyCode::Char('G') => match app.selected_tab {
                SelectedTab::Queue => app.queue_list.index = app.queue_list.list.len() - 1,
                SelectedTab::DirectoryBrowser => {
                    app.browser.selected = app.browser.filetree.len() - 1
                }
                SelectedTab::Playlists => app.pl_list.index = app.pl_list.list.len() - 1,
            },

            // Change playlist name
            KeyCode::Char('e') => app.change_playlist_name()?,
            _ => {}
        }
    }
    Ok(())
}