Prototype "Open at Login"

Aware/AppDelegate.swift

@@ -7,6 +7,7 @@
 //
 
 import Cocoa
+import ServiceManagement
 
 @NSApplicationMain
 class AppDelegate: NSObject, NSApplicationDelegate {
@@ -29,12 +30,27 @@ class AppDelegate: NSObject, NSApplicationDelegate {
         return defaultsValue ?? type(of: self).defaultUserIdleSeconds
     }
 
+    let launchAtLoginMenuItem = NSMenuItem(
+        title: "Open at Login",
+        action: #selector(openAtLogin),
+        keyEquivalent: ""
+    )
+
     let statusItem = NSStatusBar.system.statusItem(withLength: NSStatusItem.variableLength)
 
     func applicationDidFinishLaunching(_: Notification) {
         userIdleSeconds = readUserIdleSeconds()
 
+        if SMAppService.mainApp.status == .enabled {
+            launchAtLoginMenuItem.title = "Open at Login"
+            launchAtLoginMenuItem.action = #selector(openAtLogin)
+        } else {
+            launchAtLoginMenuItem.title = "Don't Open at Login"
+            launchAtLoginMenuItem.action = #selector(removeOpenAtLogin)
+        }
+
         let menu = NSMenu()
+        menu.addItem(launchAtLoginMenuItem)
         menu.addItem(
             NSMenuItem(
                 title: "Quit Aware",
@@ -52,6 +68,18 @@ class AppDelegate: NSObject, NSApplicationDelegate {
         notificationCenter.addObserver(forName: NSWorkspace.didWakeNotification, object: nil, queue: nil) { _ in self.resetTimer() }
     }
 
+    @objc func openAtLogin() {
+        try! SMAppService.mainApp.register()
+        launchAtLoginMenuItem.title = "Don't Open at Login"
+        launchAtLoginMenuItem.action = #selector(removeOpenAtLogin)
+    }
+
+    @objc func removeOpenAtLogin() {
+        try! SMAppService.mainApp.unregister()
+        launchAtLoginMenuItem.title = "Open at Login"
+        launchAtLoginMenuItem.action = #selector(openAtLogin)
+    }
+
     func resetTimer() {
         timerStart = Date()
         updateButton()