[Catalyst] Move PreferenceRestrictionMixin

NO_IFTTT=Catalyst only

Bug: 377600992
Flag: EXEMPT refactor
Test: atest
Change-Id: I3c26a5a56d8fd6e887e4ff4d38d8e8eb97844672

src/com/android/settings/restriction/PreferenceRestrictionMixin.kt

@@ -0,0 +1,59 @@
+/*
+ * 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.restriction
+
+import android.content.Context
+import android.os.UserHandle
+import android.os.UserManager
+import androidx.annotation.CallSuper
+import com.android.settingslib.RestrictedLockUtils.EnforcedAdmin
+import com.android.settingslib.RestrictedLockUtilsInternal
+import com.android.settingslib.metadata.PreferenceRestrictionProvider
+
+/** Mixin to support restriction. */
+interface PreferenceRestrictionMixin : PreferenceRestrictionProvider {
+
+    /**
+     * Keys for restriction.
+     *
+     * Preference is restricted when **ANY** key in the list is restricted.
+     */
+    val restrictionKeys: Array<String>
+
+    val useAdminDisabledSummary: Boolean
+        get() = false
+
+    @CallSuper fun isEnabled(context: Context) = !context.hasBaseUserRestriction(restrictionKeys)
+
+    override fun isRestricted(context: Context) =
+        context.getRestrictionEnforcedAdmin(restrictionKeys) != null
+}
+
+/** Returns the admin that has enforced restriction on given keys. */
+fun Context.getRestrictionEnforcedAdmin(restrictionKeys: Array<String>): EnforcedAdmin? {
+    val userId = UserHandle.myUserId()
+    return restrictionKeys.firstNotNullOfOrNull {
+        RestrictedLockUtilsInternal.checkIfRestrictionEnforced(this, it, userId)
+    }
+}
+
+/** Returns if there is **any** base user restriction on given keys. */
+fun Context.hasBaseUserRestriction(restrictionKeys: Array<String>): Boolean {
+    val userManager = getSystemService(UserManager::class.java) ?: return false
+    val userHandle = UserHandle.of(UserHandle.myUserId())
+    return restrictionKeys.any { userManager.hasBaseUserRestriction(it, userHandle) }
+}

src/com/android/settings/restriction/UserRestrictionBindingHelper.kt

@@ -17,7 +17,6 @@
 package com.android.settings.restriction
 
 import android.content.Context
-import com.android.settings.PreferenceRestrictionMixin
 import com.android.settingslib.datastore.HandlerExecutor
 import com.android.settingslib.datastore.KeyedObserver
 import com.android.settingslib.metadata.PreferenceChangeReason