features/rearrange-columns: if only one column remains, might as well…

… close the sheet
chucker · Apr 6, 2024 · a52a6c0 · a52a6c0
1 parent efa6fee
commit a52a6c0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Mastonaut/Features/ArrangeColumns/ArrangeColumnsWindowController.swift b/Mastonaut/Features/ArrangeColumns/ArrangeColumnsWindowController.swift
@@ -95,9 +95,14 @@ class ArrangeColumnsWindowController: NSWindowController, NSCollectionViewDelega
 
         return true
     }
-    
+
     func reloadData() {
         collectionView.reloadData()
+
+        // if only one column remains, there's nothing left for the user to do in the sheet
+        if let viewControllers = getColumnViewControllers?(), viewControllers.count == 1 {
+            close()
+        }
     }
 
     @IBAction func done(_ sender: Any) {