/*
 * Copyright 2025 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.permissioncontroller.appfunctions

import android.content.Context
import android.content.Intent
import android.content.res.Resources
import android.net.Uri
import android.os.UserHandle
import android.provider.Settings
import com.android.permissioncontroller.R
import com.android.permissioncontroller.appfunctions.AppFunctionsUtil.EXTRA_DEVICE_STATE_KEY
import com.android.permissioncontroller.permission.model.livedatatypes.AppPermGroupUiInfo.PermGrantState
import com.android.permissioncontroller.permission.ui.model.UnusedAppsViewModel.UnusedPeriod
import com.android.permissioncontroller.permission.utils.KotlinUtils.getPermGroupLabel
import com.android.permissioncontroller.permission.utils.Utils
import com.android.permissioncontroller.role.ui.RoleApplicationItem
import com.android.role.controller.model.Role
import com.google.android.appfunctions.schema.common.v1.devicestate.DeviceStateItem
import com.google.android.appfunctions.schema.common.v1.devicestate.LocalizedString
import com.google.android.appfunctions.schema.common.v1.devicestate.PerScreenDeviceStates

abstract class PerScreenDeviceState(val context: Context) {
    /** A unique key that identifies a screen. Used only internally */
    abstract val key: String

    /** Description of the screen. */
    abstract val description: String

    /** Intent of the screen to be used by deeplinking */
    abstract val intent: Intent?

    /** Intent Uri of the screen */
    val intentUri: String?
        get() =
            intent
                ?.putExtra(EXTRA_DEVICE_STATE_KEY, AppFunctionsUtil.getPasswordForIntent(context))
                ?.toUri(Intent.URI_INTENT_SCHEME)

    open fun getDeviceStateItems(): List<DeviceStateItem> {
        return emptyList()
    }

    fun toPerScreenDeviceStates(): PerScreenDeviceStates {
        return PerScreenDeviceStates(
            description = description,
            intentUri = intentUri,
            deviceStateItems = getDeviceStateItems(),
        )
    }

    companion object {
        const val DEFAULT_PACKAGE_LABEL = "Unknown"
    }
}

class PermissionManagerScreen(context: Context) : PerScreenDeviceState(context) {
    override val key: String
        get() = KEY

    override val description: String
        get() = DESCRIPTION

    override val intent: Intent
        get() = Intent(AppFunctionsUtil.ACTION_MANAGE_PERMISSIONS)

    companion object {
        const val KEY = "permission_manager"
        const val DESCRIPTION = "Permission Manager"
    }
}

class PermissionAppsScreen(context: Context, val permissionGroup: String) :
    PerScreenDeviceState(context) {
    private val permissionGroupLabel: String =
        getPermGroupLabel(context, permissionGroup).toString()

    override val key: String
        get() = KEY

    override val description: String
        get() = "Permission Manager: $permissionGroupLabel"

    override val intent: Intent
        get() =
            Intent(AppFunctionsUtil.ACTION_MANAGE_PERMISSION_APPS).apply {
                putExtra(Intent.EXTRA_PERMISSION_GROUP_NAME, permissionGroup)
            }

    companion object {
        const val KEY = "permission_apps"
    }
}

class AppPermissionsScreen(context: Context) : PerScreenDeviceState(context) {
    override val key: String
        get() = KEY

    override val description: String
        get() =
            "App Permissions Screen. Note that to get to the app permissions for a given " +
                "package, the intent uri is $intentWithPlaceholder"

    override val intent = null

    private val intentWithPlaceholder =
        Intent(AppFunctionsUtil.ACTION_MANAGE_APP_PERMISSIONS)
            .apply {
                putExtra(Intent.EXTRA_PACKAGE_NAME, "\$packageName")
                putExtra(EXTRA_DEVICE_STATE_KEY, AppFunctionsUtil.getPasswordForIntent(context))
            }
            .toUri(Intent.URI_INTENT_SCHEME)

    companion object {
        const val KEY = "app_permissions"
    }
}

class AppPermissionSettingScreen(context: Context, val permissionGroup: String) :
    PerScreenDeviceState(context) {
    private val permissionGroupLabel: String =
        getPermGroupLabel(context, permissionGroup).toString()

    override val key: String
        get() = KEY

    override val description: String
        get() =
            "$permissionGroupLabel permission setting screen for a package. The intent uri is $intentWithPlaceholder"

    override val intent = null

    private val intentWithPlaceholder =
        Intent(AppFunctionsUtil.ACTION_MANAGE_APP_PERMISSIONS)
            .apply {
                putExtra(Intent.EXTRA_PERMISSION_GROUP_NAME, permissionGroup)
                putExtra(Intent.EXTRA_PACKAGE_NAME, "\$packageName")
                putExtra(EXTRA_DEVICE_STATE_KEY, AppFunctionsUtil.getPasswordForIntent(context))
            }
            .toUri(Intent.URI_INTENT_SCHEME)

    companion object {
        const val KEY = "app_permission_setting"
    }
}

class AppPermissionGrantStateScreen(
    context: Context,
    val permissionGroup: String,
    val packageName: String,
    val userHandle: UserHandle,
    val permissionGrantState: PermGrantState,
    val lastAccessTime: Long,
    val usePreciseLocation: Boolean?,
) : PerScreenDeviceState(context) {
    private val permissionGroupLabel: String =
        getPermGroupLabel(context, permissionGroup).toString()

    override val key: String
        get() = KEY

    override val description: String
        get() = "$permissionGroupLabel Permission: $packageName"

    override val intent = null

    override fun getDeviceStateItems(): List<DeviceStateItem> {
        val result: MutableList<DeviceStateItem> = mutableListOf()
        val permissionGroupLabel: String = getPermGroupLabel(context, permissionGroup).toString()
        val permissionStateItem =
            DeviceStateItem(
                key = "${permissionGroupLabel.lowercase()}_permission_state",
                name = LocalizedString(english = "$permissionGroupLabel access for this app"),
                jsonValue = translatePermissionGrantState(),
            )
        result.add(permissionStateItem)

        if (lastAccessTime > 0) {
            val summaryTimestamp =
                Utils.getPermissionLastAccessSummaryTimestamp(
                    lastAccessTime,
                    context,
                    permissionGroup,
                )
            val recentAccessItem =
                DeviceStateItem(
                    key = "recent_access",
                    name = LocalizedString(english = "Recent access"),
                    jsonValue = getRecentAccessSummary(summaryTimestamp),
                )
            result.add(recentAccessItem)
        }

        if (usePreciseLocation != null) {
            val usePreciseLocation =
                DeviceStateItem(
                    key = "use_precise_location",
                    name = LocalizedString(english = "Use precise location"),
                    jsonValue = usePreciseLocation.toString(),
                )
            result.add(usePreciseLocation)
        }

        return result
    }

    private fun translatePermissionGrantState(): String {
        return when (permissionGrantState) {
            PermGrantState.PERMS_DENIED -> "Not Allowed"
            PermGrantState.PERMS_ALLOWED -> "Allowed"
            PermGrantState.PERMS_ALLOWED_FOREGROUND_ONLY -> "Allowed while using the app"
            PermGrantState.PERMS_ALLOWED_ALWAYS -> "Always Allowed"
            PermGrantState.PERMS_ASK -> "Ask every time"
        }
    }

    private fun getRecentAccessSummary(summaryTimestamp: Triple<String, Int, String>): String {
        val res: Resources = context.resources

        return when (summaryTimestamp.second) {
            Utils.LAST_24H_CONTENT_PROVIDER ->
                res.getString(R.string.app_perms_content_provider_24h)
            Utils.LAST_7D_CONTENT_PROVIDER -> res.getString(R.string.app_perms_content_provider_7d)
            Utils.LAST_24H_SENSOR_TODAY ->
                res.getString(R.string.app_perms_24h_access, summaryTimestamp.first)
            Utils.LAST_24H_SENSOR_YESTERDAY ->
                res.getString(R.string.app_perms_24h_access_yest, summaryTimestamp.first)
            Utils.LAST_7D_SENSOR ->
                res.getString(
                    R.string.app_perms_7d_access,
                    summaryTimestamp.third,
                    summaryTimestamp.first,
                )
            else -> ""
        }
    }

    companion object {
        const val KEY = "app_permission_grant_state"
    }
}

class UnusedAppsScreen(context: Context) : PerScreenDeviceState(context) {
    override val key: String
        get() = KEY

    override val description: String
        get() = DESCRIPTION

    override val intent: Intent
        get() = Intent(AppFunctionsUtil.ACTION_MANAGE_UNUSED_APPS)

    companion object {
        const val KEY = "unused_apps"
        const val DESCRIPTION = "Unused apps"
    }
}

class UnusedAppLastUsageScreen(
    context: Context,
    val packageName: String,
    private val lastUsageTime: Long,
) : PerScreenDeviceState(context) {
    override val key: String
        get() = KEY

    override val description: String
        get() = "Unused app details: $packageName"

    override val intent: Intent
        get() =
            Intent(Settings.ACTION_APPLICATION_DETAILS_SETTINGS).apply {
                setData(Uri.fromParts("package", packageName, null))
            }

    override fun getDeviceStateItems(): List<DeviceStateItem> {
        val usagePeriod = UnusedPeriod.findLongestValidPeriod(lastUsageTime)

        val lastUsed =
            DeviceStateItem(
                key = "last_used",
                name = LocalizedString(english = "Last used"),
                jsonValue = translateUnusedPeriod(usagePeriod),
            )
        return listOf(lastUsed)
    }

    private fun translateUnusedPeriod(usagePeriod: UnusedPeriod): String {
        return when (usagePeriod) {
            UnusedPeriod.ONE_MONTH -> "1 month ago"
            UnusedPeriod.THREE_MONTHS -> "3 months ago"
            UnusedPeriod.SIX_MONTHS -> "6 months ago"
        }
    }

    companion object {
        const val KEY = "unused_app_last_usage"
    }
}

class AdditionalPermissionsScreen(context: Context) : PerScreenDeviceState(context) {
    override val key: String
        get() = KEY

    override val description: String
        get() = DESCRIPTION

    override val intent: Intent
        get() = Intent(AppFunctionsUtil.ACTION_ADDITIONAL_PERMISSIONS)

    companion object {
        const val KEY = "additional_permissions"
        const val DESCRIPTION = "Additional Permissions"
    }
}

class DefaultAppListScreen(context: Context) : PerScreenDeviceState(context) {
    override val key: String
        get() = KEY

    override val description: String
        get() = DESCRIPTION

    override val intent: Intent
        get() = Intent(Settings.ACTION_MANAGE_DEFAULT_APPS_SETTINGS)

    companion object {
        const val KEY = "default_app_list"
        const val DESCRIPTION = "Default Apps"
    }
}

class DefaultAppScreen(
    context: Context,
    private val role: Role,
    private val roleApplications: List<RoleApplicationItem>,
    private val isWorkProfile: Boolean,
    private val user: UserHandle,
) : PerScreenDeviceState(context) {
    private val defaultAppName = context.getString(role.shortLabelResource)

    override val key: String
        get() = KEY

    override val description: String
        get() =
            context.getString(role.labelResource).let {
                return if (isWorkProfile) "Work profile: $it" else it
            }

    override val intent: Intent
        get() =
            Intent(AppFunctionsUtil.ACTION_MANAGE_DEFAULT_APP).apply {
                putExtra(Intent.EXTRA_ROLE_NAME, role.name)
                putExtra(Intent.EXTRA_USER, user)
            }

    override fun getDeviceStateItems(): List<DeviceStateItem> {
        val result = mutableListOf<DeviceStateItem>()

        roleApplications.forEach { roleApplicationInfo ->
            val packageName = roleApplicationInfo.applicationInfo.packageName
            if (roleApplicationInfo.isHolderApplication) {
                result.add(
                    DeviceStateItem(
                        key = "current_default_app",
                        name = LocalizedString(english = "Current default app"),
                        jsonValue = packageName,
                    )
                )
            } else {
                result.add(
                    DeviceStateItem(
                        key = "candidate_default_app",
                        name = LocalizedString(english = "Candidate default app"),
                        jsonValue = packageName,
                    )
                )
            }
        }

        return result
    }

    companion object {
        const val KEY = "default_app"
    }
}

val deviceStateScreenKeys: List<String> =
    listOf(
        PermissionManagerScreen.KEY,
        PermissionAppsScreen.KEY,
        AppPermissionsScreen.KEY,
        AppPermissionSettingScreen.KEY,
        AppPermissionGrantStateScreen.KEY,
        UnusedAppsScreen.KEY,
        UnusedAppLastUsageScreen.KEY,
        AdditionalPermissionsScreen.KEY,
        DefaultAppListScreen.KEY,
        DefaultAppScreen.KEY,
    )