/*
 * 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.launcher3.taskbar

import android.animation.AnimatorTestRule
import androidx.test.core.app.ApplicationProvider
import androidx.test.platform.app.InstrumentationRegistry.getInstrumentation
import com.android.launcher3.taskbar.TaskbarAutohideSuspendController.FLAG_AUTOHIDE_SUSPEND_BUBBLES
import com.android.launcher3.taskbar.TaskbarAutohideSuspendController.FLAG_AUTOHIDE_SUSPEND_DRAGGING
import com.android.launcher3.taskbar.TaskbarAutohideSuspendController.FLAG_AUTOHIDE_SUSPEND_IN_LAUNCHER
import com.android.launcher3.taskbar.TaskbarAutohideSuspendController.FLAG_AUTOHIDE_SUSPEND_TOUCHING
import com.android.launcher3.taskbar.TaskbarAutohideSuspendController.FLAG_AUTOHIDE_SUSPEND_TRANSIENT_TASKBAR
import com.android.launcher3.taskbar.rules.SandboxParams
import com.android.launcher3.taskbar.rules.TaskbarModeRule
import com.android.launcher3.taskbar.rules.TaskbarModeRule.Mode.TRANSIENT
import com.android.launcher3.taskbar.rules.TaskbarModeRule.TaskbarMode
import com.android.launcher3.taskbar.rules.TaskbarUnitTestRule
import com.android.launcher3.taskbar.rules.TaskbarUnitTestRule.InjectController
import com.android.launcher3.taskbar.rules.TaskbarWindowSandboxContext
import com.android.launcher3.util.Executors.MAIN_EXECUTOR
import com.android.launcher3.util.Executors.UI_HELPER_EXECUTOR
import com.android.launcher3.util.LauncherMultivalentJUnit
import com.android.launcher3.util.LauncherMultivalentJUnit.EmulatedDevices
import com.android.quickstep.SystemUiProxy
import com.google.common.truth.Truth.assertThat
import org.junit.Rule
import org.junit.Test
import org.junit.runner.RunWith
import org.mockito.kotlin.anyOrNull
import org.mockito.kotlin.doAnswer
import org.mockito.kotlin.spy
import org.mockito.kotlin.whenever

@RunWith(LauncherMultivalentJUnit::class)
@EmulatedDevices(["pixelTablet2023"])
class TaskbarAutohideSuspendControllerTest {

    @get:Rule(order = 0)
    val context =
        TaskbarWindowSandboxContext.create(
            SandboxParams({
                spy(SystemUiProxy(
                        ApplicationProvider.getApplicationContext(),
                        MAIN_EXECUTOR,
                        UI_HELPER_EXECUTOR,
                    )) { proxy ->
                    doAnswer { latestSuspendNotification = it.getArgument(0) }
                        .whenever(proxy)
                        .notifyTaskbarAutohideSuspend(anyOrNull())
                }
            })
        )
    @get:Rule(order = 1) val animatorTestRule = AnimatorTestRule(this)
    @get:Rule(order = 2) val taskbarModeRule = TaskbarModeRule(context)
    @get:Rule(order = 3) val taskbarUnitTestRule = TaskbarUnitTestRule(this, context)

    @InjectController lateinit var autohideSuspendController: TaskbarAutohideSuspendController
    @InjectController lateinit var stashController: TaskbarStashController

    private var latestSuspendNotification: Boolean? = null

    @Test
    fun testUpdateFlag_suspendInLauncher_notifiesSuspend() {
        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_IN_LAUNCHER, true)
        }
        assertThat(latestSuspendNotification).isTrue()
    }

    @Test
    fun testUpdateFlag_toggleSuspendDraggingTwice_notifiesUnsuspend() {
        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_DRAGGING, true)
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_DRAGGING, false)
        }
        assertThat(latestSuspendNotification).isFalse()
    }

    @Test
    fun testUpdateFlag_resetsAlreadyUnsetFlag_noNotifyUnsuspend() {
        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_DRAGGING, false)
        }
        assertThat(latestSuspendNotification).isNull()
    }

    @Test
    @TaskbarMode(TRANSIENT)
    fun testUpdateFlag_suspendTransientTaskbarForTouch_cancelsAutoStashTimeout() {
        // Unstash and verify alarm.
        getInstrumentation().runOnMainSync {
            stashController.updateAndAnimateTransientTaskbar(false)
            animatorTestRule.advanceTimeBy(stashController.stashDuration)
        }
        assertThat(stashController.timeoutAlarm.alarmPending()).isTrue()

        // EDU opens while unstashed.
        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_TOUCHING, true)
        }
        assertThat(stashController.timeoutAlarm.alarmPending()).isFalse()
    }

    @Test
    fun isSuspended_withBubblesFlag_returnsCorrectValue() {
        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_BUBBLES, true)
        }
        assertThat(autohideSuspendController.isSuspended).isTrue()

        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_BUBBLES, false)
        }
        assertThat(autohideSuspendController.isSuspended).isFalse()
    }

    @Test
    @TaskbarMode(TRANSIENT)
    fun suspendAutoHideForBubbles_doesNotSuspendTransientTaskbarStashing() {
        // Unstash and verify alarm.
        getInstrumentation().runOnMainSync {
            stashController.updateAndAnimateTransientTaskbar(false)
            animatorTestRule.advanceTimeBy(stashController.stashDuration)
        }
        assertThat(stashController.timeoutAlarm.alarmPending()).isTrue()

        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_BUBBLES, true)
        }
        assertThat(stashController.timeoutAlarm.alarmPending()).isTrue()
        assertThat(autohideSuspendController.isTransientTaskbarStashingSuspended).isFalse()
    }

    @Test
    fun suspendedForBubbles_shouldForceVisible() {
        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_BUBBLES, true)
        }
        assertThat(autohideSuspendController.shouldForceVisible()).isTrue()

        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_BUBBLES, false)
        }
        assertThat(autohideSuspendController.shouldForceVisible()).isFalse()
    }

    @Test
    fun suspendedForUnstashedTaskbar_shouldForceVisible() {
        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_TRANSIENT_TASKBAR, true)
        }
        assertThat(autohideSuspendController.shouldForceVisible()).isTrue()

        getInstrumentation().runOnMainSync {
            autohideSuspendController.updateFlag(FLAG_AUTOHIDE_SUSPEND_TRANSIENT_TASKBAR, false)
        }
        assertThat(autohideSuspendController.shouldForceVisible()).isFalse()
    }
}