/*
 * 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.settings.wifi

import android.content.Context
import android.net.wifi.WifiManager
import android.security.advancedprotection.AdvancedProtectionManager
import androidx.compose.foundation.clickable
import androidx.compose.foundation.layout.Box
import androidx.compose.foundation.layout.fillMaxWidth
import androidx.compose.material3.Text
import androidx.compose.runtime.Composable
import androidx.compose.runtime.getValue
import androidx.compose.runtime.mutableStateOf
import androidx.compose.runtime.saveable.rememberSaveable
import androidx.compose.runtime.setValue
import androidx.compose.ui.Modifier
import androidx.compose.ui.res.stringResource
import androidx.compose.ui.semantics.Role
import androidx.compose.ui.text.style.TextAlign
import androidx.lifecycle.compose.collectAsStateWithLifecycle
import com.android.settings.R
import com.android.settings.spa.preference.ComposePreferenceController
import com.android.settingslib.spa.framework.compose.OverridableFlow
import com.android.settingslib.spa.widget.dialog.AlertDialogButton
import com.android.settingslib.spa.widget.dialog.SettingsAlertDialogWithIcon
import com.android.settingslib.spa.widget.preference.SwitchPreference
import com.android.settingslib.spa.widget.preference.SwitchPreferenceModel
import com.android.wifi.flags.Flags
import com.android.wifitrackerlib.WifiEntry
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.asExecutor
import kotlinx.coroutines.channels.awaitClose
import kotlinx.coroutines.flow.callbackFlow
import kotlinx.coroutines.flow.flow
import kotlinx.coroutines.flow.flowOn

/** Controller that controls whether the WEP network can be connected. */
class WepNetworksPreferenceController(context: Context, preferenceKey: String) :
    ComposePreferenceController(context, preferenceKey) {

    var wifiManager = context.getSystemService(WifiManager::class.java)!!
    var aapmManager = if (android.security.Flags.aapmApi() && Flags.wepDisabledInApm())
        context.getSystemService(AdvancedProtectionManager::class.java)!!
    else null

    override fun getAvailabilityStatus() =
        if (Flags.androidVWifiApi()) AVAILABLE else UNSUPPORTED_ON_DEVICE

    @Composable
    override fun Content() {
        val isWepSupported: Boolean? =
            isWepSupportedFlow.collectAsStateWithLifecycle(initialValue = null).value
        val isWepAllowed: Boolean? =
            wepAllowedFlow.flow.collectAsStateWithLifecycle(initialValue = null).value
        val isAapmEnabled: Boolean? = if (android.security.Flags.aapmApi()
            && Flags.wepDisabledInApm())
                isAapmEnabledFlow.collectAsStateWithLifecycle(initialValue = null).value
        else false

        var openDialog by rememberSaveable { mutableStateOf(false) }

        RestrictionWrapper(
            restricted = isAapmEnabled == true
        ) {
            SwitchPreference(
                object : SwitchPreferenceModel {
                    override val title = stringResource(R.string.wifi_allow_wep_networks)
                    override val summary = { getSummary(isWepSupported) }
                    override val checked = {
                        when {
                            isWepSupported == false -> false
                            isAapmEnabled == true -> false
                            else -> isWepAllowed
                        }
                    }
                    override val changeable: () -> Boolean
                        get() = { isWepSupported == true && isAapmEnabled == false }

                    override val onCheckedChange: ((Boolean) -> Unit)? =
                        if (isAapmEnabled == true) {
                            null
                        } else {
                            { newChecked ->
                                val wifiInfo = wifiManager.connectionInfo
                                if (!newChecked &&
                                    wifiInfo.currentSecurityType == WifiEntry.SECURITY_WEP
                                ) {
                                    openDialog = true
                                } else {
                                    wifiManager.setWepAllowed(newChecked)
                                    wepAllowedFlow.override(newChecked)
                                }
                            }
                        }
                }
            )
        }
        if (openDialog) {
            SettingsAlertDialogWithIcon(
                onDismissRequest = { openDialog = false },
                confirmButton =
                    AlertDialogButton(stringResource(R.string.sim_action_yes)) {
                        wifiManager.setWepAllowed(false)
                        wepAllowedFlow.override(false)
                        openDialog = false
                    },
                dismissButton =
                    AlertDialogButton(stringResource(R.string.wifi_cancel)) { openDialog = false },
                title = stringResource(R.string.wifi_settings_wep_networks_disconnect_title),
                text = {
                    Text(
                        stringResource(R.string.wifi_settings_wep_networks_disconnect_summary),
                        modifier = Modifier.fillMaxWidth(),
                        textAlign = TextAlign.Center,
                    )
                })
        }
    }

    @Composable
    private fun RestrictionWrapper(restricted: Boolean, content: @Composable () -> Unit) {
        if (restricted) {
            Box(
                Modifier.clickable(
                    enabled = true,
                    role = Role.Switch,
                    onClick = ::startSupportIntent
                )
            ) { content() }
        } else {
            content()
        }
    }

    private fun getSummary(isWepSupported: Boolean?): String =
        mContext.getString(
            when (isWepSupported) {
                true -> R.string.wifi_allow_wep_networks_summary
                false -> R.string.wifi_allow_wep_networks_summary_carrier_not_allow
                null -> R.string.summary_placeholder
            })

    private val isWepSupportedFlow =
        flow { emit(wifiManager.isWepSupported) }.flowOn(Dispatchers.Default)

    private val isAapmEnabledFlow = flow {
        emit(aapmManager?.isAdvancedProtectionEnabled ?: false) }.flowOn(Dispatchers.Default)

    private fun startSupportIntent() {
        aapmManager?.createSupportIntent(
            AdvancedProtectionManager.FEATURE_ID_DISALLOW_WEP,
            AdvancedProtectionManager.SUPPORT_DIALOG_TYPE_DISABLED_SETTING
        )?.let { mContext.startActivity(it) }
    }

    val wepAllowedFlow =
        OverridableFlow(
            callbackFlow {
                wifiManager.queryWepAllowed(Dispatchers.Default.asExecutor(), ::trySend)

                awaitClose {}
            })
}