Add APIs for slice pinning

These APIs allow apps to know when they should be sending updates
to any given slice uri.

Test: update-api
Bug: 68378571
Change-Id: I4fa218c8d376692fa843f21777c87d592169a377
diff --git a/core/java/android/app/slice/SliceManager.java b/core/java/android/app/slice/SliceManager.java
index f8e19c1..0c5f225d 100644
--- a/core/java/android/app/slice/SliceManager.java
+++ b/core/java/android/app/slice/SliceManager.java
@@ -16,24 +16,37 @@
 
 package android.app.slice;
 
+import android.annotation.NonNull;
 import android.annotation.SystemService;
-import android.app.slice.ISliceListener.Stub;
 import android.content.Context;
 import android.net.Uri;
 import android.os.Handler;
 import android.os.RemoteException;
 import android.os.ServiceManager;
 import android.os.ServiceManager.ServiceNotFoundException;
+import android.util.ArrayMap;
+import android.util.Pair;
+
+import java.util.Arrays;
+import java.util.List;
+import java.util.concurrent.Executor;
 
 /**
- * @hide
+ * Class to handle interactions with {@link Slice}s.
+ * <p>
+ * The SliceManager manages permissions and pinned state for slices.
  */
 @SystemService(Context.SLICE_SERVICE)
 public class SliceManager {
 
     private final ISliceManager mService;
     private final Context mContext;
+    private final ArrayMap<Pair<Uri, SliceCallback>, ISliceListener> mListenerLookup =
+            new ArrayMap<>();
 
+    /**
+     * @hide
+     */
     public SliceManager(Context context, Handler handler) throws ServiceNotFoundException {
         mContext = context;
         mService = ISliceManager.Stub.asInterface(
@@ -41,38 +54,142 @@
     }
 
     /**
+     * Adds a callback to a specific slice uri.
+     * <p>
+     * This is a convenience that performs a few slice actions at once. It will put
+     * the slice in a pinned state since there is a callback attached. It will also
+     * listen for content changes, when a content change observes, the android system
+     * will bind the new slice and provide it to all registered {@link SliceCallback}s.
+     *
+     * @param uri The uri of the slice being listened to.
+     * @param callback The listener that should receive the callbacks.
+     * @param specs The list of supported {@link SliceSpec}s of the callback.
+     * @see SliceProvider#onSlicePinned(Uri)
      */
-    public void addSliceListener(Uri uri, SliceListener listener, SliceSpec[] specs) {
+    public void registerSliceCallback(@NonNull Uri uri, @NonNull SliceCallback callback,
+            @NonNull List<SliceSpec> specs) {
+        registerSliceCallback(uri, callback, specs, Handler.getMain());
+    }
+
+    /**
+     * Adds a callback to a specific slice uri.
+     * <p>
+     * This is a convenience that performs a few slice actions at once. It will put
+     * the slice in a pinned state since there is a callback attached. It will also
+     * listen for content changes, when a content change observes, the android system
+     * will bind the new slice and provide it to all registered {@link SliceCallback}s.
+     *
+     * @param uri The uri of the slice being listened to.
+     * @param callback The listener that should receive the callbacks.
+     * @param specs The list of supported {@link SliceSpec}s of the callback.
+     * @see SliceProvider#onSlicePinned(Uri)
+     */
+    public void registerSliceCallback(@NonNull Uri uri, @NonNull SliceCallback callback,
+            @NonNull List<SliceSpec> specs, Handler handler) {
         try {
-            mService.addSliceListener(uri, mContext.getPackageName(), listener.mStub, specs);
+            mService.addSliceListener(uri, mContext.getPackageName(),
+                    getListener(uri, callback, new ISliceListener.Stub() {
+                        @Override
+                        public void onSliceUpdated(Slice s) throws RemoteException {
+                            handler.post(() -> callback.onSliceUpdated(s));
+                        }
+                    }), specs.toArray(new SliceSpec[specs.size()]));
         } catch (RemoteException e) {
             throw e.rethrowFromSystemServer();
         }
     }
 
     /**
+     * Adds a callback to a specific slice uri.
+     * <p>
+     * This is a convenience that performs a few slice actions at once. It will put
+     * the slice in a pinned state since there is a callback attached. It will also
+     * listen for content changes, when a content change observes, the android system
+     * will bind the new slice and provide it to all registered {@link SliceCallback}s.
+     *
+     * @param uri The uri of the slice being listened to.
+     * @param callback The listener that should receive the callbacks.
+     * @param specs The list of supported {@link SliceSpec}s of the callback.
+     * @see SliceProvider#onSlicePinned(Uri)
      */
-    public void removeSliceListener(Uri uri, SliceListener listener) {
+    public void registerSliceCallback(@NonNull Uri uri, @NonNull SliceCallback callback,
+            @NonNull List<SliceSpec> specs, Executor executor) {
         try {
-            mService.removeSliceListener(uri, mContext.getPackageName(), listener.mStub);
+            mService.addSliceListener(uri, mContext.getPackageName(),
+                    getListener(uri, callback, new ISliceListener.Stub() {
+                        @Override
+                        public void onSliceUpdated(Slice s) throws RemoteException {
+                            executor.execute(() -> callback.onSliceUpdated(s));
+                        }
+                    }), specs.toArray(new SliceSpec[specs.size()]));
+        } catch (RemoteException e) {
+            throw e.rethrowFromSystemServer();
+        }
+    }
+
+    private ISliceListener getListener(Uri uri, SliceCallback callback,
+            ISliceListener listener) {
+        Pair<Uri, SliceCallback> key = new Pair<>(uri, callback);
+        if (mListenerLookup.containsKey(key)) {
+            try {
+                mService.removeSliceListener(uri, mContext.getPackageName(),
+                        mListenerLookup.get(key));
+            } catch (RemoteException e) {
+                throw e.rethrowFromSystemServer();
+            }
+        }
+        mListenerLookup.put(key, listener);
+        return listener;
+    }
+
+    /**
+     * Removes a callback for a specific slice uri.
+     * <p>
+     * Removes the app from the pinned state (if there are no other apps/callbacks pinning it)
+     * in addition to removing the callback.
+     *
+     * @param uri The uri of the slice being listened to
+     * @param callback The listener that should no longer receive callbacks.
+     * @see #registerSliceCallback
+     */
+    public void unregisterSliceCallback(@NonNull Uri uri, @NonNull SliceCallback callback) {
+        try {
+            mService.removeSliceListener(uri, mContext.getPackageName(),
+                    mListenerLookup.remove(new Pair<>(uri, callback)));
         } catch (RemoteException e) {
             throw e.rethrowFromSystemServer();
         }
     }
 
     /**
+     * Ensures that a slice is in a pinned state.
+     * <p>
+     * Pinned state is not persisted across reboots, so apps are expected to re-pin any slices
+     * they still care about after a reboot.
+     *
+     * @param uri The uri of the slice being pinned.
+     * @param specs The list of supported {@link SliceSpec}s of the callback.
+     * @see SliceProvider#onSlicePinned(Uri)
      */
-    public void pinSlice(Uri uri, SliceSpec[] specs) {
+    public void pinSlice(@NonNull Uri uri, @NonNull List<SliceSpec> specs) {
         try {
-            mService.pinSlice(mContext.getPackageName(), uri, specs);
+            mService.pinSlice(mContext.getPackageName(), uri,
+                    specs.toArray(new SliceSpec[specs.size()]));
         } catch (RemoteException e) {
             throw e.rethrowFromSystemServer();
         }
     }
 
     /**
+     * Remove a pin for a slice.
+     * <p>
+     * If the slice has no other pins/callbacks then the slice will be unpinned.
+     *
+     * @param uri The uri of the slice being unpinned.
+     * @see #pinSlice
+     * @see SliceProvider#onSliceUnpinned(Uri)
      */
-    public void unpinSlice(Uri uri) {
+    public void unpinSlice(@NonNull Uri uri) {
         try {
             mService.unpinSlice(mContext.getPackageName(), uri);
         } catch (RemoteException e) {
@@ -81,6 +198,7 @@
     }
 
     /**
+     * @hide
      */
     public boolean hasSliceAccess() {
         try {
@@ -91,41 +209,31 @@
     }
 
     /**
+     * Get the current set of specs for a pinned slice.
+     * <p>
+     * This is the set of specs supported for a specific pinned slice. It will take
+     * into account all clients and returns only specs supported by all.
+     * @see SliceSpec
      */
-    public SliceSpec[] getPinnedSpecs(Uri uri) {
+    public @NonNull List<SliceSpec> getPinnedSpecs(Uri uri) {
         try {
-            return mService.getPinnedSpecs(uri, mContext.getPackageName());
+            return Arrays.asList(mService.getPinnedSpecs(uri, mContext.getPackageName()));
         } catch (RemoteException e) {
             throw e.rethrowFromSystemServer();
         }
     }
 
     /**
+     * Class that listens to changes in {@link Slice}s.
      */
-    public abstract static class SliceListener {
-        private final Handler mHandler;
+    public interface SliceCallback {
 
         /**
+         * Called when slice is updated.
+         *
+         * @param s The updated slice.
+         * @see #registerSliceCallback
          */
-        public SliceListener() {
-            this(Handler.getMain());
-        }
-
-        /**
-         */
-        public SliceListener(Handler h) {
-            mHandler = h;
-        }
-
-        /**
-         */
-        public abstract void onSliceUpdated(Slice s);
-
-        private final ISliceListener.Stub mStub = new Stub() {
-            @Override
-            public void onSliceUpdated(Slice s) throws RemoteException {
-                mHandler.post(() -> SliceListener.this.onSliceUpdated(s));
-            }
-        };
+        void onSliceUpdated(Slice s);
     }
 }