// hikari-components/src/basic/avatar.rs
//! Avatar component for user profile images

use hikari_palette::classes::{AvatarClass, ClassesBuilder, TypedClass};

use crate::prelude::*;
use crate::style_builder::{CssProperty, StyleStringBuilder};

#[derive(Clone, Copy, PartialEq, Eq, Debug, Default)]
pub enum AvatarSize {
    #[default]
    Xs,
    Sm,
    Md,
    Lg,
    Xl,
}

impl AvatarSize {
    pub fn pixels(&self) -> u32 {
        match self {
            AvatarSize::Xs => 24,
            AvatarSize::Sm => 32,
            AvatarSize::Md => 40,
            AvatarSize::Lg => 48,
            AvatarSize::Xl => 64,
        }
    }

    pub fn icon_size(&self) -> u32 {
        match self {
            AvatarSize::Xs => 14,
            AvatarSize::Sm => 18,
            AvatarSize::Md => 22,
            AvatarSize::Lg => 26,
            AvatarSize::Xl => 34,
        }
    }

    pub fn font_size(&self) -> &'static str {
        match self {
            AvatarSize::Xs => "0.625rem",
            AvatarSize::Sm => "0.75rem",
            AvatarSize::Md => "0.875rem",
            AvatarSize::Lg => "1rem",
            AvatarSize::Xl => "1.25rem",
        }
    }
}

#[derive(Clone, Copy, PartialEq, Eq, Debug, Default)]
pub enum AvatarVariant {
    #[default]
    Circular,
    Rounded,
    Square,
}

impl AvatarVariant {
    pub fn as_str(&self) -> &'static str {
        match self {
            AvatarVariant::Circular => "50%",
            AvatarVariant::Rounded => "8px",
            AvatarVariant::Square => "0",
        }
    }
}

#[derive(Clone, Copy, PartialEq, Eq, Default, Debug)]
pub enum AvatarFallbackMode {
    #[default]
    Initial,
    Icon,
    None,
}

#[define_props]
pub struct AvatarProps {
    pub src: Option<String>,
    #[default("Avatar".to_string())]
    pub alt: String,
    pub size: AvatarSize,
    pub variant: AvatarVariant,
    pub fallback: Option<String>,
    pub fallback_mode: AvatarFallbackMode,
    pub class: String,
}

#[component]
pub fn Avatar(props: AvatarProps) -> Element {
    let size_px = props.size.pixels();
    let icon_size = props.size.icon_size();
    let font_size = props.size.font_size();

    let container_style = StyleStringBuilder::new()
        .add_px(CssProperty::Width, size_px)
        .add_px(CssProperty::Height, size_px)
        .add(CssProperty::BorderRadius, props.variant.as_str())
        .add(CssProperty::Display, "flex")
        .add(CssProperty::AlignItems, "center")
        .add(CssProperty::JustifyContent, "center")
        .add(CssProperty::Overflow, "hidden")
        .build_clean();

    let img_style = StyleStringBuilder::new()
        .add_px(CssProperty::Width, size_px)
        .add_px(CssProperty::Height, size_px)
        .add(CssProperty::ObjectFit, "cover")
        .build_clean();

    let base_class = ClassesBuilder::new()
        .add_typed(AvatarClass::Avatar)
        .add(&props.class)
        .build();

    let fallback_text = props.fallback.clone().unwrap_or_else(|| {
        props
            .alt
            .chars()
            .find(|c| c.is_alphabetic() || c.is_numeric())
            .map(|c| c.to_uppercase().to_string())
            .unwrap_or_else(|| "?".to_string())
    });

    let has_src = props.src.is_some();
    let is_icon_mode = props.fallback_mode == AvatarFallbackMode::Icon;
    let is_initial_mode = props.fallback_mode == AvatarFallbackMode::Initial;
    let src_val = props.src.clone();

    // Build fallback content outside rsx!
    let fallback_content = if is_icon_mode {
        rsx! {
            svg {
                class: AvatarClass::AvatarIcon.class_name(),
                width: icon_size,
                height: icon_size,
                view_box: "0 0 24 24",
                fill: "currentColor",
                path { d: "M12 4a4 4 0 0 1 4 4 4 4 0 0 1-4 4 4 4 0 0 1-4-4 4 4 0 0 1 4-4m0 10c4.42 0 8 1.79 8 4v2H4v-2c0-2.21 3.58-4 8-4z" }
            }
        }
    } else {
        rsx! {
            span {
                class: AvatarClass::AvatarFallback.class_name(),
                style: "font-size: {font_size};",
                "{fallback_text}"
            }
        }
    };

    // Build content based on condition
    let inner_content = if has_src {
        rsx! {
            img {
                class: AvatarClass::AvatarImg.class_name(),
                src: src_val.unwrap_or_default(),
                alt: props.alt.clone(),
                style: img_style,
            }
        }
    } else {
        fallback_content
    };

    rsx! {
        div { class: base_class, style: container_style, {inner_content} }
    }
}
