From fc8e92eef1e1086e343c3c5645cfee612d62f415 Mon Sep 17 00:00:00 2001
From: Stefan Ceriu <stefan.ceriu@gmail.com>
Date: Thu, 10 Mar 2022 16:40:16 +0200
Subject: [PATCH] Separated encrypted from non-encrypted rooms. Moved DMs to
 the bottom.

---
 .../Screens/HomeScreen/HomeScreenModels.swift | 20 +++++++++++--------
 .../Screens/HomeScreen/View/HomeScreen.swift  | 16 +++++++--------
 2 files changed, 20 insertions(+), 16 deletions(-)

diff --git a/ElementX/Sources/Screens/HomeScreen/HomeScreenModels.swift b/ElementX/Sources/Screens/HomeScreen/HomeScreenModels.swift
index 9db2e4846..4aae45339 100644
--- a/ElementX/Sources/Screens/HomeScreen/HomeScreenModels.swift
+++ b/ElementX/Sources/Screens/HomeScreen/HomeScreenModels.swift
@@ -37,20 +37,24 @@ struct HomeScreenViewState: BindableState {
     var rooms: [HomeScreenRoom] = []
     var isLoadingRooms: Bool = false
     
-    var firstDirectRooms: [HomeScreenRoom] {
-        Array(rooms.filter { $0.isDirect }.prefix(5))
+    var sortedRooms: [HomeScreenRoom] {
+        rooms.sorted(by: { ($0.displayName ?? $0.id).lowercased() < ($1.displayName ?? $1.id).lowercased() })
     }
     
-    var otherDirectRooms: [HomeScreenRoom] {
-        Array(rooms.filter { $0.isDirect }.dropFirst(5))
+    var unencryptedDMs: [HomeScreenRoom] {
+        Array(sortedRooms.filter { $0.isDirect && !$0.isEncrypted })
     }
     
-    var firstNondirectRooms: [HomeScreenRoom] {
-        Array(rooms.filter { !$0.isDirect }.prefix(5))
+    var encryptedDMs: [HomeScreenRoom] {
+        Array(sortedRooms.filter { $0.isDirect && $0.isEncrypted})
     }
     
-    var otherNondirectRooms: [HomeScreenRoom] {
-        Array(rooms.filter { !$0.isDirect }.dropFirst(5))
+    var unencryptedRooms: [HomeScreenRoom] {
+        Array(sortedRooms.filter { !$0.isDirect && !$0.isEncrypted })
+    }
+    
+    var encryptedRooms: [HomeScreenRoom] {
+        Array(sortedRooms.filter { !$0.isDirect && $0.isEncrypted })
     }
 }
 
diff --git a/ElementX/Sources/Screens/HomeScreen/View/HomeScreen.swift b/ElementX/Sources/Screens/HomeScreen/View/HomeScreen.swift
index af0bd571f..b1feda51c 100644
--- a/ElementX/Sources/Screens/HomeScreen/View/HomeScreen.swift
+++ b/ElementX/Sources/Screens/HomeScreen/View/HomeScreen.swift
@@ -33,15 +33,15 @@ struct HomeScreen: View {
                     }
                 } else {
                     List {
-                        Section("People") {
-                            ForEach(context.viewState.firstDirectRooms) { room in
+                        Section("Rooms") {
+                            ForEach(context.viewState.unencryptedRooms) { room in
                                 RoomCell(room: room, context: context)
                             }
                             
-                            let other = context.viewState.otherDirectRooms
+                            let other = context.viewState.encryptedRooms
                             
                             if other.count > 0 {
-                                DisclosureGroup("See more") {
+                                DisclosureGroup("Encrypted") {
                                     ForEach(other) { room in
                                         RoomCell(room: room, context: context)
                                     }
@@ -49,15 +49,15 @@ struct HomeScreen: View {
                             }
                         }
                         
-                        Section("Rooms") {
-                            ForEach(context.viewState.firstNondirectRooms) { room in
+                        Section("People") {
+                            ForEach(context.viewState.unencryptedDMs) { room in
                                 RoomCell(room: room, context: context)
                             }
                             
-                            let other = context.viewState.otherNondirectRooms
+                            let other = context.viewState.encryptedDMs
                             
                             if other.count > 0 {
-                                DisclosureGroup("See more") {
+                                DisclosureGroup("Encrypted") {
                                     ForEach(other) { room in
                                         RoomCell(room: room, context: context)
                                     }