diff --git a/app/src/main/java/com/fankes/coloros/notify/hook/HookEntry.kt b/app/src/main/java/com/fankes/coloros/notify/hook/HookEntry.kt index fee23f8..b15f4ec 100644 --- a/app/src/main/java/com/fankes/coloros/notify/hook/HookEntry.kt +++ b/app/src/main/java/com/fankes/coloros/notify/hook/HookEntry.kt @@ -22,433 +22,19 @@ */ package com.fankes.coloros.notify.hook -import android.app.WallpaperManager -import android.content.BroadcastReceiver -import android.content.Context -import android.content.Intent -import android.content.IntentFilter -import android.graphics.Bitmap -import android.graphics.Color -import android.graphics.Outline -import android.graphics.drawable.BitmapDrawable -import android.graphics.drawable.Drawable -import android.graphics.drawable.Icon -import android.graphics.drawable.VectorDrawable -import android.service.notification.StatusBarNotification -import android.util.ArraySet -import android.view.View -import android.view.ViewOutlineProvider -import android.widget.ImageView -import androidx.core.graphics.drawable.toBitmap -import com.fankes.coloros.notify.bean.IconDataBean -import com.fankes.coloros.notify.const.Const -import com.fankes.coloros.notify.hook.HookConst.ENABLE_ANDROID12_STYLE import com.fankes.coloros.notify.hook.HookConst.ENABLE_MODULE -import com.fankes.coloros.notify.hook.HookConst.ENABLE_MODULE_LOG -import com.fankes.coloros.notify.hook.HookConst.ENABLE_NOTIFY_ICON_FIX -import com.fankes.coloros.notify.hook.HookConst.ENABLE_NOTIFY_ICON_FIX_NOTIFY -import com.fankes.coloros.notify.hook.HookConst.REMOVE_CHANGECP_NOTIFY -import com.fankes.coloros.notify.hook.HookConst.REMOVE_DEV_NOTIFY -import com.fankes.coloros.notify.hook.HookConst.REMOVE_DNDALERT_NOTIFY import com.fankes.coloros.notify.hook.HookConst.SYSTEMUI_PACKAGE_NAME -import com.fankes.coloros.notify.hook.factory.isAppNotifyHookAllOf -import com.fankes.coloros.notify.hook.factory.isAppNotifyHookOf -import com.fankes.coloros.notify.param.IconPackParams -import com.fankes.coloros.notify.utils.drawable.drawabletoolbox.DrawableBuilder -import com.fankes.coloros.notify.utils.factory.* -import com.fankes.coloros.notify.utils.tool.IconAdaptationTool -import com.fankes.coloros.notify.utils.tool.IconRuleManagerTool +import com.fankes.coloros.notify.hook.entity.SystemUIHooker +import com.fankes.coloros.notify.utils.factory.isNotColorOS import com.highcapable.yukihookapi.annotation.xposed.InjectYukiHookWithXposed -import com.highcapable.yukihookapi.hook.bean.VariousClass import com.highcapable.yukihookapi.hook.factory.configs import com.highcapable.yukihookapi.hook.factory.encase -import com.highcapable.yukihookapi.hook.factory.field -import com.highcapable.yukihookapi.hook.factory.method -import com.highcapable.yukihookapi.hook.log.loggerD -import com.highcapable.yukihookapi.hook.log.loggerE import com.highcapable.yukihookapi.hook.log.loggerW -import com.highcapable.yukihookapi.hook.param.PackageParam -import com.highcapable.yukihookapi.hook.type.android.* -import com.highcapable.yukihookapi.hook.type.java.BooleanType -import com.highcapable.yukihookapi.hook.type.java.FloatType -import com.highcapable.yukihookapi.hook.type.java.IntType -import com.highcapable.yukihookapi.hook.type.java.LongType import com.highcapable.yukihookapi.hook.xposed.proxy.YukiHookXposedInitProxy @InjectYukiHookWithXposed class HookEntry : YukiHookXposedInitProxy { - companion object { - - /** 原生存在的类 */ - private const val ContrastColorUtilClass = "com.android.internal.util.ContrastColorUtil" - - /** 原生存在的类 */ - private const val NotificationUtilsClass = "$SYSTEMUI_PACKAGE_NAME.statusbar.notification.NotificationUtils" - - /** 原生存在的类 */ - private const val NotificationEntryClass = "$SYSTEMUI_PACKAGE_NAME.statusbar.notification.collection.NotificationEntry" - - /** 原生存在的类 */ - private const val StatusBarIconClass = "com.android.internal.statusbar.StatusBarIcon" - - /** 原生存在的类 */ - private const val StatusBarIconViewClass = "$SYSTEMUI_PACKAGE_NAME.statusbar.StatusBarIconView" - - /** 原生存在的类 */ - private const val IconBuilderClass = "$SYSTEMUI_PACKAGE_NAME.statusbar.notification.icon.IconBuilder" - - /** 原生存在的类 */ - private const val IconManagerClass = "$SYSTEMUI_PACKAGE_NAME.statusbar.notification.icon.IconManager" - - /** ColorOS 存在的类 - 旧版本不存在 */ - private const val OplusContrastColorUtilClass = "com.oplusos.util.OplusContrastColorUtil" - - /** 原生存在的类 */ - private const val PluginManagerImplClass = "$SYSTEMUI_PACKAGE_NAME.shared.plugins.PluginManagerImpl" - - /** 根据多个版本存在不同的包名相同的类 */ - private val SystemPromptControllerClass = VariousClass( - "com.oplusos.systemui.statusbar.policy.SystemPromptController", - "com.coloros.systemui.statusbar.policy.SystemPromptController" - ) - - /** 根据多个版本存在不同的包名相同的类 */ - private val RoundRectDrawableUtilClass = VariousClass( - "com.oplusos.systemui.notification.util.RoundRectDrawableUtil", - "com.coloros.systemui.notification.util.RoundRectDrawableUtil" - ) - - /** 根据多个版本存在不同的包名相同的类 */ - private val RoundRectDrawableUtil_CompanionClass = VariousClass( - "com.oplusos.systemui.notification.util.RoundRectDrawableUtil\$Companion", - "com.oplusos.systemui.notification.util.RoundRectDrawableUtil\$Companion" - ) - - /** 根据多个版本存在不同的包名相同的类 */ - private val DndAlertHelperClass = VariousClass( - "com.oplusos.systemui.notification.helper.DndAlertHelper", - "com.coloros.systemui.notification.helper.DndAlertHelper" - ) - - /** 根据多个版本存在不同的包名相同的类 */ - private val OplusPowerNotificationWarningsClass = VariousClass( - "com.oplusos.systemui.notification.power.OplusPowerNotificationWarnings", - "com.coloros.systemui.notification.power.ColorosPowerNotificationWarnings" - ) - - /** 根据多个版本存在不同的包名相同的类 */ - private val AbstractReceiverClass = VariousClass( - "com.oplusos.systemui.common.receiver.AbstractReceiver", - "com.coloros.systemui.common.receiver.AbstractReceiver" - ) - - /** 根据多个版本存在不同的包名相同的类 */ - private val ExpandableNotificationRowClass = VariousClass( - "$SYSTEMUI_PACKAGE_NAME.statusbar.notification.row.ExpandableNotificationRow", - "$SYSTEMUI_PACKAGE_NAME.statusbar.ExpandableNotificationRow" - ) - - /** 根据多个版本存在不同的包名相同的类 */ - private val NotificationViewWrapperClass = VariousClass( - "$SYSTEMUI_PACKAGE_NAME.statusbar.notification.row.wrapper.NotificationViewWrapper", - "$SYSTEMUI_PACKAGE_NAME.statusbar.notification.NotificationViewWrapper" - ) - - /** 根据多个版本存在不同的包名相同的类 */ - private val NotificationHeaderViewWrapperClass = VariousClass( - "$SYSTEMUI_PACKAGE_NAME.statusbar.notification.row.wrapper.NotificationHeaderViewWrapper", - "$SYSTEMUI_PACKAGE_NAME.statusbar.notification.NotificationHeaderViewWrapper" - ) - } - - /** 缓存的通知优化图标数组 */ - private var iconDatas = ArrayList() - - /** 缓存的状态栏小图标实例 */ - private var statusBarIconViews = ArraySet() - - /** 缓存的通知小图标包装纸实例 */ - private var notificationViewWrappers = HashSet() - - /** 仅监听一次主题壁纸颜色变化 */ - private var isWallpaperColorListenerSetUp = false - - /** 是否已经注册广播 */ - private var isRegisterModuleReceiver = false - - /** 模块广播接收器 */ - private val moduleReceiver by lazy { - object : BroadcastReceiver() { - override fun onReceive(context: Context?, intent: Intent?) { - context?.sendBroadcast(Intent().apply { - action = Const.MODULE_HANDLER_RECEIVER_TAG - putExtra("isAction", true) - putExtra("isValied", intent?.getStringExtra(Const.MODULE_VERSION_VERIFY_TAG) == Const.MODULE_VERSION_VERIFY) - }) - } - } - } - - /** - * 注册模块广播接收器 - * @param context 实例 - */ - private fun registerModuleReceiver(context: Context) { - if (isRegisterModuleReceiver) return - context.registerReceiver(moduleReceiver, IntentFilter().apply { addAction(Const.MODULE_CHECKING_RECEIVER_TAG) }) - isRegisterModuleReceiver = true - } - - /** - * 是否启用通知图标优化功能 - * @param isHooking 是否判断启用通知功能 - 默认：是 - * @return [Boolean] - */ - private fun PackageParam.isEnableHookColorNotifyIcon(isHooking: Boolean = true) = - prefs.getBoolean(ENABLE_NOTIFY_ICON_FIX, default = true) && - (if (isHooking) prefs.getBoolean(ENABLE_NOTIFY_ICON_FIX_NOTIFY, default = true) else true) - - /** - * 打印日志 - * @param tag 标识 - * @param context 实例 - * @param packageName APP 包名 - * @param isCustom 是否为通知优化生效图标 - * @param isGrayscale 是否为灰度图标 - */ - private fun PackageParam.printLogcat( - tag: String, - context: Context, - packageName: String, - isCustom: Boolean, - isGrayscale: Boolean - ) { - if (prefs.getBoolean(ENABLE_MODULE_LOG)) loggerD( - msg = "$tag --> [${context.findAppName(packageName)}][$packageName] " + - "custom [$isCustom] " + - "grayscale [$isGrayscale]" - ) - } - - /** - * 注册主题壁纸改变颜色监听 - * - * - 仅限在 Android 12 上注册 - * @param view 实例 - */ - private fun PackageParam.registerWallpaperColorChanged(view: View) = runInSafe { - if (!isWallpaperColorListenerSetUp && isUpperOfAndroidS) view.apply { - WallpaperManager.getInstance(context).addOnColorsChangedListener({ _, _ -> refreshNotificationIcons() }, handler) - } - isWallpaperColorListenerSetUp = true - } - - /** 刷新状态栏小图标 */ - private fun PackageParam.refreshStatusBarIcons() = runInSafe { - val nfField = StatusBarIconViewClass.clazz.field { name = "mNotification" } - val sRadiusField = StatusBarIconViewClass.clazz.field { name = "sIconRadiusFraction" } - val sNfSizeField = StatusBarIconViewClass.clazz.field { name = "sNotificationRoundIconSize" } - val roundUtil = RoundRectDrawableUtil_CompanionClass.clazz.method { - name = "getRoundRectDrawable" - param(DrawableClass, FloatType, IntType, IntType, ContextClass) - }.onNoSuchMethod { loggerE(msg = "Your system not support \"getRoundRectDrawable\"!", e = it) } - .get(RoundRectDrawableUtilClass.clazz.field { name = "Companion" }.get().self) - /** 启动一个线程防止卡顿 */ - Thread { - statusBarIconViews.takeIf { it.isNotEmpty() }?.forEach { - runInSafe { - /** 得到通知实例 */ - val nf = nfField.of(it) ?: return@Thread - - /** 得到原始通知图标 */ - val iconDrawable = nf.notification.smallIcon.loadDrawable(it.context) - /** 获取优化后的状态栏通知图标 */ - compatStatusIcon( - context = it.context, - isGrayscaleIcon = isGrayscaleIcon(it.context, iconDrawable), - packageName = nf.packageName, - drawable = iconDrawable - ).also { pair -> - /** 得到图标圆角 */ - val sRadius = sRadiusField.ofFloat(it) - - /** 得到缩放大小 */ - val sNfSize = sNfSizeField.ofInt(it) - /** 在主线程设置图标 */ - it.post { it.setImageDrawable(roundUtil.invoke(pair.first, sRadius, sNfSize, sNfSize, it.context)) } - } - } - } - }.start() - } - - /** 刷新通知小图标 */ - private fun PackageParam.refreshNotificationIcons() = runInSafe { - NotificationHeaderViewWrapperClass.clazz.method { name = "resolveHeaderViews" }.also { result -> - notificationViewWrappers.takeIf { it.isNotEmpty() }?.forEach { result.get(it).call() } - } - } - - /** - * - 这个是修复彩色图标的关键核心代码判断 - * - * 判断是否为灰度图标 - 反射执行系统方法 - * @param context 实例 - * @param drawable 要判断的图标 - * @return [Boolean] - */ - private fun PackageParam.isGrayscaleIcon(context: Context?, drawable: Drawable?) = - ContrastColorUtilClass.clazz.let { - drawable is VectorDrawable || it.method { - name = "isGrayscaleIcon" - param(DrawableClass) - }.get(it.method { - name = "getInstance" - param(ContextClass) - }.get().invoke(context)).callBoolean(drawable) - } - - /** - * 自动适配状态栏、通知栏自定义小图标 - * @param isGrayscaleIcon 是否为灰度图标 - * @param packageName APP 包名 - * @return [Pair] - ([Bitmap] 位图,[Int] 颜色) - */ - private fun PackageParam.compatCustomIcon(isGrayscaleIcon: Boolean, packageName: String): Pair { - var customPair: Pair? = null - when { - /** 替换系统图标为 Android 默认 */ - (packageName == "android" || packageName == "com.android.systemui") && !isGrayscaleIcon -> customPair = - Pair(if (isUpperOfAndroidS) IconPackParams.android12IconBitmap else IconPackParams.android11IconBitmap, 0) - /** 替换自定义通知图标 */ - prefs.getBoolean(ENABLE_NOTIFY_ICON_FIX, default = true) -> run { - if (iconDatas.isNotEmpty()) - iconDatas.forEach { - if (packageName == it.packageName && isAppNotifyHookOf(it)) { - if (!isGrayscaleIcon || isAppNotifyHookAllOf(it)) - customPair = Pair(it.iconBitmap, it.iconColor) - return@run - } - } - } - } - return customPair ?: Pair(null, 0) - } - - /** - * 自动适配状态栏小图标 - * @param context 实例 - * @param isGrayscaleIcon 是否为灰度图标 - * @param packageName APP 包名 - * @param drawable 原始图标 - * @return [Pair] - ([Drawable] 图标,[Boolean] 是否替换) - */ - private fun PackageParam.compatStatusIcon(context: Context, isGrayscaleIcon: Boolean, packageName: String, drawable: Drawable) = - compatCustomIcon(isGrayscaleIcon, packageName).first.also { - /** 打印日志 */ - printLogcat(tag = "StatusIcon", context, packageName, isCustom = it != null, isGrayscaleIcon) - }?.let { Pair(BitmapDrawable(context.resources, it), true) } ?: Pair(drawable, false) - - /** - * 自动适配通知栏小图标 - * @param isGrayscaleIcon 是否为灰度图标 - * @param packageName APP 包名 - * @param drawable 原始图标 - * @param iconColor 原生图标颜色 - * @param iconView 图标 [ImageView] - */ - private fun PackageParam.compatNotifyIcon( - isGrayscaleIcon: Boolean, - packageName: String, - drawable: Drawable, - iconColor: Int, - iconView: ImageView - ) = runInSafe { - compatCustomIcon(isGrayscaleIcon, packageName).also { customPair -> - when { - customPair.first != null || isGrayscaleIcon -> iconView.apply { - /** 设置不要裁切到边界 */ - clipToOutline = false - /** 重新设置图标 */ - setImageBitmap(customPair.first ?: drawable.toBitmap()) - - /** 是否开启 Android 12 风格 */ - val isA12Style = prefs.getBoolean(ENABLE_ANDROID12_STYLE, isUpperOfAndroidS) - - /** 旧版风格 */ - val oldStyle = (if (context.isSystemInDarkMode) 0xffdcdcdc else 0xff707173).toInt() - - /** 新版风格 */ - val newStyle = (if (context.isSystemInDarkMode) 0xffdcdcdc else Color.WHITE).toInt() - - /** 原生着色 */ - val a12Style = if (isUpperOfAndroidS) context.wallpaperColor else - (if (context.isSystemInDarkMode) 0xff707173 else oldStyle).toInt() - - /** 旧版图标着色 */ - val oldApplyColor = customPair.second.takeIf { it != 0 } ?: iconColor.takeIf { it != 0 } ?: oldStyle - - /** 新版图标着色 */ - val newApplyColor = customPair.second.takeIf { it != 0 } ?: iconColor.takeIf { it != 0 } ?: a12Style - - /** 判断风格并开始 Hook */ - if (isA12Style) { - background = DrawableBuilder().rounded().solidColor(newApplyColor).build() - setColorFilter(newStyle) - setPadding(2.dp(context), 2.dp(context), 2.dp(context), 2.dp(context)) - } else { - background = null - setColorFilter(oldApplyColor) - setPadding(0, 0, 0, 0) - } - } - else -> iconView.apply { - /** 重新设置图标 */ - setImageDrawable(drawable) - /** 设置裁切到边界 */ - clipToOutline = true - /** 设置一个圆角轮廓裁切 */ - outlineProvider = object : ViewOutlineProvider() { - override fun getOutline(view: View, out: Outline) { - out.setRoundRect( - 0, 0, - view.width, view.height, 3.dpFloat(context) - ) - } - } - /** 清除图标间距 */ - setPadding(0, 0, 0, 0) - /** 清除背景 */ - background = null - /** 清除着色 */ - colorFilter = null - } - } - /** 打印日志 */ - printLogcat(tag = "NotifyIcon", iconView.context, packageName, isCustom = customPair.first != null, isGrayscaleIcon) - } - } - - /** 缓存图标数据 */ - private fun PackageParam.cachingIconDatas() { - iconDatas.clear() - IconPackParams(param = this).iconDatas.apply { - when { - isNotEmpty() -> forEach { iconDatas.add(it) } - isEmpty() && isEnableHookColorNotifyIcon(isHooking = false) -> loggerW(msg = "NotifyIconSupportData is empty!") - } - } - } - - /** 刷新缓存数据 */ - private fun PackageParam.recachingPrefs() { - prefs.clearCache() - cachingIconDatas() - refreshStatusBarIcons() - refreshNotificationIcons() - } - override fun onInit() = configs { debugTag = "ColorOSNotify" isDebug = false @@ -462,185 +48,7 @@ class HookEntry : YukiHookXposedInitProxy { /** Hook 被手动关闭停止 Hook */ !prefs.getBoolean(ENABLE_MODULE, default = true) -> loggerW(msg = "Aborted Hook -> Hook Closed") /** 开始 Hook */ - else -> { - /** 缓存图标数据 */ - cachingIconDatas() - /** 移除开发者警告通知 */ - SystemPromptControllerClass.hook { - injectMember { - method { name = "updateDeveloperMode" } - beforeHook { - /** 是否移除 */ - if (prefs.getBoolean(REMOVE_DEV_NOTIFY, default = true)) resultNull() - } - } - } - /** 移除充电完成通知 */ - OplusPowerNotificationWarningsClass.hook { - injectMember { - method { - name = "showChargeErrorDialog" - param(IntType) - } - beforeHook { - /** 是否移除 */ - if (firstArgs() == 7 && prefs.getBoolean(REMOVE_CHANGECP_NOTIFY)) resultNull() - } - } - } - /** 移除免打扰通知 */ - DndAlertHelperClass.hook { - injectMember { - method { - name = "sendNotificationWithEndtime" - param(LongType) - } - beforeHook { - /** 是否移除 */ - if (prefs.getBoolean(REMOVE_DNDALERT_NOTIFY)) resultNull() - } - } - } - /** 修复并替换新版本 ColorOS 原生灰度图标色彩判断*/ - NotificationUtilsClass.hook { - injectMember { - method { - name = "isGrayscaleOplus" - param(ImageViewClass, OplusContrastColorUtilClass.clazz) - } - replaceAny { firstArgs()?.let { isGrayscaleIcon(it.context, it.drawable) } } - }.ignoredHookingFailure() - } - /** 替换状态栏图标 */ - IconManagerClass.hook { - injectMember { - method { - name = "getIconDescriptor" - param(NotificationEntryClass.clazz, BooleanType) - } - afterHook { - IconBuilderClass.clazz.field { name = "context" } - .of(field { name = "iconBuilder" }.of(instance))?.also { context -> - NotificationEntryClass.clazz.method { - name = "getSbn" - }.get(firstArgs).invoke()?.also { nf -> - nf.notification.smallIcon.loadDrawable(context).also { iconDrawable -> - compatStatusIcon( - context = context, - isGrayscaleIcon = isGrayscaleIcon(context, iconDrawable), - packageName = nf.packageName, - drawable = iconDrawable - ).also { pair -> - if (pair.second) StatusBarIconClass.clazz.field { - name = "icon" - type = IconClass - }.get(result).set(Icon.createWithBitmap(pair.first.toBitmap())) - /** 刷新缓存 */ - if (nf.packageName == Const.MODULE_PACKAGE_NAME && - nf.notification.channelId == IconRuleManagerTool.NOTIFY_CHANNEL - ) recachingPrefs() - } - } - } - } - } - } - } - /** 得到状态栏图标实例 */ - StatusBarIconViewClass.hook { - injectMember { - method { - name = "setNotification" - param(StatusBarNotificationClass) - } - afterHook { - if (firstArgs != null) instance().also { - registerWallpaperColorChanged(it) - registerModuleReceiver(it.context) - statusBarIconViews.add(it) - } - } - } - } - /** 替换通知图标和样式 */ - NotificationHeaderViewWrapperClass.hook { - injectMember { - method { name = "resolveHeaderViews" } - afterHook { - NotificationHeaderViewWrapperClass.clazz - .field { name = "mIcon" }.of(instance)?.apply { - ExpandableNotificationRowClass.clazz - .method { name = "getEntry" } - .get(NotificationViewWrapperClass.clazz.field { - name = "mRow" - }.get(instance).self).call()?.let { - it.javaClass.method { - name = "getSbn" - }.get(it).invoke() - }?.notification?.also { nf -> - nf.smallIcon.loadDrawable(context).also { iconDrawable -> - compatNotifyIcon( - isGrayscaleIcon = isGrayscaleIcon(context, iconDrawable), - packageName = context.packageName, - drawable = iconDrawable, - iconColor = nf.color, - iconView = this - ) - } - } - } - } - } - /** 记录实例 */ - injectMember { - constructor { param(ContextClass, ViewClass, ExpandableNotificationRowClass.clazz) } - afterHook { notificationViewWrappers.add(instance) } - } - } - /** 发送适配新的 APP 图标通知 */ - PluginManagerImplClass.hook { - injectMember { - method { - name = "onReceive" - param(ContextClass, IntentClass) - } - afterHook { - if (isEnableHookColorNotifyIcon()) (lastArgs as? Intent)?.also { - if (!it.action.equals(Intent.ACTION_PACKAGE_REPLACED) && - it.getBooleanExtra(Intent.EXTRA_REPLACING, false) - ) return@also - when (it.action) { - Intent.ACTION_PACKAGE_ADDED -> - it.data?.schemeSpecificPart?.also { newPkgName -> - if (iconDatas.takeIf { e -> e.isNotEmpty() } - ?.filter { e -> e.packageName == newPkgName } - .isNullOrEmpty() - ) IconAdaptationTool.pushNewAppSupportNotify(firstArgs()!!, newPkgName) - } - Intent.ACTION_PACKAGE_REMOVED -> - IconAdaptationTool.removeNewAppSupportNotify( - context = firstArgs()!!, - packageName = it.data?.schemeSpecificPart ?: "" - ) - } - } - } - } - } - /** 自动检查通知图标优化更新的注入监听 */ - AbstractReceiverClass.hook { - injectMember { - method { - name = "onReceive" - param(ContextClass, IntentClass) - } - afterHook { - // TODO 待实现 - loggerD(msg = "当前时间：${System.currentTimeMillis()}") - } - } - } - } + else -> loadHooker(SystemUIHooker()) } } } diff --git a/app/src/main/java/com/fankes/coloros/notify/hook/entity/SystemUIHooker.kt b/app/src/main/java/com/fankes/coloros/notify/hook/entity/SystemUIHooker.kt new file mode 100644 index 0000000..5421607 --- /dev/null +++ b/app/src/main/java/com/fankes/coloros/notify/hook/entity/SystemUIHooker.kt @@ -0,0 +1,643 @@ +/* + * ColorOSNotifyIcon - Optimize notification icons for ColorOS and adapt to native notification icon specifications. + * Copyright (C) 2019-2022 Fankes Studio(qzmmcn@163.com) + * https://github.com/fankes/ColorOSNotifyIcon + * + * This software is non-free but opensource software: you can redistribute it + * and/or modify it under the terms of the GNU Affero General Public License + * as published by the Free Software Foundation; either + * version 3 of the License, or any later version. + *

+ * + * This software is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * and eula along with this software. If not, see + * + * + * This file is Created by fankes on 2022/3/25. + */ +package com.fankes.coloros.notify.hook.entity + +import android.app.WallpaperManager +import android.content.BroadcastReceiver +import android.content.Context +import android.content.Intent +import android.content.IntentFilter +import android.graphics.Bitmap +import android.graphics.Color +import android.graphics.Outline +import android.graphics.drawable.BitmapDrawable +import android.graphics.drawable.Drawable +import android.graphics.drawable.Icon +import android.graphics.drawable.VectorDrawable +import android.service.notification.StatusBarNotification +import android.util.ArraySet +import android.view.View +import android.view.ViewOutlineProvider +import android.widget.ImageView +import androidx.core.graphics.drawable.toBitmap +import com.fankes.coloros.notify.bean.IconDataBean +import com.fankes.coloros.notify.const.Const +import com.fankes.coloros.notify.hook.HookConst.ENABLE_ANDROID12_STYLE +import com.fankes.coloros.notify.hook.HookConst.ENABLE_MODULE_LOG +import com.fankes.coloros.notify.hook.HookConst.ENABLE_NOTIFY_ICON_FIX +import com.fankes.coloros.notify.hook.HookConst.ENABLE_NOTIFY_ICON_FIX_NOTIFY +import com.fankes.coloros.notify.hook.HookConst.REMOVE_CHANGECP_NOTIFY +import com.fankes.coloros.notify.hook.HookConst.REMOVE_DEV_NOTIFY +import com.fankes.coloros.notify.hook.HookConst.REMOVE_DNDALERT_NOTIFY +import com.fankes.coloros.notify.hook.HookConst.SYSTEMUI_PACKAGE_NAME +import com.fankes.coloros.notify.hook.factory.isAppNotifyHookAllOf +import com.fankes.coloros.notify.hook.factory.isAppNotifyHookOf +import com.fankes.coloros.notify.param.IconPackParams +import com.fankes.coloros.notify.utils.drawable.drawabletoolbox.DrawableBuilder +import com.fankes.coloros.notify.utils.factory.* +import com.fankes.coloros.notify.utils.tool.IconAdaptationTool +import com.fankes.coloros.notify.utils.tool.IconRuleManagerTool +import com.highcapable.yukihookapi.hook.bean.VariousClass +import com.highcapable.yukihookapi.hook.entity.YukiBaseHooker +import com.highcapable.yukihookapi.hook.factory.field +import com.highcapable.yukihookapi.hook.factory.method +import com.highcapable.yukihookapi.hook.log.loggerD +import com.highcapable.yukihookapi.hook.log.loggerE +import com.highcapable.yukihookapi.hook.log.loggerW +import com.highcapable.yukihookapi.hook.type.android.* +import com.highcapable.yukihookapi.hook.type.java.BooleanType +import com.highcapable.yukihookapi.hook.type.java.FloatType +import com.highcapable.yukihookapi.hook.type.java.IntType +import com.highcapable.yukihookapi.hook.type.java.LongType + +/** + * 系统界面核心 Hook 类 + */ +class SystemUIHooker : YukiBaseHooker() { + + companion object { + + /** 原生存在的类 */ + private const val ContrastColorUtilClass = "com.android.internal.util.ContrastColorUtil" + + /** 原生存在的类 */ + private const val NotificationUtilsClass = "${SYSTEMUI_PACKAGE_NAME}.statusbar.notification.NotificationUtils" + + /** 原生存在的类 */ + private const val NotificationEntryClass = "${SYSTEMUI_PACKAGE_NAME}.statusbar.notification.collection.NotificationEntry" + + /** 原生存在的类 */ + private const val StatusBarIconClass = "com.android.internal.statusbar.StatusBarIcon" + + /** 原生存在的类 */ + private const val StatusBarIconViewClass = "${SYSTEMUI_PACKAGE_NAME}.statusbar.StatusBarIconView" + + /** 原生存在的类 */ + private const val IconBuilderClass = "${SYSTEMUI_PACKAGE_NAME}.statusbar.notification.icon.IconBuilder" + + /** 原生存在的类 */ + private const val IconManagerClass = "${SYSTEMUI_PACKAGE_NAME}.statusbar.notification.icon.IconManager" + + /** ColorOS 存在的类 - 旧版本不存在 */ + private const val OplusContrastColorUtilClass = "com.oplusos.util.OplusContrastColorUtil" + + /** 原生存在的类 */ + private const val PluginManagerImplClass = "${SYSTEMUI_PACKAGE_NAME}.shared.plugins.PluginManagerImpl" + + /** 根据多个版本存在不同的包名相同的类 */ + private val SystemPromptControllerClass = VariousClass( + "com.oplusos.systemui.statusbar.policy.SystemPromptController", + "com.coloros.systemui.statusbar.policy.SystemPromptController" + ) + + /** 根据多个版本存在不同的包名相同的类 */ + private val RoundRectDrawableUtilClass = VariousClass( + "com.oplusos.systemui.notification.util.RoundRectDrawableUtil", + "com.coloros.systemui.notification.util.RoundRectDrawableUtil" + ) + + /** 根据多个版本存在不同的包名相同的类 */ + private val RoundRectDrawableUtil_CompanionClass = VariousClass( + "com.oplusos.systemui.notification.util.RoundRectDrawableUtil\$Companion", + "com.oplusos.systemui.notification.util.RoundRectDrawableUtil\$Companion" + ) + + /** 根据多个版本存在不同的包名相同的类 */ + private val DndAlertHelperClass = VariousClass( + "com.oplusos.systemui.notification.helper.DndAlertHelper", + "com.coloros.systemui.notification.helper.DndAlertHelper" + ) + + /** 根据多个版本存在不同的包名相同的类 */ + private val OplusPowerNotificationWarningsClass = VariousClass( + "com.oplusos.systemui.notification.power.OplusPowerNotificationWarnings", + "com.coloros.systemui.notification.power.ColorosPowerNotificationWarnings" + ) + + /** 根据多个版本存在不同的包名相同的类 */ + private val AbstractReceiverClass = VariousClass( + "com.oplusos.systemui.common.receiver.AbstractReceiver", + "com.coloros.systemui.common.receiver.AbstractReceiver" + ) + + /** 根据多个版本存在不同的包名相同的类 */ + private val ExpandableNotificationRowClass = VariousClass( + "${SYSTEMUI_PACKAGE_NAME}.statusbar.notification.row.ExpandableNotificationRow", + "${SYSTEMUI_PACKAGE_NAME}.statusbar.ExpandableNotificationRow" + ) + + /** 根据多个版本存在不同的包名相同的类 */ + private val NotificationViewWrapperClass = VariousClass( + "${SYSTEMUI_PACKAGE_NAME}.statusbar.notification.row.wrapper.NotificationViewWrapper", + "${SYSTEMUI_PACKAGE_NAME}.statusbar.notification.NotificationViewWrapper" + ) + + /** 根据多个版本存在不同的包名相同的类 */ + private val NotificationHeaderViewWrapperClass = VariousClass( + "${SYSTEMUI_PACKAGE_NAME}.statusbar.notification.row.wrapper.NotificationHeaderViewWrapper", + "${SYSTEMUI_PACKAGE_NAME}.statusbar.notification.NotificationHeaderViewWrapper" + ) + } + + /** 缓存的通知优化图标数组 */ + private var iconDatas = ArrayList() + + /** 缓存的状态栏小图标实例 */ + private var statusBarIconViews = ArraySet() + + /** 缓存的通知小图标包装纸实例 */ + private var notificationViewWrappers = HashSet() + + /** 仅监听一次主题壁纸颜色变化 */ + private var isWallpaperColorListenerSetUp = false + + /** 是否已经使用过缓存刷新功能 */ + private var isUsingCachingMethod = false + + /** 是否已经注册广播 */ + private var isRegisterReceiver = false + + /** 用户解锁屏幕广播接收器 */ + private val userPresentReceiver by lazy { + object : BroadcastReceiver() { + override fun onReceive(context: Context?, intent: Intent?) { + /** 解锁后重新刷新状态栏图标防止系统重新设置它 */ + if (isUsingCachingMethod) refreshStatusBarIcons() + } + } + } + + /** 模块广播接收器 */ + private val moduleReceiver by lazy { + object : BroadcastReceiver() { + override fun onReceive(context: Context?, intent: Intent?) { + context?.sendBroadcast(Intent().apply { + action = Const.MODULE_HANDLER_RECEIVER_TAG + putExtra("isAction", true) + putExtra("isValied", intent?.getStringExtra(Const.MODULE_VERSION_VERIFY_TAG) == Const.MODULE_VERSION_VERIFY) + }) + } + } + } + + /** + * 注册广播接收器 + * @param context 实例 + */ + private fun registerReceiver(context: Context) { + if (isRegisterReceiver) return + context.registerReceiver(userPresentReceiver, IntentFilter().apply { addAction(Intent.ACTION_USER_PRESENT) }) + context.registerReceiver(moduleReceiver, IntentFilter().apply { addAction(Const.MODULE_CHECKING_RECEIVER_TAG) }) + isRegisterReceiver = true + } + + /** + * 是否启用通知图标优化功能 + * @param isHooking 是否判断启用通知功能 - 默认：是 + * @return [Boolean] + */ + private fun isEnableHookColorNotifyIcon(isHooking: Boolean = true) = + prefs.getBoolean(ENABLE_NOTIFY_ICON_FIX, default = true) && + (if (isHooking) prefs.getBoolean(ENABLE_NOTIFY_ICON_FIX_NOTIFY, default = true) else true) + + /** + * 打印日志 + * @param tag 标识 + * @param context 实例 + * @param packageName APP 包名 + * @param isCustom 是否为通知优化生效图标 + * @param isGrayscale 是否为灰度图标 + */ + private fun printLogcat( + tag: String, + context: Context, + packageName: String, + isCustom: Boolean, + isGrayscale: Boolean + ) { + if (prefs.getBoolean(ENABLE_MODULE_LOG)) loggerD( + msg = "$tag --> [${context.findAppName(packageName)}][$packageName] " + + "custom [$isCustom] " + + "grayscale [$isGrayscale]" + ) + } + + /** + * 注册主题壁纸改变颜色监听 + * + * - 仅限在 Android 12 上注册 + * @param view 实例 + */ + private fun registerWallpaperColorChanged(view: View) = runInSafe { + if (!isWallpaperColorListenerSetUp && isUpperOfAndroidS) view.apply { + WallpaperManager.getInstance(context).addOnColorsChangedListener({ _, _ -> refreshNotificationIcons() }, handler) + } + isWallpaperColorListenerSetUp = true + } + + /** 刷新状态栏小图标 */ + private fun refreshStatusBarIcons() = runInSafe { + val nfField = StatusBarIconViewClass.clazz.field { name = "mNotification" } + val sRadiusField = StatusBarIconViewClass.clazz.field { name = "sIconRadiusFraction" } + val sNfSizeField = StatusBarIconViewClass.clazz.field { name = "sNotificationRoundIconSize" } + val roundUtil = RoundRectDrawableUtil_CompanionClass.clazz.method { + name = "getRoundRectDrawable" + param(DrawableClass, FloatType, IntType, IntType, ContextClass) + }.onNoSuchMethod { loggerE(msg = "Your system not support \"getRoundRectDrawable\"!", e = it) } + .get(RoundRectDrawableUtilClass.clazz.field { name = "Companion" }.get().self) + /** 启动一个线程防止卡顿 */ + Thread { + statusBarIconViews.takeIf { it.isNotEmpty() }?.forEach { + runInSafe { + /** 得到通知实例 */ + val nf = nfField.of(it) ?: return@Thread + + /** 得到原始通知图标 */ + val iconDrawable = nf.notification.smallIcon.loadDrawable(it.context) + /** 获取优化后的状态栏通知图标 */ + compatStatusIcon( + context = it.context, + isGrayscaleIcon = isGrayscaleIcon(it.context, iconDrawable), + packageName = nf.packageName, + drawable = iconDrawable + ).also { pair -> + /** 得到图标圆角 */ + val sRadius = sRadiusField.ofFloat(it) + + /** 得到缩放大小 */ + val sNfSize = sNfSizeField.ofInt(it) + /** 在主线程设置图标 */ + it.post { it.setImageDrawable(roundUtil.invoke(pair.first, sRadius, sNfSize, sNfSize, it.context)) } + } + } + } + }.start() + } + + /** 刷新通知小图标 */ + private fun refreshNotificationIcons() = runInSafe { + NotificationHeaderViewWrapperClass.clazz.method { name = "resolveHeaderViews" }.also { result -> + notificationViewWrappers.takeIf { it.isNotEmpty() }?.forEach { result.get(it).call() } + } + } + + /** + * - 这个是修复彩色图标的关键核心代码判断 + * + * 判断是否为灰度图标 - 反射执行系统方法 + * @param context 实例 + * @param drawable 要判断的图标 + * @return [Boolean] + */ + private fun isGrayscaleIcon(context: Context?, drawable: Drawable?) = + ContrastColorUtilClass.clazz.let { + drawable is VectorDrawable || it.method { + name = "isGrayscaleIcon" + param(DrawableClass) + }.get(it.method { + name = "getInstance" + param(ContextClass) + }.get().invoke(context)).callBoolean(drawable) + } + + /** + * 自动适配状态栏、通知栏自定义小图标 + * @param isGrayscaleIcon 是否为灰度图标 + * @param packageName APP 包名 + * @return [Pair] - ([Bitmap] 位图,[Int] 颜色) + */ + private fun compatCustomIcon(isGrayscaleIcon: Boolean, packageName: String): Pair { + var customPair: Pair? = null + when { + /** 替换系统图标为 Android 默认 */ + (packageName == "android" || packageName == "com.android.systemui") && !isGrayscaleIcon -> customPair = + Pair(if (isUpperOfAndroidS) IconPackParams.android12IconBitmap else IconPackParams.android11IconBitmap, 0) + /** 替换自定义通知图标 */ + prefs.getBoolean(ENABLE_NOTIFY_ICON_FIX, default = true) -> run { + if (iconDatas.isNotEmpty()) + iconDatas.forEach { + if (packageName == it.packageName && isAppNotifyHookOf(it)) { + if (!isGrayscaleIcon || isAppNotifyHookAllOf(it)) + customPair = Pair(it.iconBitmap, it.iconColor) + return@run + } + } + } + } + return customPair ?: Pair(null, 0) + } + + /** + * 自动适配状态栏小图标 + * @param context 实例 + * @param isGrayscaleIcon 是否为灰度图标 + * @param packageName APP 包名 + * @param drawable 原始图标 + * @return [Pair] - ([Drawable] 图标,[Boolean] 是否替换) + */ + private fun compatStatusIcon(context: Context, isGrayscaleIcon: Boolean, packageName: String, drawable: Drawable) = + compatCustomIcon(isGrayscaleIcon, packageName).first.also { + /** 打印日志 */ + printLogcat(tag = "StatusIcon", context, packageName, isCustom = it != null, isGrayscaleIcon) + }?.let { Pair(BitmapDrawable(context.resources, it), true) } ?: Pair(drawable, false) + + /** + * 自动适配通知栏小图标 + * @param isGrayscaleIcon 是否为灰度图标 + * @param packageName APP 包名 + * @param drawable 原始图标 + * @param iconColor 原生图标颜色 + * @param iconView 图标 [ImageView] + */ + private fun compatNotifyIcon( + isGrayscaleIcon: Boolean, + packageName: String, + drawable: Drawable, + iconColor: Int, + iconView: ImageView + ) = runInSafe { + compatCustomIcon(isGrayscaleIcon, packageName).also { customPair -> + when { + customPair.first != null || isGrayscaleIcon -> iconView.apply { + /** 设置不要裁切到边界 */ + clipToOutline = false + /** 重新设置图标 */ + setImageBitmap(customPair.first ?: drawable.toBitmap()) + + /** 是否开启 Android 12 风格 */ + val isA12Style = prefs.getBoolean(ENABLE_ANDROID12_STYLE, isUpperOfAndroidS) + + /** 旧版风格 */ + val oldStyle = (if (context.isSystemInDarkMode) 0xffdcdcdc else 0xff707173).toInt() + + /** 新版风格 */ + val newStyle = (if (context.isSystemInDarkMode) 0xffdcdcdc else Color.WHITE).toInt() + + /** 原生着色 */ + val a12Style = if (isUpperOfAndroidS) context.wallpaperColor else + (if (context.isSystemInDarkMode) 0xff707173 else oldStyle).toInt() + + /** 旧版图标着色 */ + val oldApplyColor = customPair.second.takeIf { it != 0 } ?: iconColor.takeIf { it != 0 } ?: oldStyle + + /** 新版图标着色 */ + val newApplyColor = customPair.second.takeIf { it != 0 } ?: iconColor.takeIf { it != 0 } ?: a12Style + + /** 判断风格并开始 Hook */ + if (isA12Style) { + background = DrawableBuilder().rounded().solidColor(newApplyColor).build() + setColorFilter(newStyle) + setPadding(2.dp(context), 2.dp(context), 2.dp(context), 2.dp(context)) + } else { + background = null + setColorFilter(oldApplyColor) + setPadding(0, 0, 0, 0) + } + } + else -> iconView.apply { + /** 重新设置图标 */ + setImageDrawable(drawable) + /** 设置裁切到边界 */ + clipToOutline = true + /** 设置一个圆角轮廓裁切 */ + outlineProvider = object : ViewOutlineProvider() { + override fun getOutline(view: View, out: Outline) { + out.setRoundRect( + 0, 0, + view.width, view.height, 3.dpFloat(context) + ) + } + } + /** 清除图标间距 */ + setPadding(0, 0, 0, 0) + /** 清除背景 */ + background = null + /** 清除着色 */ + colorFilter = null + } + } + /** 打印日志 */ + printLogcat(tag = "NotifyIcon", iconView.context, packageName, isCustom = customPair.first != null, isGrayscaleIcon) + } + } + + /** 缓存图标数据 */ + private fun cachingIconDatas() { + iconDatas.clear() + IconPackParams(param = this).iconDatas.apply { + when { + isNotEmpty() -> forEach { iconDatas.add(it) } + isEmpty() && isEnableHookColorNotifyIcon(isHooking = false) -> loggerW(msg = "NotifyIconSupportData is empty!") + } + } + } + + /** 刷新缓存数据 */ + private fun recachingPrefs() { + isUsingCachingMethod = true + prefs.clearCache() + cachingIconDatas() + refreshStatusBarIcons() + refreshNotificationIcons() + } + + override fun onHook() { + /** 缓存图标数据 */ + cachingIconDatas() + /** 移除开发者警告通知 */ + SystemPromptControllerClass.hook { + injectMember { + method { name = "updateDeveloperMode" } + beforeHook { + /** 是否移除 */ + if (prefs.getBoolean(REMOVE_DEV_NOTIFY, default = true)) resultNull() + } + } + } + /** 移除充电完成通知 */ + OplusPowerNotificationWarningsClass.hook { + injectMember { + method { + name = "showChargeErrorDialog" + param(IntType) + } + beforeHook { + /** 是否移除 */ + if (firstArgs() == 7 && prefs.getBoolean(REMOVE_CHANGECP_NOTIFY)) resultNull() + } + } + } + /** 移除免打扰通知 */ + DndAlertHelperClass.hook { + injectMember { + method { + name = "sendNotificationWithEndtime" + param(LongType) + } + beforeHook { + /** 是否移除 */ + if (prefs.getBoolean(REMOVE_DNDALERT_NOTIFY)) resultNull() + } + } + } + /** 修复并替换新版本 ColorOS 原生灰度图标色彩判断*/ + NotificationUtilsClass.hook { + injectMember { + method { + name = "isGrayscaleOplus" + param(ImageViewClass, OplusContrastColorUtilClass.clazz) + } + replaceAny { firstArgs()?.let { isGrayscaleIcon(it.context, it.drawable) } } + }.ignoredHookingFailure() + } + /** 替换状态栏图标 */ + IconManagerClass.hook { + injectMember { + method { + name = "getIconDescriptor" + param(NotificationEntryClass.clazz, BooleanType) + } + afterHook { + IconBuilderClass.clazz.field { name = "context" } + .of(field { name = "iconBuilder" }.of(instance))?.also { context -> + NotificationEntryClass.clazz.method { + name = "getSbn" + }.get(firstArgs).invoke()?.also { nf -> + nf.notification.smallIcon.loadDrawable(context).also { iconDrawable -> + compatStatusIcon( + context = context, + isGrayscaleIcon = isGrayscaleIcon(context, iconDrawable), + packageName = nf.packageName, + drawable = iconDrawable + ).also { pair -> + if (pair.second) StatusBarIconClass.clazz.field { + name = "icon" + type = IconClass + }.get(result).set(Icon.createWithBitmap(pair.first.toBitmap())) + /** 刷新缓存 */ + if (nf.packageName == Const.MODULE_PACKAGE_NAME && + nf.notification.channelId == IconRuleManagerTool.NOTIFY_CHANNEL + ) recachingPrefs() + } + } + } + } + } + } + } + /** 得到状态栏图标实例 */ + StatusBarIconViewClass.hook { + injectMember { + method { + name = "setNotification" + param(StatusBarNotificationClass) + } + afterHook { + if (firstArgs != null) instance().also { + registerWallpaperColorChanged(it) + registerReceiver(it.context) + statusBarIconViews.add(it) + } + } + } + } + /** 替换通知图标和样式 */ + NotificationHeaderViewWrapperClass.hook { + injectMember { + method { name = "resolveHeaderViews" } + afterHook { + NotificationHeaderViewWrapperClass.clazz + .field { name = "mIcon" }.of(instance)?.apply { + ExpandableNotificationRowClass.clazz + .method { name = "getEntry" } + .get(NotificationViewWrapperClass.clazz.field { + name = "mRow" + }.get(instance).self).call()?.let { + it.javaClass.method { + name = "getSbn" + }.get(it).invoke() + }?.notification?.also { nf -> + nf.smallIcon.loadDrawable(context).also { iconDrawable -> + compatNotifyIcon( + isGrayscaleIcon = isGrayscaleIcon(context, iconDrawable), + packageName = context.packageName, + drawable = iconDrawable, + iconColor = nf.color, + iconView = this + ) + } + } + } + } + } + /** 记录实例 */ + injectMember { + constructor { param(ContextClass, ViewClass, ExpandableNotificationRowClass.clazz) } + afterHook { notificationViewWrappers.add(instance) } + } + } + /** 发送适配新的 APP 图标通知 */ + PluginManagerImplClass.hook { + injectMember { + method { + name = "onReceive" + param(ContextClass, IntentClass) + } + afterHook { + if (isEnableHookColorNotifyIcon()) (lastArgs as? Intent)?.also { + if (!it.action.equals(Intent.ACTION_PACKAGE_REPLACED) && + it.getBooleanExtra(Intent.EXTRA_REPLACING, false) + ) return@also + when (it.action) { + Intent.ACTION_PACKAGE_ADDED -> + it.data?.schemeSpecificPart?.also { newPkgName -> + if (iconDatas.takeIf { e -> e.isNotEmpty() } + ?.filter { e -> e.packageName == newPkgName } + .isNullOrEmpty() + ) IconAdaptationTool.pushNewAppSupportNotify(firstArgs()!!, newPkgName) + } + Intent.ACTION_PACKAGE_REMOVED -> + IconAdaptationTool.removeNewAppSupportNotify( + context = firstArgs()!!, + packageName = it.data?.schemeSpecificPart ?: "" + ) + } + } + } + } + } + /** 自动检查通知图标优化更新的注入监听 */ + AbstractReceiverClass.hook { + injectMember { + method { + name = "onReceive" + param(ContextClass, IntentClass) + } + afterHook { + // TODO 待实现 + loggerD(msg = "当前时间：${System.currentTimeMillis()}") + } + } + } + } +} \ No newline at end of file