From c5e5b70e08a743497f24fb6a77eb372fd40de522 Mon Sep 17 00:00:00 2001
From: Viktor De Pasquale <info@depasquale.wiki>
Date: Tue, 21 Jan 2020 22:01:55 +0100
Subject: [PATCH] Added safe mode notice to modules

---
 .../topjohnwu/magisk/ui/module/ModuleViewModel.kt    | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/app/src/main/java/com/topjohnwu/magisk/ui/module/ModuleViewModel.kt b/app/src/main/java/com/topjohnwu/magisk/ui/module/ModuleViewModel.kt
index 7c9528c10..97c04cf19 100644
--- a/app/src/main/java/com/topjohnwu/magisk/ui/module/ModuleViewModel.kt
+++ b/app/src/main/java/com/topjohnwu/magisk/ui/module/ModuleViewModel.kt
@@ -81,12 +81,14 @@ class ModuleViewModel(
         it.add(itemNoneUpdatable)
     }
 
+    private val itemsCoreOnly = ObservableArrayList<SafeModeNotice>()
     private val itemsInstalled = diffListOf<ModuleItem>()
     private val itemsUpdatable = diffListOf<RepoItem.Update>()
     private val itemsRemote = diffListOf<RepoItem.Remote>()
 
     val adapter = adapterOf<ComparableRvItem<*>>()
     val items = MergeObservableList<ComparableRvItem<*>>()
+        .insertList(itemsCoreOnly)
         .insertItem(sectionActive)
         .insertList(itemsInstalledHelpers)
         .insertList(itemsInstalled)
@@ -159,6 +161,7 @@ class ModuleViewModel(
     // ---
 
     override fun refresh(): Disposable {
+        updateCoreOnlyWarning()
         if (itemsRemote.isEmpty())
             loadRemote()
         return loadInstalled().subscribeK()
@@ -268,6 +271,15 @@ class ModuleViewModel(
             .subscribeK { it.progress.value = progress }
             .add()
 
+    private fun updateCoreOnlyWarning() {
+        if (Config.coreOnly) {
+            if (itemsCoreOnly.isNotEmpty()) return
+            itemsCoreOnly.add(SafeModeNotice)
+        } else {
+            itemsCoreOnly.clear()
+        }
+    }
+
     // ---
 
     @WorkerThread