| /* |
| * Copyright (C) 2020 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package com.android.systemui.statusbar.notification.collection.coordinator; |
| |
| import static com.android.systemui.statusbar.notification.stack.NotificationChildrenContainer.NUMBER_OF_CHILDREN_WHEN_CHILDREN_EXPANDED; |
| |
| import android.annotation.IntDef; |
| import android.os.RemoteException; |
| import android.service.notification.StatusBarNotification; |
| import android.util.ArrayMap; |
| import android.util.ArraySet; |
| |
| import com.android.internal.annotations.VisibleForTesting; |
| import com.android.internal.statusbar.IStatusBarService; |
| import com.android.systemui.statusbar.notification.collection.GroupEntry; |
| import com.android.systemui.statusbar.notification.collection.ListEntry; |
| import com.android.systemui.statusbar.notification.collection.NotifInflaterImpl; |
| import com.android.systemui.statusbar.notification.collection.NotifPipeline; |
| import com.android.systemui.statusbar.notification.collection.NotifViewBarn; |
| import com.android.systemui.statusbar.notification.collection.NotificationEntry; |
| import com.android.systemui.statusbar.notification.collection.ShadeListBuilder; |
| import com.android.systemui.statusbar.notification.collection.inflation.NotifInflater; |
| import com.android.systemui.statusbar.notification.collection.listbuilder.OnBeforeFinalizeFilterListener; |
| import com.android.systemui.statusbar.notification.collection.listbuilder.pluggable.NotifFilter; |
| import com.android.systemui.statusbar.notification.collection.notifcollection.NotifCollectionListener; |
| import com.android.systemui.statusbar.notification.row.NotifInflationErrorManager; |
| |
| import java.lang.annotation.Retention; |
| import java.lang.annotation.RetentionPolicy; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Objects; |
| import java.util.Set; |
| |
| import javax.inject.Inject; |
| import javax.inject.Singleton; |
| |
| /** |
| * Kicks off core notification inflation and view rebinding when a notification is added or updated. |
| * Aborts inflation when a notification is removed. |
| * |
| * If a notification was uninflated, this coordinator will filter the notification out from the |
| * {@link ShadeListBuilder} until it is inflated. |
| */ |
| @Singleton |
| public class PreparationCoordinator implements Coordinator { |
| private static final String TAG = "PreparationCoordinator"; |
| |
| private final PreparationCoordinatorLogger mLogger; |
| private final NotifInflater mNotifInflater; |
| private final NotifInflationErrorManager mNotifErrorManager; |
| private final NotifViewBarn mViewBarn; |
| private final Map<NotificationEntry, Integer> mInflationStates = new ArrayMap<>(); |
| |
| /** |
| * The set of notifications that are currently inflating something. Note that this is |
| * separate from inflation state as a view could either be uninflated or inflated and still be |
| * inflating something. |
| */ |
| private final Set<NotificationEntry> mInflatingNotifs = new ArraySet<>(); |
| |
| private final IStatusBarService mStatusBarService; |
| |
| /** |
| * The number of children in a group we actually keep inflated since we don't actually show |
| * all the children and don't need every child inflated at all times. |
| */ |
| private final int mChildBindCutoff; |
| |
| @Inject |
| public PreparationCoordinator( |
| PreparationCoordinatorLogger logger, |
| NotifInflaterImpl notifInflater, |
| NotifInflationErrorManager errorManager, |
| NotifViewBarn viewBarn, |
| IStatusBarService service) { |
| this(logger, notifInflater, errorManager, viewBarn, service, CHILD_BIND_CUTOFF); |
| } |
| |
| @VisibleForTesting |
| PreparationCoordinator( |
| PreparationCoordinatorLogger logger, |
| NotifInflaterImpl notifInflater, |
| NotifInflationErrorManager errorManager, |
| NotifViewBarn viewBarn, |
| IStatusBarService service, |
| int childBindCutoff) { |
| mLogger = logger; |
| mNotifInflater = notifInflater; |
| mNotifErrorManager = errorManager; |
| mNotifErrorManager.addInflationErrorListener(mInflationErrorListener); |
| mViewBarn = viewBarn; |
| mStatusBarService = service; |
| mChildBindCutoff = childBindCutoff; |
| } |
| |
| @Override |
| public void attach(NotifPipeline pipeline) { |
| pipeline.addCollectionListener(mNotifCollectionListener); |
| // Inflate after grouping/sorting since that affects what views to inflate. |
| pipeline.addOnBeforeFinalizeFilterListener(mOnBeforeFinalizeFilterListener); |
| pipeline.addFinalizeFilter(mNotifInflationErrorFilter); |
| pipeline.addFinalizeFilter(mNotifInflatingFilter); |
| } |
| |
| private final NotifCollectionListener mNotifCollectionListener = new NotifCollectionListener() { |
| |
| @Override |
| public void onEntryInit(NotificationEntry entry) { |
| mInflationStates.put(entry, STATE_UNINFLATED); |
| } |
| |
| @Override |
| public void onEntryUpdated(NotificationEntry entry) { |
| abortInflation(entry, "entryUpdated"); |
| mInflatingNotifs.remove(entry); |
| @InflationState int state = getInflationState(entry); |
| if (state == STATE_INFLATED) { |
| mInflationStates.put(entry, STATE_INFLATED_INVALID); |
| } else if (state == STATE_ERROR) { |
| // Updated so maybe it won't error out now. |
| mInflationStates.put(entry, STATE_UNINFLATED); |
| } |
| } |
| |
| @Override |
| public void onEntryRemoved(NotificationEntry entry, int reason) { |
| abortInflation(entry, "entryRemoved reason=" + reason); |
| } |
| |
| @Override |
| public void onEntryCleanUp(NotificationEntry entry) { |
| mInflationStates.remove(entry); |
| mInflatingNotifs.remove(entry); |
| mViewBarn.removeViewForEntry(entry); |
| } |
| }; |
| |
| private final OnBeforeFinalizeFilterListener mOnBeforeFinalizeFilterListener = |
| entries -> inflateAllRequiredViews(entries); |
| |
| private final NotifFilter mNotifInflationErrorFilter = new NotifFilter( |
| TAG + "InflationError") { |
| /** |
| * Filters out notifications that threw an error when attempting to inflate. |
| */ |
| @Override |
| public boolean shouldFilterOut(NotificationEntry entry, long now) { |
| return getInflationState(entry) == STATE_ERROR; |
| } |
| }; |
| |
| private final NotifFilter mNotifInflatingFilter = new NotifFilter(TAG + "Inflating") { |
| /** |
| * Filters out notifications that aren't inflated |
| */ |
| @Override |
| public boolean shouldFilterOut(NotificationEntry entry, long now) { |
| return !isInflated(entry); |
| } |
| }; |
| |
| private final NotifInflationErrorManager.NotifInflationErrorListener mInflationErrorListener = |
| new NotifInflationErrorManager.NotifInflationErrorListener() { |
| @Override |
| public void onNotifInflationError(NotificationEntry entry, Exception e) { |
| mViewBarn.removeViewForEntry(entry); |
| mInflationStates.put(entry, STATE_ERROR); |
| try { |
| final StatusBarNotification sbn = entry.getSbn(); |
| // report notification inflation errors back up |
| // to notification delegates |
| mStatusBarService.onNotificationError( |
| sbn.getPackageName(), |
| sbn.getTag(), |
| sbn.getId(), |
| sbn.getUid(), |
| sbn.getInitialPid(), |
| e.getMessage(), |
| sbn.getUserId()); |
| } catch (RemoteException ex) { |
| } |
| mNotifInflationErrorFilter.invalidateList(); |
| } |
| |
| @Override |
| public void onNotifInflationErrorCleared(NotificationEntry entry) { |
| mNotifInflationErrorFilter.invalidateList(); |
| } |
| }; |
| |
| private void inflateAllRequiredViews(List<ListEntry> entries) { |
| for (int i = 0, size = entries.size(); i < size; i++) { |
| ListEntry entry = entries.get(i); |
| if (entry instanceof GroupEntry) { |
| GroupEntry groupEntry = (GroupEntry) entry; |
| groupEntry.setUntruncatedChildCount(groupEntry.getChildren().size()); |
| inflateRequiredGroupViews(groupEntry); |
| } else { |
| NotificationEntry notifEntry = (NotificationEntry) entry; |
| inflateRequiredNotifViews(notifEntry); |
| } |
| } |
| } |
| |
| private void inflateRequiredGroupViews(GroupEntry groupEntry) { |
| NotificationEntry summary = groupEntry.getSummary(); |
| List<NotificationEntry> children = groupEntry.getChildren(); |
| inflateRequiredNotifViews(summary); |
| for (int j = 0; j < children.size(); j++) { |
| NotificationEntry child = children.get(j); |
| boolean childShouldBeBound = j < mChildBindCutoff; |
| if (childShouldBeBound) { |
| inflateRequiredNotifViews(child); |
| } else { |
| if (mInflatingNotifs.contains(child)) { |
| abortInflation(child, "Past last visible group child"); |
| } |
| if (isInflated(child)) { |
| // TODO: May want to put an animation hint here so view manager knows to treat |
| // this differently from a regular removal animation |
| freeNotifViews(child); |
| } |
| } |
| } |
| } |
| |
| private void inflateRequiredNotifViews(NotificationEntry entry) { |
| if (mInflatingNotifs.contains(entry)) { |
| // Already inflating this entry |
| return; |
| } |
| @InflationState int state = mInflationStates.get(entry); |
| switch (state) { |
| case STATE_UNINFLATED: |
| inflateEntry(entry, "entryAdded"); |
| break; |
| case STATE_INFLATED_INVALID: |
| rebind(entry, "entryUpdated"); |
| break; |
| case STATE_INFLATED: |
| case STATE_ERROR: |
| default: |
| // Nothing to do. |
| } |
| } |
| |
| private void inflateEntry(NotificationEntry entry, String reason) { |
| abortInflation(entry, reason); |
| mInflatingNotifs.add(entry); |
| mNotifInflater.inflateViews(entry, this::onInflationFinished); |
| } |
| |
| private void rebind(NotificationEntry entry, String reason) { |
| mInflatingNotifs.add(entry); |
| mNotifInflater.rebindViews(entry, this::onInflationFinished); |
| } |
| |
| private void abortInflation(NotificationEntry entry, String reason) { |
| mLogger.logInflationAborted(entry.getKey(), reason); |
| entry.abortTask(); |
| mInflatingNotifs.remove(entry); |
| } |
| |
| private void onInflationFinished(NotificationEntry entry) { |
| mLogger.logNotifInflated(entry.getKey()); |
| mInflatingNotifs.remove(entry); |
| mViewBarn.registerViewForEntry(entry, entry.getRow()); |
| mInflationStates.put(entry, STATE_INFLATED); |
| mNotifInflatingFilter.invalidateList(); |
| } |
| |
| private void freeNotifViews(NotificationEntry entry) { |
| mViewBarn.removeViewForEntry(entry); |
| entry.setRow(null); |
| mInflationStates.put(entry, STATE_UNINFLATED); |
| } |
| |
| private boolean isInflated(NotificationEntry entry) { |
| @InflationState int state = getInflationState(entry); |
| return (state == STATE_INFLATED) || (state == STATE_INFLATED_INVALID); |
| } |
| |
| private @InflationState int getInflationState(NotificationEntry entry) { |
| Integer stateObj = mInflationStates.get(entry); |
| Objects.requireNonNull(stateObj, |
| "Asking state of a notification preparation coordinator doesn't know about"); |
| return stateObj; |
| } |
| |
| @Retention(RetentionPolicy.SOURCE) |
| @IntDef(prefix = {"STATE_"}, |
| value = {STATE_UNINFLATED, STATE_INFLATED_INVALID, STATE_INFLATED, STATE_ERROR}) |
| @interface InflationState {} |
| |
| /** The notification has no views attached. */ |
| private static final int STATE_UNINFLATED = 0; |
| |
| /** The notification is inflated. */ |
| private static final int STATE_INFLATED = 1; |
| |
| /** |
| * The notification is inflated, but its content may be out-of-date since the notification has |
| * been updated. |
| */ |
| private static final int STATE_INFLATED_INVALID = 2; |
| |
| /** The notification errored out while inflating */ |
| private static final int STATE_ERROR = -1; |
| |
| /** |
| * How big the buffer of extra views we keep around to be ready to show when we do need to |
| * dynamically inflate a row. |
| */ |
| private static final int EXTRA_VIEW_BUFFER_COUNT = 1; |
| |
| private static final int CHILD_BIND_CUTOFF = |
| NUMBER_OF_CHILDREN_WHEN_CHILDREN_EXPANDED + EXTRA_VIEW_BUFFER_COUNT; |
| } |