From 02753a7e2997d58a4dea0b9bc6c6760020ab4997 Mon Sep 17 00:00:00 2001
From: Jacky Wang <jiannan@google.com>
Date: Fri, 8 Nov 2024 04:09:05 +0800
Subject: [PATCH] [Catalyst] Add restriction for Internet

Also add the missing settings:userRestriction in xml.

Bug: 377600992
Flag: com.android.settings.flags.catalyst
Test: testdpc
Change-Id: I179dbba19867d864ec2df5989857083511777514
---
 res/xml/network_provider_internet.xml                  |  1 +
 .../android/settings/network/NetworkProviderScreen.kt  | 10 +++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/res/xml/network_provider_internet.xml b/res/xml/network_provider_internet.xml
index 1437db67da5..1d1bf16dbb0 100644
--- a/res/xml/network_provider_internet.xml
+++ b/res/xml/network_provider_internet.xml
@@ -30,6 +30,7 @@
         settings:allowDividerAbove="true"
         settings:keywords="@string/keywords_internet"
         settings:useAdminDisabledSummary="true"
+        settings:userRestriction="no_config_wifi"
         settings:controller="com.android.settings.network.InternetPreferenceControllerV2" />
 
     <com.android.settingslib.RestrictedPreference
diff --git a/src/com/android/settings/network/NetworkProviderScreen.kt b/src/com/android/settings/network/NetworkProviderScreen.kt
index fa3ed788b2a..77ecf3aaa85 100644
--- a/src/com/android/settings/network/NetworkProviderScreen.kt
+++ b/src/com/android/settings/network/NetworkProviderScreen.kt
@@ -16,6 +16,8 @@
 package com.android.settings.network
 
 import android.content.Context
+import android.os.UserManager
+import com.android.settings.PreferenceRestrictionMixin
 import com.android.settings.R
 import com.android.settings.flags.Flags
 import com.android.settingslib.metadata.PreferenceAvailabilityProvider
@@ -24,7 +26,8 @@ import com.android.settingslib.metadata.preferenceHierarchy
 import com.android.settingslib.preference.PreferenceScreenCreator
 
 @ProvidePreferenceScreen
-class NetworkProviderScreen : PreferenceScreenCreator, PreferenceAvailabilityProvider {
+class NetworkProviderScreen :
+    PreferenceScreenCreator, PreferenceAvailabilityProvider, PreferenceRestrictionMixin {
     override val key: String
         get() = KEY
 
@@ -40,6 +43,11 @@ class NetworkProviderScreen : PreferenceScreenCreator, PreferenceAvailabilityPro
     override fun isAvailable(context: Context) =
         context.resources.getBoolean(R.bool.config_show_internet_settings)
 
+    override fun isEnabled(context: Context) = super<PreferenceRestrictionMixin>.isEnabled(context)
+
+    override val restrictionKey: String
+        get() = UserManager.DISALLOW_CONFIG_WIFI
+
     override fun isFlagEnabled(context: Context) = Flags.catalystInternetSettings()
 
     override fun hasCompleteHierarchy() = false