OpenAppLock/OpenAppLockTests/SchedulingTests.swift

//
//  SchedulingTests.swift
//  OpenAppLockTests
//

import Foundation
import SwiftData
import Testing

@testable import OpenAppLock

private func freshDefaults() -> UserDefaults {
    let name = "scheduling-tests-\(UUID().uuidString)"
    let defaults = UserDefaults(suiteName: name)!
    defaults.removePersistentDomain(forName: name)
    return defaults
}

@MainActor
@Suite("Rule snapshots")
struct RuleSnapshotTests {
    @Test("Snapshots round-trip through the shared store")
    func storeRoundTrip() throws {
        let store = RuleSnapshotStore(defaults: freshDefaults())
        let snapshot = RuleSnapshot(
            id: UUID(), name: "Time Keeper", kindRaw: "timeLimit", isEnabled: true,
            hardMode: false, blockAdultContent: false, selectionModeRaw: "block",
            selectionData: Data([1]), dayNumbers: [2, 3], startMinutes: 540, endMinutes: 1020,
            dailyLimitMinutes: 45, maxOpens: 5, pausedUntil: nil
        )
        store.save([snapshot])
        #expect(store.load() == [snapshot])
        #expect(store.snapshot(for: snapshot.id) == snapshot)
        #expect(store.snapshot(for: snapshot.id)?.startMinutes == 540)
        #expect(store.snapshot(for: snapshot.id)?.endMinutes == 1020)
        #expect(store.snapshot(for: UUID()) == nil)
    }

    @Test("Snapshots mirror rules and their app lists")
    func mirrorsRule() throws {
        let context = try makeInMemoryContext()
        let list = AppList(name: "Distractions", selectionData: Data([9]), selectionCount: 1)
        let rule = BlockingRule(
            name: "Gate Keeper", kind: .openLimit, days: Weekday.weekends, maxOpens: 3)
        context.insert(list)
        context.insert(rule)
        rule.appList = list

        let snapshot = RuleSnapshot(rule: rule)
        #expect(snapshot.id == rule.id)
        #expect(snapshot.kind == .openLimit)
        #expect(snapshot.selectionData == Data([9]))
        #expect(snapshot.days == Weekday.weekends)
        #expect(snapshot.maxOpens == 3)
        #expect(snapshot.startMinutes == rule.startMinutes)
        #expect(snapshot.endMinutes == rule.endMinutes)
    }

    @Test("Snapshots saved before the window fields decode with a zeroed window")
    func decodesLegacySnapshotWithoutWindowFields() throws {
        // A blob shaped like a pre-schedule-window snapshot: no start/endMinutes.
        let id = UUID()
        let legacy = """
            [{"id":"\(id.uuidString)","name":"Old","kindRaw":"timeLimit","isEnabled":true,\
            "hardMode":false,"blockAdultContent":false,"selectionModeRaw":"block",\
            "dayNumbers":[2,3],"dailyLimitMinutes":45,"maxOpens":5}]
            """
        let defaults = freshDefaults()
        defaults.set(Data(legacy.utf8), forKey: "ruleSnapshots")
        let store = RuleSnapshotStore(defaults: defaults)

        let loaded = store.load()
        #expect(loaded.count == 1)
        #expect(loaded.first?.id == id)
        #expect(loaded.first?.startMinutes == 0)
        #expect(loaded.first?.endMinutes == 0)
        #expect(loaded.first?.dailyLimitMinutes == 45)
    }
}

@MainActor
@Suite("Monitoring plan")
struct MonitoringPlanTests {
    @Test("Activity names round-trip rule IDs")
    func nameRoundTrip() {
        let id = UUID()
        #expect(
            MonitoringPlan.ruleID(
                fromDailyActivityName: MonitoringPlan.dailyActivityName(for: id)) == id)
        #expect(
            MonitoringPlan.ruleID(
                fromSessionActivityName: MonitoringPlan.sessionActivityName(for: id)) == id)
        #expect(MonitoringPlan.ruleID(fromDailyActivityName: "garbage") == nil)
        #expect(
            MonitoringPlan.ruleID(
                fromSessionActivityName: MonitoringPlan.dailyActivityName(for: id)) == nil)
    }

    @Test("Schedule-window activity names round-trip rule IDs")
    func scheduleWindowNameRoundTrip() {
        let id = UUID()
        let primary = MonitoringPlan.scheduleWindowName(for: id)
        let late = MonitoringPlan.scheduleWindowLateName(for: id)
        #expect(primary != late)
        #expect(MonitoringPlan.ruleID(fromScheduleWindowName: primary) == id)
        #expect(MonitoringPlan.ruleID(fromScheduleWindowName: late) == id)
        // Daily / session / garbage names are not schedule-window names.
        #expect(MonitoringPlan.ruleID(fromScheduleWindowName: MonitoringPlan.dailyActivityName(for: id)) == nil)
        #expect(MonitoringPlan.ruleID(fromScheduleWindowName: "garbage") == nil)
        // Schedule-window names are not mistaken for daily activities.
        #expect(MonitoringPlan.ruleID(fromDailyActivityName: primary) == nil)
        #expect(MonitoringPlan.ruleID(fromDailyActivityName: late) == nil)
    }

    @Test("Minute checkpoints cover every minute up to the budget")
    func minuteEvents() {
        let events = MonitoringPlan.minuteEvents(forLimit: 45)
        #expect(events.count == 45)
        #expect(events[MonitoringPlan.minuteEventName(for: 1)] == 1)
        #expect(events[MonitoringPlan.minuteEventName(for: 45)] == 45)
        #expect(MonitoringPlan.minutes(fromEventName: MonitoringPlan.minuteEventName(for: 17)) == 17)
        #expect(MonitoringPlan.minutes(fromEventName: "nope") == nil)
    }
}

@MainActor
@Suite("Rule scheduler → DeviceActivity")
struct RuleSchedulerTests {
    private func makeScheduler() -> (RuleScheduler, MockActivityMonitor, RuleSnapshotStore) {
        let monitor = MockActivityMonitor()
        let store = RuleSnapshotStore(defaults: freshDefaults())
        return (RuleScheduler(monitor: monitor, snapshots: store), monitor, store)
    }

    private func limitRule(kind: RuleKind, name: String) throws -> BlockingRule {
        let context = try makeInMemoryContext()
        let list = AppList(name: "Apps", selectionData: Data([1]), selectionCount: 1)
        let rule = BlockingRule(name: name, kind: kind, days: Weekday.everyDay)
        context.insert(list)
        context.insert(rule)
        rule.appList = list
        return rule
    }

    private func scheduleRule(
        name: String, start: Int, end: Int, days: Set<Weekday> = Weekday.everyDay,
        withApps: Bool = true
    ) throws -> BlockingRule {
        let context = try makeInMemoryContext()
        let rule = BlockingRule(
            name: name, kind: .schedule, days: days, startMinutes: start, endMinutes: end)
        context.insert(rule)
        if withApps {
            let list = AppList(name: "Apps", selectionData: Data([7]), selectionCount: 1)
            context.insert(list)
            rule.appList = list
        }
        return rule
    }

    @Test("Enabled limit rules with apps start daily monitoring")
    func startsMonitoring() throws {
        let (scheduler, monitor, store) = makeScheduler()
        let rule = try limitRule(kind: .timeLimit, name: "Time Keeper")

        scheduler.sync(rules: [rule])

        let name = MonitoringPlan.dailyActivityName(for: rule.id)
        #expect(monitor.monitoredNames == [name])
        #expect(monitor.startedEvents[name]?.count == rule.dailyLimitMinutes)
        #expect(store.snapshot(for: rule.id) != nil)
    }

    @Test("Open-limit rules monitor the day without usage checkpoints")
    func openLimitHasNoCheckpoints() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try limitRule(kind: .openLimit, name: "Gate Keeper")

        scheduler.sync(rules: [rule])

        #expect(monitor.startedEvents[MonitoringPlan.dailyActivityName(for: rule.id)]?.isEmpty == true)
    }

    @Test("App-less rules (schedule or limit) are not monitored")
    func skipsUnmonitorable() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let context = try makeInMemoryContext()
        let applessSchedule = BlockingRule(name: "Work Time")
        let applessLimit = BlockingRule(name: "Empty", kind: .timeLimit)
        context.insert(applessSchedule)
        context.insert(applessLimit)

        scheduler.sync(rules: [applessSchedule, applessLimit])

        #expect(monitor.monitoredNames.isEmpty)
    }

    @Test("A non-crossing schedule rule registers one window activity")
    func schedulesNonCrossingWindow() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try scheduleRule(name: "Work Time", start: 9 * 60, end: 17 * 60)

        scheduler.sync(rules: [rule])

        let primary = MonitoringPlan.scheduleWindowName(for: rule.id)
        #expect(monitor.monitoredNames == [primary])
        #expect(monitor.startedWindows[primary]?.start == 9 * 60)
        #expect(monitor.startedWindows[primary]?.end == 17 * 60)
        // A schedule window carries no usage-threshold events.
        #expect(monitor.startedEvents[primary] == nil)
    }

    @Test("A dayless schedule rule is not monitored")
    func skipsDaylessSchedule() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try scheduleRule(name: "No Days", start: 9 * 60, end: 17 * 60, days: [])

        scheduler.sync(rules: [rule])

        #expect(monitor.monitoredNames.isEmpty)
    }

    @Test("A midnight-crossing schedule rule registers two window activities")
    func schedulesCrossingWindow() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try scheduleRule(name: "Deep Sleep", start: 22 * 60, end: 6 * 60)

        scheduler.sync(rules: [rule])

        let primary = MonitoringPlan.scheduleWindowName(for: rule.id)
        let late = MonitoringPlan.scheduleWindowLateName(for: rule.id)
        #expect(Set(monitor.monitoredNames) == [primary, late])
        // The evening half runs to the end of the day; the morning half from midnight.
        #expect(monitor.startedWindows[primary]?.start == 22 * 60)
        #expect(monitor.startedWindows[primary]?.end == 24 * 60 - 1)
        #expect(monitor.startedWindows[late]?.start == 0)
        #expect(monitor.startedWindows[late]?.end == 6 * 60)
    }

    @Test("A window ending exactly at midnight registers a single evening activity")
    func scheduleWindowEndingAtMidnight() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try scheduleRule(name: "Late", start: 22 * 60, end: 0)

        scheduler.sync(rules: [rule])

        let primary = MonitoringPlan.scheduleWindowName(for: rule.id)
        let late = MonitoringPlan.scheduleWindowLateName(for: rule.id)
        #expect(monitor.monitoredNames == [primary])
        #expect(monitor.startedWindows[primary]?.start == 22 * 60)
        #expect(monitor.startedWindows[primary]?.end == 24 * 60 - 1)
        #expect(monitor.startedWindows[late] == nil)
    }

    @Test("A 24-hour window registers a single all-day activity")
    func scheduleFullDayWindow() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try scheduleRule(name: "Always", start: 0, end: 0)

        scheduler.sync(rules: [rule])

        let primary = MonitoringPlan.scheduleWindowName(for: rule.id)
        #expect(monitor.monitoredNames == [primary])
        #expect(monitor.startedWindows[primary]?.start == 0)
        #expect(monitor.startedWindows[primary]?.end == 24 * 60 - 1)
    }

    @Test("Switching a window from crossing to non-crossing stops the morning half")
    func dropsLateActivityWhenWindowStopsCrossing() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try scheduleRule(name: "Deep Sleep", start: 22 * 60, end: 6 * 60)
        scheduler.sync(rules: [rule])

        let primary = MonitoringPlan.scheduleWindowName(for: rule.id)
        let late = MonitoringPlan.scheduleWindowLateName(for: rule.id)
        #expect(Set(monitor.monitoredNames) == [primary, late])

        // Now a normal daytime window — the post-midnight half must be stopped.
        rule.startMinutes = 9 * 60
        rule.endMinutes = 17 * 60
        scheduler.sync(rules: [rule])
        #expect(monitor.monitoredNames == [primary])
        #expect(monitor.startedWindows[late] == nil)
    }

    @Test("Changing only the days does not restart the window activity")
    func keepsWindowWhenOnlyDaysChange() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try scheduleRule(
            name: "Work Time", start: 9 * 60, end: 17 * 60, days: Weekday.weekdays)

        scheduler.sync(rules: [rule])
        #expect(monitor.startCallCount == 1)

        // The window interval is unchanged, so the DeviceActivity activity that
        // only encodes start/end need not restart — reconcile() reads days fresh.
        rule.days = Weekday.everyDay
        scheduler.sync(rules: [rule])
        #expect(monitor.startCallCount == 1)
    }

    @Test("Disabling a schedule rule stops its window monitoring")
    func stopsScheduleWindowWhenDisabled() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try scheduleRule(name: "Work Time", start: 9 * 60, end: 17 * 60)

        scheduler.sync(rules: [rule])
        #expect(!monitor.monitoredNames.isEmpty)

        rule.isEnabled = false
        scheduler.sync(rules: [rule])
        #expect(monitor.monitoredNames.isEmpty)
    }

    @Test("Changing a schedule window restarts monitoring")
    func restartsOnWindowChange() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try scheduleRule(name: "Work Time", start: 9 * 60, end: 17 * 60)

        scheduler.sync(rules: [rule])
        scheduler.sync(rules: [rule])
        #expect(monitor.startCallCount == 1)

        rule.endMinutes = 18 * 60
        scheduler.sync(rules: [rule])
        #expect(monitor.startCallCount == 2)
        let primary = MonitoringPlan.scheduleWindowName(for: rule.id)
        #expect(monitor.startedWindows[primary]?.end == 18 * 60)
    }

    @Test("Monitoring stops when a rule is disabled or removed")
    func stopsStaleMonitoring() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try limitRule(kind: .timeLimit, name: "Time Keeper")
        scheduler.sync(rules: [rule])

        rule.isEnabled = false
        scheduler.sync(rules: [rule])
        #expect(monitor.monitoredNames.isEmpty)

        rule.isEnabled = true
        scheduler.sync(rules: [rule])
        scheduler.sync(rules: [])
        #expect(monitor.monitoredNames.isEmpty)
    }

    @Test("Unchanged rules are not restarted (checkpoints would reset)")
    func avoidsRestartChurn() throws {
        let (scheduler, monitor, _) = makeScheduler()
        let rule = try limitRule(kind: .timeLimit, name: "Time Keeper")

        scheduler.sync(rules: [rule])
        scheduler.sync(rules: [rule])
        #expect(monitor.startCallCount == 1)

        rule.dailyLimitMinutes = 60
        scheduler.sync(rules: [rule])
        #expect(monitor.startCallCount == 2)
    }
}

@MainActor
@Suite("Limit enforcement reactions")
struct LimitEnforcementTests {
    let monday = date(2025, 1, 6, 10, 0)

    private func makeEnforcement() -> (LimitEnforcement, MockShieldController, UsageLedger, RuleSnapshotStore) {
        let shields = MockShieldController()
        let ledger = UsageLedger(defaults: freshDefaults())
        let store = RuleSnapshotStore(defaults: freshDefaults())
        return (LimitEnforcement(snapshots: store, ledger: ledger, shields: shields), shields, ledger, store)
    }

    private func snapshot(
        kind: RuleKind, limit: Int = 45, maxOpens: Int = 5, pausedUntil: Date? = nil
    ) -> RuleSnapshot {
        RuleSnapshot(
            id: UUID(), name: "Rule", kindRaw: kind.rawValue, isEnabled: true,
            hardMode: false, blockAdultContent: false, selectionModeRaw: "block",
            selectionData: Data([1]), dayNumbers: Weekday.everyDay.map(\.rawValue),
            startMinutes: 0, endMinutes: 0,
            dailyLimitMinutes: limit, maxOpens: maxOpens, pausedUntil: pausedUntil
        )
    }

    @Test("Day start shields open-limit rules so opens can be counted")
    func dayStartShieldsOpenLimit() {
        let (enforcement, shields, _, store) = makeEnforcement()
        let snap = snapshot(kind: .openLimit)
        store.save([snap])

        enforcement.handleDayStart(ruleID: snap.id, now: monday, calendar: utc)

        #expect(shields.shieldedRuleIDs == [snap.id])
    }

    @Test("Day start clears time-limit shields for the fresh budget")
    func dayStartClearsTimeLimit() {
        let (enforcement, shields, _, store) = makeEnforcement()
        let snap = snapshot(kind: .timeLimit)
        store.save([snap])
        shields.applyShield(
            ruleID: snap.id, selectionData: nil, mode: .block, blockAdultContent: false)

        enforcement.handleDayStart(ruleID: snap.id, now: monday, calendar: utc)

        #expect(shields.shieldedRuleIDs.isEmpty)
    }

    @Test("Usage checkpoints record minutes and shield at the limit")
    func usageCheckpointsShieldAtLimit() {
        let (enforcement, shields, ledger, store) = makeEnforcement()
        let snap = snapshot(kind: .timeLimit, limit: 45)
        store.save([snap])

        enforcement.handleUsageMinutes(20, ruleID: snap.id, now: monday, calendar: utc)
        #expect(ledger.usage(for: snap.id, onDayContaining: monday, calendar: utc).minutesUsed == 20)
        #expect(shields.shieldedRuleIDs.isEmpty)

        enforcement.handleUsageMinutes(45, ruleID: snap.id, now: monday, calendar: utc)
        #expect(shields.shieldedRuleIDs == [snap.id])
    }

    @Test("An Open press spends one open and lifts the shield")
    func openRequestSpendsAndLifts() {
        let (enforcement, shields, ledger, store) = makeEnforcement()
        let snap = snapshot(kind: .openLimit, maxOpens: 2)
        store.save([snap])
        enforcement.handleDayStart(ruleID: snap.id, now: monday, calendar: utc)

        #expect(enforcement.handleOpenRequest(ruleID: snap.id, now: monday, calendar: utc))
        #expect(ledger.usage(for: snap.id, onDayContaining: monday, calendar: utc).opensUsed == 1)
        #expect(shields.shieldedRuleIDs.isEmpty)
    }

    @Test("Open presses are refused once opens are exhausted")
    func openRequestRefusedWhenExhausted() {
        let (enforcement, shields, ledger, store) = makeEnforcement()
        let snap = snapshot(kind: .openLimit, maxOpens: 1)
        store.save([snap])
        ledger.recordOpen(for: snap.id, onDayContaining: monday, calendar: utc)
        enforcement.handleDayStart(ruleID: snap.id, now: monday, calendar: utc)

        #expect(!enforcement.handleOpenRequest(ruleID: snap.id, now: monday, calendar: utc))
        #expect(shields.shieldedRuleIDs == [snap.id])
    }

    @Test("Session end re-shields the rule for the next open")
    func sessionEndReshields() {
        let (enforcement, shields, _, store) = makeEnforcement()
        let snap = snapshot(kind: .openLimit)
        store.save([snap])

        enforcement.handleOpenSessionEnded(ruleID: snap.id, now: monday, calendar: utc)

        #expect(shields.shieldedRuleIDs == [snap.id])
    }

    @Test("A paused (unblocked) rule is left alone until midnight")
    func pausedRuleLeftAlone() {
        let (enforcement, shields, _, store) = makeEnforcement()
        let snap = snapshot(kind: .openLimit, pausedUntil: date(2025, 1, 7, 0, 0))
        store.save([snap])

        enforcement.handleDayStart(ruleID: snap.id, now: monday, calendar: utc)
        enforcement.handleOpenSessionEnded(ruleID: snap.id, now: monday, calendar: utc)

        #expect(shields.shieldedRuleIDs.isEmpty)
    }
}

@MainActor
@Suite("Schedule-window enforcement reactions")
struct ScheduleEnforcementTests {
    /// Monday 2025-01-06 10:00 UTC sits inside a 09:00–17:00 weekday window.
    let mondayMorning = date(2025, 1, 6, 10, 0)
    let mondayEvening = date(2025, 1, 6, 19, 0)

    private func makeEnforcement() -> (ScheduleEnforcement, MockShieldController, RuleSnapshotStore) {
        let shields = MockShieldController()
        let store = RuleSnapshotStore(defaults: freshDefaults())
        return (ScheduleEnforcement(snapshots: store, shields: shields), shields, store)
    }

    private func snapshot(
        start: Int = 9 * 60, end: Int = 17 * 60, days: Set<Weekday> = Weekday.everyDay,
        isEnabled: Bool = true, mode: SelectionMode = .block, pausedUntil: Date? = nil
    ) -> RuleSnapshot {
        RuleSnapshot(
            id: UUID(), name: "Work Time", kindRaw: RuleKind.schedule.rawValue,
            isEnabled: isEnabled, hardMode: false, blockAdultContent: false,
            selectionModeRaw: mode.rawValue, selectionData: Data([1]),
            dayNumbers: days.map(\.rawValue), startMinutes: start, endMinutes: end,
            dailyLimitMinutes: 45, maxOpens: 5, pausedUntil: pausedUntil
        )
    }

    @Test("Reconcile shields a rule whose window is active now")
    func shieldsActiveWindow() {
        let (enforcement, shields, store) = makeEnforcement()
        let snap = snapshot()
        store.save([snap])

        enforcement.reconcile(ruleID: snap.id, now: mondayMorning, calendar: utc)

        #expect(shields.shieldedRuleIDs == [snap.id])
    }

    @Test("Reconcile clears a rule whose window is over")
    func clearsInactiveWindow() {
        let (enforcement, shields, store) = makeEnforcement()
        let snap = snapshot()
        store.save([snap])
        shields.applyShield(
            ruleID: snap.id, selectionData: nil, mode: .block, blockAdultContent: false)

        enforcement.reconcile(ruleID: snap.id, now: mondayEvening, calendar: utc)

        #expect(shields.shieldedRuleIDs.isEmpty)
    }

    @Test("Reconcile forwards the rule's selection mode")
    func forwardsSelectionMode() {
        let (enforcement, shields, store) = makeEnforcement()
        let snap = snapshot(mode: .allowOnly)
        store.save([snap])

        enforcement.reconcile(ruleID: snap.id, now: mondayMorning, calendar: utc)

        #expect(shields.appliedModes[snap.id] == .allowOnly)
    }

    @Test("A disabled schedule rule is never shielded")
    func skipsDisabled() {
        let (enforcement, shields, store) = makeEnforcement()
        let snap = snapshot(isEnabled: false)
        store.save([snap])

        enforcement.reconcile(ruleID: snap.id, now: mondayMorning, calendar: utc)

        #expect(shields.shieldedRuleIDs.isEmpty)
    }

    @Test("A paused (unblocked) schedule rule is left clear")
    func skipsPaused() {
        let (enforcement, shields, store) = makeEnforcement()
        let snap = snapshot(pausedUntil: date(2025, 1, 6, 17, 0))
        store.save([snap])

        enforcement.reconcile(ruleID: snap.id, now: mondayMorning, calendar: utc)

        #expect(shields.shieldedRuleIDs.isEmpty)
    }

    @Test("A midnight-crossing window is active in the early morning")
    func shieldsCrossingWindowAfterMidnight() {
        let (enforcement, shields, store) = makeEnforcement()
        // 22:00–06:00 every day; 02:00 belongs to the window that started the
        // previous evening (so the previous day must be enabled — it is).
        let snap = snapshot(start: 22 * 60, end: 6 * 60)
        store.save([snap])

        enforcement.reconcile(ruleID: snap.id, now: date(2025, 1, 6, 2, 0), calendar: utc)

        #expect(shields.shieldedRuleIDs == [snap.id])
    }

    @Test("A weekday window is inactive on the weekend")
    func clearsOnDisabledDay() {
        let (enforcement, shields, store) = makeEnforcement()
        let snap = snapshot(days: Weekday.weekdays)
        store.save([snap])
        // 2025-01-11 is a Saturday.
        enforcement.reconcile(ruleID: snap.id, now: date(2025, 1, 11, 10, 0), calendar: utc)

        #expect(shields.shieldedRuleIDs.isEmpty)
    }
}