radiate-ui 1.3.1

use super::chart::MetricChartType;
use radiate_engines::Metric;

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum UiMode {
    Dashboard,
    MetricModal,
    Search,
    Help,
}

#[derive(Clone, Copy, PartialEq, Eq)]
pub enum DashboardTab {
    Stats,
    Time,
    Distribution,
    Species,
    Log,
    Front,
    Events,
}

impl DashboardTab {
    pub fn next(self) -> Self {
        match self {
            DashboardTab::Stats => DashboardTab::Time,
            DashboardTab::Time => DashboardTab::Distribution,
            DashboardTab::Distribution => DashboardTab::Species,
            DashboardTab::Species => DashboardTab::Log,
            DashboardTab::Log => DashboardTab::Front,
            DashboardTab::Front => DashboardTab::Events,
            DashboardTab::Events => DashboardTab::Stats,
        }
    }

    pub fn previous(self) -> Self {
        match self {
            DashboardTab::Stats => DashboardTab::Events,
            DashboardTab::Time => DashboardTab::Stats,
            DashboardTab::Distribution => DashboardTab::Time,
            DashboardTab::Species => DashboardTab::Distribution,
            DashboardTab::Log => DashboardTab::Species,
            DashboardTab::Front => DashboardTab::Log,
            DashboardTab::Events => DashboardTab::Front,
        }
    }

    pub fn supports_metric_modal(self) -> bool {
        !matches!(
            self,
            DashboardTab::Species | DashboardTab::Log | DashboardTab::Front | DashboardTab::Events
        )
    }

    /// The focusable panes this tab lays out, in `Tab`-cycle order. Every tab
    /// currently has the same shape: a list, a chart, and a detail panel.
    /// Events is a single full-width list with no chart pane.
    pub fn panes(self) -> &'static [Pane] {
        match self {
            DashboardTab::Events => &[Pane::List],
            _ => &[Pane::List, Pane::Chart],
        }
    }
}

/// A focusable region within the active dashboard tab.
#[derive(Clone, Copy, PartialEq, Eq)]
pub enum Pane {
    List,
    Chart,
}

pub struct SearchState {
    pub query: String,
    pub active: bool,
}

pub struct NavState {
    pub mode: UiMode,
    pub dashboard_tab: DashboardTab,
    pub focus: Pane,
    pub search: SearchState,
    chart_tabs: [MetricChartType; 7],
}

impl NavState {
    pub fn is_pane_focused(&self, pane: Pane) -> bool {
        self.focus == pane && matches!(self.mode, UiMode::Dashboard | UiMode::MetricModal)
    }

    /// The chart view remembered for the active dashboard tab.
    pub fn chart_tab(&self) -> MetricChartType {
        self.chart_tabs[self.dashboard_tab_index()]
    }

    /// Remember `view` for the active dashboard tab.
    pub fn set_chart_tab(&mut self, view: MetricChartType) {
        let idx = self.dashboard_tab_index();
        self.chart_tabs[idx] = view;
    }

    fn clamp_focus(&mut self) {
        let panes = self.dashboard_tab.panes();
        if !panes.contains(&self.focus) {
            self.focus = panes[0];
        }
    }

    pub fn is_search_focused(&self) -> bool {
        self.mode == UiMode::Search
    }

    pub fn open_search(&mut self) {
        if !matches!(self.mode, UiMode::Dashboard) {
            return;
        }

        self.mode = UiMode::Search;
        self.search.active = true;
    }

    pub fn close_search(&mut self) {
        self.mode = UiMode::Dashboard;
        self.search.active = false;
    }

    pub fn clear_search(&mut self) {
        self.search.query.clear();
    }

    pub fn push_search_char(&mut self, c: char) {
        self.search.query.push(c);
    }

    pub fn pop_search_char(&mut self) {
        self.search.query.pop();
    }

    pub fn toggle_help(&mut self) {
        match self.mode {
            UiMode::Help => self.mode = UiMode::Dashboard,
            _ => self.mode = UiMode::Help,
        }
    }

    pub fn toggle_metric_modal(&mut self) {
        match self.mode {
            UiMode::Dashboard if self.dashboard_tab.supports_metric_modal() => {
                self.mode = UiMode::MetricModal;
                self.focus = Pane::Chart;
            }
            UiMode::MetricModal => {
                self.mode = UiMode::Dashboard;
                self.focus = Pane::List;
            }
            _ => {}
        }
    }

    pub fn next_tab(&mut self, has_species: bool, is_multi: bool) {
        if let UiMode::Dashboard = self.mode {
            let mut next = self.dashboard_tab.next();
            while !tab_available(next, has_species, is_multi) {
                next = next.next();
            }
            self.dashboard_tab = next;
            self.clamp_focus();
        }
    }

    pub fn previous_tab(&mut self, has_species: bool, is_multi: bool) {
        if let UiMode::Dashboard = self.mode {
            let mut prev = self.dashboard_tab.previous();
            while !tab_available(prev, has_species, is_multi) {
                prev = prev.previous();
            }
            self.dashboard_tab = prev;
            self.clamp_focus();
        }
    }

    pub fn ensure_tab_available(&mut self, has_species: bool, is_multi: bool) {
        if !tab_available(self.dashboard_tab, has_species, is_multi) {
            self.dashboard_tab = DashboardTab::Stats;
            self.clamp_focus();
        }
    }

    pub fn dashboard_tab_index(&self) -> usize {
        match self.dashboard_tab {
            DashboardTab::Stats => 0,
            DashboardTab::Time => 1,
            DashboardTab::Distribution => 2,
            DashboardTab::Species => 3,
            DashboardTab::Log => 4,
            DashboardTab::Front => 5,
            DashboardTab::Events => 6,
        }
    }

    pub fn clear_search_query(&mut self) {
        if !self.search.active {
            self.search.query.clear();
        }
    }

    pub fn metric_matches_search(&self, metric: &Metric) -> bool {
        let query = self.search.query.trim();
        if query.is_empty() {
            return true;
        }
        let q = query.to_lowercase();
        metric.name().to_lowercase().contains(&q)
            || metric
                .iter_tags()
                .any(|tag| tag.as_str().to_lowercase().contains(&q))
    }
}

impl Default for NavState {
    fn default() -> Self {
        Self {
            mode: UiMode::Dashboard,
            dashboard_tab: DashboardTab::Stats,
            focus: Pane::List,
            chart_tabs: [
                MetricChartType::Mean,
                MetricChartType::Mean,
                MetricChartType::BoxWhisker,
                MetricChartType::Mean,
                MetricChartType::Mean,
                MetricChartType::Mean,
                MetricChartType::Mean,
            ],
            search: SearchState {
                query: String::new(),
                active: false,
            },
        }
    }
}

/// Whether a tab can currently be shown given current run state.
/// Stats/Time/Distribution are always available (guarantees the skip-loops terminate).
fn tab_available(tab: DashboardTab, has_species: bool, is_multi: bool) -> bool {
    match tab {
        DashboardTab::Species => has_species,
        DashboardTab::Log => !is_multi,
        DashboardTab::Front => is_multi,
        _ => true,
    }
}

// /// Step through `panes` from `current` by `dir` (±1), wrapping.
// fn cycle(panes: &[Pane], current: Pane, dir: isize) -> Pane {
//     let n = panes.len() as isize;
//     let i = panes.iter().position(|p| *p == current).unwrap_or(0) as isize;
//     panes[(((i + dir) % n + n) % n) as usize]
// }