feat(mac): auto-update checker and Plan pane button cleanup

iamtoruk · iamtoruk · commit bc92b49c1b20 · 2026-04-19T03:33:37.000-07:00
Remove the broken "Connect Claude" / "Reconnect Claude" buttons from
the Plan pane -- they opened a terminal session that did nothing useful
for already-logged-in users. Keep only the "Retry" button.

Add an auto-update checker that queries GitHub releases every 2 days in
the background. When a newer menubar build is available, an "Update"
pill appears in the header. Clicking it runs the existing installer
flow (download, replace, relaunch) with no manual steps.
diff --git a/mac/Sources/CodeBurnMenubar/CodeBurnApp.swift b/mac/Sources/CodeBurnMenubar/CodeBurnApp.swift
@@ -28,6 +28,7 @@ final class AppDelegate: NSObject, NSApplicationDelegate, NSPopoverDelegate {
     private var statusItem: NSStatusItem!
     private var popover: NSPopover!
     private let store = AppStore()
+    let updateChecker = UpdateChecker()
     private var refreshTask: Task<Void, Never>?
 
     func applicationDidFinishLaunching(_ notification: Notification) {
@@ -39,8 +40,7 @@ final class AppDelegate: NSObject, NSApplicationDelegate, NSPopoverDelegate {
         setupPopover()
         observeStore()
         startRefreshLoop()
-        // Subscription is fetched lazily when the user opens the Plan pill, so the macOS
-        // Keychain prompt never fires until the user explicitly asks for it.
+        Task { await updateChecker.checkIfNeeded() }
     }
 
     /// Loads the currency code persisted by `codeburn currency` so a relaunch picks up where
@@ -161,6 +161,7 @@ final class AppDelegate: NSObject, NSApplicationDelegate, NSPopoverDelegate {
 
         let content = MenuBarContent()
             .environment(store)
+            .environment(updateChecker)
             .frame(width: popoverWidth)
 
         popover.contentViewController = NSHostingController(rootView: content)
diff --git a/mac/Sources/CodeBurnMenubar/Data/UpdateChecker.swift b/mac/Sources/CodeBurnMenubar/Data/UpdateChecker.swift
@@ -0,0 +1,87 @@
+import Foundation
+import Observation
+
+private let releasesAPI = "https://api.github.com/repos/getagentseal/codeburn/releases/latest"
+private let checkIntervalSeconds: TimeInterval = 2 * 24 * 60 * 60
+private let lastCheckKey = "UpdateChecker.lastCheckDate"
+private let cachedVersionKey = "UpdateChecker.latestVersion"
+
+@MainActor
+@Observable
+final class UpdateChecker {
+    var latestVersion: String?
+    var isUpdating = false
+    var updateError: String?
+
+    var updateAvailable: Bool {
+        guard let latest = latestVersion else { return false }
+        let current = currentVersion
+        return !current.isEmpty && current != "dev" && latest != current
+    }
+
+    var currentVersion: String {
+        Bundle.main.infoDictionary?["CFBundleShortVersionString"] as? String ?? ""
+    }
+
+    func checkIfNeeded() async {
+        let lastCheck = UserDefaults.standard.double(forKey: lastCheckKey)
+        let now = Date().timeIntervalSince1970
+        if now - lastCheck < checkIntervalSeconds {
+            latestVersion = UserDefaults.standard.string(forKey: cachedVersionKey)
+            return
+        }
+        await check()
+    }
+
+    func check() async {
+        guard let url = URL(string: releasesAPI) else { return }
+        var request = URLRequest(url: url)
+        request.setValue("codeburn-menubar-updater", forHTTPHeaderField: "User-Agent")
+        request.setValue("application/vnd.github+json", forHTTPHeaderField: "Accept")
+
+        do {
+            let (data, _) = try await URLSession.shared.data(for: request)
+            let release = try JSONDecoder().decode(GitHubRelease.self, from: data)
+            guard let asset = release.assets.first(where: {
+                $0.name.hasPrefix("CodeBurnMenubar-") && $0.name.hasSuffix(".zip")
+            }) else { return }
+
+            let version = asset.name
+                .replacingOccurrences(of: "CodeBurnMenubar-", with: "")
+                .replacingOccurrences(of: ".zip", with: "")
+
+            latestVersion = version
+            UserDefaults.standard.set(Date().timeIntervalSince1970, forKey: lastCheckKey)
+            UserDefaults.standard.set(version, forKey: cachedVersionKey)
+        } catch {
+            NSLog("CodeBurn: update check failed: \(error)")
+        }
+    }
+
+    func performUpdate() {
+        isUpdating = true
+        updateError = nil
+
+        let process = CodeburnCLI.makeProcess(subcommand: ["menubar", "--force"])
+        process.standardOutput = FileHandle.nullDevice
+        process.standardError = FileHandle.nullDevice
+
+        do {
+            try process.run()
+        } catch {
+            isUpdating = false
+            updateError = error.localizedDescription
+            NSLog("CodeBurn: update spawn failed: \(error)")
+        }
+    }
+}
+
+private struct GitHubRelease: Decodable {
+    let tag_name: String
+    let assets: [GitHubAsset]
+}
+
+private struct GitHubAsset: Decodable {
+    let name: String
+    let browser_download_url: String
+}
diff --git a/mac/Sources/CodeBurnMenubar/Views/HeatmapSection.swift b/mac/Sources/CodeBurnMenubar/Views/HeatmapSection.swift
@@ -1041,7 +1041,6 @@ private struct PlanLoadingView: View {
 
 private struct PlanNoCredentialsView: View {
     @Environment(AppStore.self) private var store
-    @State private var showManualFallback = false
 
     var body: some View {
         VStack(spacing: 8) {
@@ -1051,32 +1050,17 @@ private struct PlanNoCredentialsView: View {
             Text("No Claude subscription connected")
                 .font(.system(size: 12, weight: .semibold))
                 .foregroundStyle(.primary)
-            if showManualFallback {
-                Text("Terminal.app isn't available. Open your terminal and run `claude login`, then click Retry.")
-                    .font(.system(size: 10.5))
-                    .foregroundStyle(.secondary)
-                    .multilineTextAlignment(.center)
-                    .frame(maxWidth: 280)
-            } else {
-                Text("Click Connect to sign in with Claude, then return here.")
-                    .font(.system(size: 10.5))
-                    .foregroundStyle(.secondary)
-                    .multilineTextAlignment(.center)
-                    .frame(maxWidth: 260)
-            }
-            HStack(spacing: 8) {
-                Button("Connect Claude") {
-                    if !TerminalLauncher.openClaudeLogin() { showManualFallback = true }
-                }
-                .controlSize(.small)
-                .buttonStyle(.borderedProminent)
-                .tint(Theme.brandAccent)
-                Button("Retry") {
-                    Task { await store.refreshSubscription() }
-                }
-                .controlSize(.small)
-                .buttonStyle(.bordered)
+            Text("Sign in with Claude Code, then click Retry.")
+                .font(.system(size: 10.5))
+                .foregroundStyle(.secondary)
+                .multilineTextAlignment(.center)
+                .frame(maxWidth: 260)
+            Button("Retry") {
+                Task { await store.refreshSubscription() }
             }
+            .controlSize(.small)
+            .buttonStyle(.borderedProminent)
+            .tint(Theme.brandAccent)
         }
         .frame(maxWidth: .infinity)
         .padding(.vertical, 14)
@@ -1086,7 +1070,6 @@ private struct PlanNoCredentialsView: View {
 private struct PlanFailedView: View {
     @Environment(AppStore.self) private var store
     let error: String?
-    @State private var showManualFallback = false
 
     var body: some View {
         VStack(spacing: 8) {
@@ -1096,33 +1079,20 @@ private struct PlanFailedView: View {
             Text("Couldn't load plan data")
                 .font(.system(size: 12, weight: .semibold))
                 .foregroundStyle(.primary)
-            if showManualFallback {
-                Text("Terminal.app isn't available. Open your terminal and run `claude login`, then click Retry.")
-                    .font(.system(size: 10.5))
-                    .foregroundStyle(.secondary)
-                    .multilineTextAlignment(.center)
-                    .frame(maxWidth: 280)
-            } else if let error {
+            if let error {
                 Text(error)
                     .font(.system(size: 10))
                     .foregroundStyle(.tertiary)
                     .multilineTextAlignment(.center)
                     .frame(maxWidth: 280)
                     .lineLimit(3)
             }
-            HStack(spacing: 8) {
-                Button("Reconnect Claude") {
-                    if !TerminalLauncher.openClaudeLogin() { showManualFallback = true }
-                }
-                .controlSize(.small)
-                .buttonStyle(.borderedProminent)
-                .tint(Theme.brandAccent)
-                Button("Retry") {
-                    Task { await store.refreshSubscription() }
-                }
-                .controlSize(.small)
-                .buttonStyle(.bordered)
+            Button("Retry") {
+                Task { await store.refreshSubscription() }
             }
+            .controlSize(.small)
+            .buttonStyle(.borderedProminent)
+            .tint(Theme.brandAccent)
         }
         .frame(maxWidth: .infinity)
         .padding(.vertical, 14)
diff --git a/mac/Sources/CodeBurnMenubar/Views/MenuBarContent.swift b/mac/Sources/CodeBurnMenubar/Views/MenuBarContent.swift
@@ -183,25 +183,61 @@ private struct BurnFlame: View {
 }
 
 private struct Header: View {
+    @Environment(UpdateChecker.self) private var updateChecker
+
     var body: some View {
-        VStack(alignment: .leading, spacing: 1) {
-            (
-                Text("Code").foregroundStyle(.primary)
-                + Text("Burn").foregroundStyle(Theme.brandAccent)
-            )
-            .font(.system(size: 13, weight: .semibold))
-            .tracking(-0.15)
-            Text("AI Coding Cost Tracker")
-                .font(.system(size: 10.5))
-                .foregroundStyle(.secondary)
+        HStack {
+            VStack(alignment: .leading, spacing: 1) {
+                (
+                    Text("Code").foregroundStyle(.primary)
+                    + Text("Burn").foregroundStyle(Theme.brandAccent)
+                )
+                .font(.system(size: 13, weight: .semibold))
+                .tracking(-0.15)
+                Text("AI Coding Cost Tracker")
+                    .font(.system(size: 10.5))
+                    .foregroundStyle(.secondary)
+            }
+            Spacer()
+            if updateChecker.updateAvailable {
+                UpdateBadge()
+            }
         }
-        .frame(maxWidth: .infinity, alignment: .leading)
         .padding(.horizontal, 14)
         .padding(.top, 10)
         .padding(.bottom, 8)
     }
 }
 
+private struct UpdateBadge: View {
+    @Environment(UpdateChecker.self) private var updateChecker
+
+    var body: some View {
+        Button {
+            updateChecker.performUpdate()
+        } label: {
+            HStack(spacing: 4) {
+                if updateChecker.isUpdating {
+                    ProgressView()
+                        .controlSize(.mini)
+                        .scaleEffect(0.7)
+                } else {
+                    Image(systemName: "arrow.down.circle.fill")
+                        .font(.system(size: 10))
+                }
+                Text(updateChecker.isUpdating ? "Updating..." : "Update")
+                    .font(.system(size: 10, weight: .medium))
+            }
+            .padding(.horizontal, 8)
+            .padding(.vertical, 4)
+        }
+        .buttonStyle(.borderedProminent)
+        .tint(Theme.brandAccent)
+        .controlSize(.mini)
+        .disabled(updateChecker.isUpdating)
+    }
+}
+
 struct FlameMark: View {
     var body: some View {
         ZStack {
