From f87c2b6cb0369ad801ed3b3f00f089cc84c14914 Mon Sep 17 00:00:00 2001
From: hosizoraru <hosizoraru@gmail.com>
Date: Tue, 23 May 2023 20:00:26 +0800
Subject: [PATCH] Magisk WhiteList/KernelSU SU Permission Toast

---
 .../sky/voyager/activity/pages/MenuPage.kt    | 100 +++++++++++++-----
 app/src/main/res/values-zh-rCN/strings.xml    |   1 +
 app/src/main/res/values/strings.xml           |   1 +
 3 files changed, 73 insertions(+), 29 deletions(-)

diff --git a/app/src/main/kotlin/star/sky/voyager/activity/pages/MenuPage.kt b/app/src/main/kotlin/star/sky/voyager/activity/pages/MenuPage.kt
index 04eca558..935f2500 100644
--- a/app/src/main/kotlin/star/sky/voyager/activity/pages/MenuPage.kt
+++ b/app/src/main/kotlin/star/sky/voyager/activity/pages/MenuPage.kt
@@ -46,15 +46,24 @@ class MenuPage : BasePage() {
                         dismiss()
                     }
                     setRButton(R.string.done) {
-                        PACKAGE_NAME_HOOKED.forEach {
-                            if (it != "android") Terminal.exec("killall $it")
+                        try {
+                            PACKAGE_NAME_HOOKED.forEach {
+                                if (it != "android") Terminal.exec("killall $it")
+                            }
+                            Toast.makeText(
+                                activity,
+                                getString(R.string.finished),
+                                Toast.LENGTH_SHORT
+                            ).show()
+                            dismiss()
+                        } catch (_: Throwable) {
+                            Toast.makeText(
+                                activity,
+                                getString(R.string.su_permission),
+                                Toast.LENGTH_LONG
+                            ).show()
+                            dismiss()
                         }
-                        Toast.makeText(
-                            activity,
-                            getString(R.string.finished),
-                            Toast.LENGTH_SHORT
-                        ).show()
-                        dismiss()
                     }
                 }.show()
             }
@@ -70,13 +79,22 @@ class MenuPage : BasePage() {
                         dismiss()
                     }
                     setRButton(R.string.done) {
-                        Terminal.exec("killall com.android.systemui")
-                        Toast.makeText(
-                            activity,
-                            getString(R.string.finished),
-                            Toast.LENGTH_SHORT
-                        ).show()
-                        dismiss()
+                        try {
+                            Terminal.exec("killall com.android.systemui")
+                            Toast.makeText(
+                                activity,
+                                getString(R.string.finished),
+                                Toast.LENGTH_SHORT
+                            ).show()
+                            dismiss()
+                        } catch (_: Throwable) {
+                            Toast.makeText(
+                                activity,
+                                getString(R.string.su_permission),
+                                Toast.LENGTH_LONG
+                            ).show()
+                            dismiss()
+                        }
                     }
                 }.show()
             }
@@ -92,25 +110,44 @@ class MenuPage : BasePage() {
                         dismiss()
                     }
                     setRButton(R.string.done) {
-                        Terminal.exec("killall com.miui.home")
-                        Toast.makeText(
-                            activity,
-                            getString(R.string.finished),
-                            Toast.LENGTH_SHORT
-                        ).show()
-                        dismiss()
+                        try {
+                            Terminal.exec("killall com.miui.home")
+                            Toast.makeText(
+                                activity,
+                                getString(R.string.finished),
+                                Toast.LENGTH_SHORT
+                            ).show()
+                            dismiss()
+                        } catch (_: Throwable) {
+                            Toast.makeText(
+                                activity,
+                                getString(R.string.su_permission),
+                                Toast.LENGTH_LONG
+                            ).show()
+                            dismiss()
+                        }
                     }
                 }.show()
             }
         )
         Line()
-        TextSummaryWithArrow(TextSummaryV(
-            textId = R.string.backup, onClickListener = {
-                BackupUtils.backup(activity, activity.createDeviceProtectedStorageContext().getSharedPreferences("voyager_config", Context.MODE_WORLD_READABLE))
-            }))
+        TextSummaryWithArrow(
+            TextSummaryV(
+                textId = R.string.backup, onClickListener = {
+                    BackupUtils.backup(
+                        activity,
+                        activity.createDeviceProtectedStorageContext()
+                            .getSharedPreferences("voyager_config", Context.MODE_WORLD_READABLE)
+                    )
+                })
+        )
 
         TextSummaryWithArrow(TextSummaryV(textId = R.string.recovery, onClickListener = {
-            BackupUtils.recovery(activity, activity.createDeviceProtectedStorageContext().getSharedPreferences("voyager_config", Context.MODE_WORLD_READABLE))
+            BackupUtils.recovery(
+                activity,
+                activity.createDeviceProtectedStorageContext()
+                    .getSharedPreferences("voyager_config", Context.MODE_WORLD_READABLE)
+            )
         }))
 
         TextWithArrow(TextV(textId = R.string.ResetModule, onClickListener = {
@@ -118,8 +155,13 @@ class MenuPage : BasePage() {
                 setTitle(R.string.ResetModuleDialog)
                 setMessage(R.string.ResetModuleDialogTips)
                 setLButton(R.string.done) {
-                    activity.getSharedPreferences("voyager_config", Activity.MODE_WORLD_READABLE).edit().clear().apply()
-                    Toast.makeText(activity, activity.getString(R.string.ResetSuccess), Toast.LENGTH_LONG).show()
+                    activity.getSharedPreferences("voyager_config", Activity.MODE_WORLD_READABLE)
+                        .edit().clear().apply()
+                    Toast.makeText(
+                        activity,
+                        activity.getString(R.string.ResetSuccess),
+                        Toast.LENGTH_LONG
+                    ).show()
                 }
                 setRButton(R.string.cancel)
                 finally { dismiss() }
diff --git a/app/src/main/res/values-zh-rCN/strings.xml b/app/src/main/res/values-zh-rCN/strings.xml
index 657dc49e..6a636593 100644
--- a/app/src/main/res/values-zh-rCN/strings.xml
+++ b/app/src/main/res/values-zh-rCN/strings.xml
@@ -43,6 +43,7 @@
     <string name="restart_scope_miui_home_tips">确定重启系统桌面？</string>
 
     <string name="main_switch">模块总开关</string>
+    <string name="su_permission">请给予本模块Su/Root权限</string>
     <string name="about">关于</string>
     <string name="about_module">关于模块</string>
     <string name="about_module_summary">查看模块相关信息</string>
diff --git a/app/src/main/res/values/strings.xml b/app/src/main/res/values/strings.xml
index 61e882c5..97f817bf 100644
--- a/app/src/main/res/values/strings.xml
+++ b/app/src/main/res/values/strings.xml
@@ -46,6 +46,7 @@
     <string name="restart_scope_miui_home_tips">Are you sure to restart MiuiHome?</string>
 
     <string name="main_switch">Module main switch</string>
+    <string name="su_permission">Please give this module Su/Root permission</string>
     <string name="about">About</string>
     <string name="about_module">About Module</string>
     <string name="about_module_summary">View module related information</string>