Data Storage

private let statsKey = "typing_stats_daily"
private let hourlyKey = "typing_stats_hourly"
private let minuteKey = "typing_stats_minute"
private let appStatsKey = "typing_stats_apps"
private let projectStatsKey = "typing_stats_projects"
private let appBundleMappingKey = "typing_app_bundles"
private let notifiedKey = "typing_notified_milestones"

@AppStorage("wakatime_api_key") var wakaTimeApiKey: String = ""
@AppStorage("daily_goal") private var dailyGoal = 5000
@AppStorage("app_theme") private var appTheme = 0
@AppStorage("has_completed_onboarding") private var hasCompletedOnboarding = false

@Published var dailyStats: [String: Int] = [:]

{
  "2026-03-01": 12543,
  "2026-03-02": 8932,
  "2026-03-03": 4521
}

@Published var hourlyStats: [String: Int] = [:]

{
  "2026-03-03-09": 543,
  "2026-03-03-10": 1234,
  "2026-03-03-11": 892
}

@Published var minuteStats: [String: Int] = [:]

if minuteStats.count > 120 {
    let keys = minuteStats.keys.sorted().prefix(minuteStats.count - 120)
    for key in keys { minuteStats.removeValue(forKey: key) }
}

@Published var appStats: [String: Int] = [:]

{
  "Xcode": 5432,
  "Visual Studio Code": 3210,
  "Cursor": 2100
}

@Published var projectStats: [String: Int] = [:]

{
  "typesteps": 8543,
  "my-web-app": 3210
}

@Published var appBundleMapping: [String: String] = [:]

{
  "Xcode": "com.apple.dt.xcode",
  "Visual Studio Code": "com.visualstudio.code",
  "Cursor": "com.todesktop.230313m78xv9jbu"
}

func incrementCount(for date: Date = Date(), 
                   appName: String? = nil, 
                   bundleId: String? = nil, 
                   projectName: String? = nil) {
    let dayString = dateFormatter.string(from: date)
    let hourString = hourlyFormatter.string(from: date)
    let minuteString = minuteFormatter.string(from: date)
    
    dailyStats[dayString] = (dailyStats[dayString] ?? 0) + 1
    hourlyStats[hourString] = (hourlyStats[hourString] ?? 0) + 1
    minuteStats[minuteString] = (minuteStats[minuteString] ?? 0) + 1
    
    if let appName = appName {
        appStats[appName] = (appStats[appName] ?? 0) + 1
        if let bundleId = bundleId { 
            appBundleMapping[appName] = bundleId 
        }
    }
    
    if let projectName = projectName { 
        projectStats[projectName] = (projectStats[projectName] ?? 0) + 1 
    }
    
    // Prune old minute data
    if minuteStats.count > 120 {
        let keys = minuteStats.keys.sorted().prefix(minuteStats.count - 120)
        for key in keys { minuteStats.removeValue(forKey: key) }
    }
    
    checkMilestones(count: dailyStats[dayString] ?? 0, day: dayString)
    saveStats()
}

private func saveStats() {
    UserDefaults.standard.set(dailyStats, forKey: statsKey)
    UserDefaults.standard.set(hourlyStats, forKey: hourlyKey)
    UserDefaults.standard.set(minuteStats, forKey: minuteKey)
    UserDefaults.standard.set(appStats, forKey: appStatsKey)
    UserDefaults.standard.set(projectStats, forKey: projectStatsKey)
    UserDefaults.standard.set(appBundleMapping, forKey: appBundleMappingKey)
}

func loadStats() {
    if let daily = UserDefaults.standard.dictionary(forKey: statsKey) as? [String: Int] { 
        self.dailyStats = daily 
    }
    if let hourly = UserDefaults.standard.dictionary(forKey: hourlyKey) as? [String: Int] { 
        self.hourlyStats = hourly 
    }
    if let minute = UserDefaults.standard.dictionary(forKey: minuteKey) as? [String: Int] { 
        self.minuteStats = minute 
    }
    if let apps = UserDefaults.standard.dictionary(forKey: appStatsKey) as? [String: Int] { 
        self.appStats = apps 
    }
    if let projects = UserDefaults.standard.dictionary(forKey: projectStatsKey) as? [String: Int] { 
        self.projectStats = projects 
    }
    if let mapping = UserDefaults.standard.dictionary(forKey: appBundleMappingKey) as? [String: String] { 
        self.appBundleMapping = mapping 
    }
}

struct BackupData: Codable {
    let dailyStats: [String: Int]
    let hourlyStats: [String: Int]
    let minuteStats: [String: Int]
    let appStats: [String: Int]
    let projectStats: [String: Int]
    let appBundleMapping: [String: String]
    let notifiedMilestones: [String: [Double]]
    let wakaTimeApiKey: String
    let dailyGoal: Int
}

func exportData() -> Data? {
    let notified = UserDefaults.standard.dictionary(forKey: notifiedKey) as? [String: [Double]] ?? [:]
    let goal = UserDefaults.standard.integer(forKey: "daily_goal")
    
    let backup = BackupData(
        dailyStats: dailyStats,
        hourlyStats: hourlyStats,
        minuteStats: minuteStats,
        appStats: appStats,
        projectStats: projectStats,
        appBundleMapping: appBundleMapping,
        notifiedMilestones: notified,
        wakaTimeApiKey: wakaTimeApiKey,
        dailyGoal: goal
    )
    
    let encoder = JSONEncoder()
    encoder.outputFormatting = .prettyPrinted
    return try? encoder.encode(backup)
}

func importData(from data: Data) -> Bool {
    guard let backup = try? JSONDecoder().decode(BackupData.self, from: data) else { 
        return false 
    }
    
    DispatchQueue.main.async {
        self.dailyStats = backup.dailyStats
        self.hourlyStats = backup.hourlyStats
        self.minuteStats = backup.minuteStats
        self.appStats = backup.appStats
        self.projectStats = backup.projectStats
        self.appBundleMapping = backup.appBundleMapping
        self.wakaTimeApiKey = backup.wakaTimeApiKey
        
        UserDefaults.standard.set(backup.dailyGoal, forKey: "daily_goal")
        UserDefaults.standard.set(backup.notifiedMilestones, forKey: self.notifiedKey)
        
        self.saveStats()
    }
    return true
}