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 start()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 handleIntent(Intent intent)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 BroadcastRelay relay = getAndRemoveRelay(uri); 63 if (relay != null) { 64 relay.unregister(mContext); 65 } 66 } 67 } 68 getOrCreateRelay(Uri uri)69 private BroadcastRelay getOrCreateRelay(Uri uri) { 70 BroadcastRelay ret = mRelays.get(uri); 71 if (ret == null) { 72 ret = new BroadcastRelay(uri); 73 mRelays.put(uri, ret); 74 } 75 return ret; 76 } 77 getAndRemoveRelay(Uri uri)78 private BroadcastRelay getAndRemoveRelay(Uri uri) { 79 return mRelays.remove(uri); 80 } 81 82 private final BroadcastReceiver mReceiver = new BroadcastReceiver() { 83 @Override 84 public void onReceive(Context context, Intent intent) { 85 handleIntent(intent); 86 } 87 }; 88 89 private static class BroadcastRelay extends BroadcastReceiver { 90 91 private final ArraySet<ComponentName> mReceivers = new ArraySet<>(); 92 private final UserHandle mUserId; 93 private final Uri mUri; 94 BroadcastRelay(Uri uri)95 public BroadcastRelay(Uri uri) { 96 mUserId = new UserHandle(ContentProvider.getUserIdFromUri(uri)); 97 mUri = uri; 98 } 99 register(Context context, ComponentName receiver, IntentFilter filter)100 public void register(Context context, ComponentName receiver, IntentFilter filter) { 101 mReceivers.add(receiver); 102 context.registerReceiver(this, filter); 103 } 104 unregister(Context context)105 public void unregister(Context context) { 106 context.unregisterReceiver(this); 107 } 108 109 @Override onReceive(Context context, Intent intent)110 public void onReceive(Context context, Intent intent) { 111 intent.addFlags(Intent.FLAG_RECEIVER_FOREGROUND); 112 for (ComponentName receiver : mReceivers) { 113 intent.setComponent(receiver); 114 intent.putExtra(SliceBroadcastRelay.EXTRA_URI, mUri.toString()); 115 if (DEBUG) Log.d(TAG, "Forwarding " + receiver + " " + intent + " " + mUserId); 116 context.sendBroadcastAsUser(intent, mUserId); 117 } 118 } 119 } 120 } 121