Deselect all sessions when leaving select mode

vector/src/main/java/im/vector/app/features/settings/devices/v2/othersessions/OtherSessionsFragment.kt

@@ -154,7 +154,7 @@ class OtherSessionsFragment :
 
     override fun invalidate() = withState(viewModel) { state ->
         if (state.devices is Success) {
-            val devices = state.devices().orEmpty()
+            val devices = state.devices.invoke()
             renderDevices(devices, state.currentFilter)
             updateToolbar(devices, state.isSelectModeEnabled)
         }

vector/src/main/java/im/vector/app/features/settings/devices/v2/othersessions/OtherSessionsViewModel.kt

@@ -85,16 +85,15 @@ class OtherSessionsViewModel @AssistedInject constructor(
     }
 
     private fun handleDisableSelectMode() {
-        // TODO deselect all selected sessions
-        setState { copy(isSelectModeEnabled = false) }
+        setSelectionForAllDevices(isSelected = false, enableSelectMode = false)
     }
 
     private fun handleEnableSelectMode(deviceId: String?) {
-        toggleSelectionForDevice(deviceId, true)
+        toggleSelectionForDevice(deviceId, enableSelectMode = true)
     }
 
     private fun handleToggleSelectionForDevice(deviceId: String) = withState { state ->
-        toggleSelectionForDevice(deviceId, state.isSelectModeEnabled)
+        toggleSelectionForDevice(deviceId, enableSelectMode = state.isSelectModeEnabled)
     }
 
     private fun toggleSelectionForDevice(deviceId: String?, enableSelectMode: Boolean) = withState { state ->
@@ -118,4 +117,20 @@ class OtherSessionsViewModel @AssistedInject constructor(
             )
         }
     }
+
+    private fun setSelectionForAllDevices(isSelected: Boolean, enableSelectMode: Boolean) = withState { state ->
+        val updatedDevices = if (state.devices is Success) {
+            val updatedDevices = state.devices.invoke().map { it.copy(isSelected = isSelected) }
+            Success(updatedDevices)
+        } else {
+            state.devices
+        }
+
+        setState {
+            copy(
+                    devices = updatedDevices,
+                    isSelectModeEnabled = enableSelectMode
+            )
+        }
+    }
 }