commit ccc8000f4fd9d04abdeacac1e25c5ec286fb0fa9
parent ae2e4c16ae04acd19d715783424d3bfed15e92bf
Author: rhunk <101876869+rhunk@users.noreply.github.com>
Date:   Sat,  7 Sep 2024 22:44:13 +0200

fix(core): stealth mode indicator

Diffstat:

Mcore/src/main/kotlin/me/rhunk/snapenhance/core/features/impl/ui/StealthModeIndicator.kt | 5+++--


1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/core/src/main/kotlin/me/rhunk/snapenhance/core/features/impl/ui/StealthModeIndicator.kt b/core/src/main/kotlin/me/rhunk/snapenhance/core/features/impl/ui/StealthModeIndicator.kt
@@ -9,6 +9,7 @@ import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.Job
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.withContext
+import me.rhunk.snapenhance.common.data.RuleState
 import me.rhunk.snapenhance.core.event.events.impl.BindViewEvent
 import me.rhunk.snapenhance.core.features.Feature
 import me.rhunk.snapenhance.core.features.impl.spying.StealthMode
@@ -30,7 +31,7 @@ class StealthModeIndicator : Feature("StealthModeIndicator") {
         private fun requestUpdate(conversationId: String) {
             fetchJob?.cancel()
             fetchJob = context.coroutineScope.launch {
-                val isStealth = stealthMode.getState(conversationId)
+                val isStealth = stealthMode.canUseRule(conversationId)
                 withContext(Dispatchers.Main) {
                     listener(isStealth)
                 }
@@ -58,7 +59,7 @@ class StealthModeIndicator : Feature("StealthModeIndicator") {
 
             stealthMode.addStateListener { conversationId, state ->
                 runCatching {
-                    listeners[conversationId]?.invoke(state)
+                    listeners[conversationId]?.invoke(stealthMode.getRuleState()?.let { if (it == RuleState.BLACKLIST) !state else state } ?: state)
                 }.onFailure {
                     context.log.error("Failed to update stealth mode indicator", it)
                 }