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 import com.android.systemui.broadcast.BroadcastDispatcher;
32 import com.android.systemui.dagger.SysUISingleton;
33 
34 import javax.inject.Inject;
35 
36 /**
37  * Allows settings to register certain broadcasts to launch the settings app for pinned slices.
38  * @see SliceBroadcastRelay
39  */
40 @SysUISingleton
41 public class SliceBroadcastRelayHandler extends SystemUI {
42     private static final String TAG = "SliceBroadcastRelay";
43     private static final boolean DEBUG = false;
44 
45     private final ArrayMap<Uri, BroadcastRelay> mRelays = new ArrayMap<>();
46     private final BroadcastDispatcher mBroadcastDispatcher;
47 
48     @Inject
SliceBroadcastRelayHandler(Context context, BroadcastDispatcher broadcastDispatcher)49     public SliceBroadcastRelayHandler(Context context, BroadcastDispatcher broadcastDispatcher) {
50         super(context);
51         mBroadcastDispatcher = broadcastDispatcher;
52     }
53 
54     @Override
start()55     public void start() {
56         if (DEBUG) Log.d(TAG, "Start");
57         IntentFilter filter = new IntentFilter(SliceBroadcastRelay.ACTION_REGISTER);
58         filter.addAction(SliceBroadcastRelay.ACTION_UNREGISTER);
59         mBroadcastDispatcher.registerReceiver(mReceiver, filter);
60     }
61 
62     // This does not use BroadcastDispatcher as the filter may have schemas or mime types.
63     @VisibleForTesting
handleIntent(Intent intent)64     void handleIntent(Intent intent) {
65         if (SliceBroadcastRelay.ACTION_REGISTER.equals(intent.getAction())) {
66             Uri uri = intent.getParcelableExtra(SliceBroadcastRelay.EXTRA_URI);
67             ComponentName receiverClass =
68                     intent.getParcelableExtra(SliceBroadcastRelay.EXTRA_RECEIVER);
69             IntentFilter filter = intent.getParcelableExtra(SliceBroadcastRelay.EXTRA_FILTER);
70             if (DEBUG) Log.d(TAG, "Register " + uri + " " + receiverClass + " " + filter);
71             getOrCreateRelay(uri).register(mContext, receiverClass, filter);
72         } else if (SliceBroadcastRelay.ACTION_UNREGISTER.equals(intent.getAction())) {
73             Uri uri = intent.getParcelableExtra(SliceBroadcastRelay.EXTRA_URI);
74             if (DEBUG) Log.d(TAG, "Unregister " + uri);
75             BroadcastRelay relay = getAndRemoveRelay(uri);
76             if (relay != null) {
77                 relay.unregister(mContext);
78             }
79         }
80     }
81 
getOrCreateRelay(Uri uri)82     private BroadcastRelay getOrCreateRelay(Uri uri) {
83         BroadcastRelay ret = mRelays.get(uri);
84         if (ret == null) {
85             ret = new BroadcastRelay(uri);
86             mRelays.put(uri, ret);
87         }
88         return ret;
89     }
90 
getAndRemoveRelay(Uri uri)91     private BroadcastRelay getAndRemoveRelay(Uri uri) {
92         return mRelays.remove(uri);
93     }
94 
95     private final BroadcastReceiver mReceiver = new BroadcastReceiver() {
96         @Override
97         public void onReceive(Context context, Intent intent) {
98             handleIntent(intent);
99         }
100     };
101 
102     private static class BroadcastRelay extends BroadcastReceiver {
103 
104         private final ArraySet<ComponentName> mReceivers = new ArraySet<>();
105         private final UserHandle mUserId;
106         private final Uri mUri;
107 
BroadcastRelay(Uri uri)108         public BroadcastRelay(Uri uri) {
109             mUserId = new UserHandle(ContentProvider.getUserIdFromUri(uri));
110             mUri = uri;
111         }
112 
register(Context context, ComponentName receiver, IntentFilter filter)113         public void register(Context context, ComponentName receiver, IntentFilter filter) {
114             mReceivers.add(receiver);
115             context.registerReceiver(this, filter);
116         }
117 
unregister(Context context)118         public void unregister(Context context) {
119             context.unregisterReceiver(this);
120         }
121 
122         @Override
onReceive(Context context, Intent intent)123         public void onReceive(Context context, Intent intent) {
124             intent.addFlags(Intent.FLAG_RECEIVER_FOREGROUND);
125             for (ComponentName receiver : mReceivers) {
126                 intent.setComponent(receiver);
127                 intent.putExtra(SliceBroadcastRelay.EXTRA_URI, mUri.toString());
128                 if (DEBUG) Log.d(TAG, "Forwarding " + receiver + " " + intent + " " + mUserId);
129                 context.sendBroadcastAsUser(intent, mUserId);
130             }
131         }
132     }
133 }
134