From 519d26d8f0e258dd7e13c0cfdf9582737871b20c Mon Sep 17 00:00:00 2001
From: Kyle Zhang <kelzhan@google.com>
Date: Tue, 31 Oct 2023 20:30:13 +0000
Subject: [PATCH] Add feature flag for force L3 dev option

Bug: 301669353
Change-Id: I289bed846982e94584fdb067e84186a61587d578
---
 Android.bp                                    |  1 +
 aconfig/Android.bp                            | 11 +++++++++++
 ...ettings_mediadrm_flag_declarations.aconfig |  8 ++++++++
 .../ForceL3FallbackPreferenceController.java  | 19 +++++++++++++++++++
 4 files changed, 39 insertions(+)
 create mode 100644 aconfig/media_drm/settings_mediadrm_flag_declarations.aconfig

diff --git a/Android.bp b/Android.bp
index 5a1224ca71d..cc2c2ae41ea 100644
--- a/Android.bp
+++ b/Android.bp
@@ -82,6 +82,7 @@ android_library {
 
         // Settings dependencies
         "FingerprintManagerInteractor",
+        "MediaDrmSettingsFlagsLib",
         "Settings-change-ids",
         "SettingsLib",
         "SettingsLibActivityEmbedding",
diff --git a/aconfig/Android.bp b/aconfig/Android.bp
index eb9a6b4155b..37b03bafb14 100644
--- a/aconfig/Android.bp
+++ b/aconfig/Android.bp
@@ -25,3 +25,14 @@ java_aconfig_library {
     name: "factory_reset_flags_lib",
     aconfig_declarations: "factory_reset_flags",
 }
+
+aconfig_declarations {
+    name: "media_drm_flags",
+    package: "com.android.settings.media_drm",
+    srcs: ["media_drm/*.aconfig"],
+}
+
+java_aconfig_library {
+    name: "MediaDrmSettingsFlagsLib",
+    aconfig_declarations: "media_drm_flags",
+}
\ No newline at end of file
diff --git a/aconfig/media_drm/settings_mediadrm_flag_declarations.aconfig b/aconfig/media_drm/settings_mediadrm_flag_declarations.aconfig
new file mode 100644
index 00000000000..06d75f1e34c
--- /dev/null
+++ b/aconfig/media_drm/settings_mediadrm_flag_declarations.aconfig
@@ -0,0 +1,8 @@
+package: "com.android.settings.media_drm"
+
+flag {
+    name: "force_l3_enabled"
+    namespace: "media_drm"
+    description: "Feature flag of forcing L3"
+    bug: "301669353"
+}
\ No newline at end of file
diff --git a/src/com/android/settings/development/widevine/ForceL3FallbackPreferenceController.java b/src/com/android/settings/development/widevine/ForceL3FallbackPreferenceController.java
index 05f2dc1f67c..78468c19f5f 100644
--- a/src/com/android/settings/development/widevine/ForceL3FallbackPreferenceController.java
+++ b/src/com/android/settings/development/widevine/ForceL3FallbackPreferenceController.java
@@ -18,17 +18,22 @@ package com.android.settings.development.widevine;
 
 import android.content.Context;
 import android.sysprop.WidevineProperties;
+import android.util.Log;
+
+import androidx.preference.Preference;
 
 import com.android.internal.annotations.VisibleForTesting;
 import com.android.settings.R;
 import com.android.settings.core.TogglePreferenceController;
 import com.android.settingslib.development.DevelopmentSettingsEnabler;
+import com.android.settings.media_drm.Flags;
 
 /**
  * The controller (in the Media Widevine settings) enforces L3 security level
 * of Widevine CDM.
 */
 public class ForceL3FallbackPreferenceController extends TogglePreferenceController {
+    private static final String TAG = "ForceL3FallbackPreferenceController";
 
     public ForceL3FallbackPreferenceController(Context context, String preferenceKey) {
         super(context, preferenceKey);
@@ -45,6 +50,20 @@ public class ForceL3FallbackPreferenceController extends TogglePreferenceControl
         return true;
     }
 
+    @Override
+    public void updateState(Preference preference) {
+        if (Flags.forceL3Enabled()) {
+            preference.setEnabled(true);
+            Log.i(TAG, "forceL3Enabled is on");
+        } else {
+            preference.setEnabled(false);
+            // In case of flag rollback, the controller should be unchecked.
+            WidevineProperties.forcel3_enabled(false);
+            Log.i(TAG, "forceL3Enabled is off");
+        }
+        super.updateState(preference);
+    }
+
     @Override
     public int getAvailabilityStatus() {
         if (DevelopmentSettingsEnabler.isDevelopmentSettingsEnabled(mContext)) {