From 2f740c70656b89467b9b5a7e082079aa757b131f Mon Sep 17 00:00:00 2001
From: Ryan ZHAO <>
Date: Wed, 31 Jan 2024 16:41:34 +1100
Subject: [PATCH] WIP: refactor global search screen into SwiftUI

---
 .../GlobalSearch/GlobalSearchScreen.swift     | 188 ++++++++++++++++--
 Session/Home/HomeVC.swift                     |   2 +-
 .../Shared/SessionHostingViewController.swift |  18 +-
 .../ProfilePictureView+Convenience.swift      |  30 +++
 .../Components/SwiftUI/SessionSearchBar.swift |   2 +-
 5 files changed, 217 insertions(+), 23 deletions(-)

diff --git a/Session/Home/GlobalSearch/GlobalSearchScreen.swift b/Session/Home/GlobalSearch/GlobalSearchScreen.swift
index d81d18a29..b72c04bbd 100644
--- a/Session/Home/GlobalSearch/GlobalSearchScreen.swift
+++ b/Session/Home/GlobalSearch/GlobalSearchScreen.swift
@@ -14,44 +14,192 @@ struct GlobalSearchScreen: View {
 
     enum SearchSection: Int, Differentiable {
         case noResults
-        case contactsAndGroups
+        case contacts
         case messages
     }
     
     @EnvironmentObject var host: HostWrapper
     
-    @State var searchText: String = ""
-    @State private var searchResultSet: [SectionModel] = []
+    @State private var searchText: String = ""
+    @State private var searchResultSet: [SectionModel] = Self.defaultSearchResults
+    @State private var readConnection: Atomic<Database?> = Atomic(nil)
+    @State private var termForCurrentSearchResultSet: String = ""
+    @State private var lastSearchText: String?
+    @State private var isLoading = false
+    
+    fileprivate static var defaultSearchResults: [SectionModel] = {
+        let result: SessionThreadViewModel? = Storage.shared.read { db -> SessionThreadViewModel? in
+            try SessionThreadViewModel
+                .noteToSelfOnlyQuery(userPublicKey: getUserHexEncodedPublicKey(db))
+                .fetchOne(db)
+        }
+        
+        return [ result.map { ArraySection(model: .contacts, elements: [$0]) } ]
+            .compactMap { $0 }
+    }()
     
     var body: some View {
-        ZStack(alignment: .topLeading) {
-            ScrollView(.vertical, showsIndicators: false) {
-                VStack(
-                    alignment: .leading,
-                    spacing: Values.smallSpacing
-                ) {
-                    SessionSearchBar(
-                        searchText: $searchText.onChange{ updatedSearchText in 
-                            onSearchTextChange(updatedSearchText: updatedSearchText)
-                        },
-                        cancelAction: {
-                            self.host.controller?.navigationController?.popViewController(animated: true)
+        VStack(alignment: .leading) {
+            SessionSearchBar(
+                searchText: $searchText.onChange{ updatedSearchText in
+                    onSearchTextChange(rawSearchText: updatedSearchText)
+                },
+                cancelAction: {
+                    self.host.controller?.navigationController?.popViewController(animated: true)
+                }
+            )
+            
+            List{
+                ForEach(0..<searchResultSet.count, id: \.self) { sectionIndex in
+                    let section = searchResultSet[sectionIndex]
+                    let sectionTitle: String = {
+                        switch section.model {
+                            case .noResults: return ""
+                            case .contacts: return (section.elements.isEmpty ? "" : "NEW_CONVERSATION_CONTACTS_SECTION_TITLE".localized())
+                            case .messages:return (section.elements.isEmpty ? "" : "SEARCH_SECTION_MESSAGES".localized())
                         }
-                    )
-                    
-                    
+                    }()
+                    Section(
+                        header: Text(sectionTitle)
+                            .bold()
+                            .font(.system(size: Values.mediumLargeFontSize))
+                            .foregroundColor(themeColor: .textPrimary)
+                    ) {
+                        ForEach(0..<section.elements.count, id: \.self) { rowIndex in
+                            let row = section.elements[rowIndex]
+                            SearchResultCell(searchText: searchText, viewModel: row)
+                        }
+                    }
                 }
             }
+            .transparentScrolling()
+            .listStyle(.plain)
+            .padding(.top, -Values.mediumSpacing)
         }
         .backgroundColor(themeColor: .backgroundPrimary)
     }
     
-    func onSearchTextChange(updatedSearchText: String) {
-        guard updatedSearchText != searchText else { return }
+    func onSearchTextChange(rawSearchText: String, force: Bool = false) {
+        let searchText = rawSearchText.stripped
+        
+        guard searchText.count > 0 else {
+            guard searchText != (lastSearchText ?? "") else { return }
+            
+            searchResultSet = Self.defaultSearchResults
+            lastSearchText = nil
+            return
+        }
+        guard force || lastSearchText != searchText else { return }
+
+        lastSearchText = searchText
+        
+        DispatchQueue.global(qos: .default).async {
+            self.readConnection.wrappedValue?.interrupt()
+            
+            let result: Result<[SectionModel], Error>? = Storage.shared.read { db -> Result<[SectionModel], Error> in
+                self.readConnection.mutate { $0 = db }
+                
+                do {
+                    let userPublicKey: String = getUserHexEncodedPublicKey(db)
+                    let contactsResults: [SessionThreadViewModel] = try SessionThreadViewModel // TODO: Remove group search results
+                        .contactsAndGroupsQuery(
+                            userPublicKey: userPublicKey,
+                            pattern: try SessionThreadViewModel.pattern(db, searchTerm: searchText),
+                            searchTerm: searchText
+                        )
+                        .fetchAll(db)
+                    let messageResults: [SessionThreadViewModel] = try SessionThreadViewModel
+                        .messagesQuery(
+                            userPublicKey: userPublicKey,
+                            pattern: try SessionThreadViewModel.pattern(db, searchTerm: searchText)
+                        )
+                        .fetchAll(db)
+                    
+                    return .success([
+                        ArraySection(model: .contacts, elements: contactsResults),
+                        ArraySection(model: .messages, elements: messageResults)
+                    ])
+                }
+                catch {
+                    // Don't log the 'interrupt' error as that's just the user typing too fast
+                    if (error as? DatabaseError)?.resultCode != DatabaseError.SQLITE_INTERRUPT {
+                        SNLog("[GlobalSearch] Failed to find results due to error: \(error)")
+                    }
+                    
+                    return .failure(error)
+                }
+            }
+            
+            DispatchQueue.main.async {
+                switch result {
+                    case .success(let sections):
+                        let hasResults: Bool = (
+                            !searchText.isEmpty &&
+                            (sections.map { $0.elements.count }.reduce(0, +) > 0)
+                        )
+                        
+                        self.termForCurrentSearchResultSet = searchText
+                        self.searchResultSet = [
+                            (hasResults ? nil : [
+                                ArraySection(
+                                    model: .noResults,
+                                    elements: [
+                                        SessionThreadViewModel(threadId: SessionThreadViewModel.invalidId)
+                                    ]
+                                )
+                            ]),
+                            (hasResults ? sections : nil)
+                        ]
+                        .compactMap { $0 }
+                        .flatMap { $0 }
+                        self.isLoading = false
+                        
+                    default: break
+                }
+            }
+        }
         
     }
 }
 
+struct SearchResultCell: View {
+    var searchText: String
+    var viewModel: SessionThreadViewModel
+    
+    var body: some View {
+        HStack(
+            alignment: .center,
+            spacing: Values.mediumSpacing
+        ) {
+            let size: ProfilePictureView.Size = .list
+            
+            ProfilePictureSwiftUI(
+                size: size,
+                publicKey: viewModel.threadId,
+                threadVariant: viewModel.threadVariant,
+                customImageData: viewModel.openGroupProfilePictureData,
+                profile: viewModel.profile,
+                additionalProfile: viewModel.additionalProfile
+            )
+            .frame(
+                width: size.viewSize,
+                height: size.viewSize,
+                alignment: .topLeading
+            )
+            
+            VStack(
+                alignment: .leading,
+                spacing: Values.verySmallSpacing
+            ) {
+                Text(viewModel.displayName)
+                    .bold()
+                    .font(.system(size: Values.mediumFontSize))
+                    .foregroundColor(themeColor: .textPrimary)
+            }
+        }
+    }
+}
+
 #Preview {
     GlobalSearchScreen()
 }
diff --git a/Session/Home/HomeVC.swift b/Session/Home/HomeVC.swift
index 26f9bdf35..d7b1abcd8 100644
--- a/Session/Home/HomeVC.swift
+++ b/Session/Home/HomeVC.swift
@@ -864,7 +864,7 @@ final class HomeVC: BaseVC, SessionUtilRespondingViewController, UITableViewData
             presentedVC.dismiss(animated: false, completion: nil)
         }
 //        let searchController = GlobalSearchViewController()
-        let searchController = SessionHostingViewController(rootView: GlobalSearchScreen())
+        let searchController = SessionHostingViewController(rootView: GlobalSearchScreen(), shouldHideNavigationBar: true)
         self.navigationController?.setViewControllers([ self, searchController ], animated: true)
     }
     
diff --git a/Session/Shared/SessionHostingViewController.swift b/Session/Shared/SessionHostingViewController.swift
index b8e24740d..9bedae05f 100644
--- a/Session/Shared/SessionHostingViewController.swift
+++ b/Session/Shared/SessionHostingViewController.swift
@@ -19,6 +19,7 @@ public class SessionHostingViewController<Content>: UIHostingController<Modified
     
     public var navigationBackground: ThemeValue { customizedNavigationBackground ?? .backgroundPrimary }
     private let customizedNavigationBackground: ThemeValue?
+    private let shouldHideNavigationBar: Bool
 
     lazy var navBarTitleLabel: UILabel = {
         let result = UILabel()
@@ -40,8 +41,9 @@ public class SessionHostingViewController<Content>: UIHostingController<Modified
         return result
     }()
     
-    public init(rootView:Content, customizedNavigationBackground: ThemeValue? = nil) {
+    public init(rootView:Content, customizedNavigationBackground: ThemeValue? = nil, shouldHideNavigationBar: Bool = false) {
         self.customizedNavigationBackground = customizedNavigationBackground
+        self.shouldHideNavigationBar = shouldHideNavigationBar
         let container = HostWrapper()
         let modified = rootView.environmentObject(container) as! ModifiedContent<Content, _EnvironmentKeyWritingModifier<HostWrapper?>>
         super.init(rootView: modified)
@@ -61,6 +63,20 @@ public class SessionHostingViewController<Content>: UIHostingController<Modified
         
         setNeedsStatusBarAppearanceUpdate()
     }
+    
+    public override func viewWillAppear(_ animated: Bool) {
+        if shouldHideNavigationBar {
+            self.navigationController?.setNavigationBarHidden(true, animated: animated)
+        }
+        super.viewWillAppear(animated)
+    }
+    
+    public override func viewWillDisappear(_ animated: Bool) {
+        if shouldHideNavigationBar {
+            self.navigationController?.setNavigationBarHidden(false, animated: animated)
+        }
+        super.viewWillDisappear(animated)
+    }
 
     internal func setNavBarTitle(_ title: String, customFontSize: CGFloat? = nil) {
         let container = UIView()
diff --git a/SessionMessagingKit/Utilities/ProfilePictureView+Convenience.swift b/SessionMessagingKit/Utilities/ProfilePictureView+Convenience.swift
index a5a6ecc17..796e12f1e 100644
--- a/SessionMessagingKit/Utilities/ProfilePictureView+Convenience.swift
+++ b/SessionMessagingKit/Utilities/ProfilePictureView+Convenience.swift
@@ -140,3 +140,33 @@ public extension ProfilePictureView {
         }
     }
 }
+
+public extension ProfilePictureSwiftUI {
+    init?(
+        size: ProfilePictureView.Size,
+        publicKey: String,
+        threadVariant: SessionThread.Variant,
+        customImageData: Data?,
+        profile: Profile?,
+        profileIcon: ProfilePictureView.ProfileIcon = .none,
+        additionalProfile: Profile? = nil,
+        additionalProfileIcon: ProfilePictureView.ProfileIcon = .none
+    ) {
+        let (info, additionalInfo) = ProfilePictureView.getProfilePictureInfo(
+            size: size,
+            publicKey: publicKey,
+            threadVariant: threadVariant,
+            customImageData: customImageData,
+            profile: profile,
+            profileIcon: profileIcon,
+            additionalProfile: additionalProfile,
+            additionalProfileIcon: additionalProfileIcon
+        )
+        
+        if let info = info {
+            self.init(size: size, info: info, additionalInfo: additionalInfo)
+        } else {
+            return nil
+        }
+    }
+}
diff --git a/SessionUIKit/Components/SwiftUI/SessionSearchBar.swift b/SessionUIKit/Components/SwiftUI/SessionSearchBar.swift
index 5f51360c6..050008c15 100644
--- a/SessionUIKit/Components/SwiftUI/SessionSearchBar.swift
+++ b/SessionUIKit/Components/SwiftUI/SessionSearchBar.swift
@@ -64,7 +64,7 @@ public struct SessionSearchBar: View {
                     .padding(.leading, Values.mediumSpacing)
             }
         }
-        .padding(.horizontal, Values.mediumSpacing)
+        .padding(.all, Values.mediumSpacing)
     }
 }