fankes/MIUINativeNotifyIcon
2
0
Fork 0
mirror of https://github.com/fankes/MIUINativeNotifyIcon.git synced 2026-03-26 21:34:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add AOD hook (#253)

Browse Source
This commit is contained in:
Midori Kochiya
2026-03-13 17:23:13 +08:00
committed by GitHub
parent d40dfc128e
commit ef795edf8b
1 changed files with 57 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
app/src/main/java/com/fankes/miui/notify/hook/entity/SystemUIHooker.kt
View File

@@ -128,6 +128,8 @@ object SystemUIHooker : YukiBaseHooker() {
/** MIUI 新版本存在的类 */
private val NotificationStatClass by lazyClassOrNull("${PackageName.SYSTEMUI}.statusbar.notification.analytics.NotificationStat")
private val MiuiDozeServiceClass by lazyClassOrNull("com.android.keyguard.doze.MiuiDozeService")
/** 原生存在的类 */
private val NotificationChildrenContainerClass by lazyClass("${PackageName.SYSTEMUI}.statusbar.notification.stack.NotificationChildrenContainer")
@@ -475,7 +477,9 @@ object SystemUIHooker : YukiBaseHooker() {
* @param iconDrawable 小图标 [Drawable]
* @return [Pair] 回调小图标 - ([Drawable] 小图标,[Boolean] 是否替换)
*/
private fun compatStatusIcon(context: Context, nf: StatusBarNotification?, iconDrawable: Drawable?) = nf?.let { notifyInstance ->
private fun compatStatusIcon(
context: Context, nf: StatusBarNotification?, iconDrawable: Drawable?, tag: String = "Status Bar Icon"
) = nf?.let { notifyInstance ->
if (iconDrawable == null) return@let Pair(null, false)
/** 判断是否不是灰度图标 */
val isGrayscaleIcon = notifyInstance.isXmsf.not() && isGrayscaleIcon(context, iconDrawable)
@@ -490,7 +494,7 @@ object SystemUIHooker : YukiBaseHooker() {
/** 是否为通知优化生效图标 */
val isCustom = customTriple.first != null && customTriple.third.not()
/** 打印日志 */
loggerDebug(tag = "Status Bar Icon", context, notifyInstance, isCustom = isCustom, isGrayscaleIcon)
loggerDebug(tag = tag, context, notifyInstance, isCustom = isCustom, isGrayscaleIcon)
when {
/** 处理自定义通知图标优化 */
customTriple.first != null -> Pair(customTriple.first, true)
@@ -1289,5 +1293,56 @@ object SystemUIHooker : YukiBaseHooker() {
sbn.packageName == PackageName.SYSTEMUI && sbn.tag == "UNIMPORTANT"
} ?: false
}
@Suppress("LocalVariableName")
MiuiDozeServiceClass?.resolve()?.optional()?.apply {
firstMethodOrNull { name = "onPluginConnected" }?.hook()?.before {
val plugin = args(0).any()
?: return@before YLog.warn("AOD got null plugin")
val pluginClassLoader = plugin::class.java.classLoader
/**
* AOD 是插件，用传入的 ClassLoader 去解析
*/
val NotificationControllerClass = "com.miui.aod.notification.NotificationController".toClassOrNull(pluginClassLoader)
val NotificationDataClass = $$"com.miui.aod.notification.NotificationController$NotificationData".toClassOrNull(pluginClassLoader)
val NotificationDataIconDrawableField = NotificationDataClass?.resolve()?.firstFieldOrNull { name = "mIconDrawable" }
if (NotificationControllerClass == null || NotificationDataClass == null || NotificationDataIconDrawableField == null) {
return@before YLog.warn("AOD got null NotificationControllerClass or NotificationDataClass")
}
NotificationControllerClass.resolve().firstMethodOrNull {
name = "fillNotificationData"
parameters(
NotificationDataClass,
StatusBarNotification::class,
Boolean::class,
Boolean::class,
Boolean::class,
)
}?.hook()?.after {
val context = instance.asResolver().firstFieldOrNull { name = "mContext" }?.get<Context>()
?: return@after YLog.warn("fillNotificationData got null context")
val notificationData = args(0).any()
?: return@after YLog.warn("fillNotificationData got null notificationData")
val notifyInstance = args(1).cast<StatusBarNotification>()
?: return@after YLog.warn("fillNotificationData got null notifyInstance")
val iconDrawableField = notificationData.asResolver().firstFieldOrNull { name = "mIconDrawable" }
?: return@after YLog.warn("fillNotificationData got null iconDrawableField")
val smallIconDrawable = notifyInstance.notification?.smallIcon?.loadDrawable(context)
compatStatusIcon(context, notifyInstance, smallIconDrawable, "AOD Icon").also { pair ->
if (pair.second) {
iconDrawableField.set(pair.first)
} else {
iconDrawableField.set(smallIconDrawable)
}
}
}
}
}
}
}