#308 [fix] 저장 가능, 대표 룰 수정 로직 변경

Hous-Release · Sep 3, 2023 · 88980e5 · 88980e5
1 parent 190a1b6
commit 88980e5
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/...main/java/hous/release/android/presentation/our_rules/viewmodel/RepresentRuleViewModel.kt b/...main/java/hous/release/android/presentation/our_rules/viewmodel/RepresentRuleViewModel.kt
@@ -43,11 +43,11 @@ class RepresentRuleViewModel @Inject constructor(
         }
     }
 
-    val isSavable get() = uiState.value.rules == uiState.value.originRules
+    val isSavable get() = uiState.value.rules != uiState.value.originRules
 
     fun updateRuleBy(id: Int) {
         viewModelScope.launch {
-            if (canAddRepresentRule()) {
+            if (canUpdateRuleBy(id)) {
                 uiEvents.send(RepresentRulesEvent.UpdateRule(id))
                 return@launch
             }
@@ -78,8 +78,10 @@ class RepresentRuleViewModel @Inject constructor(
         }
     }
 
-    private fun canAddRepresentRule(): Boolean {
-        return uiState.value.rules.filter { it.isRepresent }.size < 3
+    private fun canUpdateRuleBy(id: Int): Boolean {
+        val isRepresent = uiState.value.rules.find { it.id == id }?.isRepresent ?: false
+        if (isRepresent) return true
+        return uiState.value.rules.count { it.isRepresent } + 1 <= 3
     }
 
     private companion object {