Better onboarding!

n0shake · Jul 21, 2021 · aa0259f · aa0259f
1 parent 0f6107e
commit aa0259f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/Clocker/Onboarding/OnboardingSearchController.swift b/Clocker/Onboarding/OnboardingSearchController.swift
@@ -84,8 +84,9 @@ class OnboardingSearchController: NSViewController {
             let selectedType = searchResultsDataSource.placeForRow(resultsTableView.selectedRow)
             switch selectedType {
             case .city:
-                let filteredGoogleResult = searchResultsDataSource.retrieveFilteredResultFromGoogleAPI(resultsTableView.selectedRow)
-                addTimezoneToDefaults(filteredGoogleResult!)
+                if let filteredGoogleResult = searchResultsDataSource.retrieveFilteredResultFromGoogleAPI(resultsTableView.selectedRow) {
+                    addTimezoneToDefaults(filteredGoogleResult)
+                }
                 return
             case .timezone:
                 cleanupAfterInstallingTimezone()
@@ -422,6 +423,8 @@ class OnboardingSearchController: NSViewController {
 
     private func resetSearchView() {
         searchResultsDataSource.cleanupFilterArray()
+        searchResultsDataSource.timezoneFilteredArray = []
+        searchResultsDataSource.calculateChangesets()
         resultsTableView.reloadData()
         searchBar.stringValue = CLEmptyString
         searchBar.placeholderString = "Press Enter to Search"
@@ -513,4 +516,8 @@ extension OnboardingSearchController: NSSearchFieldDelegate {
         // return true if the action was handled; otherwise false
         return false
     }
+
+    func searchFieldDidEndSearching(_ sender: NSSearchField) {
+        self.search(sender)
+    }
 }
diff --git a/Clocker/Preferences/General/SearchDataSource.swift b/Clocker/Preferences/General/SearchDataSource.swift
@@ -75,6 +75,10 @@ class SearchDataSource: NSObject {
     }
 
     func retrieveFilteredResultFromGoogleAPI(_ index: Int) -> TimezoneData? {
+        if index >= filteredArray.count {
+            return nil
+        }
+
         return filteredArray[index % filteredArray.count]
     }