// src/components/auth/register.rs

use crate::{
    config::colors::themes::Theme,
    state::pages::auth::RegisterState, // Use RegisterState
    components::common::{dialog, autocomplete},    
    state::state::AppState,
    state::canvas_state::CanvasState,
    modes::handlers::mode_manager::AppMode,
};
use ratatui::{
    layout::{Alignment, Constraint, Direction, Layout, Rect, Margin},
    style::{Style, Modifier, Color},
    widgets::{Block, BorderType, Borders, Paragraph, Wrap},
    Frame,
};

pub fn render_register(
    f: &mut Frame,
    area: Rect,
    theme: &Theme,
    state: &RegisterState, // Use RegisterState
    app_state: &AppState,
    is_edit_mode: bool,
) {
    let block = Block::default()
        .borders(Borders::ALL)
        .border_type(BorderType::Plain)
        .border_style(Style::default().fg(theme.border))
        .title(" Register ") // Update title
        .style(Style::default().bg(theme.bg));

    f.render_widget(block, area);

    let inner_area = area.inner(Margin {
        horizontal: 1,
        vertical: 1,
    });

    // Adjust constraints for 4 fields + error + buttons
    let chunks = Layout::default()
        .direction(Direction::Vertical)
        .constraints([
            Constraint::Length(7), // Form (5 fields + padding)
            Constraint::Length(1), // Help text line
            Constraint::Length(1), // Error message
            Constraint::Length(3), // Buttons
        ])
        .split(inner_area);

    // --- FORM RENDERING (Using render_canvas) ---
    let active_field_rect = crate::components::handlers::canvas::render_canvas(
        f,
        chunks[0], // Area for the canvas
        state,     // The state object (RegisterState)
        &[         // Field labels
            "Username",
            "Email*",
            "Password*",
            "Confirm Password",
            "Role* (Tab)",
        ],
        &state.current_field(), // Pass current field index
        &state.inputs().iter().map(|s| *s).collect::<Vec<&String>>(), // Pass inputs directly
        theme,
        is_edit_mode,
    );

    // --- HELP TEXT ---
    let help_text = Paragraph::new("* are optional fields")
        .style(Style::default().fg(theme.fg))
        .alignment(Alignment::Center);
    f.render_widget(help_text, chunks[1]);

   
    // --- ERROR MESSAGE ---
    if let Some(err) = &state.error_message {
        f.render_widget(
            Paragraph::new(err.as_str())
                .style(Style::default().fg(Color::Red))
                .alignment(Alignment::Center),
            chunks[2],
        );
    }

    // --- BUTTONS ---
    let button_chunks = Layout::default()
        .direction(Direction::Horizontal)
        .constraints([Constraint::Percentage(50), Constraint::Percentage(50)])
        .split(chunks[3]);

    // Register Button
    let register_button_index = 0;
    let register_active = if app_state.ui.focus_outside_canvas {
        app_state.general.selected_item == register_button_index
    } else {
        false
    };
    let mut register_style = Style::default().fg(theme.fg);
    let mut register_border = Style::default().fg(theme.border);
    if register_active {
        register_style = register_style.fg(theme.highlight).add_modifier(Modifier::BOLD);
        register_border = register_border.fg(theme.accent);
    }

    f.render_widget(
        Paragraph::new("Register") // Update button text
            .style(register_style)
            .alignment(Alignment::Center)
            .block(
                Block::default()
                    .borders(Borders::ALL)
                    .border_type(BorderType::Plain)
                    .border_style(register_border),
            ),
        button_chunks[0],
    );

    // Return Button (logic remains similar)
    let return_button_index = 1;
    let return_active = if app_state.ui.focus_outside_canvas {
        app_state.general.selected_item == return_button_index
    } else {
        false
    };
    let mut return_style = Style::default().fg(theme.fg);
    let mut return_border = Style::default().fg(theme.border);
    if return_active {
        return_style = return_style.fg(theme.highlight).add_modifier(Modifier::BOLD);
        return_border = return_border.fg(theme.accent);
    }

    f.render_widget(
        Paragraph::new("Return")
            .style(return_style)
            .alignment(Alignment::Center)
            .block(
                Block::default()
                    .borders(Borders::ALL)
                    .border_type(BorderType::Plain)
                    .border_style(return_border),
            ),
        button_chunks[1],
    );

    // --- Render Autocomplete Dropdown (Draw AFTER buttons) ---
    if app_state.current_mode == AppMode::Edit {
        if let Some(suggestions) = state.get_suggestions() {
            let selected = state.get_selected_suggestion_index();
            if !suggestions.is_empty() {
                if let Some(input_rect) = active_field_rect {
                    autocomplete::render_autocomplete_dropdown(f, input_rect, f.size(), theme, suggestions, selected);
                }    
            }
        }
    }

    // --- DIALOG --- (Keep dialog logic)
    if app_state.ui.dialog.dialog_show {
        dialog::render_dialog(
            f,
            f.area(),
            theme,
            &app_state.ui.dialog.dialog_title,
            &app_state.ui.dialog.dialog_message,
            &app_state.ui.dialog.dialog_buttons,
            app_state.ui.dialog.dialog_active_button_index,
        );
    }
}