add update checker (#2)

* update plist version with tag

* check for updates

Author: karansangha

.github/workflows/release.yml

@@ -21,6 +21,16 @@ jobs:
         with:
           xcode-version: latest-stable
 
+      - name: Set App Version
+        run: |
+          VERSION=${GITHUB_REF#refs/tags/v}
+          echo "Setting version to $VERSION"
+          plutil -replace CFBundleShortVersionString -string "$VERSION" insomnia/Info.plist
+          plutil -replace CFBundleVersion -string "$VERSION" insomnia/Info.plist
+
+      - name: Run Tests
+        run: swift test
+
       - name: Build App and Create DMG
         run: |
           chmod +x install_app.sh

Package.swift

@@ -14,7 +14,7 @@ let package = Package(
         .target(
             name: "InsomniaCore",
             path: "insomnia",
-            exclude: ["InsomniaApp.swift", "Info.plist", "insomnia.entitlements", "Assets.xcassets"]
+            exclude: ["InsomniaApp.swift", "UpdateChecker.swift", "Info.plist", "insomnia.entitlements", "Assets.xcassets"]
         ),
         .executableTarget(
             name: "Insomnia",

insomnia/Info.plist

@@ -17,9 +17,9 @@
 	<key>CFBundlePackageType</key>
 	<string>APPL</string>
 	<key>CFBundleShortVersionString</key>
-	<string>1.1</string>
+	<string>2.0.1</string>
 	<key>CFBundleVersion</key>
-	<string>1</string>
+	<string>201</string>
 	<key>LSMinimumSystemVersion</key>
 	<string>$(MACOSX_DEPLOYMENT_TARGET)</string>
 	<key>LSUIElement</key>

insomnia/InsomniaApp.swift

@@ -4,9 +4,18 @@ import InsomniaCore
 @main
 struct InsomniaApp: App {
     @StateObject private var sleepManager = SleepManager()
+    @StateObject private var updateChecker = UpdateChecker()
 
     var body: some Scene {
         MenuBarExtra {
+            // Update Banner
+            if updateChecker.updateAvailable, let url = updateChecker.releaseURL {
+                Button("Update Available: \(updateChecker.latestVersion)") {
+                    NSWorkspace.shared.open(url)
+                }
+                Divider()
+            }
+
             // Unconditional keep awake
             Button {
                 sleepManager.toggle()
@@ -59,5 +68,17 @@ struct InsomniaApp: App {
         } label: {
             Image(sleepManager.iconName)
         }
+        .menuBarExtraStyle(.menu) // Ensure standard menu style
+    }
+    
+    init() {
+        // Trigger update check on launch
+        let checker = UpdateChecker()
+        checker.checkForUpdates()
+        _updateChecker = StateObject(wrappedValue: checker)
+        
+        // Initialize Core
+        let manager = SleepManager()
+        _sleepManager = StateObject(wrappedValue: manager)
     }
 }

insomnia/UpdateChecker.swift

@@ -0,0 +1,42 @@
+import Foundation
+import SwiftUI
+
+class UpdateChecker: ObservableObject {
+    @Published var updateAvailable: Bool = false
+    @Published var latestVersion: String = ""
+    @Published var releaseURL: URL? = nil
+    
+    private var currentVersion: String {
+        return Bundle.main.infoDictionary?["CFBundleShortVersionString"] as? String ?? "1.0.0"
+    }
+    private let repoOwner = "karansangha"
+    private let repoName = "Insomnia"
+    
+    func checkForUpdates() {
+        guard let url = URL(string: "https://api.github.com/repos/\(repoOwner)/\(repoName)/releases/latest") else { return }
+        
+        URLSession.shared.dataTask(with: url) { [weak self] data, response, error in
+            guard let data = data, error == nil else { return }
+            
+            do {
+                if let json = try JSONSerialization.jsonObject(with: data, options: []) as? [String: Any],
+                   let tagName = json["tag_name"] as? String {
+                    
+                    let cleanTag = tagName.replacingOccurrences(of: "v", with: "")
+                    
+                    DispatchQueue.main.async {
+                        if cleanTag.compare(self?.currentVersion ?? "", options: .numeric) == .orderedDescending {
+                            self?.latestVersion = tagName
+                            self?.updateAvailable = true
+                            if let htmlUrl = json["html_url"] as? String {
+                                self?.releaseURL = URL(string: htmlUrl)
+                            }
+                        }
+                    }
+                }
+            } catch {
+                print("Failed to check for updates: \(error)")
+            }
+        }.resume()
+    }
+}