use galaxy_core::ui::appearance::Appearance;
use galaxy_core::ui::color::contrast::MinimumAllowedContrast;
use galaxy_core::ui::color::ContrastingColor;
use galaxy_core::ui::theme::Fill;
use pathfinder_color::ColorU;

use super::DriveObjectType;
use crate::ui_components::blended_colors;

pub fn warp_drive_icon_color(
    appearance: &Appearance,
    cloud_object_type: DriveObjectType,
) -> ColorU {
    match cloud_object_type {
        DriveObjectType::Workflow => {
            let color: Fill = appearance.theme().terminal_colors().normal.red.into();
            color
                .on_background(
                    appearance.theme().surface_1(),
                    MinimumAllowedContrast::NonText,
                )
                .into()
        }
        DriveObjectType::Notebook { .. } => {
            let color: Fill = appearance.theme().terminal_colors().normal.blue.into();
            color
                .on_background(
                    appearance.theme().surface_1(),
                    MinimumAllowedContrast::NonText,
                )
                .into()
        }
        DriveObjectType::EnvVarCollection => {
            let color: Fill = appearance.theme().terminal_colors().normal.magenta.into();
            color
                .on_background(
                    appearance.theme().surface_1(),
                    MinimumAllowedContrast::NonText,
                )
                .into()
        }
        DriveObjectType::Folder => {
            // Match File Tree styling - use text_sub color
            blended_colors::text_sub(appearance.theme(), appearance.theme().background())
        }
        DriveObjectType::AIFactCollection
        | DriveObjectType::AIFact
        | DriveObjectType::MCPServer
        | DriveObjectType::MCPServerCollection
        | DriveObjectType::AgentModeWorkflow => appearance
            .theme()
            .main_text_color(appearance.theme().background())
            .into(),
    }
}