/*
 * Copyright (C) 2023 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 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 android.platform.uiautomator_helpers.DurationUtils.platformAdjust
import androidx.test.uiautomator.By
import com.android.app.tracing.traceSection
import com.android.systemui.Flags.migrateClocksToBlueprint
import java.time.Duration
import java.util.regex.Pattern

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

    private val sysuiProcessUtils = ProcessUtil(UI_PACKAGE_NAME_SYSUI)

    // https://hsv.googleplex.com/5130837462876160?node=117
    private val PAGE_TITLE_SELECTOR_PATTERN = if (migrateClocksToBlueprint()) {
        Pattern.compile(String.format("com.android.systemui:id/%s", "keyguard_indication_area"))
    } else {
        Pattern.compile(String.format("com.android.systemui:id/%s", "keyguard_clock_container"))
    }
    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) {
        traceSection("restartSystemUI") {
            // 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,
            timeout = Duration.ofSeconds(10).platformAdjust(),
            errorProvider = errorMessageProvider
        )
    }
}