mirror of
https://github.com/BigMarc/grammargem.git
synced 2026-10-13 17:01:55 +02:00
f5faa27f88
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>
102 lines
3.3 KiB
Swift
102 lines
3.3 KiB
Swift
import Foundation
|
|
|
|
/// The runtime gate every paid path goes through. Combines current `Entitlements`
|
|
/// (from the license) with a local daily counter for the free tier's AI allowance.
|
|
@MainActor
|
|
final class FeatureGate: ObservableObject {
|
|
enum Decision: Equatable {
|
|
case allowed
|
|
case denied(String)
|
|
}
|
|
|
|
private let license: LicenseManager
|
|
private let counter = DailyCounter()
|
|
|
|
/// Bumped to nudge SwiftUI when the counter changes (it's stored externally).
|
|
@Published private(set) var usageTick = 0
|
|
|
|
init(license: LicenseManager) {
|
|
self.license = license
|
|
}
|
|
|
|
var entitlements: Entitlements { license.entitlements }
|
|
|
|
// MARK: - AI gating (rewrite / tone / Ask / translate / paid Modes share this)
|
|
|
|
func authorizeAI(_ action: AIAction) -> Decision {
|
|
let ent = entitlements
|
|
|
|
// Paid Modes are unavailable on free (only Polish ships free).
|
|
if case .applyMode(let mode) = action, mode.isPaid, !ent.allModes {
|
|
return .denied("“\(mode.name)” mode is part of the lifetime upgrade.")
|
|
}
|
|
|
|
if ent.unlimitedAIActions { return .allowed }
|
|
|
|
guard remainingAIActionsToday > 0 else {
|
|
return .denied("You've used today's \(ent.dailyAIActionCap) free AI actions. Upgrade for unlimited.")
|
|
}
|
|
return .allowed
|
|
}
|
|
|
|
func recordAIActionUsed() {
|
|
guard !entitlements.unlimitedAIActions else { return }
|
|
counter.increment()
|
|
usageTick &+= 1
|
|
}
|
|
|
|
var aiActionsUsedToday: Int { counter.countToday() }
|
|
|
|
var remainingAIActionsToday: Int {
|
|
let ent = entitlements
|
|
if ent.unlimitedAIActions { return Int.max }
|
|
return max(0, ent.dailyAIActionCap - counter.countToday())
|
|
}
|
|
|
|
// MARK: - Other gates
|
|
|
|
func availableModes() -> [WritingMode] {
|
|
ModeRegistry.available(for: license.tier)
|
|
}
|
|
|
|
var appAwareEnabled: Bool { entitlements.appAwareSwitching }
|
|
var snippetsEnabled: Bool { entitlements.snippets }
|
|
|
|
/// Can the user add another personal-dictionary entry given the current count?
|
|
func canAddDictionaryEntry(currentCount: Int) -> Decision {
|
|
if currentCount < entitlements.dictionaryCap { return .allowed }
|
|
return .denied("Free dictionary holds \(entitlements.dictionaryCap) entries. Upgrade for unlimited.")
|
|
}
|
|
}
|
|
|
|
/// Free-tier AI action counter. Persists `{date, count}` in UserDefaults and
|
|
/// resets at local midnight (compares the stored local date string to today's).
|
|
final class DailyCounter {
|
|
private let key = "GrammaGem.aiActions.daily"
|
|
private let defaults = UserDefaults.standard
|
|
|
|
private static let formatter: DateFormatter = {
|
|
let f = DateFormatter()
|
|
f.dateFormat = "yyyy-MM-dd"
|
|
f.locale = Locale(identifier: "en_US_POSIX")
|
|
f.timeZone = .current // local midnight reset
|
|
return f
|
|
}()
|
|
|
|
private func today() -> String { Self.formatter.string(from: Date()) }
|
|
|
|
func countToday() -> Int {
|
|
guard let stored = defaults.dictionary(forKey: key),
|
|
let date = stored["date"] as? String, date == today()
|
|
else { return 0 }
|
|
return stored["count"] as? Int ?? 0
|
|
}
|
|
|
|
func increment() {
|
|
let next = countToday() + 1
|
|
defaults.set(["date": today(), "count": next], forKey: key)
|
|
}
|
|
|
|
func reset() { defaults.removeObject(forKey: key) }
|
|
}
|