fix: space numbers are correctly removed if single space

lwouis · Dec 30, 2019 · 45ad43f · 45ad43f
1 parent 1e210c3
commit 45ad43f
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/alt-tab-macos/logic/TrackedWindows.swift b/alt-tab-macos/logic/TrackedWindows.swift
@@ -17,13 +17,24 @@ class TrackedWindows {
         list.removeAll()
         focusedWindowIndex = 0
         let spaces = Spaces.allIdsAndIndexes()
-        Spaces.singleSpace = spaces.count == 1
         Spaces.currentSpaceId = CGSManagedDisplayGetCurrentSpace(cgsMainConnectionId, Screen.mainUuid())
         Spaces.currentSpaceIndex = spaces.first { $0.0 == Spaces.currentSpaceId }!.1
         filterAndAddToList(mapWindowsWithRankAndSpace(spaces))
+        isSingleSpace()
         sortList()
     }
 
+    private class func isSingleSpace() {
+        let firstSpaceIndex = list[0].spaceIndex
+        for window in list {
+            if window.spaceIndex != firstSpaceIndex {
+                Spaces.singleSpace = false
+                return
+            }
+        }
+        Spaces.singleSpace = true
+    }
+
     private static func mapWindowsWithRankAndSpace(_ spaces: [(CGSSpaceID, SpaceIndex)]) -> [CGWindowID: (CGSSpaceID, SpaceIndex, WindowRank)] {
         var windowSpaceMap: [CGWindowID: (CGSSpaceID, SpaceIndex, WindowRank?)] = [:]
         for (spaceId, spaceIndex) in spaces {