package com.android.systemui.scene

import android.view.View
import com.android.compose.animation.scene.ObservableTransitionState
import com.android.systemui.classifier.domain.interactor.falsingInteractor
import com.android.systemui.desktop.domain.interactor.desktopInteractor
import com.android.systemui.deviceentry.domain.interactor.deviceUnlockedInteractor
import com.android.systemui.haptics.msdl.msdlPlayer
import com.android.systemui.keyguard.domain.interactor.keyguardInteractor
import com.android.systemui.keyguard.ui.viewmodel.aodBurnInViewModel
import com.android.systemui.keyguard.ui.viewmodel.keyguardClockViewModel
import com.android.systemui.keyguard.ui.viewmodel.lightRevealScrimViewModel
import com.android.systemui.kosmos.Kosmos
import com.android.systemui.kosmos.Kosmos.Fixture
import com.android.systemui.power.domain.interactor.powerInteractor
import com.android.systemui.qs.panels.ui.viewmodel.animateQsTilesViewModelFactory
import com.android.systemui.scene.domain.interactor.onBootTransitionInteractor
import com.android.systemui.scene.domain.interactor.sceneInteractor
import com.android.systemui.scene.shared.logger.sceneLogger
import com.android.systemui.scene.shared.model.Overlays
import com.android.systemui.scene.shared.model.SceneContainerConfig
import com.android.systemui.scene.shared.model.Scenes
import com.android.systemui.scene.ui.FakeOverlay
import com.android.systemui.scene.ui.composable.ConstantSceneContainerTransitionsBuilder
import com.android.systemui.scene.ui.viewmodel.SceneContainerHapticsViewModel
import com.android.systemui.scene.ui.viewmodel.SceneContainerViewModel
import com.android.systemui.scene.ui.viewmodel.dualShadeEducationalTooltipsViewModelFactory
import com.android.systemui.shade.domain.interactor.shadeInteractor
import com.android.systemui.shade.domain.interactor.shadeModeInteractor
import com.android.systemui.statusbar.domain.interactor.remoteInputInteractor
import com.android.systemui.statusbar.notification.stack.domain.interactor.notificationContainerInteractor
import com.android.systemui.wallpapers.ui.viewmodel.wallpaperViewModel
import kotlinx.coroutines.flow.MutableStateFlow
import org.mockito.kotlin.mock

var Kosmos.sceneKeys by Fixture {
    listOf(
        Scenes.QuickSettings,
        Scenes.Shade,
        Scenes.Lockscreen,
        Scenes.Gone,
        Scenes.Communal,
        Scenes.Dream,
        Scenes.Occluded,
    )
}

val Kosmos.initialSceneKey by Fixture { Scenes.Lockscreen }

var Kosmos.overlayKeys by Fixture {
    listOf(Overlays.NotificationsShade, Overlays.QuickSettingsShade, Overlays.Bouncer)
}

val Kosmos.fakeOverlaysByKeys by Fixture { overlayKeys.associateWith { FakeOverlay(it) } }

val Kosmos.fakeOverlays by Fixture { fakeOverlaysByKeys.values.toSet() }

val Kosmos.overlays by Fixture { fakeOverlays }

val Kosmos.sceneTransitionsBuilder by Fixture { ConstantSceneContainerTransitionsBuilder() }

var Kosmos.sceneContainerConfig by Fixture {
    val navigationDistances =
        mapOf(
            Scenes.Gone to 0,
            Scenes.Lockscreen to 0,
            Scenes.Occluded to 1,
            Scenes.Communal to 1,
            Scenes.Dream to 2,
            Scenes.Shade to 3,
            Scenes.QuickSettings to 4,
        )

    SceneContainerConfig(
        sceneKeys = sceneKeys,
        initialSceneKey = initialSceneKey,
        overlayKeys = overlayKeys,
        navigationDistances = navigationDistances,
        transitionsBuilder = sceneTransitionsBuilder,
    )
}

val Kosmos.transitionState by Fixture {
    MutableStateFlow<ObservableTransitionState>(
        ObservableTransitionState.Idle(sceneContainerConfig.initialSceneKey)
    )
}

val Kosmos.sceneContainerViewModel by Fixture {
    sceneContainerViewModelFactory
        .create(mock<View>()) {}
        .apply { setTransitionState(transitionState) }
}

val Kosmos.sceneContainerViewModelFactory by Fixture {
    object : SceneContainerViewModel.Factory {
        override fun create(
            view: View,
            motionEventHandlerReceiver: (SceneContainerViewModel.MotionEventHandler?) -> Unit,
        ): SceneContainerViewModel =
            SceneContainerViewModel(
                sceneInteractor = sceneInteractor,
                desktopInteractor = desktopInteractor,
                deviceUnlockedInteractor = deviceUnlockedInteractor,
                falsingInteractor = falsingInteractor,
                powerInteractor = powerInteractor,
                shadeModeInteractor = shadeModeInteractor,
                remoteInputInteractor = remoteInputInteractor,
                logger = sceneLogger,
                hapticsViewModelFactory = sceneContainerHapticsViewModelFactory,
                view = view,
                motionEventHandlerReceiver = motionEventHandlerReceiver,
                lightRevealScrim = lightRevealScrimViewModel,
                wallpaperViewModel = wallpaperViewModel,
                keyguardInteractor = keyguardInteractor,
                burnIn = aodBurnInViewModel,
                clock = keyguardClockViewModel,
                onBootTransitionInteractor = onBootTransitionInteractor,
                notificationContainerInteractor = notificationContainerInteractor,
                dualShadeEducationalTooltipsViewModelFactory =
                    dualShadeEducationalTooltipsViewModelFactory,
                animateQsTilesViewModelFactory = animateQsTilesViewModelFactory,
            )
    }
}

val Kosmos.sceneContainerHapticsViewModelFactory by Fixture {
    object : SceneContainerHapticsViewModel.Factory {
        override fun create(view: View): SceneContainerHapticsViewModel {
            return SceneContainerHapticsViewModel(
                view = view,
                sceneInteractor = sceneInteractor,
                shadeInteractor = shadeInteractor,
                msdlPlayer = msdlPlayer,
            )
        }
    }
}