Full-fledged date picker

LunarBar-app · Jan 8, 2024 · 407db8a · 407db8a
1 parent 6c6bf33
commit 407db8a
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 2 deletions.
diff --git a/LunarBarMac/Modules/Sources/AppKitExtensions/NSControl+Extension.swift b/LunarBarMac/Modules/Sources/AppKitExtensions/NSControl+Extension.swift
@@ -31,6 +31,7 @@ public extension ClosureActionable {
 
 extension NSButton: ClosureActionable {}
 extension NSMenuItem: ClosureActionable {}
+extension NSDatePicker: ClosureActionable {}
 
 // MARK: - Private
 

diff --git a/LunarBarMac/Sources/Main/AppMainVC+Menu.swift b/LunarBarMac/Sources/Main/AppMainVC+Menu.swift
@@ -6,6 +6,7 @@
 //
 
 import AppKit
+import AppKitExtensions
 import LunarBarKit
 import ServiceManagement
 
@@ -54,8 +55,8 @@ private extension AppMainVC {
     let menu = NSMenu()
     let current = Calendar.solar.year(from: monthDate)
 
-    // Generate 20 years before and after the current year
-    for year in (current - 10)...(current + 10) {
+    // Quick picker that supports 12 years around the current year
+    for year in (current - 6)...(current + 6) {
       let item = menu.addItem(withTitle: String(year))
       item.submenu = NSMenu()
 
@@ -75,6 +76,41 @@ private extension AppMainVC {
       }
     }
 
+    menu.addSeparator()
+
+    // Full-fledged picker that supports any year
+    menu.addItem({ [weak self] in
+      let item = NSMenuItem()
+      let picker = NSDatePicker()
+      picker.isBezeled = false
+      picker.isBordered = false
+      picker.datePickerStyle = .textFieldAndStepper
+      picker.datePickerElements = .yearMonth
+      picker.translatesAutoresizingMaskIntoConstraints = false
+      picker.dateValue = self?.monthDate ?? .now
+      picker.sizeToFit()
+
+      picker.addAction { [weak picker] in
+        guard let date = picker?.dateValue else {
+          return
+        }
+
+        self?.updateCalendar(targetDate: date)
+      }
+
+      let wrapper = NSView(frame: CGRect(origin: .zero, size: CGSize(width: 0, height: picker.frame.height)))
+      wrapper.autoresizingMask = .width
+      wrapper.addSubview(picker)
+
+      NSLayoutConstraint.activate([
+        picker.centerXAnchor.constraint(equalTo: wrapper.centerXAnchor),
+        picker.centerYAnchor.constraint(equalTo: wrapper.centerYAnchor),
+      ])
+
+      item.view = wrapper
+      return item
+    }())
+
     let item = NSMenuItem(title: Localized.UI.menuTitleGotoDate)
     item.submenu = menu
     return item