/*
 * 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 android.healthconnect.cts.ui.permissions

import android.Manifest
import android.app.Activity
import android.content.Intent
import android.content.pm.PackageManager
import android.health.connect.HealthConnectManager
import android.health.connect.HealthPermissions.*
import android.healthconnect.cts.ui.HealthConnectBaseTest
import android.healthconnect.testing.cts.testapphelpers.ProxyActivity
import android.healthconnect.testing.cts.ui.ActivityLauncher.launchRequestPermissionActivity
import android.healthconnect.testing.cts.ui.UiTestUtils.TEST_APP_PACKAGE_NAME
import android.healthconnect.testing.cts.ui.UiTestUtils.clickOnTextAndWaitForNewWindow
import android.healthconnect.testing.cts.ui.UiTestUtils.findText
import android.healthconnect.testing.cts.ui.UiTestUtils.findTextAndClick
import android.healthconnect.testing.cts.ui.UiTestUtils.grantPermissionViaPackageManager
import android.healthconnect.testing.cts.ui.UiTestUtils.revokePermissionViaPackageManager
import android.healthconnect.testing.cts.ui.UiTestUtils.scrollDownToAndFindText
import android.healthconnect.testing.cts.ui.UiTestUtils.skipOnboardingIfAppears
import android.os.Build
import androidx.test.filters.SdkSuppress
import com.android.compatibility.common.util.SystemUtil
import com.android.compatibility.common.util.UiAutomatorUtils2.getUiDevice
import com.google.common.truth.Truth
import org.junit.After
import org.junit.Before
import org.junit.Ignore
import org.junit.Test

class AdditionalPermissionsRequestUITest : HealthConnectBaseTest() {

    @Before
    fun setup() {
        revokePermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_HISTORY)
        revokePermissionViaPackageManager(
            context,
            TEST_APP_PACKAGE_NAME,
            READ_HEALTH_DATA_IN_BACKGROUND,
        )
        with(getUiDevice()) { executeShellCommand("settings put system font_scale 0.85") }
    }

    @After
    fun tearDown() {
        revokePermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_HISTORY)
        revokePermissionViaPackageManager(
            context,
            TEST_APP_PACKAGE_NAME,
            READ_HEALTH_DATA_IN_BACKGROUND,
        )
        revokePermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEIGHT)
    }

    @Test
    fun requestAdditionalPermissions_showsBothAdditionalPermissions_grantsBoth() {
        val permissions = listOf(READ_HEALTH_DATA_HISTORY, READ_HEALTH_DATA_IN_BACKGROUND)
        grantPermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEIGHT)
        context.launchRequestPermissionActivity(
            packageName = TEST_APP_PACKAGE_NAME,
            permissions = permissions,
        ) {
            findText("Allow additional access for Health Connect cts test app?")
            scrollDownToAndFindText("Access past data")
            findTextAndClick("Access past data")
            scrollDownToAndFindText("Access data in the background")
            findTextAndClick("Access data in the background")
            clickOnTextAndWaitForNewWindow("Allow")

            assertPermGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_HISTORY)
            assertPermGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_IN_BACKGROUND)
        }
    }

    @Test
    fun requestAdditionalPermissions_showsBothAdditionalPermissions_deniesBoth() {
        val permissions = listOf(READ_HEALTH_DATA_HISTORY, READ_HEALTH_DATA_IN_BACKGROUND)
        grantPermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEIGHT)
        context.launchRequestPermissionActivity(
            packageName = TEST_APP_PACKAGE_NAME,
            permissions = permissions,
        ) {
            findText("Allow additional access for Health Connect cts test app?")
            scrollDownToAndFindText("Access past data")
            findTextAndClick("Access past data")
            scrollDownToAndFindText("Access data in the background")
            findTextAndClick("Access data in the background")
            clickOnTextAndWaitForNewWindow("Don't allow")

            assertPermNotGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_HISTORY)
            assertPermNotGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_IN_BACKGROUND)
        }
    }

    @Test
    fun requestAdditionalPermissions_showsBothAdditionalPermissions_grantsOnlySelected() {
        val permissions = listOf(READ_HEALTH_DATA_HISTORY, READ_HEALTH_DATA_IN_BACKGROUND)
        grantPermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEIGHT)
        context.launchRequestPermissionActivity(
            packageName = TEST_APP_PACKAGE_NAME,
            permissions = permissions,
        ) {
            findText("Allow additional access for Health Connect cts test app?")
            scrollDownToAndFindText("Access past data")
            findTextAndClick("Access past data")
            clickOnTextAndWaitForNewWindow("Allow")

            assertPermGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_HISTORY)
            assertPermNotGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_IN_BACKGROUND)
        }
    }

    @Test
    fun requestAdditionalPermissions_showsOnlyNotGrantedPermissions() {
        val permissions = listOf(READ_HEALTH_DATA_HISTORY, READ_HEALTH_DATA_IN_BACKGROUND)
        grantPermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEIGHT)
        grantPermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_HISTORY)
        context.launchRequestPermissionActivity(
            packageName = TEST_APP_PACKAGE_NAME,
            permissions = permissions,
        ) {
            findText("Allow Health Connect cts test app to access data in the background?")
            clickOnTextAndWaitForNewWindow("Allow")

            assertPermGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_HISTORY)
            assertPermGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_IN_BACKGROUND)
        }
    }

    @Test
    @Ignore("b/382029765")
    fun requestAdditionalPermissions_noReadPermissions_returnsResultCanceled() {
        revokePermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEIGHT)
        revokePermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_MINDFULNESS)
        val permissions = listOf(READ_HEALTH_DATA_HISTORY, READ_HEALTH_DATA_IN_BACKGROUND)

        val intent =
            Intent(HealthConnectManager.ACTION_REQUEST_HEALTH_PERMISSIONS).apply {
                addFlags(Intent.FLAG_ACTIVITY_NEW_TASK)
                addFlags(Intent.FLAG_ACTIVITY_CLEAR_TASK)
                putExtra(PackageManager.EXTRA_REQUEST_PERMISSIONS_NAMES, permissions.toTypedArray())
                putExtra(Intent.EXTRA_PACKAGE_NAME, TEST_APP_PACKAGE_NAME)
            }

        SystemUtil.runWithShellPermissionIdentity(
            {
                val result =
                    ProxyActivity.launchActivityForResult(intent) { skipOnboardingIfAppears() }
                Truth.assertThat(result.resultCode).isEqualTo(Activity.RESULT_CANCELED)
            },
            Manifest.permission.GRANT_RUNTIME_PERMISSIONS,
        )
    }

    @SdkSuppress(maxSdkVersion = Build.VERSION_CODES.VANILLA_ICE_CREAM)
    @Test
    fun requestCombinedPermissions_showsDataTypeAndAdditionalPermissions_heathConnectBrand() {
        revokePermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEIGHT)
        revokePermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, WRITE_BODY_FAT)
        val permissions =
            listOf(
                READ_HEIGHT,
                WRITE_BODY_FAT,
                READ_HEALTH_DATA_HISTORY,
                READ_HEALTH_DATA_IN_BACKGROUND,
            )
        context.launchRequestPermissionActivity(
            packageName = TEST_APP_PACKAGE_NAME,
            permissions = permissions,
        ) {
            findText("Allow Health Connect cts test app to access Health Connect?")
            scrollDownToAndFindText("Height")
            findTextAndClick("Height")
            clickOnTextAndWaitForNewWindow("Allow")

            findText("Allow additional access for Health Connect cts test app?")
            scrollDownToAndFindText("Access past data")
            findTextAndClick("Access past data")
            clickOnTextAndWaitForNewWindow("Allow")

            assertPermGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEIGHT)
            assertPermNotGrantedForApp(TEST_APP_PACKAGE_NAME, WRITE_BODY_FAT)
            assertPermGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_HISTORY)
            assertPermNotGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_IN_BACKGROUND)
        }
    }

    @SdkSuppress(minSdkVersion = Build.VERSION_CODES.BAKLAVA, codeName = "Baklava")
    @Test
    fun requestCombinedPermissions_showsDataTypeAndAdditionalPermissions_healthFitnessBrand() {
        revokePermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, READ_HEIGHT)
        revokePermissionViaPackageManager(context, TEST_APP_PACKAGE_NAME, WRITE_BODY_FAT)
        val permissions =
            listOf(
                READ_HEIGHT,
                WRITE_BODY_FAT,
                READ_HEALTH_DATA_HISTORY,
                READ_HEALTH_DATA_IN_BACKGROUND,
            )
        context.launchRequestPermissionActivity(
            packageName = TEST_APP_PACKAGE_NAME,
            permissions = permissions,
        ) {
            findText("Allow Health Connect cts test app to access your fitness and wellness data?")
            scrollDownToAndFindText("Height")
            findTextAndClick("Height")
            clickOnTextAndWaitForNewWindow("Allow")

            findText("Allow additional access for Health Connect cts test app?")
            scrollDownToAndFindText("Access past data")
            findTextAndClick("Access past data")
            clickOnTextAndWaitForNewWindow("Allow")

            assertPermGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEIGHT)
            assertPermNotGrantedForApp(TEST_APP_PACKAGE_NAME, WRITE_BODY_FAT)
            assertPermGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_HISTORY)
            assertPermNotGrantedForApp(TEST_APP_PACKAGE_NAME, READ_HEALTH_DATA_IN_BACKGROUND)
        }
    }

    @Throws(Exception::class)
    private fun assertPermGrantedForApp(packageName: String, permName: String) {
        Truth.assertThat(context.packageManager.checkPermission(permName, packageName))
            .isEqualTo(PackageManager.PERMISSION_GRANTED)
    }

    @Throws(Exception::class)
    private fun assertPermNotGrantedForApp(packageName: String, permName: String) {
        Truth.assertThat(context.packageManager.checkPermission(permName, packageName))
            .isEqualTo(PackageManager.PERMISSION_DENIED)
    }
}