summaryrefslogtreecommitdiff
path: root/Hutch/Views/Lookup/UserProfileViewModel.swift
blob: b88d36ab3f11547e223bdcda5bd9f694daadb7c2 (plain) (blame)
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
import Foundation

@Observable
@MainActor
final class UserProfileViewModel {
    private(set) var repositories: [RepositorySummary] = []
    private(set) var trackers: [TrackerSummary] = []
    private(set) var contributionCalendar: ContributionCalendarResponse?
    private(set) var contributionStats: ContributionStatsResponse?
    private(set) var isLoadingRepositories = false
    private(set) var isLoadingTrackers = false
    private(set) var isLoadingContributions = false
    var repositoriesError: String?
    var trackersError: String?
    var contributionsError: String?

    private let client: SRHTClient
    private let statsService: HutchStatsService
    let ownerUsername: String
    let actor: String

    var isContributionActivityIndexedButEmpty: Bool {
        contributionDisplayState != .populated && contributionDisplayState != .unavailable
    }

    var contributionStatusText: String? {
        switch contributionDisplayState {
        case .indexing:
            return "Activity is being indexed."
        case .empty:
            return "No contribution activity found."
        case .unavailable:
            return "Contribution activity is unavailable."
        case .populated:
            return nil
        }
    }

    init(ownerUsername: String, actor: String, client: SRHTClient, statsService: HutchStatsService) {
        self.ownerUsername = ownerUsername
        self.actor = actor
        self.client = client
        self.statsService = statsService
    }

    func loadRepositories() async {
        isLoadingRepositories = true
        repositoriesError = nil
        defer { isLoadingRepositories = false }

        do {
            let result = try await client.execute(
                service: .git,
                query: Self.repositoriesQuery,
                variables: ["owner": ownerUsername],
                responseType: UserRepositoriesResponse.self
            )
            repositories = result.user.repositories.results.map { $0.repositorySummary(service: .git) }
        } catch {
            repositoriesError = error.userFacingMessage
        }
    }

    func updateRepository(_ repository: RepositorySummary) {
        guard let index = repositories.firstIndex(where: { $0.id == repository.id }) else { return }
        repositories[index] = repository
        repositories.sort { lhs, rhs in
            if lhs.updated == rhs.updated {
                return lhs.name.localizedCaseInsensitiveCompare(rhs.name) == .orderedAscending
            }
            return lhs.updated > rhs.updated
        }
    }

    func loadTrackers() async {
        isLoadingTrackers = true
        trackersError = nil
        defer { isLoadingTrackers = false }

        do {
            let result = try await client.execute(
                service: .todo,
                query: Self.trackersQuery,
                variables: ["owner": ownerUsername],
                responseType: UserTrackersResponse.self
            )
            trackers = result.user.trackers.results
        } catch {
            trackersError = error.userFacingMessage
        }
    }

    func loadContributions(endingOn endDate: Date? = nil) async {
        isLoadingContributions = true
        contributionsError = nil
        defer { isLoadingContributions = false }

        let resolvedEndDate = Calendar.contributionCalendar.startOfDay(for: endDate ?? Date())
        do {
            async let contributionCalendar = statsService.fetchContributionCalendar(actor: actor, endingOn: resolvedEndDate)
            async let contributionStats = statsService.fetchContributionStats(actor: actor, endingOn: resolvedEndDate)

            self.contributionCalendar = try await contributionCalendar
            self.contributionStats = try await contributionStats
            if contributionDisplayState != .unavailable {
                contributionsError = nil
            }
        } catch {
            contributionsError = error.userFacingMessage
        }
    }

    private static let repositoriesQuery = """
    query userRepositories($owner: String!) {
        user(username: $owner) {
            repositories {
                results {
                    id
                    rid
                    name
                    description
                    visibility
                    updated
                    owner { canonicalName }
                    HEAD { name target }
                }
                cursor
            }
        }
    }
    """

    private static let trackersQuery = """
    query userTrackers($owner: String!) {
        user(username: $owner) {
            trackers {
                results {
                    id
                    rid
                    name
                    description
                    visibility
                    updated
                    owner { canonicalName }
                }
                cursor
            }
        }
    }
    """

    private struct UserRepositoriesResponse: Decodable, Sendable {
        let user: UserRepositoriesContainer
    }

    private struct UserRepositoriesContainer: Decodable, Sendable {
        let repositories: RepositoriesPage
    }

    private struct RepositoriesPage: Decodable, Sendable {
        let results: [RepositoryPayload]
        let cursor: String?
    }

    private struct RepositoryPayload: Decodable, Sendable {
        let id: Int
        let rid: String
        let name: String
        let description: String?
        let visibility: Visibility
        let updated: Date
        let owner: Entity
        let head: Reference?

        enum CodingKeys: String, CodingKey {
            case id, rid, name, description, visibility, updated, owner
            case head = "HEAD"
        }

        func repositorySummary(service: SRHTService) -> RepositorySummary {
            RepositorySummary(
                id: id,
                rid: rid,
                service: service,
                name: name,
                description: description,
                visibility: visibility,
                updated: updated,
                owner: owner,
                head: head
            )
        }
    }

    private struct UserTrackersResponse: Decodable, Sendable {
        let user: UserTrackersContainer
    }

    private struct UserTrackersContainer: Decodable, Sendable {
        let trackers: TrackersPage
    }

    private struct TrackersPage: Decodable, Sendable {
        let results: [TrackerSummary]
        let cursor: String?
    }

    private enum ContributionDisplayState {
        case populated
        case indexing
        case empty
        case unavailable
    }

    private var contributionDisplayState: ContributionDisplayState {
        if let contributionStats, contributionStats.totalEvents > 0 {
            return .populated
        }

        if let contributionCalendar, !contributionCalendar.isEmpty {
            return .populated
        }

        let indexingState = contributionStats?.indexingState ?? contributionCalendar?.indexingState
        switch indexingState {
        case .pending:
            return .indexing
        case .error:
            return .unavailable
        case .indexed, nil:
            return .empty
        }
    }

}