package android.platform.helpers

import android.platform.helpers.CommonUtils.assertScreenOn
import android.platform.helpers.Constants.UI_PACKAGE_NAME_SYSUI
import android.platform.helpers.LockscreenUtils.LockscreenType
import android.platform.helpers.features.common.HomeLockscreenPage
import android.platform.uiautomator_helpers.DeviceHelpers.assertVisibility
import android.platform.uiautomator_helpers.DeviceHelpers.uiDevice
import androidx.test.uiautomator.By
import java.util.regex.Pattern

/** Restarts system ui. */
object SysuiRestarter {

    private val sysuiProcessUtils = ProcessUtil(UI_PACKAGE_NAME_SYSUI)

    private val PAGE_TITLE_SELECTOR_PATTERN =
        Pattern.compile(
            String.format(
                "com.android.systemui:id/(%s|%s)",
                "lockscreen_clock_view",
                "lockscreen_clock_view_large"
            )
        )
    private val PAGE_TITLE_SELECTOR = By.res(PAGE_TITLE_SELECTOR_PATTERN)

    /**
     * Restart System UI by running `am crash com.android.systemui`.
     *
     * This is sometimes necessary after changing flags, configs, or settings ensure that systemui
     * is properly initialized with the new changes. This method will wait until the home screen is
     * visible, then it will optionally dismiss the home screen via swipe.
     *
     * @param swipeUp whether to call [HomeLockscreenPage.swipeUp] after restarting System UI
     */
    @JvmStatic
    fun restartSystemUI(swipeUp: Boolean) {
        // This method assumes the screen is on.
        assertScreenOn("restartSystemUI needs the screen to be on.")
        // make sure the lock screen is enable.
        LockscreenUtils.setLockscreen(
            LockscreenType.SWIPE,
            /* lockscreenCode= */ null,
            /* expectedResult= */ false
        )
        sysuiProcessUtils.restart()
        assertLockscreenVisibility(true) { "Lockscreen not visible after restart" }
        if (swipeUp) {
            HomeLockscreenPage().swipeUp()
            assertLockscreenVisibility(false) { "Lockscreen still visible after swiping up." }
        }
    }

    private fun assertLockscreenVisibility(visible: Boolean, errorMessageProvider: () -> String) {
        uiDevice.assertVisibility(
            PAGE_TITLE_SELECTOR,
            visible,
            errorProvider = errorMessageProvider
        )
    }
}