wehub-resource-sync
80 lines
2.7 KiB
Swift
80 lines
2.7 KiB
Swift
import Foundation
|
|
|
|
public struct OllamaUsageSnapshot: Sendable {
|
|
public let planName: String?
|
|
public let accountEmail: String?
|
|
public let sessionUsedPercent: Double?
|
|
public let weeklyUsedPercent: Double?
|
|
public let sessionResetsAt: Date?
|
|
public let weeklyResetsAt: Date?
|
|
public let sessionWindowMinutes: Int?
|
|
public let updatedAt: Date
|
|
|
|
public init(
|
|
planName: String?,
|
|
accountEmail: String?,
|
|
sessionUsedPercent: Double?,
|
|
weeklyUsedPercent: Double?,
|
|
sessionResetsAt: Date?,
|
|
weeklyResetsAt: Date?,
|
|
sessionWindowMinutes: Int? = nil,
|
|
updatedAt: Date)
|
|
{
|
|
self.planName = planName
|
|
self.accountEmail = accountEmail
|
|
self.sessionUsedPercent = sessionUsedPercent
|
|
self.weeklyUsedPercent = weeklyUsedPercent
|
|
self.sessionResetsAt = sessionResetsAt
|
|
self.weeklyResetsAt = weeklyResetsAt
|
|
self.sessionWindowMinutes = sessionWindowMinutes
|
|
self.updatedAt = updatedAt
|
|
}
|
|
}
|
|
|
|
extension OllamaUsageSnapshot {
|
|
public func toUsageSnapshot() -> UsageSnapshot {
|
|
let sessionWindow = self.makeSessionWindow(
|
|
usedPercent: self.sessionUsedPercent,
|
|
resetsAt: self.sessionResetsAt)
|
|
let weeklyWindow = self.makeWeeklyWindow(
|
|
usedPercent: self.weeklyUsedPercent,
|
|
resetsAt: self.weeklyResetsAt)
|
|
|
|
let plan = self.planName?.trimmingCharacters(in: .whitespacesAndNewlines)
|
|
let email = self.accountEmail?.trimmingCharacters(in: .whitespacesAndNewlines)
|
|
let identity = ProviderIdentitySnapshot(
|
|
providerID: .ollama,
|
|
accountEmail: email?.isEmpty == false ? email : nil,
|
|
accountOrganization: nil,
|
|
loginMethod: plan?.isEmpty == false ? plan : nil)
|
|
|
|
return UsageSnapshot(
|
|
primary: sessionWindow,
|
|
secondary: weeklyWindow,
|
|
tertiary: nil,
|
|
providerCost: nil,
|
|
updatedAt: self.updatedAt,
|
|
identity: identity)
|
|
}
|
|
|
|
private func makeSessionWindow(usedPercent: Double?, resetsAt: Date?) -> RateWindow? {
|
|
guard let usedPercent else { return nil }
|
|
let clamped = min(100, max(0, usedPercent))
|
|
return RateWindow(
|
|
usedPercent: clamped,
|
|
windowMinutes: self.sessionWindowMinutes,
|
|
resetsAt: resetsAt,
|
|
resetDescription: nil)
|
|
}
|
|
|
|
private func makeWeeklyWindow(usedPercent: Double?, resetsAt: Date?) -> RateWindow? {
|
|
guard let usedPercent else { return nil }
|
|
let clamped = min(100, max(0, usedPercent))
|
|
return RateWindow(
|
|
usedPercent: clamped,
|
|
windowMinutes: 7 * 24 * 60,
|
|
resetsAt: resetsAt,
|
|
resetDescription: nil)
|
|
}
|
|
}
|