From 230b3b4e3800cfbf130fd4b45e9b7436f73057ea Mon Sep 17 00:00:00 2001
From: Fankesyooni <qzmmcn@163.com>
Date: Tue, 29 Mar 2022 21:45:19 +0800
Subject: [PATCH] Merge code

---
 .../notify/hook/entity/SystemUIHooker.kt       | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

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
index aa4e971..7e55ddf 100644
--- 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
@@ -607,8 +607,11 @@ class SystemUIHooker : YukiBaseHooker() {
                 }
                 afterHook {
                     if (firstArgs != null) instance<ImageView>().also {
+                        /** 注册壁纸颜色监听 */
                         registerWallpaperColorChanged(it)
+                        /** 注册广播 */
                         registerReceiver(it.context)
+                        /** 缓存实例 */
                         statusBarIconViews.add(it)
                     }
                 }
@@ -691,11 +694,16 @@ class SystemUIHooker : YukiBaseHooker() {
                     param(ContextClass, IntentClass)
                 }
                 afterHook {
-                    if (isEnableHookColorNotifyIcon() && prefs.get(DataConst.ENABLE_NOTIFY_ICON_FIX_AUTO))
-                        IconAdaptationTool.prepareAutoUpdateIconRule(
-                            context = firstArgs()!!,
-                            timeSet = prefs.get(DataConst.NOTIFY_ICON_FIX_AUTO_TIME)
-                        )
+                    firstArgs<Context>()?.also {
+                        /** 注册广播 */
+                        registerReceiver(it)
+                        /** 注册定时监听 */
+                        if (isEnableHookColorNotifyIcon() && prefs.get(DataConst.ENABLE_NOTIFY_ICON_FIX_AUTO))
+                            IconAdaptationTool.prepareAutoUpdateIconRule(
+                                context = it,
+                                timeSet = prefs.get(DataConst.NOTIFY_ICON_FIX_AUTO_TIME)
+                            )
+                    }
                 }
             }
         }