/*
 * Copyright (C) 2022 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.systemui.statusbar.pipeline.mobile.ui

import com.android.app.tracing.coroutines.launchTraced as launch
import com.android.systemui.CoreStartable
import com.android.systemui.dagger.SysUISingleton
import com.android.systemui.dagger.qualifiers.Application
import com.android.systemui.shade.carrier.ShadeCarrierGroupController
import com.android.systemui.statusbar.phone.ui.StatusBarIconController
import com.android.systemui.statusbar.pipeline.mobile.domain.interactor.MobileIconsInteractor
import com.android.systemui.statusbar.pipeline.mobile.ui.viewmodel.MobileIconsViewModel
import java.io.PrintWriter
import javax.inject.Inject
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.flow.collectLatest
import kotlinx.coroutines.flow.combine

/**
 * This class is intended to provide a context to collect on the
 * [MobileIconsInteractor.filteredSubscriptions] data source and supply a state flow that can
 * control [StatusBarIconController] to keep the old UI in sync with the new data source.
 *
 * It also provides a mechanism to create a top-level view model for each IconManager to know about
 * the list of available mobile lines of service for which we want to show icons.
 */
@Suppress("EXPERIMENTAL_IS_NOT_ENABLED")
@SysUISingleton
class MobileUiAdapter
@Inject
constructor(
    private val iconController: StatusBarIconController,
    val mobileIconsViewModel: MobileIconsViewModel,
    private val logger: MobileViewLogger,
    @Application private val scope: CoroutineScope,
) : CoreStartable {
    private var isCollecting: Boolean = false
    private var lastValue: List<Int>? = null

    private var shadeCarrierGroupController: ShadeCarrierGroupController? = null

    override fun start() {
        // Start notifying the icon controller of subscriptions
        scope.launch {
            isCollecting = true
            combine(
                    mobileIconsViewModel.subscriptionIdsFlow,
                    mobileIconsViewModel.isStackable,
                    ::Pair,
                )
                .collectLatest { (subIds, isStackable) ->
                    logger.logUiAdapterSubIdsSentToIconController(subIds, isStackable)
                    lastValue = subIds
                    if (isStackable) {
                        // Passing an empty list to remove pre-existing mobile icons.
                        // StackedMobileBindableIcon will show the stacked icon instead.
                        iconController.setNewMobileIconSubIds(emptyList())
                    } else {
                        iconController.setNewMobileIconSubIds(subIds)
                    }
                    shadeCarrierGroupController?.updateModernMobileIcons(subIds)
                }
        }
    }

    /** Set the [ShadeCarrierGroupController] to notify of subscription updates */
    fun setShadeCarrierGroupController(controller: ShadeCarrierGroupController) {
        shadeCarrierGroupController = controller
    }

    override fun dump(pw: PrintWriter, args: Array<out String>) {
        pw.println("isCollecting=$isCollecting")
        pw.println("Last values sent to icon controller: $lastValue")
    }
}