From 2a5dd0e2ada5cb9c2252429ec257939d5c305eb3 Mon Sep 17 00:00:00 2001
From: Evan Severson <evanseverson@google.com>
Date: Wed, 23 Jun 2021 21:44:21 -0700
Subject: [PATCH] Disable the mic/cam toggles when admin disallows

Test: Use test DPC app
Bug: 184927615
Change-Id: Ib3825967b2e7ddb6d4768abbc2e17fd39b936371
---
 res/xml/privacy_dashboard_settings.xml             |  4 ++--
 .../settings/privacy/CameraToggleController.java   |  7 +++++++
 .../settings/privacy/MicToggleController.java      |  6 ++++++
 .../settings/privacy/SensorToggleController.java   | 14 ++++++++++++++
 4 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/res/xml/privacy_dashboard_settings.xml b/res/xml/privacy_dashboard_settings.xml
index 954d3ef74d0..f4e7062908f 100644
--- a/res/xml/privacy_dashboard_settings.xml
+++ b/res/xml/privacy_dashboard_settings.xml
@@ -64,14 +64,14 @@
     </Preference>
 
     <!-- Camera toggle -->
-    <SwitchPreference
+    <com.android.settingslib.RestrictedSwitchPreference
         android:key="privacy_camera_toggle"
         android:title="@string/camera_toggle_title"
         android:summary="@string/sensor_toggle_description"
         settings:controller="com.android.settings.privacy.CameraToggleController"/>
 
     <!-- Microphone toggle -->
-    <SwitchPreference
+    <com.android.settingslib.RestrictedSwitchPreference
         android:key="privacy_mic_toggle"
         android:title="@string/mic_toggle_title"
         android:summary="@string/sensor_toggle_description"
diff --git a/src/com/android/settings/privacy/CameraToggleController.java b/src/com/android/settings/privacy/CameraToggleController.java
index a9229943fb5..1a5da76869c 100644
--- a/src/com/android/settings/privacy/CameraToggleController.java
+++ b/src/com/android/settings/privacy/CameraToggleController.java
@@ -16,6 +16,8 @@
 
 package com.android.settings.privacy;
 
+import static android.os.UserManager.DISALLOW_CAMERA_TOGGLE;
+
 import static com.android.settings.utils.SensorPrivacyManagerHelper.CAMERA;
 
 import android.content.Context;
@@ -40,4 +42,9 @@ public class CameraToggleController extends SensorToggleController {
                 && DeviceConfig.getBoolean(DeviceConfig.NAMESPACE_PRIVACY, "camera_toggle_enabled",
                 true) ? AVAILABLE_UNSEARCHABLE : UNSUPPORTED_ON_DEVICE;
     }
+
+    @Override
+    protected String getRestriction() {
+        return DISALLOW_CAMERA_TOGGLE;
+    }
 }
diff --git a/src/com/android/settings/privacy/MicToggleController.java b/src/com/android/settings/privacy/MicToggleController.java
index 419cbddccac..23d41d65cc8 100644
--- a/src/com/android/settings/privacy/MicToggleController.java
+++ b/src/com/android/settings/privacy/MicToggleController.java
@@ -16,6 +16,8 @@
 
 package com.android.settings.privacy;
 
+import static android.os.UserManager.DISALLOW_MICROPHONE_TOGGLE;
+
 import static com.android.settings.utils.SensorPrivacyManagerHelper.MICROPHONE;
 
 import android.content.Context;
@@ -41,4 +43,8 @@ public class MicToggleController extends SensorToggleController {
                 true) ? AVAILABLE_UNSEARCHABLE : UNSUPPORTED_ON_DEVICE;
     }
 
+    @Override
+    protected String getRestriction() {
+        return DISALLOW_MICROPHONE_TOGGLE;
+    }
 }
diff --git a/src/com/android/settings/privacy/SensorToggleController.java b/src/com/android/settings/privacy/SensorToggleController.java
index 9e8aca2de2d..1b9d57f3a12 100644
--- a/src/com/android/settings/privacy/SensorToggleController.java
+++ b/src/com/android/settings/privacy/SensorToggleController.java
@@ -22,6 +22,8 @@ import androidx.preference.PreferenceScreen;
 
 import com.android.settings.core.TogglePreferenceController;
 import com.android.settings.utils.SensorPrivacyManagerHelper;
+import com.android.settingslib.RestrictedLockUtilsInternal;
+import com.android.settingslib.RestrictedSwitchPreference;
 
 import java.util.concurrent.Executor;
 
@@ -44,6 +46,10 @@ public abstract class SensorToggleController extends TogglePreferenceController
      */
     public abstract int getSensor();
 
+    protected String getRestriction() {
+        return null;
+    }
+
     @Override
     public boolean isChecked() {
         return !mSensorPrivacyManagerHelper.isSensorBlocked(getSensor());
@@ -58,6 +64,14 @@ public abstract class SensorToggleController extends TogglePreferenceController
     @Override
     public void displayPreference(PreferenceScreen screen) {
         super.displayPreference(screen);
+
+        RestrictedSwitchPreference preference =
+                (RestrictedSwitchPreference) screen.findPreference(getPreferenceKey());
+        if (preference != null) {
+            preference.setDisabledByAdmin(RestrictedLockUtilsInternal
+                    .checkIfRestrictionEnforced(mContext, getRestriction(), mContext.getUserId()));
+        }
+
         mSensorPrivacyManagerHelper.addSensorBlockedListener(
                 getSensor(),
                 (sensor, blocked) -> updateState(screen.findPreference(mPreferenceKey)),