app_Settings/src/com/android/settings/bluetooth/ui/viewmodel/BluetoothDeviceDetailsViewModel.kt

/*
 * Copyright (C) 2024 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.settings.bluetooth.ui.viewmodel

import android.app.Application
import androidx.lifecycle.AndroidViewModel
import androidx.lifecycle.ViewModel
import androidx.lifecycle.ViewModelProvider
import androidx.lifecycle.viewModelScope
import com.android.settings.R
import com.android.settings.bluetooth.domain.interactor.SpatialAudioInteractor
import com.android.settings.bluetooth.ui.layout.DeviceSettingLayout
import com.android.settings.bluetooth.ui.layout.DeviceSettingLayoutColumn
import com.android.settings.bluetooth.ui.layout.DeviceSettingLayoutRow
import com.android.settings.bluetooth.ui.model.DeviceSettingPreferenceModel
import com.android.settings.bluetooth.ui.model.FragmentTypeModel
import com.android.settingslib.bluetooth.CachedBluetoothDevice
import com.android.settingslib.bluetooth.devicesettings.DeviceSettingId
import com.android.settingslib.bluetooth.devicesettings.data.repository.DeviceSettingRepository
import com.android.settingslib.bluetooth.devicesettings.shared.model.DeviceSettingConfigItemModel
import com.android.settingslib.bluetooth.devicesettings.shared.model.DeviceSettingIcon
import com.android.settingslib.bluetooth.devicesettings.shared.model.DeviceSettingModel
import com.android.settingslib.bluetooth.devicesettings.shared.model.DeviceSettingStateModel
import kotlin.coroutines.CoroutineContext
import kotlinx.coroutines.CoroutineStart
import kotlinx.coroutines.async
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.SharingStarted
import kotlinx.coroutines.flow.combine
import kotlinx.coroutines.flow.flowOf
import kotlinx.coroutines.flow.map
import kotlinx.coroutines.flow.stateIn

class BluetoothDeviceDetailsViewModel(
    private val application: Application,
    private val deviceSettingRepository: DeviceSettingRepository,
    private val spatialAudioInteractor: SpatialAudioInteractor,
    private val cachedDevice: CachedBluetoothDevice,
    backgroundCoroutineContext: CoroutineContext,
) : AndroidViewModel(application) {

    private val items =
        viewModelScope.async(backgroundCoroutineContext, start = CoroutineStart.LAZY) {
            deviceSettingRepository.getDeviceSettingsConfig(cachedDevice)
        }

    suspend fun getItems(fragment: FragmentTypeModel): List<DeviceSettingConfigItemModel>? =
        when (fragment) {
            is FragmentTypeModel.DeviceDetailsMainFragment -> items.await()?.mainItems
            is FragmentTypeModel.DeviceDetailsMoreSettingsFragment ->
                items.await()?.moreSettingsItems
        }

    suspend fun getHelpItem(fragment: FragmentTypeModel): DeviceSettingConfigItemModel? =
        when (fragment) {
            is FragmentTypeModel.DeviceDetailsMainFragment -> null
            is FragmentTypeModel.DeviceDetailsMoreSettingsFragment ->
                items.await()?.moreSettingsHelpItem
        }

    fun getDeviceSetting(
        cachedDevice: CachedBluetoothDevice,
        @DeviceSettingId settingId: Int,
    ): Flow<DeviceSettingPreferenceModel?> {
        if (settingId == DeviceSettingId.DEVICE_SETTING_ID_MORE_SETTINGS) {
            return flowOf(DeviceSettingPreferenceModel.MoreSettingsPreference(settingId))
        }
        return when (settingId) {
            DeviceSettingId.DEVICE_SETTING_ID_SPATIAL_AUDIO_MULTI_TOGGLE ->
                spatialAudioInteractor.getDeviceSetting(cachedDevice)
            else -> deviceSettingRepository.getDeviceSetting(cachedDevice, settingId)
        }.map { it?.toPreferenceModel() }
    }

    private fun DeviceSettingModel.toPreferenceModel(): DeviceSettingPreferenceModel? {
        return when (this) {
            is DeviceSettingModel.ActionSwitchPreference -> {
                if (switchState != null) {
                    DeviceSettingPreferenceModel.SwitchPreference(
                        id = id,
                        title = title,
                        summary = summary,
                        icon = icon,
                        checked = switchState?.checked ?: false,
                        onCheckedChange = { newState ->
                            updateState?.invoke(
                                DeviceSettingStateModel.ActionSwitchPreferenceState(newState)
                            )
                        },
                        disabled = !isAllowedChangingState,
                        action = action,
                    )
                } else {
                    DeviceSettingPreferenceModel.PlainPreference(
                        id = id,
                        title = title,
                        summary = summary,
                        icon = icon,
                        action = action,
                    )
                }
            }
            is DeviceSettingModel.FooterPreference ->
                DeviceSettingPreferenceModel.FooterPreference(id = id, footerText = footerText)
            is DeviceSettingModel.HelpPreference ->
                DeviceSettingPreferenceModel.HelpPreference(
                    id = id,
                    icon = DeviceSettingIcon.ResourceIcon(R.drawable.ic_help),
                    intent = intent,
                )
            is DeviceSettingModel.MultiTogglePreference ->
                DeviceSettingPreferenceModel.MultiTogglePreference(
                    id = id,
                    title = title,
                    toggles = toggles,
                    isActive = isActive,
                    selectedIndex = state.selectedIndex,
                    isAllowedChangingState = isAllowedChangingState,
                    onSelectedChange = { newState ->
                        updateState(DeviceSettingStateModel.MultiTogglePreferenceState(newState))
                    },
                )
            is DeviceSettingModel.Unknown -> null
        }
    }

    suspend fun getLayout(fragment: FragmentTypeModel): DeviceSettingLayout? {
        val configItems = getItems(fragment) ?: return null
        val idToDeviceSetting =
            configItems
                .filterIsInstance<DeviceSettingConfigItemModel.AppProvidedItem>()
                .associateBy({ it.settingId }, { getDeviceSetting(cachedDevice, it.settingId) })

        val configDeviceSetting =
            configItems.map { idToDeviceSetting[it.settingId] ?: flowOf(null) }
        val positionToSettingIds =
            combine(configDeviceSetting) { settings ->
                    val positionMapping = mutableMapOf<Int, List<DeviceSettingLayoutColumn>>()
                    var multiToggleSettingIds: MutableList<DeviceSettingLayoutColumn>? = null
                    for (i in settings.indices) {
                        val configItem = configItems[i]
                        val setting = settings[i]
                        val isXmlPreference = configItem is DeviceSettingConfigItemModel.BuiltinItem
                        if (!isXmlPreference && setting == null) {
                            continue
                        }
                        if (setting !is DeviceSettingPreferenceModel.MultiTogglePreference) {
                            multiToggleSettingIds = null
                            positionMapping[i] =
                                listOf(
                                    DeviceSettingLayoutColumn(
                                        configItem.settingId,
                                        configItem.highlighted,
                                    )
                                )
                            continue
                        }

                        if (multiToggleSettingIds != null) {
                            multiToggleSettingIds.add(
                                DeviceSettingLayoutColumn(
                                    configItem.settingId,
                                    configItem.highlighted,
                                )
                            )
                        } else {
                            multiToggleSettingIds =
                                mutableListOf(
                                    DeviceSettingLayoutColumn(
                                        configItem.settingId,
                                        configItem.highlighted,
                                    )
                                )
                            positionMapping[i] = multiToggleSettingIds
                        }
                    }
                    positionMapping
                }
                .stateIn(viewModelScope, SharingStarted.WhileSubscribed(), initialValue = mapOf())
        return DeviceSettingLayout(
            configItems.indices.map { idx ->
                DeviceSettingLayoutRow(positionToSettingIds.map { it[idx] ?: emptyList() })
            }
        )
    }

    class Factory(
        private val application: Application,
        private val deviceSettingRepository: DeviceSettingRepository,
        private val spatialAudioInteractor: SpatialAudioInteractor,
        private val cachedDevice: CachedBluetoothDevice,
        private val backgroundCoroutineContext: CoroutineContext,
    ) : ViewModelProvider.Factory {
        override fun <T : ViewModel> create(modelClass: Class<T>): T {
            @Suppress("UNCHECKED_CAST")
            return BluetoothDeviceDetailsViewModel(
                application,
                deviceSettingRepository,
                spatialAudioInteractor,
                cachedDevice,
                backgroundCoroutineContext,
            )
                as T
        }
    }

    companion object {
        private const val TAG = "BluetoothDeviceDetailsViewModel"
    }
}