grammargem/Sources/GrammaGem/Licensing/LicenseManager.swift
BigMarc f5faa27f88 GrammaGem — open-source, on-device macOS writing assistant
Native SwiftUI menu-bar app: global hotkey → capture selected text (AX API /
clipboard fallback) → grammar (Harper) / local AI (MLX) → replace in place.
Free/paid entitlement gate + Lemon Squeezy device-cap licensing, $0 marginal
cost per user. Heavyweight engines behind protocol stubs so it builds out of the box.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
2026-06-16 13:56:26 +02:00

101 lines
3.8 KiB
Swift

import Foundation
/// Owns license state and maps it to an `Entitlements` value the rest of the app
/// gates on. Implements the two gaps Lemon Squeezy leaves to us (spec §6):
/// device fingerprinting (via `DeviceFingerprint`) and **offline grace**.
@MainActor
final class LicenseManager: ObservableObject {
@Published private(set) var record: LicenseRecord?
@Published private(set) var lastError: String?
@Published private(set) var isWorking = false
private let client: LemonSqueezyClient
private let account = "license"
init(client: LemonSqueezyClient = LemonSqueezyClient()) {
self.client = client
self.record = Keychain.getObject(LicenseRecord.self, account: account)
}
/// Effective tier *right now*, honoring the 30-day offline grace: if the
/// device hasn't validated within the window, downgrade to Free rather than
/// hard-lock (writing features never need the network anyway).
var tier: Tier {
guard let record else { return .free }
let grace = TimeInterval(AppConfig.Limits.offlineGraceDays * 86_400)
if Date().timeIntervalSince(record.lastValidated) > grace {
return .free
}
return record.tier
}
var isLicensed: Bool { tier.isPaid }
var entitlements: Entitlements { Entitlements(tier: tier) }
// MARK: - Flows
/// Activate a pasted key against this device.
func activate(key: String) async {
let key = key.trimmingCharacters(in: .whitespacesAndNewlines)
guard !key.isEmpty else { lastError = "Enter a license key."; return }
isWorking = true; lastError = nil
defer { isWorking = false }
do {
let instanceName = DeviceFingerprint.stableID()
let (instanceID, tier) = try await client.activate(key: key, instanceName: instanceName)
let rec = LicenseRecord(
key: key, instanceID: instanceID, tier: tier,
lastValidated: Date(), activatedDevices: tier.deviceCap)
Keychain.setObject(rec, account: account)
record = rec
Log.licensing.info("Activated tier \(tier.rawValue, privacy: .public)")
} catch {
lastError = error.localizedDescription
}
}
/// On launch + every ~7 days. Network failures are non-fatal (offline grace).
func validateIfNeeded(force: Bool = false) async {
guard var rec = record else { return }
let interval = TimeInterval(AppConfig.Limits.validationIntervalDays * 86_400)
guard force || Date().timeIntervalSince(rec.lastValidated) > interval else { return }
do {
let (valid, tier) = try await client.validate(key: rec.key, instanceID: rec.instanceID)
if valid {
rec.lastValidated = Date()
rec.tier = tier
Keychain.setObject(rec, account: account)
record = rec
} else {
// Explicitly invalid (refunded / disabled) → drop the license.
Log.licensing.notice("License reported invalid by server; clearing.")
clearLocal()
}
} catch {
// Offline / transient: keep the cached license under the grace window.
Log.licensing.notice("Validation deferred (offline). Cached license stands under grace.")
}
}
/// Free a device slot so the license can move to another Mac.
func deactivateThisDevice() async {
guard let rec = record else { return }
isWorking = true; lastError = nil
defer { isWorking = false }
do {
_ = try await client.deactivate(key: rec.key, instanceID: rec.instanceID)
clearLocal()
} catch {
lastError = error.localizedDescription
}
}
func clearLocal() {
Keychain.delete(account: account)
record = nil
}
}