imp-tui 0.2.0

use ratatui::buffer::Buffer;
use ratatui::layout::Rect;
use ratatui::style::Style;
use ratatui::text::{Line, Span};
use ratatui::widgets::{Block, Borders, Clear, Widget};

use imp_llm::model::ProviderRegistry;

use crate::theme::Theme;

#[derive(Debug, Clone)]
pub struct LoginProviderOption {
    pub id: &'static str,
    pub label: &'static str,
    pub description: &'static str,
    pub logged_in: bool,
}

pub fn login_providers(_registry: &ProviderRegistry) -> Vec<LoginProviderOption> {
    vec![
        LoginProviderOption {
            id: "anthropic",
            label: "Anthropic",
            description: "Claude Max/Pro subscription (OAuth)",
            logged_in: false,
        },
        LoginProviderOption {
            id: "openai",
            label: "OpenAI",
            description: "OpenAI / ChatGPT account (OAuth)",
            logged_in: false,
        },
        LoginProviderOption {
            id: "kimi-code",
            label: "Kimi Code",
            description: "Kimi Code subscription (OAuth)",
            logged_in: false,
        },
    ]
}

#[derive(Debug, Clone)]
pub struct LoginPickerState {
    pub providers: Vec<LoginProviderOption>,
    pub selected: usize,
}

impl LoginPickerState {
    pub fn new(providers: Vec<LoginProviderOption>) -> Self {
        Self {
            providers,
            selected: 0,
        }
    }

    pub fn move_up(&mut self) {
        if self.selected > 0 {
            self.selected -= 1;
        }
    }

    pub fn move_down(&mut self) {
        if self.selected + 1 < self.providers.len() {
            self.selected += 1;
        }
    }

    pub fn selected_provider(&self) -> Option<&LoginProviderOption> {
        self.providers.get(self.selected)
    }
}

pub struct LoginPickerView<'a> {
    state: &'a LoginPickerState,
    theme: &'a Theme,
}

impl<'a> LoginPickerView<'a> {
    pub fn new(state: &'a LoginPickerState, theme: &'a Theme) -> Self {
        Self { state, theme }
    }
}

impl Widget for LoginPickerView<'_> {
    fn render(self, area: Rect, buf: &mut Buffer) {
        if area.height < 6 || area.width < 20 {
            return;
        }

        Clear.render(area, buf);
        let block = Block::default()
            .title(" Provider Login ")
            .borders(Borders::ALL)
            .border_style(self.theme.accent_style());
        let inner = block.inner(area);
        block.render(area, buf);

        if self.state.providers.is_empty() {
            let line = Line::from(Span::styled(
                "  No providers available",
                self.theme.muted_style(),
            ));
            buf.set_line(inner.x, inner.y, &line, inner.width);
            return;
        }

        let footer = "Enter: configure provider · Esc: cancel";
        let footer_y = inner.y + inner.height.saturating_sub(1);

        for (i, provider) in self.state.providers.iter().enumerate() {
            if inner.y + i as u16 >= footer_y {
                break;
            }

            let is_selected = i == self.state.selected;
            let row_style = if is_selected {
                self.theme.selected_style()
            } else {
                Style::default()
            };

            let status = if provider.logged_in {
                vec![
                    Span::raw("  "),
                    Span::styled("✓ configured", self.theme.success_style()),
                ]
            } else {
                Vec::new()
            };

            let mut spans = vec![
                Span::styled(
                    if is_selected { " ▸ " } else { "   " },
                    self.theme.accent_style(),
                ),
                Span::styled(provider.label, row_style),
                Span::raw("  "),
                Span::styled(provider.description, self.theme.muted_style()),
            ];
            spans.extend(status);
            let line = Line::from(spans);
            buf.set_line(inner.x, inner.y + i as u16, &line, inner.width);
        }

        let footer_line = Line::from(Span::styled(footer, self.theme.muted_style()));
        buf.set_line(inner.x, footer_y, &footer_line, inner.width);
    }
}

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

    #[test]
    fn login_picker_includes_oauth_providers() {
        let registry = ProviderRegistry::with_builtins();
        let state = LoginPickerState::new(login_providers(&registry));
        let ids: Vec<&str> = state.providers.iter().map(|provider| provider.id).collect();
        assert!(ids.contains(&"anthropic"));
        assert!(ids.contains(&"openai"));
    }

    #[test]
    fn login_picker_only_lists_oauth_providers() {
        let registry = ProviderRegistry::with_builtins();
        let providers = login_providers(&registry);
        let ids: Vec<&str> = providers.iter().map(|provider| provider.id).collect();
        assert!(ids.contains(&"anthropic"));
        assert!(ids.contains(&"openai"));
        assert!(!ids.contains(&"tavily"));
        assert!(!ids.contains(&"exa"));
    }

    #[test]
    fn picker_selection_moves_with_bounds() {
        let registry = ProviderRegistry::with_builtins();
        let mut state = LoginPickerState::new(login_providers(&registry));
        assert_eq!(
            state.selected_provider().map(|provider| provider.id),
            Some("anthropic")
        );

        state.move_down();
        assert_eq!(
            state.selected_provider().map(|provider| provider.id),
            Some("openai")
        );

        for _ in 0..20 {
            state.move_down();
        }
        assert!(state.selected_provider().is_some());

        state.move_up();
        assert!(state.selected_provider().is_some());
    }
}