rusticity_term/

common.rs

use chrono::{DateTime, Utc};
use ratatui::{prelude::*, widgets::*};
use std::collections::HashMap;
use std::sync::OnceLock;

use crate::ui::{filter_area, styles};

pub type ColumnId = &'static str;

static I18N: OnceLock<HashMap<String, String>> = OnceLock::new();

pub fn set_i18n(map: HashMap<String, String>) {
    I18N.set(map).ok();
}

pub fn t(key: &str) -> String {
    I18N.get()
        .and_then(|map| map.get(key))
        .cloned()
        .unwrap_or_else(|| key.to_string())
}

pub fn translate_column(key: &str, default: &str) -> String {
    let translated = t(key);
    if translated == key {
        default.to_string()
    } else {
        translated
    }
}

// Width for UTC timestamp format: "YYYY-MM-DD HH:MM:SS (UTC)"
pub const UTC_TIMESTAMP_WIDTH: u16 = 27;

pub fn format_timestamp(dt: &DateTime<Utc>) -> String {
    format!("{} (UTC)", dt.format("%Y-%m-%d %H:%M:%S"))
}

pub fn format_optional_timestamp(dt: Option<DateTime<Utc>>) -> String {
    dt.map(|t| format_timestamp(&t))
        .unwrap_or_else(|| "-".to_string())
}

pub fn format_iso_timestamp(iso_string: &str) -> String {
    if iso_string.is_empty() {
        return "-".to_string();
    }

    // Parse ISO 8601 format (e.g., "2024-01-01T12:30:45.123Z")
    if let Ok(dt) = DateTime::parse_from_rfc3339(iso_string) {
        format_timestamp(&dt.with_timezone(&Utc))
    } else {
        iso_string.to_string()
    }
}

pub fn format_unix_timestamp(unix_string: &str) -> String {
    if unix_string.is_empty() {
        return "-".to_string();
    }

    if let Ok(timestamp) = unix_string.parse::<i64>() {
        if let Some(dt) = DateTime::from_timestamp(timestamp, 0) {
            format_timestamp(&dt)
        } else {
            unix_string.to_string()
        }
    } else {
        unix_string.to_string()
    }
}

#[derive(Debug, Clone, Copy, PartialEq)]
pub enum ColumnType {
    String,
    Number,
    DateTime,
    Boolean,
}

pub fn format_bytes(bytes: i64) -> String {
    const KB: i64 = 1000;
    const MB: i64 = KB * 1000;
    const GB: i64 = MB * 1000;
    const TB: i64 = GB * 1000;

    if bytes >= TB {
        format!("{:.2} TB", bytes as f64 / TB as f64)
    } else if bytes >= GB {
        format!("{:.2} GB", bytes as f64 / GB as f64)
    } else if bytes >= MB {
        format!("{:.2} MB", bytes as f64 / MB as f64)
    } else if bytes >= KB {
        format!("{:.2} KB", bytes as f64 / KB as f64)
    } else {
        format!("{} B", bytes)
    }
}

pub fn format_memory_mb(mb: i32) -> String {
    if mb >= 1024 {
        format!("{} GB", mb / 1024)
    } else {
        format!("{} MB", mb)
    }
}

pub fn format_duration_seconds(seconds: i32) -> String {
    if seconds == 0 {
        return "0s".to_string();
    }

    let days = seconds / 86400;
    let hours = (seconds % 86400) / 3600;
    let minutes = (seconds % 3600) / 60;
    let secs = seconds % 60;

    let mut parts = Vec::new();
    if days > 0 {
        parts.push(format!("{}d", days));
    }
    if hours > 0 {
        parts.push(format!("{}h", hours));
    }
    if minutes > 0 {
        parts.push(format!("{}m", minutes));
    }
    if secs > 0 {
        parts.push(format!("{}s", secs));
    }

    parts.join(" ")
}

pub fn border_style(is_active: bool) -> Style {
    if is_active {
        styles::active_border()
    } else {
        Style::default()
    }
}

pub fn render_scrollbar(frame: &mut Frame, area: Rect, total: usize, position: usize) {
    if total == 0 {
        return;
    }
    let scrollbar = Scrollbar::new(ScrollbarOrientation::VerticalRight)
        .begin_symbol(Some("↑"))
        .end_symbol(Some("↓"));
    let mut state = ScrollbarState::new(total).position(position);
    frame.render_stateful_widget(scrollbar, area, &mut state);
}

pub fn render_vertical_scrollbar(frame: &mut Frame, area: Rect, total: usize, position: usize) {
    render_scrollbar(frame, area, total, position);
}

pub fn render_horizontal_scrollbar(frame: &mut Frame, area: Rect, position: usize, total: usize) {
    let scrollbar = Scrollbar::new(ScrollbarOrientation::HorizontalBottom)
        .begin_symbol(Some("◀"))
        .end_symbol(Some("▶"));
    let mut state = ScrollbarState::new(total).position(position);
    frame.render_stateful_widget(scrollbar, area, &mut state);
}

pub fn render_pagination(current: usize, total: usize) -> String {
    if total == 0 {
        return "[1]".to_string();
    }
    if total <= 10 {
        return (0..total)
            .map(|i| {
                if i == current {
                    format!("[{}]", i + 1)
                } else {
                    format!("{}", i + 1)
                }
            })
            .collect::<Vec<_>>()
            .join(" ");
    }
    let start = current.saturating_sub(4);
    let end = (start + 9).min(total);
    let start = if end == total {
        total.saturating_sub(9)
    } else {
        start
    };
    (start..end)
        .map(|i| {
            if i == current {
                format!("[{}]", i + 1)
            } else {
                format!("{}", i + 1)
            }
        })
        .collect::<Vec<_>>()
        .join(" ")
}

/// Renders pagination with unknown total (infinite pagination)
/// Shows: 1 ... 6 7 8 9 [10] 11 ...
pub fn render_infinite_pagination(current: usize) -> String {
    let mut parts = Vec::new();

    // Show 4 pages before current (or from page 1)
    let start = current.saturating_sub(4);

    // If start > 1, show page 1 and ...
    if start > 1 {
        parts.push("1".to_string());
        parts.push("...".to_string());
    }

    // Show pages from start to current
    for i in start..current {
        parts.push(format!("{}", i + 1));
    }

    // Current page
    parts.push(format!("[{}]", current + 1));

    // Show 1 page after current
    parts.push(format!("{}", current + 2));

    // Always show ... at the end to indicate more pages
    parts.push("...".to_string());

    parts.join(" ")
}

pub fn render_pagination_text(current: usize, total: usize) -> String {
    render_pagination(current, total)
}

/// Show only navigable, loaded pages, with an unknown remainder. Unlike infinite
/// pagination, this does not invent a next page that has not been loaded yet.
pub fn render_partial_pagination(current: usize, loaded_pages: usize) -> String {
    if loaded_pages == 0 {
        "...".to_string()
    } else {
        format!("{} ...", render_pagination_text(current, loaded_pages))
    }
}

pub fn render_dropdown<T: AsRef<str>>(
    frame: &mut ratatui::Frame,
    items: &[T],
    selected_index: usize,
    filter_area: ratatui::prelude::Rect,
    controls_after_width: u16,
) {
    use ratatui::prelude::*;
    use ratatui::widgets::{Block, BorderType, Borders, Clear, List, ListItem};

    let max_width = items
        .iter()
        .map(|item| item.as_ref().len())
        .max()
        .unwrap_or(10) as u16
        + 4;

    let dropdown_items: Vec<ListItem> = items
        .iter()
        .enumerate()
        .map(|(idx, item)| {
            let style = if idx == selected_index {
                Style::default().fg(Color::Yellow).bold()
            } else {
                Style::default().fg(Color::White)
            };
            ListItem::new(format!(" {} ", item.as_ref())).style(style)
        })
        .collect();

    let dropdown_height = dropdown_items.len() as u16 + 2;
    let dropdown_width = max_width;
    let dropdown_x = filter_area
        .x
        .saturating_add(filter_area.width)
        .saturating_sub(controls_after_width + dropdown_width);

    // Show every option when there is room below the filter row, so the whole
    // list is visible while the dropdown is open.
    let y = filter_area.y + filter_area.height;
    let space_below = frame.area().height.saturating_sub(y);
    let dropdown_area = Rect {
        x: dropdown_x,
        y,
        width: dropdown_width,
        height: dropdown_height.min(space_below.max(3)),
    };

    // Clear the background first
    frame.render_widget(Clear, dropdown_area);

    frame.render_widget(
        List::new(dropdown_items)
            .block(
                Block::default()
                    .borders(Borders::ALL)
                    .border_type(BorderType::Rounded)
                    .border_style(Style::default().fg(Color::Yellow)),
            )
            .style(Style::default().bg(Color::Black)),
        dropdown_area,
    );
}

pub struct FilterConfig<'a> {
    pub text: &'a str,
    pub placeholder: &'a str,
    pub is_active: bool,
    pub right_content: Vec<(&'a str, &'a str)>,
    pub area: Rect,
}

pub struct FilterAreaConfig<'a> {
    pub filter_text: &'a str,
    pub placeholder: &'a str,
    pub mode: crate::keymap::Mode,
    pub input_focus: FilterFocusType,
    pub controls: Vec<FilterControl>,
    pub area: Rect,
}

#[derive(Debug, Clone, Copy, PartialEq, Default)]
pub enum SortDirection {
    #[default]
    Asc,
    Desc,
}

impl SortDirection {
    pub fn as_str(&self) -> &'static str {
        match self {
            SortDirection::Asc => "ASC",
            SortDirection::Desc => "DESC",
        }
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Default)]
pub enum InputFocus {
    #[default]
    Filter,
    Pagination,
    Dropdown(&'static str),
    Checkbox(&'static str),
}

impl InputFocus {
    pub fn next(&self, controls: &[InputFocus]) -> Self {
        if controls.is_empty() {
            return *self;
        }
        let idx = controls.iter().position(|f| f == self).unwrap_or(0);
        controls[(idx + 1) % controls.len()]
    }

    pub fn prev(&self, controls: &[InputFocus]) -> Self {
        if controls.is_empty() {
            return *self;
        }
        let idx = controls.iter().position(|f| f == self).unwrap_or(0);
        controls[(idx + controls.len() - 1) % controls.len()]
    }

    /// Navigate to next page when pagination is focused
    pub fn handle_page_down(
        &self,
        selected: &mut usize,
        scroll_offset: &mut usize,
        page_size: usize,
        filtered_count: usize,
    ) {
        if *self == InputFocus::Pagination {
            // max = start of last page, e.g. 754 items / 50 per page → last page starts at 750
            let last_page_start = if filtered_count > page_size {
                ((filtered_count - 1) / page_size) * page_size
            } else {
                0
            };
            *selected = (*selected + page_size).min(last_page_start);
            *scroll_offset = *selected;
        }
    }

    /// Navigate to previous page when pagination is focused
    pub fn handle_page_up(
        &self,
        selected: &mut usize,
        scroll_offset: &mut usize,
        page_size: usize,
    ) {
        if *self == InputFocus::Pagination {
            *selected = selected.saturating_sub(page_size);
            *scroll_offset = *selected;
        }
    }
}

pub trait CyclicEnum: Copy + PartialEq + Sized + 'static {
    const ALL: &'static [Self];

    fn next(&self) -> Self {
        let idx = Self::ALL.iter().position(|x| x == self).unwrap_or(0);
        Self::ALL[(idx + 1) % Self::ALL.len()]
    }

    fn prev(&self) -> Self {
        let idx = Self::ALL.iter().position(|x| x == self).unwrap_or(0);
        Self::ALL[(idx + Self::ALL.len() - 1) % Self::ALL.len()]
    }
}

#[derive(PartialEq)]
pub enum FilterFocusType {
    Input,
    Control(usize),
}

pub struct FilterControl {
    pub text: String,
    pub is_focused: bool,
    pub style: ratatui::style::Style,
}

pub fn render_filter_area(frame: &mut Frame, config: FilterAreaConfig) {
    use crate::keymap::Mode;
    use crate::ui::get_cursor;
    use ratatui::prelude::*;

    let cursor = get_cursor(
        config.mode == Mode::FilterInput && config.input_focus == FilterFocusType::Input,
    );
    let filter_width = config.area.width.saturating_sub(4) as usize;

    // Calculate controls text
    let controls_text: String = config
        .controls
        .iter()
        .map(|c| c.text.as_str())
        .collect::<Vec<_>>()
        .join(" ⋮ ");
    let controls_len = controls_text.len();

    let placeholder_len = config.placeholder.len();
    let content_len =
        if config.filter_text.is_empty() && config.mode != Mode::FilterInput {
            placeholder_len
        } else {
            config.filter_text.len()
        } + if config.mode == Mode::FilterInput && config.input_focus == FilterFocusType::Input {
            cursor.len()
        } else {
            0
        };

    let available_space = filter_width.saturating_sub(controls_len + 1);

    let mut line_spans = vec![];
    if config.filter_text.is_empty() {
        if config.mode == Mode::FilterInput {
            line_spans.push(Span::raw(""));
        } else {
            line_spans.push(Span::styled(
                config.placeholder,
                Style::default().fg(Color::DarkGray),
            ));
        }
    } else {
        line_spans.push(Span::raw(config.filter_text));
    }

    if config.mode == Mode::FilterInput && config.input_focus == FilterFocusType::Input {
        line_spans.push(Span::styled(cursor, Style::default().fg(Color::Yellow)));
    }

    if content_len < available_space {
        line_spans.push(Span::raw(" ".repeat(available_space - content_len)));
    }

    if config.mode == Mode::FilterInput {
        for control in &config.controls {
            line_spans.push(Span::raw(" ⋮ "));
            line_spans.push(Span::styled(&control.text, control.style));
        }
    } else {
        line_spans.push(Span::styled(
            format!(" ⋮ {}", controls_text),
            Style::default(),
        ));
    }

    let filter = filter_area(line_spans, config.mode == Mode::FilterInput);
    frame.render_widget(filter, config.area);
}

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

    #[test]
    fn test_format_timestamp() {
        let dt = Utc.with_ymd_and_hms(2025, 11, 12, 14, 30, 45).unwrap();
        assert_eq!(format_timestamp(&dt), "2025-11-12 14:30:45 (UTC)");
    }

    #[test]
    fn test_format_optional_timestamp_some() {
        let dt = Utc.with_ymd_and_hms(2025, 11, 12, 14, 30, 45).unwrap();
        assert_eq!(
            format_optional_timestamp(Some(dt)),
            "2025-11-12 14:30:45 (UTC)"
        );
    }

    #[test]
    fn test_format_optional_timestamp_none() {
        assert_eq!(format_optional_timestamp(None), "-");
    }

    #[test]
    fn test_format_bytes() {
        assert_eq!(format_bytes(500), "500 B");
        assert_eq!(format_bytes(1500), "1.50 KB");
        assert_eq!(format_bytes(1_500_000), "1.50 MB");
        assert_eq!(format_bytes(1_500_000_000), "1.50 GB");
        assert_eq!(format_bytes(1_500_000_000_000), "1.50 TB");
    }

    #[test]
    fn test_format_duration_seconds_zero() {
        assert_eq!(format_duration_seconds(0), "0s");
    }

    #[test]
    fn test_render_infinite_pagination_page_1() {
        // Page 1: [1] 2 ...
        let result = render_infinite_pagination(0);
        assert_eq!(result, "[1] 2 ...");
    }

    #[test]
    fn test_render_infinite_pagination_page_5() {
        // Page 5: 1 2 3 4 [5] 6 ...
        let result = render_infinite_pagination(4);
        assert_eq!(result, "1 2 3 4 [5] 6 ...");
    }

    #[test]
    fn test_render_infinite_pagination_page_6() {
        // Page 6: 2 3 4 5 [6] 7 ... (start=2, no need for "1 ...")
        let result = render_infinite_pagination(5);
        assert_eq!(result, "2 3 4 5 [6] 7 ...");
    }

    #[test]
    fn test_render_infinite_pagination_page_7() {
        // Page 7: 1 ... 3 4 5 6 [7] 8 ...
        let result = render_infinite_pagination(6);
        assert_eq!(result, "1 ... 3 4 5 6 [7] 8 ...");
    }

    #[test]
    fn test_render_infinite_pagination_page_10() {
        // Page 10: 1 ... 6 7 8 9 [10] 11 ...
        let result = render_infinite_pagination(9);
        assert_eq!(result, "1 ... 6 7 8 9 [10] 11 ...");
    }

    #[test]
    fn test_render_infinite_pagination_page_100() {
        // Page 100: 1 ... 96 97 98 99 [100] 101 ...
        let result = render_infinite_pagination(99);
        assert_eq!(result, "1 ... 96 97 98 99 [100] 101 ...");
    }

    #[test]
    fn test_format_duration_seconds_only_seconds() {
        assert_eq!(format_duration_seconds(30), "30s");
    }

    #[test]
    fn test_format_duration_seconds_minutes_and_seconds() {
        assert_eq!(format_duration_seconds(120), "2m");
        assert_eq!(format_duration_seconds(150), "2m 30s");
    }

    #[test]
    fn test_format_duration_seconds_hours() {
        assert_eq!(format_duration_seconds(3630), "1h 30s");
        assert_eq!(format_duration_seconds(10800), "3h");
    }

    #[test]
    fn test_format_duration_seconds_days() {
        assert_eq!(format_duration_seconds(90061), "1d 1h 1m 1s");
        assert_eq!(format_duration_seconds(345600), "4d");
    }

    #[test]
    fn test_format_duration_seconds_complex() {
        assert_eq!(format_duration_seconds(1800), "30m");
        assert_eq!(format_duration_seconds(86400), "1d");
    }

    #[test]
    fn test_render_pagination_single_page() {
        assert_eq!(render_pagination(0, 1), "[1]");
    }

    #[test]
    fn test_render_pagination_two_pages() {
        assert_eq!(render_pagination(0, 2), "[1] 2");
        assert_eq!(render_pagination(1, 2), "1 [2]");
    }

    #[test]
    fn test_render_pagination_ten_pages() {
        assert_eq!(render_pagination(0, 10), "[1] 2 3 4 5 6 7 8 9 10");
        assert_eq!(render_pagination(5, 10), "1 2 3 4 5 [6] 7 8 9 10");
        assert_eq!(render_pagination(9, 10), "1 2 3 4 5 6 7 8 9 [10]");
    }

    #[test]
    fn test_format_memory_mb() {
        assert_eq!(format_memory_mb(128), "128 MB");
        assert_eq!(format_memory_mb(512), "512 MB");
        assert_eq!(format_memory_mb(1024), "1 GB");
        assert_eq!(format_memory_mb(2048), "2 GB");
    }

    #[test]
    fn test_render_pagination_many_pages() {
        assert_eq!(render_pagination(0, 20), "[1] 2 3 4 5 6 7 8 9");
        assert_eq!(render_pagination(5, 20), "2 3 4 5 [6] 7 8 9 10");
        assert_eq!(render_pagination(15, 20), "12 13 14 15 [16] 17 18 19 20");
        assert_eq!(render_pagination(19, 20), "12 13 14 15 16 17 18 19 [20]");
    }

    #[test]
    fn test_render_pagination_zero_total() {
        assert_eq!(render_pagination(0, 0), "[1]");
    }

    #[test]
    fn test_render_dropdown_items_format() {
        let items = ["us-east-1", "us-west-2", "eu-west-1"];
        assert_eq!(items.len(), 3);
        assert_eq!(items[0], "us-east-1");
        assert_eq!(items[2], "eu-west-1");
    }

    #[test]
    fn test_render_dropdown_selected_index() {
        let items = ["item1", "item2", "item3"];
        let selected = 1;
        assert_eq!(items[selected], "item2");
    }

    #[test]
    fn test_render_dropdown_controls_after_width() {
        let pagination_len = 10;
        let separator = 3;
        let controls_after = pagination_len + separator;
        assert_eq!(controls_after, 13);
    }

    #[test]
    fn test_render_dropdown_multiple_controls_after() {
        let view_nested_width = 15;
        let pagination_len = 10;
        let controls_after = view_nested_width + 3 + pagination_len + 3;
        assert_eq!(controls_after, 31);
    }

    #[test]
    fn test_render_dropdown_clears_background() {
        // This test verifies that render_dropdown uses Clear widget
        // The actual rendering is tested via integration tests
        // Here we just verify the function can be called with valid parameters
        use ratatui::backend::TestBackend;
        use ratatui::Terminal;

        let backend = TestBackend::new(80, 24);
        let mut terminal = Terminal::new(backend).unwrap();

        terminal
            .draw(|frame| {
                let area = ratatui::prelude::Rect {
                    x: 0,
                    y: 0,
                    width: 80,
                    height: 3,
                };
                let items = ["Running", "Stopped", "Terminated"];
                render_dropdown(frame, &items, 0, area, 10);
            })
            .unwrap();

        // If we get here without panic, the function works correctly
        // The Clear widget is used internally to clear the background
    }
}

pub fn render_filter(frame: &mut Frame, config: FilterConfig) {
    let cursor = if config.is_active { "█" } else { "" };
    let content = if config.text.is_empty() && !config.is_active {
        config.placeholder
    } else {
        config.text
    };

    let right_text = config
        .right_content
        .iter()
        .map(|(k, v)| format!("{}: {}", k, v))
        .collect::<Vec<_>>()
        .join(" ⋮ ");

    let width = (config.area.width as usize).saturating_sub(4);
    let right_len = right_text.len();
    let content_len = content.len() + if config.is_active { cursor.len() } else { 0 };
    let available = width.saturating_sub(right_len + 3);

    let display = if content_len > available {
        &content[content_len.saturating_sub(available)..]
    } else {
        content
    };

    let style = if config.is_active {
        styles::yellow()
    } else {
        styles::placeholder()
    };

    let mut spans = vec![Span::styled(display, style)];
    if config.is_active {
        spans.push(Span::styled(cursor, styles::cursor()));
    }

    let padding = " ".repeat(
        width
            .saturating_sub(display.len())
            .saturating_sub(if config.is_active { cursor.len() } else { 0 })
            .saturating_sub(right_len)
            .saturating_sub(3),
    );

    spans.push(Span::raw(padding));
    spans.push(Span::styled(format!(" {}", right_text), styles::cyan()));

    frame.render_widget(
        Paragraph::new(Line::from(spans)).block(
            Block::default()
                .borders(Borders::ALL)
                .border_style(border_style(config.is_active)),
        ),
        config.area,
    );
}

#[derive(Debug, Clone, Copy, PartialEq)]
pub enum PageSize {
    Ten,
    TwentyFive,
    Fifty,
    OneHundred,
}

/// Generic helper to filter items by a field that matches a filter string (case-insensitive contains)
pub fn filter_by_field<'a, T, F>(items: &'a [T], filter: &str, get_field: F) -> Vec<&'a T>
where
    F: Fn(&T) -> &str,
{
    if filter.is_empty() {
        items.iter().collect()
    } else {
        let filter_lower = filter.to_lowercase();
        items
            .iter()
            .filter(|item| get_field(item).to_lowercase().contains(&filter_lower))
            .collect()
    }
}

/// Generic helper to filter items by multiple fields (case-insensitive contains on any field)
pub fn filter_by_fields<'a, T, F>(items: &'a [T], filter: &str, get_fields: F) -> Vec<&'a T>
where
    F: Fn(&T) -> Vec<&str>,
{
    if filter.is_empty() {
        items.iter().collect()
    } else {
        let filter_lower = filter.to_lowercase();
        items
            .iter()
            .filter(|item| {
                get_fields(item)
                    .iter()
                    .any(|field| field.to_lowercase().contains(&filter_lower))
            })
            .collect()
    }
}

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

    #[test]
    fn partial_pagination_never_invents_an_unloaded_page() {
        assert_eq!(render_partial_pagination(0, 0), "...");
        assert_eq!(render_partial_pagination(0, 1), "[1] ...");
        assert_eq!(render_partial_pagination(0, 2), "[1] 2 ...");
        assert_eq!(render_partial_pagination(1, 2), "1 [2] ...");
        assert_eq!(
            render_partial_pagination(19, 20),
            "12 13 14 15 16 17 18 19 [20] ..."
        );
    }

    #[test]
    fn finite_pagination_is_unchanged() {
        assert_eq!(render_pagination_text(0, 2), "[1] 2");
        assert_eq!(render_pagination_text(3, 4), "1 2 3 [4]");
        assert_eq!(render_pagination_text(0, 0), "[1]");
    }
}