From ea81f68d2a7a7e93fd8d9d062cfe4c880020555a Mon Sep 17 00:00:00 2001
From: QQ littleice <1767523953@qq.com>
Date: Thu, 10 Mar 2022 21:57:36 +0800
Subject: [PATCH] simplify HideStatusBarNetworkSpeedSecond.kt

---
 .../HideStatusBarNetworkSpeedSecond.kt        | 34 +++++++------------
 1 file changed, 12 insertions(+), 22 deletions(-)

diff --git a/app/src/main/java/com/lt2333/simplicitytools/hook/app/systemui/HideStatusBarNetworkSpeedSecond.kt b/app/src/main/java/com/lt2333/simplicitytools/hook/app/systemui/HideStatusBarNetworkSpeedSecond.kt
index 0ff8c068..930beb00 100644
--- a/app/src/main/java/com/lt2333/simplicitytools/hook/app/systemui/HideStatusBarNetworkSpeedSecond.kt
+++ b/app/src/main/java/com/lt2333/simplicitytools/hook/app/systemui/HideStatusBarNetworkSpeedSecond.kt
@@ -1,32 +1,22 @@
 package com.lt2333.simplicitytools.hook.app.systemui
 
-import com.lt2333.simplicitytools.util.XSPUtils
+import com.lt2333.simplicitytools.util.hasEnable
+import com.lt2333.simplicitytools.util.hookBeforeMethod
+import com.lt2333.simplicitytools.util.isNonNull
 import de.robv.android.xposed.IXposedHookLoadPackage
-import de.robv.android.xposed.XC_MethodHook
-import de.robv.android.xposed.XposedHelpers
 import de.robv.android.xposed.callbacks.XC_LoadPackage
 
 class HideStatusBarNetworkSpeedSecond :IXposedHookLoadPackage {
     override fun handleLoadPackage(lpparam: XC_LoadPackage.LoadPackageParam) {
-        val classIfExists = XposedHelpers.findClassIfExists(
-            "com.android.systemui.statusbar.views.NetworkSpeedView",
-            lpparam.classLoader
-        )
-        XposedHelpers.findAndHookMethod(
-            classIfExists,
-            "setNetworkSpeed",
-            String::class.java,
-            object : XC_MethodHook() {
-                override fun beforeHookedMethod(param: MethodHookParam) {
-                    if (XSPUtils.getBoolean("hide_status_bar_network_speed_second", false)) {
-                        if (param.args[0] != null) {
-                            param.args[0] = (param.args[0] as String)
-                                .replace("/", "")
-                                .replace("s", "")
-                                .replace("'", "")
-                        }
-                    }
+        "com.android.systemui.statusbar.views.NetworkSpeedView".hookBeforeMethod(lpparam.classLoader, "setNetworkSpeed", String::class.java) {
+            hasEnable("hide_status_bar_network_speed_second") {
+                it.args[0].isNonNull { s ->
+                    it.args[0] = (s as String)
+                        .replace("/", "")
+                        .replace("s", "")
+                        .replace("'", "")
                 }
-            })
+            }
+        }
     }
 }
\ No newline at end of file