all-smi 0.26.2

// Copyright 2025 Lablup Inc. and Jeongkyu Shin
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use crossterm::{
    queue,
    style::{Color, Print},
};
use std::io::Write;

use crate::app_state::AppState;
use crate::ui::text::print_colored_text;

/// Reserved tab name for the cluster-wide Users tab (issue #189).
///
/// Stored in `AppState::tabs` as a literal string so the existing
/// `Vec<String>` representation keeps working without a breaking
/// enum-based rewrite; the renderer treats this name specially and
/// skips the GPU / storage sections when it is active.
pub const USERS_TAB_NAME: &str = "Users";

/// Reserved tab name for the per-host Topology tab (issue #190).
///
/// Like [`USERS_TAB_NAME`] this is stored as a literal string in
/// `AppState::tabs` so the renderer can special-case it without
/// touching the `Vec<String>` shape.  The Topology tab renders a single
/// host's NvLink / NUMA / PCIe topology; the event handler's `T`
/// binding jumps to this tab.
pub const TOPOLOGY_TAB_NAME: &str = "Topology";

/// Return the index of the Users tab inside `tabs`, or `None` when the
/// tab has not been inserted yet (local mode, replay streams that do
/// not carry process rows).
#[inline]
pub fn users_tab_index(tabs: &[String]) -> Option<usize> {
    tabs.iter().position(|t| t == USERS_TAB_NAME)
}

/// True when `state.current_tab` is the Users tab.
#[inline]
pub fn is_users_tab_active(state: &AppState) -> bool {
    users_tab_index(&state.tabs).is_some_and(|i| i == state.current_tab)
}

/// Return the index of the Topology tab inside `tabs`, or `None` when
/// the tab has not been inserted yet.
#[inline]
pub fn topology_tab_index(tabs: &[String]) -> Option<usize> {
    tabs.iter().position(|t| t == TOPOLOGY_TAB_NAME)
}

/// True when `state.current_tab` is the Topology tab.
#[inline]
pub fn is_topology_tab_active(state: &AppState) -> bool {
    topology_tab_index(&state.tabs).is_some_and(|i| i == state.current_tab)
}

/// True when `name` is a reserved cluster-level tab ("All", Users,
/// Topology) rather than a per-host tab. Used by callers that want to
/// count or iterate only the host tabs without hard-coding the reserved
/// names at the call site (see issue raised when 50 hosts displayed as
/// `50/52` because the dashboard counted reserved tabs as nodes).
#[inline]
pub fn is_reserved_tab(name: &str) -> bool {
    matches!(name, "All" | USERS_TAB_NAME | TOPOLOGY_TAB_NAME)
}

/// Count host tabs in `tabs`, excluding the reserved cluster-level tabs
/// ("All", Users, Topology). The dashboard's `live/total nodes` figure
/// uses this so newly added cluster-level tabs cannot inflate the
/// denominator again.
#[inline]
pub fn host_tab_count(tabs: &[String]) -> usize {
    tabs.iter().filter(|t| !is_reserved_tab(t)).count()
}

pub fn draw_tabs<W: Write>(stdout: &mut W, state: &AppState, cols: u16) {
    // Print tabs
    let mut labels: Vec<(String, Color)> = Vec::new();

    // Calculate available width for tabs
    // Reserve space for "Tabs: " prefix (6 chars) plus some padding
    let mut available_width = cols.saturating_sub(8);

    // Always show "All" tab first (index 0)
    if !state.tabs.is_empty() {
        let all_tab = &state.tabs[0];
        let tab_width = all_tab.len() as u16 + 2; // Tab name + 2 spaces padding

        if available_width >= tab_width {
            if state.current_tab == 0 {
                labels.push((format!(" {all_tab} "), Color::Black));
            } else {
                labels.push((format!(" {all_tab} "), Color::White));
            }
            available_width -= tab_width;
        }
    }

    // Show node tabs starting from scroll offset (skip "All" tab at index 0)
    let node_tabs: Vec<_> = state
        .tabs
        .iter()
        .enumerate()
        .skip(1) // Skip "All" tab
        .skip(state.tab_scroll_offset)
        .collect();

    for (i, tab) in node_tabs {
        // Get display name (instance name) while keeping tab as the key
        let connection_status = state.connection_status.get(tab);
        let display_name = if tab == "All" {
            tab.to_string()
        } else if let Some(status) = connection_status {
            // SSH mode tabs carry the `ssh://user@host` prefix in the
            // host_id; when present we show that label verbatim so the
            // operator immediately sees the transport. Issue #194.
            if tab.contains('@') && !tab.starts_with("http") {
                // SSH host_id: user@host:port. Strip the default port
                // to keep the label compact. When we have a live
                // transport chip, append it so the operator can see
                // which command is feeding each tab at a glance.
                let base = format_ssh_tab_label(tab);
                match status.transport_chip.as_deref() {
                    Some(chip) if !chip.is_empty() => format!("{base}·{chip}"),
                    _ => base,
                }
            } else {
                status.actual_hostname.as_ref().unwrap_or(tab).clone()
            }
        } else {
            tab.to_string()
        };

        let tab_width = display_name.len() as u16 + 2; // Display name + 2 spaces padding
        if available_width < tab_width {
            break; // No more space
        }

        // Determine color based on connection status and selection
        let color = if state.current_tab == i {
            Color::Black // Selected tab (will get blue background)
        } else {
            // Check if this tab represents a disconnected node
            let is_connected = if tab != "All" {
                state
                    .connection_status
                    .get(tab)
                    .map(|status| status.is_connected)
                    .unwrap_or(true) // Default to connected for local mode
            } else {
                true // "All" tab is always "connected"
            };

            if is_connected {
                Color::White // Connected: normal white text
            } else {
                Color::DarkGrey // Disconnected: dimmed grey text
            }
        };

        labels.push((format!(" {display_name} "), color));

        available_width -= tab_width;
    }

    // Render tabs
    render_tab_labels(stdout, labels);
    render_tab_separator(stdout, cols);
}

fn render_tab_labels<W: Write>(stdout: &mut W, labels: Vec<(String, Color)>) {
    queue!(stdout, Print("Tabs: ")).unwrap();
    for (text, color) in labels {
        if color == Color::Black {
            // Selected tab: white text on blue background for good visibility
            print_colored_text(stdout, &text, Color::White, Some(Color::Blue), None);
        } else {
            print_colored_text(stdout, &text, color, None, None);
        }
    }
    queue!(stdout, Print("\r\n")).unwrap();
}

fn render_tab_separator<W: Write>(stdout: &mut W, cols: u16) {
    // Print separator
    let separator = "─".repeat(cols as usize);
    print_colored_text(stdout, &separator, Color::DarkGrey, None, None);
    queue!(stdout, Print("\r\n")).unwrap();
}

/// Format an SSH host-id (`user@host:port`) as a compact tab label.
///
/// Strips the `:22` suffix because SSH's default port is rarely
/// interesting visually; everything else is preserved. Always prefixed
/// with `ssh://` so the transport is obvious in the tab strip.
pub fn format_ssh_tab_label(host_id: &str) -> String {
    let without_default_port = host_id.strip_suffix(":22").unwrap_or(host_id);
    format!("ssh://{without_default_port}")
}

#[allow(dead_code)]
pub fn calculate_tab_visibility(state: &AppState, cols: u16) -> TabVisibility {
    let mut available_width = cols.saturating_sub(8);

    // Reserve space for "All" tab (always visible)
    if !state.tabs.is_empty() {
        let all_tab_width = state.tabs[0].len() as u16 + 2;
        available_width = available_width.saturating_sub(all_tab_width);
    }

    // Calculate visible node tabs (skip "All" tab)
    let mut last_visible_node_tab = state.tab_scroll_offset;

    for (node_index, tab) in state
        .tabs
        .iter()
        .enumerate()
        .skip(1)
        .skip(state.tab_scroll_offset)
    {
        // Get display name for width calculation
        let display_name = if tab.contains('@') && !tab.starts_with("http") {
            let base = format_ssh_tab_label(tab);
            match state
                .connection_status
                .get(tab)
                .and_then(|s| s.transport_chip.as_deref())
            {
                Some(chip) if !chip.is_empty() => format!("{base}·{chip}"),
                _ => base,
            }
        } else if let Some(connection_status) = state.connection_status.get(tab) {
            connection_status
                .actual_hostname
                .as_ref()
                .unwrap_or(tab)
                .clone()
        } else {
            tab.to_string()
        };
        let tab_width = display_name.len() as u16 + 2;
        if available_width < tab_width {
            break;
        }
        available_width -= tab_width;
        last_visible_node_tab = node_index - 1; // Convert to node tab index
    }

    TabVisibility {
        first_visible: state.tab_scroll_offset,
        last_visible: last_visible_node_tab + 1, // Convert back to absolute tab index
        has_more_left: state.tab_scroll_offset > 0,
        has_more_right: last_visible_node_tab + 1 < state.tabs.len() - 1,
    }
}

#[allow(dead_code)]
pub struct TabVisibility {
    pub first_visible: usize,
    pub last_visible: usize,
    pub has_more_left: bool,
    pub has_more_right: bool,
}

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

    fn create_test_state() -> AppState {
        let mut state = AppState::new();
        state.tabs = vec![
            "All".to_string(),
            "host1".to_string(),
            "host2".to_string(),
            "host3".to_string(),
        ];
        state.is_local_mode = false; // Test state assumes remote mode
        state
    }

    #[test]
    fn test_tab_visibility_calculation() {
        let state = create_test_state();
        let visibility = calculate_tab_visibility(&state, 80);

        assert_eq!(visibility.first_visible, 0);
        assert!(!visibility.has_more_left);
        assert!(!visibility.has_more_right || state.tabs.len() > 4);
    }

    #[test]
    fn test_tab_visibility_with_scroll() {
        let mut state = create_test_state();
        state.tab_scroll_offset = 1;
        let visibility = calculate_tab_visibility(&state, 80);

        assert_eq!(visibility.first_visible, 1);
        assert!(visibility.has_more_left);
    }

    #[test]
    fn ssh_tab_label_strips_default_port() {
        assert_eq!(
            format_ssh_tab_label("admin@dgx-01:22"),
            "ssh://admin@dgx-01"
        );
    }

    #[test]
    fn ssh_tab_label_preserves_custom_port() {
        assert_eq!(
            format_ssh_tab_label("admin@dgx-01:2222"),
            "ssh://admin@dgx-01:2222"
        );
    }

    #[test]
    fn host_tab_count_excludes_reserved_tabs() {
        // Mirrors the real remote-mode layout from
        // `remote_collector.rs`: [All, Users, Topology, host1, host2, ...]
        // 50 hosts must register as 50, not 52 (issue: `50/52`).
        let mut tabs = vec![
            "All".to_string(),
            USERS_TAB_NAME.to_string(),
            TOPOLOGY_TAB_NAME.to_string(),
        ];
        for i in 0..50 {
            tabs.push(format!("host-{i}"));
        }
        assert_eq!(host_tab_count(&tabs), 50);

        // Local mode style: only "All" — zero hosts.
        assert_eq!(host_tab_count(&["All".to_string()]), 0);

        // Defensive: empty tab list returns 0.
        assert_eq!(host_tab_count(&[]), 0);
    }

    #[test]
    fn is_reserved_tab_matches_cluster_level_tabs() {
        assert!(is_reserved_tab("All"));
        assert!(is_reserved_tab(USERS_TAB_NAME));
        assert!(is_reserved_tab(TOPOLOGY_TAB_NAME));
        assert!(!is_reserved_tab("dgx-01"));
        assert!(!is_reserved_tab("admin@dgx-01:22"));
    }
}