From d1d9798a5dd46bc06df681f05b9a6d926956e3a1 Mon Sep 17 00:00:00 2001
From: changbetty <changbetty@google.com>
Date: Sat, 9 Apr 2022 05:04:17 +0000
Subject: [PATCH] [LE Audio] Add entry point in Media Volume slice for
 broadcast sink

Bug: 228274114
Test: make RunSettingsRoboTests
Test: Manual test
Change-Id: I331232c30291348faf7166d4de8060a1cfe12bff
---
 res/values/styles.xml                         |  3 +
 .../core/SliderPreferenceController.java      |  8 ++
 .../media/MediaOutputIndicatorWorker.java     | 14 +++-
 .../MediaVolumePreferenceController.java      | 80 ++++++++++++++++++-
 .../settings/panel/PanelSlicesAdapter.java    |  3 +
 .../settings/slices/SliceBuilderUtils.java    |  5 ++
 6 files changed, 110 insertions(+), 3 deletions(-)

diff --git a/res/values/styles.xml b/res/values/styles.xml
index 5e4e5a48f98..e2a8423cd12 100644
--- a/res/values/styles.xml
+++ b/res/values/styles.xml
@@ -579,6 +579,9 @@
         <!-- Padding for indeterminate progress bar -->
         <item name="progressBarStartPadding">12dp</item>
         <item name="progressBarEndPadding">16dp</item>
+
+        <!-- Title icon size -->
+        <item name="iconSize">25dp</item>
     </style>
 
     <style name="SliceRow.Slider.LargeIcon">
diff --git a/src/com/android/settings/core/SliderPreferenceController.java b/src/com/android/settings/core/SliderPreferenceController.java
index 0a7ece25d4b..8c936f2a9b6 100644
--- a/src/com/android/settings/core/SliderPreferenceController.java
+++ b/src/com/android/settings/core/SliderPreferenceController.java
@@ -17,6 +17,7 @@ package com.android.settings.core;
 import android.content.Context;
 
 import androidx.preference.Preference;
+import androidx.slice.builders.SliceAction;
 
 import com.android.settings.slices.SliceData;
 
@@ -70,4 +71,11 @@ public abstract class SliderPreferenceController extends BasePreferenceControlle
     public int getSliceType() {
         return SliceData.SliceType.SLIDER;
     }
+
+    /**
+     * @return the SliceAction for the end item of the slice.
+     */
+    public SliceAction getSliceEndItem(Context context) {
+        return null;
+    }
 }
diff --git a/src/com/android/settings/media/MediaOutputIndicatorWorker.java b/src/com/android/settings/media/MediaOutputIndicatorWorker.java
index ef3a7bc02d7..a87b0eae0d5 100644
--- a/src/com/android/settings/media/MediaOutputIndicatorWorker.java
+++ b/src/com/android/settings/media/MediaOutputIndicatorWorker.java
@@ -34,6 +34,7 @@ import androidx.annotation.Nullable;
 import com.android.settings.bluetooth.Utils;
 import com.android.settings.slices.SliceBackgroundWorker;
 import com.android.settingslib.bluetooth.BluetoothCallback;
+import com.android.settingslib.bluetooth.LocalBluetoothLeBroadcast;
 import com.android.settingslib.bluetooth.LocalBluetoothManager;
 import com.android.settingslib.media.LocalMediaManager;
 import com.android.settingslib.media.MediaDevice;
@@ -126,7 +127,7 @@ public class MediaOutputIndicatorWorker extends SliceBackgroundWorker implements
     }
 
     @Nullable
-    MediaController getActiveLocalMediaController() {
+    public MediaController getActiveLocalMediaController() {
         return MediaOutputUtils.getActiveLocalMediaController(mContext.getSystemService(
                 MediaSessionManager.class));
     }
@@ -156,7 +157,7 @@ public class MediaOutputIndicatorWorker extends SliceBackgroundWorker implements
         return mMediaDevices;
     }
 
-    MediaDevice getCurrentConnectedMediaDevice() {
+    public MediaDevice getCurrentConnectedMediaDevice() {
         return mLocalMediaManager.getCurrentConnectedDevice();
     }
 
@@ -164,6 +165,15 @@ public class MediaOutputIndicatorWorker extends SliceBackgroundWorker implements
         return mPackageName;
     }
 
+    public boolean isDeviceBroadcasting() {
+        LocalBluetoothLeBroadcast broadcast =
+                mLocalBluetoothManager.getProfileManager().getLeAudioBroadcastProfile();
+        if (broadcast == null) {
+            return false;
+        }
+        return broadcast.isEnabled(null);
+    }
+
     private class DevicesChangedBroadcastReceiver extends BroadcastReceiver {
         @Override
         public void onReceive(Context context, Intent intent) {
diff --git a/src/com/android/settings/notification/MediaVolumePreferenceController.java b/src/com/android/settings/notification/MediaVolumePreferenceController.java
index e18d2e5b1f1..978a4b40237 100644
--- a/src/com/android/settings/notification/MediaVolumePreferenceController.java
+++ b/src/com/android/settings/notification/MediaVolumePreferenceController.java
@@ -16,16 +16,32 @@
 
 package com.android.settings.notification;
 
+import android.app.PendingIntent;
 import android.content.Context;
+import android.content.Intent;
+import android.graphics.drawable.Drawable;
 import android.media.AudioManager;
+import android.net.Uri;
 import android.text.TextUtils;
+import android.util.Log;
+
+import androidx.core.graphics.drawable.IconCompat;
+import androidx.slice.builders.ListBuilder;
+import androidx.slice.builders.SliceAction;
 
 import com.android.settings.R;
+import com.android.settings.Utils;
+import com.android.settings.media.MediaOutputIndicatorWorker;
+import com.android.settings.slices.CustomSliceRegistry;
+import com.android.settings.slices.SliceBackgroundWorker;
+import com.android.settingslib.media.MediaDevice;
 
 public class MediaVolumePreferenceController extends VolumeSeekBarPreferenceController {
-
+    private static final String TAG = "MediaVolumePreCtrl";
     private static final String KEY_MEDIA_VOLUME = "media_volume";
 
+    private MediaOutputIndicatorWorker mWorker;
+
     public MediaVolumePreferenceController(Context context) {
         super(context, KEY_MEDIA_VOLUME);
     }
@@ -66,4 +82,66 @@ public class MediaVolumePreferenceController extends VolumeSeekBarPreferenceCont
     public int getMuteIcon() {
         return R.drawable.ic_media_stream_off;
     }
+
+    private boolean isSupportEndItem() {
+        return getWorker() != null
+            && getWorker().getActiveLocalMediaController() != null
+            && isConnectedBLEDevice();
+    }
+
+    private boolean isConnectedBLEDevice() {
+        final MediaDevice device = getWorker().getCurrentConnectedMediaDevice();
+        if (device != null) {
+            return device.isBLEDevice();
+        }
+        return false;
+    }
+
+    @Override
+    public SliceAction getSliceEndItem(Context context) {
+        if (!isSupportEndItem()) {
+            Log.d(TAG, "The slice doesn't support end item");
+            return null;
+        }
+
+        final Intent intent = new Intent();
+        if (getWorker().isDeviceBroadcasting()) {
+            // TODO(b/229577323) : Get the intent action for the Media Output Broadcast Dialog
+            //  in SystemUI
+        } else {
+            // TODO(b/229577518) : Get the intent action of the Bluetooth Broadcast Dialog
+            //  for user to choose the action
+        }
+        final PendingIntent pi = PendingIntent.getBroadcast(context, 0 /* requestCode */, intent,
+                PendingIntent.FLAG_UPDATE_CURRENT | PendingIntent.FLAG_MUTABLE);
+        final IconCompat icon = getBroadcastIcon(context);
+
+        return SliceAction.createDeeplink(pi, icon, ListBuilder.ICON_IMAGE, getPreferenceKey());
+    }
+
+    private IconCompat getBroadcastIcon(Context context) {
+        final Drawable drawable = context.getDrawable(
+                com.android.settingslib.R.drawable.settings_input_antenna);
+        if (drawable != null) {
+            drawable.setTint(Utils.getColorAccentDefaultColor(context));
+            return Utils.createIconWithDrawable(drawable);
+        }
+        return null;
+    }
+
+    private MediaOutputIndicatorWorker getWorker() {
+        if (mWorker == null) {
+            mWorker = SliceBackgroundWorker.getInstance(getUri());
+        }
+        return mWorker;
+    }
+
+    private Uri getUri() {
+        return CustomSliceRegistry.VOLUME_MEDIA_URI;
+    }
+
+    @Override
+    public Class<? extends SliceBackgroundWorker> getBackgroundWorkerClass() {
+        return MediaOutputIndicatorWorker.class;
+    }
 }
diff --git a/src/com/android/settings/panel/PanelSlicesAdapter.java b/src/com/android/settings/panel/PanelSlicesAdapter.java
index 2486361a243..9f5ffe97fa5 100644
--- a/src/com/android/settings/panel/PanelSlicesAdapter.java
+++ b/src/com/android/settings/panel/PanelSlicesAdapter.java
@@ -143,6 +143,9 @@ public class PanelSlicesAdapter
                 sliceView.setVisibility(View.GONE);
             }
 
+            // Add divider for the end icon
+            sliceView.setShowActionDividers(true);
+
             // Log Panel interaction
             sliceView.setOnSliceActionListener(
                     ((eventInfo, sliceItem) -> {
diff --git a/src/com/android/settings/slices/SliceBuilderUtils.java b/src/com/android/settings/slices/SliceBuilderUtils.java
index ce5384f8909..8d5a0b4d8c7 100644
--- a/src/com/android/settings/slices/SliceBuilderUtils.java
+++ b/src/com/android/settings/slices/SliceBuilderUtils.java
@@ -339,6 +339,11 @@ public class SliceBuilderUtils {
             inputRangeBuilder.setSubtitle(subtitleText);
         }
 
+        SliceAction endItemAction = sliderController.getSliceEndItem(context);
+        if (endItemAction != null) {
+            inputRangeBuilder.addEndItem(endItemAction);
+        }
+
         return new ListBuilder(context, sliceData.getUri(), ListBuilder.INFINITY)
                 .setAccentColor(color)
                 .addInputRange(inputRangeBuilder)