Merge "Create RestrictedSwitchPreference for SpaLib"

2 files changed, 185 insertions, 0 deletions

diff --git a/packages/SettingsLib/SpaPrivileged/src/com/android/settingslib/spaprivileged/model/enterprise/RestrictionsProvider.kt b/packages/SettingsLib/SpaPrivileged/src/com/android/settingslib/spaprivileged/model/enterprise/RestrictionsProvider.kt
new file mode 100644
index 000000000000..061580702327
--- /dev/null
+++ b/packages/SettingsLib/SpaPrivileged/src/com/android/settingslib/spaprivileged/model/enterprise/RestrictionsProvider.kt
@@ -0,0 +1,88 @@
+/*
+ * Copyright (C) 2022 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.settingslib.spaprivileged.model.enterprise
+
+import android.app.admin.DevicePolicyResources.Strings.Settings
+import android.content.Context
+import android.os.UserHandle
+import android.os.UserManager
+import androidx.lifecycle.LiveData
+import com.android.settingslib.RestrictedLockUtils.EnforcedAdmin
+import com.android.settingslib.RestrictedLockUtilsInternal
+import com.android.settingslib.spaprivileged.R
+
+data class Restrictions(
+    val userId: Int,
+    val keys: List<String>,
+)
+
+sealed class RestrictedMode
+
+object NoRestricted : RestrictedMode()
+
+object BaseUserRestricted : RestrictedMode()
+
+data class BlockedByAdmin(
+    val enterpriseRepository: EnterpriseRepository,
+    val enforcedAdmin: EnforcedAdmin,
+) : RestrictedMode() {
+    fun getSummary(checked: Boolean?): String = when (checked) {
+        true -> enterpriseRepository.getEnterpriseString(
+            Settings.ENABLED_BY_ADMIN_SWITCH_SUMMARY, R.string.enabled_by_admin
+        )
+        false -> enterpriseRepository.getEnterpriseString(
+            Settings.DISABLED_BY_ADMIN_SWITCH_SUMMARY, R.string.disabled_by_admin
+        )
+        else -> ""
+    }
+}
+
+class RestrictionsProvider(
+    private val context: Context,
+    private val restrictions: Restrictions,
+) {
+    private val userManager by lazy { UserManager.get(context) }
+    private val enterpriseRepository by lazy { EnterpriseRepository(context) }
+
+    val restrictedMode = object : LiveData<RestrictedMode>() {
+        override fun onActive() {
+            postValue(getRestrictedMode())
+        }
+
+        override fun onInactive() {
+        }
+    }
+
+    private fun getRestrictedMode(): RestrictedMode {
+        for (key in restrictions.keys) {
+            if (userManager.hasBaseUserRestriction(key, UserHandle.of(restrictions.userId))) {
+                return BaseUserRestricted
+            }
+        }
+        for (key in restrictions.keys) {
+            RestrictedLockUtilsInternal
+                .checkIfRestrictionEnforced(context, key, restrictions.userId)
+                ?.let {
+                    return BlockedByAdmin(
+                        enterpriseRepository = enterpriseRepository,
+                        enforcedAdmin = it,
+                    )
+                }
+        }
+        return NoRestricted
+    }
+}
diff --git a/packages/SettingsLib/SpaPrivileged/src/com/android/settingslib/spaprivileged/template/preference/RestrictedSwitchPreference.kt b/packages/SettingsLib/SpaPrivileged/src/com/android/settingslib/spaprivileged/template/preference/RestrictedSwitchPreference.kt
new file mode 100644
index 000000000000..165481fb3ec1
--- /dev/null
+++ b/packages/SettingsLib/SpaPrivileged/src/com/android/settingslib/spaprivileged/template/preference/RestrictedSwitchPreference.kt
@@ -0,0 +1,97 @@
+/*
+ * Copyright (C) 2022 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.settingslib.spaprivileged.template.preference
+
+import android.content.Context
+import androidx.compose.foundation.clickable
+import androidx.compose.foundation.layout.Box
+import androidx.compose.runtime.Composable
+import androidx.compose.runtime.derivedStateOf
+import androidx.compose.runtime.livedata.observeAsState
+import androidx.compose.runtime.remember
+import androidx.compose.ui.Modifier
+import androidx.compose.ui.platform.LocalContext
+import androidx.compose.ui.semantics.Role
+import com.android.settingslib.RestrictedLockUtils
+import com.android.settingslib.spa.framework.compose.stateOf
+import com.android.settingslib.spa.widget.preference.SwitchPreference
+import com.android.settingslib.spa.widget.preference.SwitchPreferenceModel
+import com.android.settingslib.spaprivileged.R
+import com.android.settingslib.spaprivileged.model.enterprise.BaseUserRestricted
+import com.android.settingslib.spaprivileged.model.enterprise.BlockedByAdmin
+import com.android.settingslib.spaprivileged.model.enterprise.NoRestricted
+import com.android.settingslib.spaprivileged.model.enterprise.RestrictedMode
+import com.android.settingslib.spaprivileged.model.enterprise.Restrictions
+import com.android.settingslib.spaprivileged.model.enterprise.RestrictionsProvider
+
+@Composable
+fun RestrictedSwitchPreference(model: SwitchPreferenceModel, restrictions: Restrictions) {
+    if (restrictions.keys.isEmpty()) {
+        SwitchPreference(model)
+        return
+    }
+    val context = LocalContext.current
+    val restrictionsProvider = remember { RestrictionsProvider(context, restrictions) }
+    val restrictedMode = restrictionsProvider.restrictedMode.observeAsState().value ?: return
+    val restrictedSwitchModel = remember(restrictedMode) {
+        RestrictedSwitchPreferenceModel(context, model, restrictedMode)
+    }
+    Box(remember { restrictedSwitchModel.getModifier() }) {
+        SwitchPreference(restrictedSwitchModel)
+    }
+}
+
+private class RestrictedSwitchPreferenceModel(
+    private val context: Context,
+    model: SwitchPreferenceModel,
+    private val restrictedMode: RestrictedMode,
+) : SwitchPreferenceModel {
+    override val title = model.title
+
+    override val summary = when (restrictedMode) {
+        is NoRestricted -> model.summary
+        is BaseUserRestricted -> stateOf(context.getString(R.string.disabled))
+        is BlockedByAdmin -> derivedStateOf { restrictedMode.getSummary(model.checked.value) }
+    }
+
+    override val checked = when (restrictedMode) {
+        is NoRestricted -> model.checked
+        is BaseUserRestricted -> stateOf(false)
+        is BlockedByAdmin -> model.checked
+    }
+
+    override val changeable = when (restrictedMode) {
+        is NoRestricted -> model.changeable
+        is BaseUserRestricted -> stateOf(false)
+        is BlockedByAdmin -> stateOf(false)
+    }
+
+    override val onCheckedChange = when (restrictedMode) {
+        is NoRestricted -> model.onCheckedChange
+        is BaseUserRestricted -> null
+        is BlockedByAdmin -> null
+    }
+
+    fun getModifier(): Modifier = when (restrictedMode) {
+        is BlockedByAdmin -> Modifier.clickable(role = Role.Switch) {
+            RestrictedLockUtils.sendShowAdminSupportDetailsIntent(
+                context, restrictedMode.enforcedAdmin
+            )
+        }
+        else -> Modifier
+    }
+}