rpi-tui 0.1.4

//! SelectList component - a selectable list with filtering.
//!
//! Provides a list of items that can be navigated and selected.

use std::any::Any;
use std::sync::{Arc, Mutex};

use super::component::Component;
use super::fuzzy::fuzzy_filter;
use crate::utils::{truncate_to_width, visible_width};

const DEFAULT_PRIMARY_COLUMN_WIDTH: usize = 32;
const PRIMARY_COLUMN_GAP: usize = 2;
const MIN_DESCRIPTION_WIDTH: usize = 10;

/// A select list item.
#[derive(Debug, Clone)]
pub struct SelectItem {
    /// The value (used for filtering and returned on selection).
    pub value: String,
    /// Display label.
    pub label: String,
    /// Optional description.
    pub description: Option<String>,
}

impl SelectItem {
    /// Create a new select item.
    pub fn new(value: &str, label: &str) -> Self {
        Self {
            value: value.to_string(),
            label: label.to_string(),
            description: None,
        }
    }

    /// Add a description.
    pub fn with_description(mut self, description: &str) -> Self {
        self.description = Some(description.to_string());
        self
    }

    /// Get the display value (label or value).
    pub fn display_value(&self) -> &str {
        if self.label.is_empty() {
            &self.value
        } else {
            &self.label
        }
    }
}

/// Theme for select list.
#[derive(Debug, Clone)]
pub struct SelectListTheme {
    /// Style for selected item prefix.
    pub selected_prefix: fn(&str) -> String,
    /// Style for selected item text.
    pub selected_text: fn(&str) -> String,
    /// Style for description text.
    pub description: fn(&str) -> String,
    /// Style for scroll info.
    pub scroll_info: fn(&str) -> String,
    /// Style for no match message.
    pub no_match: fn(&str) -> String,
}

impl Default for SelectListTheme {
    fn default() -> Self {
        Self {
            selected_prefix: |s| format!("\x1b[32m{}\x1b[0m", s),
            selected_text: |s| format!("\x1b[1;36m{}\x1b[0m", s),
            description: |s| format!("\x1b[90m{}\x1b[0m", s),
            scroll_info: |s| format!("\x1b[90m{}\x1b[0m", s),
            no_match: |s| format!("\x1b[90m{}\x1b[0m", s),
        }
    }
}

/// Layout options for select list.
#[derive(Clone)]
pub struct SelectListLayoutOptions {
    /// Minimum width for the primary column.
    pub min_primary_column_width: Option<usize>,
    /// Maximum width for the primary column.
    pub max_primary_column_width: Option<usize>,
    /// Custom truncation function.
    pub truncate_primary: Option<Arc<dyn Fn(&str, usize, usize) -> String + Send + Sync>>,
}

impl Default for SelectListLayoutOptions {
    fn default() -> Self {
        Self {
            min_primary_column_width: None,
            max_primary_column_width: None,
            truncate_primary: None,
        }
    }
}

/// SelectList component.
pub struct SelectList {
    items: Mutex<Vec<SelectItem>>,
    filtered_items: Mutex<Vec<SelectItem>>,
    selected_index: Mutex<usize>,
    max_visible: usize,
    theme: SelectListTheme,
    layout: SelectListLayoutOptions,
    on_select: Mutex<Option<Arc<dyn Fn(&SelectItem) + Send + Sync>>>,
    on_cancel: Mutex<Option<Arc<dyn Fn() + Send + Sync>>>,
    on_selection_change: Mutex<Option<Arc<dyn Fn(&SelectItem) + Send + Sync>>>,
}

impl SelectList {
    /// Create a new select list.
    pub fn new(items: Vec<SelectItem>, max_visible: usize) -> Self {
        Self {
            filtered_items: Mutex::new(items.clone()),
            items: Mutex::new(items),
            selected_index: Mutex::new(0),
            max_visible,
            theme: SelectListTheme::default(),
            layout: SelectListLayoutOptions::default(),
            on_select: Mutex::new(None),
            on_cancel: Mutex::new(None),
            on_selection_change: Mutex::new(None),
        }
    }

    /// Create a select list with a theme.
    pub fn with_theme(items: Vec<SelectItem>, max_visible: usize, theme: SelectListTheme) -> Self {
        Self {
            filtered_items: Mutex::new(items.clone()),
            items: Mutex::new(items),
            selected_index: Mutex::new(0),
            max_visible,
            theme,
            layout: SelectListLayoutOptions::default(),
            on_select: Mutex::new(None),
            on_cancel: Mutex::new(None),
            on_selection_change: Mutex::new(None),
        }
    }

    /// Set the filter text.
    pub fn set_filter(&self, filter: &str) {
        if let Ok(items) = self.items.lock() {
            let filtered = fuzzy_filter(&items, filter, |item| item.value.as_str());
            if let Ok(mut filtered_items) = self.filtered_items.lock() {
                *filtered_items = filtered;
            }
        }
        // Reset selection when filter changes
        if let Ok(mut index) = self.selected_index.lock() {
            *index = 0;
        }
    }

    /// Set the selected index.
    pub fn set_selected_index(&self, index: usize) {
        if let Ok(mut current) = self.selected_index.lock() {
            if let Ok(filtered) = self.filtered_items.lock() {
                *current = index.min(filtered.len().saturating_sub(1));
            }
        }
    }

    /// Get the selected item.
    pub fn get_selected_item(&self) -> Option<SelectItem> {
        let index = self.selected_index.lock().ok()?;
        let filtered = self.filtered_items.lock().ok()?;
        filtered.get(*index).cloned()
    }

    /// Set callback for when an item is selected.
    pub fn on_select(&self, callback: Arc<dyn Fn(&SelectItem) + Send + Sync>) {
        if let Ok(mut cb) = self.on_select.lock() {
            *cb = Some(callback);
        }
    }

    /// Set callback for when cancelled.
    pub fn on_cancel(&self, callback: Arc<dyn Fn() + Send + Sync>) {
        if let Ok(mut cb) = self.on_cancel.lock() {
            *cb = Some(callback);
        }
    }

    /// Set callback for when selection changes.
    pub fn on_selection_change(&self, callback: Arc<dyn Fn(&SelectItem) + Send + Sync>) {
        if let Ok(mut cb) = self.on_selection_change.lock() {
            *cb = Some(callback);
        }
    }

    /// Handle key input.
    pub fn handle_key(&self, key: crossterm::event::KeyEvent) {
        use crossterm::event::{KeyCode, KeyModifiers};

        match key.code {
            KeyCode::Up => {
                self.move_up();
            }
            KeyCode::Down => {
                self.move_down();
            }
            KeyCode::Enter => {
                self.confirm();
            }
            KeyCode::Esc => {
                self.cancel();
            }
            KeyCode::Char('k') if key.modifiers.contains(KeyModifiers::CONTROL) => {
                self.move_up();
            }
            KeyCode::Char('j') if key.modifiers.contains(KeyModifiers::CONTROL) => {
                self.move_down();
            }
            KeyCode::Char('p') if key.modifiers.contains(KeyModifiers::CONTROL) => {
                self.move_up();
            }
            KeyCode::Char('n') if key.modifiers.contains(KeyModifiers::CONTROL) => {
                self.move_down();
            }
            _ => {}
        }
    }

    /// Move selection up.
    fn move_up(&self) {
        if let Ok(mut index) = self.selected_index.lock() {
            if let Ok(filtered) = self.filtered_items.lock() {
                if filtered.is_empty() {
                    return;
                }
                if *index == 0 {
                    *index = filtered.len() - 1;
                } else {
                    *index -= 1;
                }
                if let Some(item) = filtered.get(*index) {
                    self.notify_selection_change(item);
                }
            }
        }
    }

    /// Move selection down.
    fn move_down(&self) {
        if let Ok(mut index) = self.selected_index.lock() {
            if let Ok(filtered) = self.filtered_items.lock() {
                if filtered.is_empty() {
                    return;
                }
                if *index >= filtered.len() - 1 {
                    *index = 0;
                } else {
                    *index += 1;
                }
                if let Some(item) = filtered.get(*index) {
                    self.notify_selection_change(item);
                }
            }
        }
    }

    /// Confirm selection.
    fn confirm(&self) {
        if let Some(item) = self.get_selected_item() {
            if let Ok(cb) = self.on_select.lock() {
                if let Some(callback) = cb.as_ref() {
                    callback(&item);
                }
            }
        }
    }

    /// Cancel selection.
    fn cancel(&self) {
        if let Ok(cb) = self.on_cancel.lock() {
            if let Some(callback) = cb.as_ref() {
                callback();
            }
        }
    }

    /// Notify selection change.
    fn notify_selection_change(&self, item: &SelectItem) {
        if let Ok(cb) = self.on_selection_change.lock() {
            if let Some(callback) = cb.as_ref() {
                callback(item);
            }
        }
    }

    /// Get primary column width.
    fn get_primary_column_width(&self, filtered: &[SelectItem]) -> usize {
        let (min_width, max_width) = self.get_primary_column_bounds();
        
        let widest = filtered
            .iter()
            .map(|item| visible_width(item.display_value()) + PRIMARY_COLUMN_GAP)
            .max()
            .unwrap_or(DEFAULT_PRIMARY_COLUMN_WIDTH);

        widest.clamp(min_width, max_width)
    }

    /// Get primary column bounds.
    fn get_primary_column_bounds(&self) -> (usize, usize) {
        let raw_min = self.layout.min_primary_column_width
            .or(self.layout.max_primary_column_width)
            .unwrap_or(DEFAULT_PRIMARY_COLUMN_WIDTH);
        let raw_max = self.layout.max_primary_column_width
            .or(self.layout.min_primary_column_width)
            .unwrap_or(DEFAULT_PRIMARY_COLUMN_WIDTH);

        let min = raw_min.min(raw_max).max(1);
        let max = raw_min.max(raw_max).max(1);
        (min, max)
    }

    /// Render an item.
    fn render_item(&self, item: &SelectItem, is_selected: bool, width: usize, primary_column_width: usize) -> String {
        let prefix = if is_selected { "→ " } else { "  " };
        let prefix_width = visible_width(prefix);

        // Handle description
        if let Some(ref description) = item.description {
            if width > 40 {
                let effective_primary = primary_column_width.min(width.saturating_sub(prefix_width + 4));
                let max_primary = effective_primary.saturating_sub(PRIMARY_COLUMN_GAP).max(1);
                
                let truncated_value = self.truncate_primary(item, is_selected, max_primary, effective_primary);
                let value_width = visible_width(&truncated_value);
                let spacing = " ".repeat(effective_primary.saturating_sub(value_width).max(1));
                
                let desc_start = prefix_width + value_width + spacing.len();
                let remaining = width.saturating_sub(desc_start + 2);

                if remaining > MIN_DESCRIPTION_WIDTH {
                    let truncated_desc = truncate_to_width(&normalize_single_line(description), remaining, "");
                    if is_selected {
                        return (self.theme.selected_text)(&format!("{}{}{}{}", prefix, truncated_value, spacing, truncated_desc));
                    }
                    let desc_text = (self.theme.description)(&format!("{}{}", spacing, truncated_desc));
                    return format!("{}{}{}", prefix, truncated_value, desc_text);
                }
            }
        }

        // No description or not enough space
        let max_width = width.saturating_sub(prefix_width + 2);
        let truncated = self.truncate_primary(item, is_selected, max_width, max_width);
        
        if is_selected {
            (self.theme.selected_text)(&format!("{}{}", prefix, truncated))
        } else {
            format!("{}{}", prefix, truncated)
        }
    }

    /// Truncate primary text.
    fn truncate_primary(&self, item: &SelectItem, _is_selected: bool, max_width: usize, column_width: usize) -> String {
        let display = item.display_value();
        
        if let Some(ref truncate_fn) = self.layout.truncate_primary {
            truncate_fn(display, max_width, column_width)
        } else {
            truncate_to_width(display, max_width, "")
        }
    }
}

impl Component for SelectList {
    fn render(&self, width: usize) -> Vec<String> {
        let filtered = if let Ok(filtered) = self.filtered_items.lock() {
            filtered.clone()
        } else {
            return Vec::new();
        };

        let mut lines = Vec::new();

        if filtered.is_empty() {
            lines.push((self.theme.no_match)("  No matching items"));
            return lines;
        }

        let selected = self.selected_index.lock().unwrap_or_else(|e| e.into_inner());
        let primary_column_width = self.get_primary_column_width(&filtered);

        // Calculate visible range with scrolling
        let start_index = if filtered.len() <= self.max_visible {
            0
        } else {
            (*selected)
                .saturating_sub(self.max_visible / 2)
                .min(filtered.len().saturating_sub(self.max_visible))
        };
        let end_index = (start_index + self.max_visible).min(filtered.len());

        // Render visible items
        for i in start_index..end_index {
            if let Some(item) = filtered.get(i) {
                let is_selected = i == *selected;
                lines.push(self.render_item(item, is_selected, width, primary_column_width));
            }
        }

        // Add scroll indicator if needed
        if start_index > 0 || end_index < filtered.len() {
            let scroll_text = format!("  ({}/{})", *selected + 1, filtered.len());
            lines.push((self.theme.scroll_info)(&truncate_to_width(&scroll_text, width.saturating_sub(2), "")));
        }

        lines
    }

    fn invalidate(&self) {
        // No cached state
    }

    fn as_any(&self) -> &dyn Any {
        self
    }
}

/// Normalize text to a single line.
fn normalize_single_line(text: &str) -> String {
    text.replace(|c: char| c == '\r' || c == '\n', " ").trim().to_string()
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_select_list_new() {
        let items = vec![
            SelectItem::new("one", "One"),
            SelectItem::new("two", "Two"),
        ];
        let list = SelectList::new(items, 5);
        
        let lines = list.render(40);
        assert_eq!(lines.len(), 2);
    }

    #[test]
    fn test_select_list_filter() {
        let items = vec![
            SelectItem::new("apple", "Apple"),
            SelectItem::new("banana", "Banana"),
            SelectItem::new("apricot", "Apricot"),
        ];
        let list = SelectList::new(items, 5);
        
        list.set_filter("ap");
        let lines = list.render(40);
        assert_eq!(lines.len(), 2); // Apple and Apricot
    }

    #[test]
    fn test_select_list_navigation() {
        let items = vec![
            SelectItem::new("one", "One"),
            SelectItem::new("two", "Two"),
        ];
        let list = SelectList::new(items, 5);
        
        assert_eq!(list.get_selected_item().unwrap().value, "one");
        
        list.set_selected_index(1);
        assert_eq!(list.get_selected_item().unwrap().value, "two");
    }

    #[test]
    fn test_select_item() {
        let item = SelectItem::new("value", "Label")
            .with_description("Description");
        
        assert_eq!(item.value, "value");
        assert_eq!(item.label, "Label");
        assert_eq!(item.description, Some("Description".to_string()));
        assert_eq!(item.display_value(), "Label");
    }
}