Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 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; |
Fabian Kozynski | 5ca7a51 | 2019-10-16 19:56:11 +0000 | [diff] [blame] | 31 | import com.android.systemui.broadcast.BroadcastDispatcher; |
| 32 | |
| 33 | import javax.inject.Inject; |
| 34 | import javax.inject.Singleton; |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 35 | |
| 36 | /** |
| 37 | * Allows settings to register certain broadcasts to launch the settings app for pinned slices. |
| 38 | * @see SliceBroadcastRelay |
| 39 | */ |
Fabian Kozynski | 5ca7a51 | 2019-10-16 19:56:11 +0000 | [diff] [blame] | 40 | @Singleton |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 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<>(); |
Fabian Kozynski | 5ca7a51 | 2019-10-16 19:56:11 +0000 | [diff] [blame] | 46 | private final BroadcastDispatcher mBroadcastDispatcher; |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 47 | |
Fabian Kozynski | 5ca7a51 | 2019-10-16 19:56:11 +0000 | [diff] [blame] | 48 | @Inject |
| 49 | public SliceBroadcastRelayHandler(Context context, BroadcastDispatcher broadcastDispatcher) { |
Dave Mankoff | a5d8a39 | 2019-10-10 12:21:09 -0400 | [diff] [blame] | 50 | super(context); |
Fabian Kozynski | 5ca7a51 | 2019-10-16 19:56:11 +0000 | [diff] [blame] | 51 | mBroadcastDispatcher = broadcastDispatcher; |
Dave Mankoff | a5d8a39 | 2019-10-10 12:21:09 -0400 | [diff] [blame] | 52 | } |
| 53 | |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 54 | @Override |
| 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); |
Fabian Kozynski | 5ca7a51 | 2019-10-16 19:56:11 +0000 | [diff] [blame] | 59 | mBroadcastDispatcher.registerReceiver(mReceiver, filter); |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 60 | } |
| 61 | |
Fabian Kozynski | 5ca7a51 | 2019-10-16 19:56:11 +0000 | [diff] [blame] | 62 | // This does not use BroadcastDispatcher as the filter may have schemas or mime types. |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 63 | @VisibleForTesting |
| 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); |
Sunny Goyal | a21570c7 | 2019-04-19 15:05:41 -0700 | [diff] [blame] | 75 | BroadcastRelay relay = getAndRemoveRelay(uri); |
| 76 | if (relay != null) { |
| 77 | relay.unregister(mContext); |
| 78 | } |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 79 | } |
| 80 | } |
| 81 | |
| 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 | |
| 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; |
Matthew Fritze | 429d27d | 2018-05-02 19:43:39 -0700 | [diff] [blame] | 106 | private final Uri mUri; |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 107 | |
| 108 | public BroadcastRelay(Uri uri) { |
| 109 | mUserId = new UserHandle(ContentProvider.getUserIdFromUri(uri)); |
Matthew Fritze | 429d27d | 2018-05-02 19:43:39 -0700 | [diff] [blame] | 110 | mUri = uri; |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 111 | } |
| 112 | |
| 113 | public void register(Context context, ComponentName receiver, IntentFilter filter) { |
| 114 | mReceivers.add(receiver); |
| 115 | context.registerReceiver(this, filter); |
| 116 | } |
| 117 | |
| 118 | public void unregister(Context context) { |
| 119 | context.unregisterReceiver(this); |
| 120 | } |
| 121 | |
| 122 | @Override |
| 123 | public void onReceive(Context context, Intent intent) { |
| 124 | intent.addFlags(Intent.FLAG_RECEIVER_FOREGROUND); |
| 125 | for (ComponentName receiver : mReceivers) { |
| 126 | intent.setComponent(receiver); |
Matthew Fritze | 429d27d | 2018-05-02 19:43:39 -0700 | [diff] [blame] | 127 | intent.putExtra(SliceBroadcastRelay.EXTRA_URI, mUri.toString()); |
Jason Monk | 0b5e52c | 2018-04-19 22:14:46 -0400 | [diff] [blame] | 128 | if (DEBUG) Log.d(TAG, "Forwarding " + receiver + " " + intent + " " + mUserId); |
| 129 | context.sendBroadcastAsUser(intent, mUserId); |
| 130 | } |
| 131 | } |
| 132 | } |
| 133 | } |