Prepare system extension support

2023-07-24 20:08:02 +08:00
parent 8a5c1a80d1
commit 12bf02070e
44 changed files with 1469 additions and 382 deletions
@@ -0,0 +1,61 @@
+import AppKit
+import ApplicationLibrary
+import Foundation
+import Libbox
+import Library
+
+open class ApplicationDelegate: NSObject, NSApplicationDelegate {
+    public func applicationDidFinishLaunching(_: Notification) {
+        NSLog("Here I stand")
+        // ServiceNotification.register() // Not work
+        let event = NSAppleEventManager.shared().currentAppleEvent
+        let launchedAsLogInItem =
+            event?.eventID == kAEOpenApplication &&
+            event?.paramDescriptor(forKeyword: keyAEPropData)?.enumCodeValue == keyAELaunchedAsLogInItem
+        if !launchedAsLogInItem || !SharedPreferences.showMenuBarExtra || !SharedPreferences.menuBarExtraInBackground {
+            NSApp.setActivationPolicy(.regular)
+            NSApp.activate(ignoringOtherApps: true)
+        } else {
+            NSApp.windows.first?.close()
+        }
+        Task.detached {
+            do {
+                if Variant.useSystemExtension {
+                    if await SystemExtension.isInstalled() {
+                        if let result = try await SystemExtension.install() {
+                            if result == .willCompleteAfterReboot {
+                                return
+                            }
+                        }
+                    }
+                }
+                try await self.postStart(launchedAsLogInItem)
+            } catch {
+                NSLog("application setup error: \(error.localizedDescription)")
+            }
+        }
+    }
+
+    private func postStart(_ launchedAsLogInItem: Bool) async throws {
+        try ProfileUpdateTask.setup()
+        if launchedAsLogInItem {
+            if SharedPreferences.startedByUser {
+                if let profile = try await ExtensionProfile.load() {
+                    try await profile.start()
+                }
+            }
+        }
+    }
+
+    public func applicationShouldTerminateAfterLastWindowClosed(_: NSApplication) -> Bool {
+        !SharedPreferences.menuBarExtraInBackground
+    }
+
+    public func applicationShouldHandleReopen(_: NSApplication, hasVisibleWindows flag: Bool) -> Bool {
+        if !flag, NSApp.activationPolicy() == .accessory {
+            NSApp.setActivationPolicy(.regular)
+            NSRunningApplication.runningApplications(withBundleIdentifier: "com.apple.dock").first?.activate()
+        }
+        return true
+    }
+}