Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit ceaf91a4 authored by Jeff DeCew's avatar Jeff DeCew Committed by Automerger Merge Worker
Browse files

Convert notification debug filter to CommandRegistry am: 1149017c 

Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/18800598



Change-Id: I2f96caa4e8400ee7f01459b90a1091c9255d8548
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents cf3641b0 1149017c

packages/SystemUI/src/com/android/systemui/statusbar/notification/collection/provider/DebugModeFilterProvider.kt

+50 −0
Original line number Diff line number Diff line
@@ -25,6 +25,8 @@ import android.util.Log 
import com.android.systemui.Dumpable

import com.android.systemui.dagger.SysUISingleton

import com.android.systemui.dump.DumpManager

import com.android.systemui.statusbar.commandline.Command

import com.android.systemui.statusbar.commandline.CommandRegistry

import com.android.systemui.statusbar.notification.collection.NotificationEntry

import com.android.systemui.util.Assert

import com.android.systemui.util.ListenerSet
@@ -43,15 +45,20 @@ import javax.inject.Inject 
 *

 * `$ adb shell am broadcast -a com.android.systemui.action.SET_NOTIF_DEBUG_MODE

 *          --esal allowed_packages <comma-separated-packages>`

 * or

 * `$ adb shell cmd statusbar notif-filter allowed-pkgs <package> ...`

 *

 * To disable filtering, send the action without a list:

 *

 * `$ adb shell am broadcast -a com.android.systemui.action.SET_NOTIF_DEBUG_MODE`

 * or

 * `$ adb shell cmd statusbar notif-filter reset`

 *

 * NOTE: this feature only works on debug builds, and when the broadcaster is root.

 */

@SysUISingleton

class DebugModeFilterProvider @Inject constructor(

    private val commandRegistry: CommandRegistry,

    private val context: Context,

    dumpManager: DumpManager

) : Dumpable {
@@ -74,6 +81,7 @@ class DebugModeFilterProvider @Inject constructor( 
        val needsInitialization = listeners.isEmpty()

        listeners.addIfAbsent(listener)

        if (needsInitialization) {

            commandRegistry.registerCommand("notif-filter") { NotifFilterCommand() }

            val filter = IntentFilter().apply { addAction(ACTION_SET_NOTIF_DEBUG_MODE) }

            val permission = NOTIF_DEBUG_MODE_PERMISSION

            context.registerReceiver(mReceiver, filter, permission, null, Context.RECEIVER_EXPORTED)
@@ -126,4 +134,46 @@ class DebugModeFilterProvider @Inject constructor( 
            "com.android.systemui.permission.NOTIF_DEBUG_MODE"

        private const val EXTRA_ALLOWED_PACKAGES = "allowed_packages"

    }



    inner class NotifFilterCommand : Command {

        override fun execute(pw: PrintWriter, args: List<String>) {

            when (args.firstOrNull()) {

                "reset" -> {

                    if (args.size > 1) {

                        return invalidCommand(pw, "Unexpected arguments for 'reset' command")

                    }

                    allowedPackages = emptyList()

                }

                "allowed-pkgs" -> {

                    allowedPackages = args.drop(1)

                }

                null -> return invalidCommand(pw, "Missing command")

                else -> return invalidCommand(pw, "Unknown command: ${args.firstOrNull()}")

            }

            Log.d(TAG, "Updated allowedPackages: $allowedPackages")

            if (allowedPackages.isEmpty()) {

                pw.print("Resetting allowedPackages ... ")

            } else {

                pw.print("Updating allowedPackages: $allowedPackages ... ")

            }

            listeners.forEach(Runnable::run)

            pw.println("DONE")

        }



        private fun invalidCommand(pw: PrintWriter, reason: String) {

            pw.println("Error: $reason")

            pw.println()

            help(pw)

        }



        override fun help(pw: PrintWriter) {

            pw.println("Usage: adb shell cmd statusbar notif-filter <command>")

            pw.println("Available commands:")

            pw.println("  reset")

            pw.println("     Restore the default system behavior.")

            pw.println("  allowed-pkgs <package> ...")

            pw.println("     Hide all notification except from packages listed here.")

            pw.println("     Providing no packages is treated as a reset.")

        }

    }

}