khm/src/gui/admin/ui.rs

use super::state::{get_key_preview, get_key_type, AdminState};
use crate::gui::api::SshKey;
use eframe::egui;
use std::collections::BTreeMap;

/// Render statistics cards
pub fn render_statistics(ui: &mut egui::Ui, admin_state: &AdminState) {
    let stats = admin_state.get_statistics();

    ui.group(|ui| {
        ui.set_min_width(ui.available_width());
        ui.vertical(|ui| {
            ui.label(egui::RichText::new("📊 Statistics").size(16.0).strong());
            ui.add_space(8.0);

            ui.horizontal(|ui| {
                ui.columns(4, |cols| {
                    // Total keys
                    cols[0].vertical_centered_justified(|ui| {
                        ui.label(egui::RichText::new("📊").size(20.0));
                        ui.label(
                            egui::RichText::new(stats.total_keys.to_string())
                                .size(24.0)
                                .strong(),
                        );
                        ui.label(
                            egui::RichText::new("Total Keys")
                                .size(11.0)
                                .color(egui::Color32::GRAY),
                        );
                    });

                    // Active keys
                    cols[1].vertical_centered_justified(|ui| {
                        ui.label(egui::RichText::new("✅").size(20.0));
                        ui.label(
                            egui::RichText::new(stats.active_keys.to_string())
                                .size(24.0)
                                .strong()
                                .color(egui::Color32::LIGHT_GREEN),
                        );
                        ui.label(
                            egui::RichText::new("Active")
                                .size(11.0)
                                .color(egui::Color32::GRAY),
                        );
                    });

                    // Deprecated keys
                    cols[2].vertical_centered_justified(|ui| {
                        ui.label(egui::RichText::new("❌").size(20.0));
                        ui.label(
                            egui::RichText::new(stats.deprecated_keys.to_string())
                                .size(24.0)
                                .strong()
                                .color(egui::Color32::LIGHT_RED),
                        );
                        ui.label(
                            egui::RichText::new("Deprecated")
                                .size(11.0)
                                .color(egui::Color32::GRAY),
                        );
                    });

                    // Servers
                    cols[3].vertical_centered_justified(|ui| {
                        ui.label(egui::RichText::new("💻").size(20.0));
                        ui.label(
                            egui::RichText::new(stats.unique_servers.to_string())
                                .size(24.0)
                                .strong()
                                .color(egui::Color32::LIGHT_BLUE),
                        );
                        ui.label(
                            egui::RichText::new("Servers")
                                .size(11.0)
                                .color(egui::Color32::GRAY),
                        );
                    });
                });
            });
        });
    });
}

/// Render search and filter controls
pub fn render_search_controls(ui: &mut egui::Ui, admin_state: &mut AdminState) -> bool {
    let mut changed = false;

    ui.group(|ui| {
        ui.set_min_width(ui.available_width());
        ui.vertical(|ui| {
            ui.label(egui::RichText::new("🔍 Search").size(16.0).strong());
            ui.add_space(8.0);

            // Search field with full width
            ui.horizontal(|ui| {
                ui.label(egui::RichText::new("🔍").size(14.0));
                let search_response = ui.add_sized(
                    [ui.available_width() * 0.6, 20.0],
                    egui::TextEdit::singleline(&mut admin_state.search_term)
                        .hint_text("Search servers or keys..."),
                );

                if admin_state.search_term.is_empty() {
                    ui.label(
                        egui::RichText::new("Type to search")
                            .size(11.0)
                            .color(egui::Color32::GRAY),
                    );
                } else {
                    ui.label(
                        egui::RichText::new(format!("{} results", admin_state.filtered_keys.len()))
                            .size(11.0),
                    );
                    if ui
                        .add(
                            egui::Button::new(
                                egui::RichText::new("❌").color(egui::Color32::WHITE),
                            )
                            .fill(egui::Color32::from_rgb(170, 170, 170))
                            .stroke(egui::Stroke::new(1.0, egui::Color32::from_rgb(89, 89, 89)))
                            .rounding(egui::Rounding::same(3.0))
                            .min_size(egui::vec2(18.0, 18.0)),
                        )
                        .on_hover_text("Clear search")
                        .clicked()
                    {
                        admin_state.search_term.clear();
                        changed = true;
                    }
                }

                // Handle search text changes
                if search_response.changed() {
                    changed = true;
                }
            });

            ui.add_space(5.0);

            // Filter controls
            ui.horizontal(|ui| {
                ui.label("Filter:");
                let show_deprecated = admin_state.show_deprecated_only;
                if ui.selectable_label(!show_deprecated, "✅ Active").clicked() {
                    admin_state.show_deprecated_only = false;
                    changed = true;
                }
                if ui
                    .selectable_label(show_deprecated, "❗ Deprecated")
                    .clicked()
                {
                    admin_state.show_deprecated_only = true;
                    changed = true;
                }
            });
        });
    });

    if changed {
        admin_state.filter_keys();
    }

    changed
}

/// Render bulk actions controls
pub fn render_bulk_actions(ui: &mut egui::Ui, admin_state: &mut AdminState) -> BulkAction {
    let selected_count = admin_state
        .selected_servers
        .values()
        .filter(|&&v| v)
        .count();

    if selected_count == 0 {
        return BulkAction::None;
    }

    let mut action = BulkAction::None;

    ui.group(|ui| {
        ui.set_min_width(ui.available_width());
        ui.vertical(|ui| {
            ui.horizontal(|ui| {
                ui.label(egui::RichText::new("📋").size(14.0));
                ui.label(
                    egui::RichText::new(format!("Selected {} servers", selected_count))
                        .size(14.0)
                        .strong()
                        .color(egui::Color32::LIGHT_BLUE),
                );
            });

            ui.add_space(5.0);

            ui.horizontal(|ui| {
                if ui
                    .add(
                        egui::Button::new(
                            egui::RichText::new("❗ Deprecate Selected")
                                .color(egui::Color32::BLACK),
                        )
                        .fill(egui::Color32::from_rgb(255, 200, 0))
                        .stroke(egui::Stroke::new(1.0, egui::Color32::from_rgb(102, 94, 72)))
                        .rounding(egui::Rounding::same(6.0))
                        .min_size(egui::vec2(130.0, 28.0)),
                    )
                    .clicked()
                {
                    action = BulkAction::DeprecateSelected;
                }

                if ui
                    .add(
                        egui::Button::new(
                            egui::RichText::new("✅ Restore Selected").color(egui::Color32::WHITE),
                        )
                        .fill(egui::Color32::from_rgb(101, 199, 40))
                        .stroke(egui::Stroke::new(1.0, egui::Color32::from_rgb(94, 105, 25)))
                        .rounding(egui::Rounding::same(6.0))
                        .min_size(egui::vec2(120.0, 28.0)),
                    )
                    .clicked()
                {
                    action = BulkAction::RestoreSelected;
                }

                if ui
                    .add(
                        egui::Button::new(
                            egui::RichText::new("X Clear Selection").color(egui::Color32::WHITE),
                        )
                        .fill(egui::Color32::from_rgb(170, 170, 170))
                        .stroke(egui::Stroke::new(1.0, egui::Color32::from_rgb(89, 89, 89)))
                        .rounding(egui::Rounding::same(6.0))
                        .min_size(egui::vec2(110.0, 28.0)),
                    )
                    .clicked()
                {
                    admin_state.clear_selection();
                    action = BulkAction::ClearSelection;
                }
            });
        });
    });

    action
}

/// Render keys table grouped by servers
pub fn render_keys_table(ui: &mut egui::Ui, admin_state: &mut AdminState) -> KeyAction {
    if admin_state.filtered_keys.is_empty() {
        render_empty_state(ui, admin_state);
        return KeyAction::None;
    }

    let mut action = KeyAction::None;

    // Group keys by server
    let mut servers: BTreeMap<String, Vec<SshKey>> = BTreeMap::new();
    for key in &admin_state.filtered_keys {
        servers
            .entry(key.server.clone())
            .or_insert_with(Vec::new)
            .push(key.clone());
    }

    // Render each server group
    for (server_name, server_keys) in servers {
        let is_expanded = admin_state
            .expanded_servers
            .get(&server_name)
            .copied()
            .unwrap_or(false);
        let active_count = server_keys.iter().filter(|k| !k.deprecated).count();
        let deprecated_count = server_keys.len() - active_count;

        // Server header
        ui.group(|ui| {
            ui.horizontal(|ui| {
                // Server selection checkbox
                let mut selected = admin_state
                    .selected_servers
                    .get(&server_name)
                    .copied()
                    .unwrap_or(false);
                if ui
                    .add(egui::Checkbox::new(&mut selected, "").indeterminate(false))
                    .changed()
                {
                    admin_state
                        .selected_servers
                        .insert(server_name.clone(), selected);
                }

                // Expand/collapse button
                let expand_icon = if is_expanded { "-" } else { "+" };
                if ui
                    .add(
                        egui::Button::new(expand_icon)
                            .fill(egui::Color32::TRANSPARENT)
                            .stroke(egui::Stroke::NONE)
                            .min_size(egui::vec2(20.0, 20.0)),
                    )
                    .clicked()
                {
                    admin_state
                        .expanded_servers
                        .insert(server_name.clone(), !is_expanded);
                }

                // Server icon and name
                ui.label(egui::RichText::new("💻").size(16.0));
                ui.label(
                    egui::RichText::new(&server_name)
                        .size(15.0)
                        .strong()
                        .color(egui::Color32::WHITE),
                );

                // Keys count badge
                render_badge(
                    ui,
                    &format!("{} keys", server_keys.len()),
                    egui::Color32::from_rgb(52, 152, 219),
                    egui::Color32::WHITE,
                );

                ui.add_space(5.0);

                // Deprecated count badge
                if deprecated_count > 0 {
                    render_badge(
                        ui,
                        &format!("{} depr", deprecated_count),
                        egui::Color32::from_rgb(231, 76, 60),
                        egui::Color32::WHITE,
                    );
                }

                ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| {
                    // Server action buttons
                    if deprecated_count > 0 {
                        if ui
                            .add(
                                egui::Button::new(
                                    egui::RichText::new("✅ Restore").color(egui::Color32::WHITE),
                                )
                                .fill(egui::Color32::from_rgb(101, 199, 40))
                                .stroke(egui::Stroke::new(
                                    1.0,
                                    egui::Color32::from_rgb(94, 105, 25),
                                ))
                                .rounding(egui::Rounding::same(4.0))
                                .min_size(egui::vec2(70.0, 24.0)),
                            )
                            .clicked()
                        {
                            action = KeyAction::RestoreServer(server_name.clone());
                        }
                    }

                    if active_count > 0 {
                        if ui
                            .add(
                                egui::Button::new(
                                    egui::RichText::new("❗ Deprecate").color(egui::Color32::BLACK),
                                )
                                .fill(egui::Color32::from_rgb(255, 200, 0))
                                .stroke(egui::Stroke::new(
                                    1.0,
                                    egui::Color32::from_rgb(102, 94, 72),
                                ))
                                .rounding(egui::Rounding::same(4.0))
                                .min_size(egui::vec2(85.0, 24.0)),
                            )
                            .clicked()
                        {
                            action = KeyAction::DeprecateServer(server_name.clone());
                        }
                    }
                });
            });
        });

        // Expanded key details
        if is_expanded {
            ui.indent("server_keys", |ui| {
                for key in &server_keys {
                    if let Some(key_action) = render_key_item(ui, key, &server_name) {
                        action = key_action;
                    }
                }
            });
        }

        ui.add_space(5.0);
    }

    action
}

/// Render empty state when no keys are available
fn render_empty_state(ui: &mut egui::Ui, admin_state: &AdminState) {
    ui.vertical_centered(|ui| {
        ui.add_space(60.0);
        if admin_state.keys.is_empty() {
            ui.label(
                egui::RichText::new("🔑")
                    .size(48.0)
                    .color(egui::Color32::GRAY),
            );
            ui.label(
                egui::RichText::new("No SSH keys available")
                    .size(18.0)
                    .color(egui::Color32::GRAY),
            );
            ui.label(
                egui::RichText::new("Keys will appear here once loaded from the server")
                    .size(14.0)
                    .color(egui::Color32::DARK_GRAY),
            );
        } else if !admin_state.search_term.is_empty() {
            ui.label(
                egui::RichText::new("🔍")
                    .size(48.0)
                    .color(egui::Color32::GRAY),
            );
            ui.label(
                egui::RichText::new("No results found")
                    .size(18.0)
                    .color(egui::Color32::GRAY),
            );
            ui.label(
                egui::RichText::new(format!(
                    "Try adjusting your search: '{}'",
                    admin_state.search_term
                ))
                .size(14.0)
                .color(egui::Color32::DARK_GRAY),
            );
        } else {
            ui.label(
                egui::RichText::new("❌")
                    .size(48.0)
                    .color(egui::Color32::GRAY),
            );
            ui.label(
                egui::RichText::new("No keys match current filters")
                    .size(18.0)
                    .color(egui::Color32::GRAY),
            );
            ui.label(
                egui::RichText::new("Try adjusting your search or filter settings")
                    .size(14.0)
                    .color(egui::Color32::DARK_GRAY),
            );
        }
    });
}

/// Render individual key item
fn render_key_item(ui: &mut egui::Ui, key: &SshKey, server_name: &str) -> Option<KeyAction> {
    let mut action = None;

    ui.group(|ui| {
        ui.horizontal(|ui| {
            // Key type badge
            let key_type = get_key_type(&key.public_key);
            let (badge_color, text_color) = match key_type.as_str() {
                "RSA" => (egui::Color32::from_rgb(52, 144, 220), egui::Color32::WHITE),
                "ED25519" => (egui::Color32::from_rgb(46, 204, 113), egui::Color32::WHITE),
                "ECDSA" => (egui::Color32::from_rgb(241, 196, 15), egui::Color32::BLACK),
                "DSA" => (egui::Color32::from_rgb(230, 126, 34), egui::Color32::WHITE),
                _ => (egui::Color32::GRAY, egui::Color32::WHITE),
            };

            render_small_badge(ui, &key_type, badge_color, text_color);
            ui.add_space(5.0);

            // Status badge
            if key.deprecated {
                ui.label(
                    egui::RichText::new("❗ DEPR")
                        .size(10.0)
                        .color(egui::Color32::from_rgb(231, 76, 60))
                        .strong(),
                );
            } else {
                ui.label(
                    egui::RichText::new("✅")
                        .size(10.0)
                        .color(egui::Color32::from_rgb(46, 204, 113))
                        .strong(),
                );
            }

            ui.add_space(5.0);

            // Key preview
            ui.label(
                egui::RichText::new(get_key_preview(&key.public_key))
                    .font(egui::FontId::monospace(10.0))
                    .color(egui::Color32::LIGHT_GRAY),
            );

            ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| {
                // Key action buttons
                if key.deprecated {
                    if ui
                        .add(
                            egui::Button::new(
                                egui::RichText::new("[R]").color(egui::Color32::WHITE),
                            )
                            .fill(egui::Color32::from_rgb(101, 199, 40))
                            .stroke(egui::Stroke::new(1.0, egui::Color32::from_rgb(94, 105, 25)))
                            .rounding(egui::Rounding::same(3.0))
                            .min_size(egui::vec2(22.0, 18.0)),
                        )
                        .on_hover_text("Restore key")
                        .clicked()
                    {
                        action = Some(KeyAction::RestoreKey(server_name.to_string()));
                    }
                    if ui
                        .add(
                            egui::Button::new(
                                egui::RichText::new("Del").color(egui::Color32::WHITE),
                            )
                            .fill(egui::Color32::from_rgb(246, 36, 71))
                            .stroke(egui::Stroke::new(1.0, egui::Color32::from_rgb(129, 18, 17)))
                            .rounding(egui::Rounding::same(3.0))
                            .min_size(egui::vec2(26.0, 18.0)),
                        )
                        .on_hover_text("Delete key")
                        .clicked()
                    {
                        action = Some(KeyAction::DeleteKey(server_name.to_string()));
                    }
                } else {
                    if ui
                        .add(
                            egui::Button::new(
                                egui::RichText::new("❗").color(egui::Color32::BLACK),
                            )
                            .fill(egui::Color32::from_rgb(255, 200, 0))
                            .stroke(egui::Stroke::new(1.0, egui::Color32::from_rgb(102, 94, 72)))
                            .rounding(egui::Rounding::same(3.0))
                            .min_size(egui::vec2(22.0, 18.0)),
                        )
                        .on_hover_text("Deprecate key")
                        .clicked()
                    {
                        action = Some(KeyAction::DeprecateKey(server_name.to_string()));
                    }
                }

                if ui
                    .add(
                        egui::Button::new(egui::RichText::new("Copy").color(egui::Color32::WHITE))
                            .fill(egui::Color32::from_rgb(0, 111, 230))
                            .stroke(egui::Stroke::new(1.0, egui::Color32::from_rgb(35, 84, 97)))
                            .rounding(egui::Rounding::same(3.0))
                            .min_size(egui::vec2(30.0, 18.0)),
                    )
                    .on_hover_text("Copy to clipboard")
                    .clicked()
                {
                    ui.output_mut(|o| o.copied_text = key.public_key.clone());
                }
            });
        });
    });

    action
}

/// Render a badge with text
fn render_badge(ui: &mut egui::Ui, text: &str, bg_color: egui::Color32, text_color: egui::Color32) {
    let (rect, _) = ui.allocate_exact_size(egui::vec2(50.0, 18.0), egui::Sense::hover());
    ui.painter()
        .rect_filled(rect, egui::Rounding::same(8.0), bg_color);
    ui.painter().text(
        rect.center(),
        egui::Align2::CENTER_CENTER,
        text,
        egui::FontId::proportional(10.0),
        text_color,
    );
}

/// Render a small badge with text
fn render_small_badge(
    ui: &mut egui::Ui,
    text: &str,
    bg_color: egui::Color32,
    text_color: egui::Color32,
) {
    let (rect, _) = ui.allocate_exact_size(egui::vec2(40.0, 16.0), egui::Sense::hover());
    ui.painter()
        .rect_filled(rect, egui::Rounding::same(3.0), bg_color);
    ui.painter().text(
        rect.center(),
        egui::Align2::CENTER_CENTER,
        text,
        egui::FontId::proportional(9.0),
        text_color,
    );
}

/// Actions that can be performed on keys
#[derive(Debug, Clone)]
pub enum KeyAction {
    None,
    DeprecateKey(String),
    RestoreKey(String),
    DeleteKey(String),
    DeprecateServer(String),
    RestoreServer(String),
}

/// Bulk actions that can be performed
#[derive(Debug, Clone)]
pub enum BulkAction {
    None,
    DeprecateSelected,
    RestoreSelected,
    ClearSelection,
}