Application Architecture

@main
struct TypeStepsApp: App {
    @NSApplicationDelegateAdaptor(AppDelegate.self) var appDelegate
    @StateObject private var storage = StorageManager.shared
    @StateObject private var listener = KeystrokeListener.shared
    @State private var showOnboarding = false
    // ...
}

class StorageManager: ObservableObject {
    static let shared = StorageManager()
    
    @Published var dailyStats: [String: Int] = [:]
    @Published var hourlyStats: [String: Int] = [:]
    @Published var minuteStats: [String: Int] = [:]
    @Published var appStats: [String: Int] = [:]
    @Published var projectStats: [String: Int] = [:]
    @Published var appBundleMapping: [String: String] = [:]
    // ...
}

class KeystrokeListener: ObservableObject {
    static let shared = KeystrokeListener()
    
    @Published var isAuthorized: Bool = false
    @Published var isPaused: Bool = false
    private var eventMonitor: Any?
    // ...
}

class WakaTimeManager: ObservableObject {
    static let shared = WakaTimeManager()
    
    @Published var totalMinutesToday: Double = 0
    @Published var isFetching: Bool = false
    @Published var lastError: String?
    // ...
}

class AppDelegate: NSObject, NSApplicationDelegate {
    func applicationDidFinishLaunching(_ notification: Notification) {
        NSApp.setActivationPolicy(.regular)
        UNUserNotificationCenter.current().requestAuthorization(options: [.alert, .sound]) { _, _ in }
        
        DispatchQueue.main.asyncAfter(deadline: .now() + 0.5) {
            self.checkAndStartTracking()
        }
    }
    
    private func checkAndStartTracking() {
        let hasCompletedOnboarding = UserDefaults.standard.bool(forKey: "has_completed_onboarding")
        
        if hasCompletedOnboarding {
            if KeystrokeListener.shared.checkPermissionsSilently() {
                KeystrokeListener.shared.startListening()
            }
        } else {
            NotificationCenter.default.post(name: .showOnboarding, object: nil)
        }
    }
}

extension Notification.Name {
    static let showOnboarding = Notification.Name("showOnboarding")
    static let onboardingCompleted = Notification.Name("onboardingCompleted")
    static let accessibilityGranted = Notification.Name("accessibilityGranted")
}

MenuBarExtra {
    Text("Today: \(storage.getCount())")
    Text("This Week: \(storage.getWeeklyTotal())")
    Text("This Month: \(storage.getMonthlyTotal())")
    
    Divider()
    
    Button(listener.isPaused ? "Resume Tracking" : "Pause Tracking") {
        listener.isPaused.toggle()
    }
    // ...
} label: {
    HStack {
        Image(systemName: listener.isPaused ? "keyboard.badge.ellipsis" : "keyboard")
        Text("\(storage.getCount())")
    }
}

Window("TypeSteps", id: "dashboard") {
    ZStack {
        DashboardView()
            .preferredColorScheme(appTheme == 0 ? nil : (appTheme == 1 ? .light : .dark))
        
        if showOnboarding {
            // Onboarding overlay
        }
    }
}
.windowStyle(.hiddenTitleBar)
.defaultSize(width: 1200, height: 850)

struct AppTheme: Identifiable, Equatable {
    let id: Int
    let name: String
    let mainBg: Color
    let secondaryBg: Color
    let border: Color
    let text: Color
    let secondaryText: Color
    let accent: Color
}

@AppStorage("app_theme") private var appTheme = 0

struct ActivityPoint: Identifiable, Equatable {
    let id = UUID()
    let label: String
    let count: Int
}

enum AppCategory: String, CaseIterable {
    case code = "Code"
    case communicate = "Communicate"
    case create = "Create"
    case browsing = "Browsing"
    case utility = "Utility"
    case other = "Other"
}

private func updateLoginItem(enabled: Bool) {
    do {
        if enabled {
            try SMAppService.mainApp.register()
        } else {
            try SMAppService.mainApp.unregister()
        }
    } catch {
        print("Failed to toggle login item: \(error)")
    }
}