Home | History | Annotate | Download | only in systemui 
      1 /*
      2  * Copyright (C) 2018 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
      5  * except in compliance with the License. You may obtain a copy of the License at
      6  *
      7  *      http://www.apache.org/licenses/LICENSE-2.0
      8  *
      9  * Unless required by applicable law or agreed to in writing, software distributed under the
     10  * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
     11  * KIND, either express or implied. See the License for the specific language governing
     12  * permissions and limitations under the License.
     13  */
     14 
     15 package com.android.systemui;
     16 
     17 import android.content.BroadcastReceiver;
     18 import android.content.ComponentName;
     19 import android.content.ContentProvider;
     20 import android.content.Context;
     21 import android.content.Intent;
     22 import android.content.IntentFilter;
     23 import android.net.Uri;
     24 import android.os.UserHandle;
     25 import android.util.ArrayMap;
     26 import android.util.ArraySet;
     27 import android.util.Log;
     28 
     29 import com.android.internal.annotations.VisibleForTesting;
     30 import com.android.settingslib.SliceBroadcastRelay;
     31 
     32 /**
     33  * Allows settings to register certain broadcasts to launch the settings app for pinned slices.
     34  * @see SliceBroadcastRelay
     35  */
     36 public class SliceBroadcastRelayHandler extends SystemUI {
     37     private static final String TAG = "SliceBroadcastRelay";
     38     private static final boolean DEBUG = false;
     39 
     40     private final ArrayMap<Uri, BroadcastRelay> mRelays = new ArrayMap<>();
     41 
     42     @Override
     43     public void start() {
     44         if (DEBUG) Log.d(TAG, "Start");
     45         IntentFilter filter = new IntentFilter(SliceBroadcastRelay.ACTION_REGISTER);
     46         filter.addAction(SliceBroadcastRelay.ACTION_UNREGISTER);
     47         mContext.registerReceiver(mReceiver, filter);
     48     }
     49 
     50     @VisibleForTesting
     51     void handleIntent(Intent intent) {
     52         if (SliceBroadcastRelay.ACTION_REGISTER.equals(intent.getAction())) {
     53             Uri uri = intent.getParcelableExtra(SliceBroadcastRelay.EXTRA_URI);
     54             ComponentName receiverClass =
     55                     intent.getParcelableExtra(SliceBroadcastRelay.EXTRA_RECEIVER);
     56             IntentFilter filter = intent.getParcelableExtra(SliceBroadcastRelay.EXTRA_FILTER);
     57             if (DEBUG) Log.d(TAG, "Register " + uri + " " + receiverClass + " " + filter);
     58             getOrCreateRelay(uri).register(mContext, receiverClass, filter);
     59         } else if (SliceBroadcastRelay.ACTION_UNREGISTER.equals(intent.getAction())) {
     60             Uri uri = intent.getParcelableExtra(SliceBroadcastRelay.EXTRA_URI);
     61             if (DEBUG) Log.d(TAG, "Unregister " + uri);
     62             getAndRemoveRelay(uri).unregister(mContext);
     63         }
     64     }
     65 
     66     private BroadcastRelay getOrCreateRelay(Uri uri) {
     67         BroadcastRelay ret = mRelays.get(uri);
     68         if (ret == null) {
     69             ret = new BroadcastRelay(uri);
     70             mRelays.put(uri, ret);
     71         }
     72         return ret;
     73     }
     74 
     75     private BroadcastRelay getAndRemoveRelay(Uri uri) {
     76         return mRelays.remove(uri);
     77     }
     78 
     79     private final BroadcastReceiver mReceiver = new BroadcastReceiver() {
     80         @Override
     81         public void onReceive(Context context, Intent intent) {
     82             handleIntent(intent);
     83         }
     84     };
     85 
     86     private static class BroadcastRelay extends BroadcastReceiver {
     87 
     88         private final ArraySet<ComponentName> mReceivers = new ArraySet<>();
     89         private final UserHandle mUserId;
     90         private final Uri mUri;
     91 
     92         public BroadcastRelay(Uri uri) {
     93             mUserId = new UserHandle(ContentProvider.getUserIdFromUri(uri));
     94             mUri = uri;
     95         }
     96 
     97         public void register(Context context, ComponentName receiver, IntentFilter filter) {
     98             mReceivers.add(receiver);
     99             context.registerReceiver(this, filter);
    100         }
    101 
    102         public void unregister(Context context) {
    103             context.unregisterReceiver(this);
    104         }
    105 
    106         @Override
    107         public void onReceive(Context context, Intent intent) {
    108             intent.addFlags(Intent.FLAG_RECEIVER_FOREGROUND);
    109             for (ComponentName receiver : mReceivers) {
    110                 intent.setComponent(receiver);
    111                 intent.putExtra(SliceBroadcastRelay.EXTRA_URI, mUri.toString());
    112                 if (DEBUG) Log.d(TAG, "Forwarding " + receiver + " " + intent + " " + mUserId);
    113                 context.sendBroadcastAsUser(intent, mUserId);
    114             }
    115         }
    116     }
    117 }
    118