Merge "Show correct last notification time for grouped notifications" into... (9cf6ecbf) · Commits · e / os / android_frameworks_base

packages/SystemUI/src/com/android/systemui/statusbar/notification/collection/coordinator/GroupWhenCoordinator.kt

0 → 100644

+108 −0

Original line number	Diff line number	Diff line
		/*
		* Copyright (C) 2023 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 android.util.ArrayMap
		import com.android.systemui.dagger.qualifiers.Main
		import com.android.systemui.statusbar.notification.collection.GroupEntry
		import com.android.systemui.statusbar.notification.collection.ListEntry
		import com.android.systemui.statusbar.notification.collection.NotifPipeline
		import com.android.systemui.statusbar.notification.collection.coordinator.dagger.CoordinatorScope
		import com.android.systemui.statusbar.notification.collection.listbuilder.pluggable.Invalidator
		import com.android.systemui.statusbar.notification.collection.render.NotifGroupController
		import com.android.systemui.util.concurrency.DelayableExecutor
		import com.android.systemui.util.time.SystemClock
		import javax.inject.Inject
		import kotlin.math.max
		import kotlin.math.min

		/** A small coordinator which finds, stores, and applies the closest notification time. */
		@CoordinatorScope
		class GroupWhenCoordinator
		@Inject
		constructor(
		@Main private val delayableExecutor: DelayableExecutor,
		private val systemClock: SystemClock
		) : Coordinator {

		private val invalidator = object : Invalidator("GroupWhenCoordinator") {}
		private val notificationGroupTimes = ArrayMap<GroupEntry, Long>()
		private var cancelInvalidateListRunnable: Runnable? = null

		private val invalidateListRunnable: Runnable = Runnable {
		invalidator.invalidateList("future notification invalidation")
		}

		override fun attach(pipeline: NotifPipeline) {
		pipeline.addOnBeforeFinalizeFilterListener(::onBeforeFinalizeFilterListener)
		pipeline.addOnAfterRenderGroupListener(::onAfterRenderGroupListener)
		pipeline.addPreRenderInvalidator(invalidator)
		}

		private fun onBeforeFinalizeFilterListener(entries: List<ListEntry>) {
		cancelListInvalidation()
		notificationGroupTimes.clear()

		val now = systemClock.currentTimeMillis()
		var closestFutureTime = Long.MAX_VALUE
		entries.asSequence().filterIsInstance<GroupEntry>().forEach { groupEntry ->
		val whenMillis = calculateGroupNotificationTime(groupEntry, now)
		notificationGroupTimes[groupEntry] = whenMillis
		if (whenMillis > now) {
		closestFutureTime = min(closestFutureTime, whenMillis)
		}
		}

		if (closestFutureTime != Long.MAX_VALUE) {
		cancelInvalidateListRunnable =
		delayableExecutor.executeDelayed(invalidateListRunnable, closestFutureTime - now)
		}
		}

		private fun cancelListInvalidation() {
		cancelInvalidateListRunnable?.run()
		cancelInvalidateListRunnable = null
		}

		private fun onAfterRenderGroupListener(group: GroupEntry, controller: NotifGroupController) {
		notificationGroupTimes[group]?.let(controller::setNotificationGroupWhen)
		}

		private fun calculateGroupNotificationTime(
		groupEntry: GroupEntry,
		currentTimeMillis: Long
		): Long {
		var pastTime = Long.MIN_VALUE
		var futureTime = Long.MAX_VALUE
		groupEntry.children
		.asSequence()
		.mapNotNull { child -> child.sbn.notification.`when`.takeIf { it > 0 } }
		.forEach { time ->
		val isInThePast = currentTimeMillis - time > 0
		if (isInThePast) {
		pastTime = max(pastTime, time)
		} else {
		futureTime = min(futureTime, time)
		}
		}

		if (pastTime == Long.MIN_VALUE && futureTime == Long.MAX_VALUE) {
		return checkNotNull(groupEntry.summary).creationTime
		}

		return if (futureTime != Long.MAX_VALUE) futureTime else pastTime
		}
		}

packages/SystemUI/src/com/android/systemui/statusbar/notification/collection/coordinator/NotifCoordinators.kt

+27 −25

Original line number	Diff line number	Diff line
		@@ -45,6 +45,7 @@ class NotifCoordinatorsImpl @Inject constructor(
		conversationCoordinator: ConversationCoordinator,
		debugModeCoordinator: DebugModeCoordinator,
		groupCountCoordinator: GroupCountCoordinator,
		groupWhenCoordinator: GroupWhenCoordinator,
		mediaCoordinator: MediaCoordinator,
		preparationCoordinator: PreparationCoordinator,
		remoteInputCoordinator: RemoteInputCoordinator,
		@@ -82,6 +83,7 @@ class NotifCoordinatorsImpl @Inject constructor(
		mCoordinators.add(debugModeCoordinator)
		mCoordinators.add(conversationCoordinator)
		mCoordinators.add(groupCountCoordinator)
		mCoordinators.add(groupWhenCoordinator)
		mCoordinators.add(mediaCoordinator)
		mCoordinators.add(rowAppearanceCoordinator)
		mCoordinators.add(stackCoordinator)

packages/SystemUI/src/com/android/systemui/statusbar/notification/collection/render/NotifGroupController.kt

+3 −0

Original line number	Diff line number	Diff line
		@@ -20,4 +20,7 @@ package com.android.systemui.statusbar.notification.collection.render
		interface NotifGroupController {
		/** Set the number of children that this group would have if not for the 8-child max */
		fun setUntruncatedChildCount(untruncatedChildCount: Int)

		/** Set the when value of notification group that reflects most important closest notification time */
		fun setNotificationGroupWhen(whenMillis: Long)
		}
		No newline at end of file

packages/SystemUI/src/com/android/systemui/statusbar/notification/row/ExpandableNotificationRow.java

+13 −0

Original line number	Diff line number	Diff line
		@@ -851,6 +851,19 @@ public class ExpandableNotificationRow extends ActivatableNotificationView
		mChildrenContainer.setUntruncatedChildCount(childCount);
		}

		/**
		* @see NotificationChildrenContainer#setNotificationGroupWhen(long)
		*/
		public void setNotificationGroupWhen(long whenMillis) {
		if (mIsSummaryWithChildren) {
		mChildrenContainer.setNotificationGroupWhen(whenMillis);
		} else {
		Log.w(TAG, "setNotificationGroupWhen( whenMillis: " + whenMillis + ")"
		+ " mIsSummaryWithChildren: false"
		+ " mChildrenContainer has not been inflated yet.");
		}
		}

		/**
		* Called after children have been attached to set the expansion states
		*/

packages/SystemUI/src/com/android/systemui/statusbar/notification/row/ExpandableNotificationRowController.java

+9 −0

Original line number	Diff line number	Diff line
		@@ -348,6 +348,15 @@ public class ExpandableNotificationRowController implements NotifViewController
		}
		}

		@Override
		public void setNotificationGroupWhen(long whenMillis) {
		if (mView.isSummaryWithChildren()) {
		mView.setNotificationGroupWhen(whenMillis);
		} else {
		Log.w(TAG, "Called setNotificationTime(" + whenMillis + ") on a leaf row");
		}
		}

		@Override
		public void setSystemExpanded(boolean systemExpanded) {
		mView.setSystemExpanded(systemExpanded);