package com.suno.android.extensions

import com.suno.android.common_networking.remote.session.Model
import com.suno.android.common_ui.components.menus.ModelSelectorDropdownState
import kotlinx.collections.immutable.ImmutableList
import kotlinx.collections.immutable.toPersistentList

fun List<Model>.toModelSelectorDropdownStates(
    selectedModelName: String?,
): ImmutableList<ModelSelectorDropdownState> =
    map { model ->
        ModelSelectorDropdownState(
            title = model.name.orEmpty(),
            showProBadge = model.badges?.contains("pro") == true,
            showBetaBadge = model.badges?.contains("beta") == true,
            description = model.description.orEmpty(),
            isSelected = selectedModelName == model.name,
        )
    }.toPersistentList()