use crate::app::{App, SelectedTab}; use ratatui::{ prelude::*, widgets::{block::Title, *}, }; #[derive(Debug, PartialEq)] pub enum InputMode { Editing, Normal, } impl InputMode { pub fn toggle_editing_states(state: &InputMode) -> InputMode { match state { InputMode::Editing => return InputMode::Normal, InputMode::Normal => return InputMode::Editing, }; } } /// Renders the user interface widgets pub fn render(app: &mut App, frame: &mut Frame) { // Layout let layout = Layout::default() .direction(Direction::Vertical) .constraints(vec![Constraint::Percentage(93), Constraint::Min(3)]) .split(frame.size()); match app.selected_tab { SelectedTab::Queue => draw_queue(frame, app, layout[0]), SelectedTab::Playlists => draw_playlists(frame, app, layout[0]), SelectedTab::DirectoryBrowser => draw_directory_browser(frame, app, layout[0]), } match app.inputmode { InputMode::Normal => { draw_progress_bar(frame, app, layout[1]); } InputMode::Editing => { draw_search_bar(frame, app, layout[1]); } } if app.playlist_popup { draw_add_to_playlist(frame, app, layout[0]); } } /// Draws the file tree browser fn draw_directory_browser(frame: &mut Frame, app: &mut App, size: Rect) { let mut song_state = ListState::default(); let total_songs = app.conn.conn.stats().unwrap().songs.to_string(); let mut list: Vec = vec![]; for (t, s) in app.browser.filetree.iter() { if t == "file" { let mut status: bool = false; for sn in app.queue_list.list.iter() { if sn.contains(s) { status = true; } } if status { list.push(ListItem::new(s.clone().magenta().bold())); } else { list.push(ListItem::new(s.clone())); } } else { list.push(ListItem::new(Line::styled( format!("[{}]", *s), Style::default(), ))); } } let list = List::new(list) .block( Block::default() .title(format!("File Browser: {}", app.browser.path.clone()).bold()) .title( Title::from(format!("Total Songs: {}", total_songs).bold().green()) .alignment(Alignment::Center), ) .title( Title::from(format!("Volume: {}%", app.conn.volume).bold().green()) .alignment(Alignment::Right), ) .borders(Borders::ALL), ) .highlight_style( Style::new() .fg(Color::Cyan) .bg(Color::Black) .add_modifier(Modifier::BOLD) .add_modifier(Modifier::REVERSED), ) .highlight_symbol(">>") .repeat_highlight_symbol(true) .scroll_padding(20); song_state.select(Some(app.browser.selected)); frame.render_stateful_widget(list, size, &mut song_state); } /// draws playing queue fn draw_queue(frame: &mut Frame, app: &mut App, size: Rect) { let mut queue_state = ListState::default(); let title = Block::default() .title(Title::from("Play Queue".green().bold())) .title( Title::from(format!("Volume: {}%", app.conn.volume).bold().green()) .alignment(Alignment::Right), ); let mut items: Vec = vec![]; for item in app.queue_list.list.iter() { if item.contains(&app.conn.current_song.file) { items.push(ListItem::new(item.clone().magenta().bold())) } else { items.push(ListItem::new(item.clone())); } } let list = List::new(items) .block(title.borders(Borders::ALL)) .highlight_style( Style::new() .fg(Color::Cyan) .bg(Color::Black) .add_modifier(Modifier::BOLD) .add_modifier(Modifier::REVERSED), ) .highlight_symbol(">>"); queue_state.select(Some(app.queue_list.index)); frame.render_stateful_widget(list, size, &mut queue_state); } /// draws all playlists fn draw_playlists(frame: &mut Frame, app: &mut App, size: Rect) { let mut state = ListState::default(); let title = Block::default() .title(Title::from("Playlist".green().bold())) .title( Title::from(format!("Volume: {}%", app.conn.volume).bold().green()) .alignment(Alignment::Right), ); let list = List::new(app.pl_list.list.clone()) .block(title.borders(Borders::ALL)) .highlight_style( Style::new() .fg(Color::Cyan) .bg(Color::Black) .add_modifier(Modifier::BOLD) .add_modifier(Modifier::REVERSED), ) .highlight_symbol(">>") .repeat_highlight_symbol(true); state.select(Some(app.pl_list.index)); frame.render_stateful_widget(list, size, &mut state); } // Draw search bar fn draw_search_bar(frame: &mut Frame, app: &mut App, size: Rect) { match app.inputmode { InputMode::Normal => // Hide the cursor. `Frame` does this by default, so we don't need to do anything here {} InputMode::Editing => { // Make the cursor visible and ask ratatui to put it at the specified coordinates after // rendering #[allow(clippy::cast_possible_truncation)] frame.set_cursor( // Draw the cursor at the current position in the input field. // This position is can be controlled via the left and right arrow key size.x + app.cursor_position as u16 + 2, // Move one line down, from the border to the input line size.y + 1, ); } } let input = Paragraph::new("/".to_string() + &app.search_input) .style(Style::default()) .block( Block::default() .borders(Borders::ALL) .title("Search Forward: ".bold().green()), ); frame.render_widget(input, size); } /// Draws Progress Bar fn draw_progress_bar(frame: &mut Frame, app: &mut App, size: Rect) { // Get the current playing song let song = app .conn .now_playing() .unwrap() .unwrap_or_else(|| "No Title Found".to_string()); // Get the current playing state let mut state: String = String::new(); if !app.queue_list.list.is_empty() { state = app.conn.state.clone(); state.push(':'); } // Get the current modes let mut modes_bottom: String = String::new(); // we do this to check if at least one mode is enabled so we can push "[]" if app.conn.repeat | app.conn.random { modes_bottom.push('r'); } if !modes_bottom.is_empty() { modes_bottom.clear(); modes_bottom.push('['); if app.conn.repeat { modes_bottom.push('r'); } if app.conn.random { modes_bottom.push('z'); } modes_bottom.push(']'); }; // get the duration let duration = if app.conn.total_duration.as_secs() != 0 { format!( "[{}/{}]", humantime::format_duration(app.conn.elapsed), humantime::format_duration(app.conn.total_duration) ) } else { "".to_string() }; // Define the title let title = Block::default() .title(Title::from(format!("{}", state).red().bold())) .title(Title::from(song.green().bold())) .title(Title::from(duration.cyan().bold()).alignment(Alignment::Right)) .title(Title::from(format!("{}", modes_bottom)).position(block::Position::Bottom)) .borders(Borders::ALL); let progress_bar = LineGauge::default() .block(title.borders(Borders::ALL)) .gauge_style( Style::default() .fg(Color::Blue) .bg(Color::Black) .add_modifier(Modifier::BOLD), ) .line_set(symbols::line::THICK) .ratio(app.conn.get_progress_ratio()); frame.render_widget(progress_bar, size); } fn draw_add_to_playlist(frame: &mut Frame, app: &mut App, area: Rect) { let area = centered_rect(40, 50, area); let mut state = ListState::default(); let title = Block::default() .title(Title::from("Add Selected Item to: ")) .title(Title::from(" to Cancel".green().bold()).alignment(Alignment::Right)); let list = List::new(app.append_list.list.clone()) .block(title.borders(Borders::ALL)) .highlight_style( Style::new() .fg(Color::Cyan) .bg(Color::Black) .add_modifier(Modifier::BOLD) .add_modifier(Modifier::REVERSED), ) .highlight_symbol(">>") .repeat_highlight_symbol(true); state.select(Some(app.append_list.index)); frame.render_widget(Clear, area); //this clears out the background frame.render_stateful_widget(list, area, &mut state); } fn centered_rect(percent_x: u16, percent_y: u16, r: Rect) -> Rect { let popup_layout = Layout::vertical([ Constraint::Percentage((100 - percent_y) / 2), Constraint::Percentage(percent_y), Constraint::Percentage((100 - percent_y) / 2), ]) .split(r); Layout::horizontal([ Constraint::Percentage((100 - percent_x) / 2), Constraint::Percentage(percent_x), Constraint::Percentage((100 - percent_x) / 2), ]) .split(popup_layout[1])[1] }