DevLog_iOS/DevLog/Widget/Heatmap/HeatmapWidgetSyncCoordinator.swift at c97285f340886787f44b63c095dcbaedd4d678de · opficdev/DevLog_iOS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
//
//  HeatmapWidgetSyncCoordinator.swift
//  DevLog
//
//  Created by opfic on 4/17/26.
//

import Foundation
import WidgetKit

final class HeatmapWidgetSyncCoordinator {
    private let fetchTodosUseCase: FetchTodosUseCase
    private let factory: HeatmapWidgetSnapshotFactory
    private let store: WidgetSnapshotStore
    private let calendar: Calendar
    private let logger = Logger(category: "HeatmapWidgetSyncCoordinator")

    init(
        fetchTodosUseCase: FetchTodosUseCase,
        factory: HeatmapWidgetSnapshotFactory = .init(),
        store: WidgetSnapshotStore = .init(),
        calendar: Calendar = .current
    ) {
        self.fetchTodosUseCase = fetchTodosUseCase
        self.factory = factory
        self.store = store
        self.calendar = calendar
    }

    func sync(
        selectedActivityKinds: Set<ActivityKind>,
        now: Date = Date()
    ) async {
        let quarterStart = startOfQuarter(for: now)
        guard let nextQuarterStart = calendar.date(byAdding: .month, value: 3, to: quarterStart) else {
            return
        }

        do {
            async let createdTodoPage = fetchTodosUseCase.execute(
                TodoQuery(
                    sortDateFrom: quarterStart,
                    sortDateTo: nextQuarterStart,
                    includesDeleted: true,
                    sortTarget: .createdAt,
                    pageSize: 100,
                    fetchAllPages: true
                ),
                cursor: nil
            )
            async let completedTodoPage = fetchTodosUseCase.execute(
                TodoQuery(
                    sortDateFrom: quarterStart,
                    sortDateTo: nextQuarterStart,
                    includesDeleted: true,
                    sortTarget: .completedAt,
                    pageSize: 100,
                    fetchAllPages: true
                ),
                cursor: nil
            )
            async let deletedTodoPage = fetchTodosUseCase.execute(
                TodoQuery(
                    sortDateFrom: quarterStart,
                    sortDateTo: nextQuarterStart,
                    includesDeleted: true,
                    sortTarget: .deletedAt,
                    pageSize: 100,
                    fetchAllPages: true
                ),
                cursor: nil
            )

            let snapshot = factory.makeSnapshot(
                createdTodos: try await createdTodoPage.items,
                completedTodos: try await completedTodoPage.items,
                deletedTodos: try await deletedTodoPage.items,
                selectedActivityKinds: selectedActivityKinds,
                quarterStart: quarterStart,
                now: now
            )

            try store.saveHeatmapSnapshot(snapshot)
            WidgetCenter.shared.reloadTimelines(ofKind: "HeatmapWidget")
        } catch is CancellationError {
            logger.debug("Heatmap widget sync cancelled.")
        } catch {
            logger.error(
                "Failed to sync heatmap widget snapshot.",
                error: error
            )
        }
    }
}

private extension HeatmapWidgetSyncCoordinator {
    func startOfQuarter(for date: Date) -> Date {
        let month = calendar.component(.month, from: date)
        let startMonth = ((month - 1) / 3) * 3 + 1
        var components = calendar.dateComponents([.year], from: date)
        components.month = startMonth
        components.day = 1
        return calendar.date(from: components) ?? calendar.startOfDay(for: date)
    }
}