package com.suno.android.extensions

import com.suno.android.common_data.badge.ModelBadgeData
import com.suno.android.common_ui.models.UiModelBadge

fun ModelBadgeData.xAsUiModelBadge(): UiModelBadge? {
    return UiModelBadge(
        displayName = displayName,
        themes = UiModelBadge.Themes(
            lightTheme = themes.lightTheme.xAsTheme() ?: return null,
            darkTheme = themes.darkTheme.xAsTheme() ?: return null,
        ),
    )
}

private fun ModelBadgeData.Themes.Theme.xAsTheme(): UiModelBadge.Themes.Theme? {
    val background = backgroundColor.xAsColor() ?: return null
    val text = textColor.xAsColor() ?: return null
    return UiModelBadge.Themes.Theme(
        backgroundColor = background,
        textColor = text,
        borderColor = borderColor?.xAsColor(),
    )
}