From dcf4135ec376d357b8fafc0101a75b674e76329a Mon Sep 17 00:00:00 2001
From: Christian Cleberg <hello@cleberg.net>
Date: Wed, 22 Apr 2026 12:39:23 -0500
Subject: feat(v3​.1​.0): add ​Store​Kit integration and ​Pro tier
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

• implement Purchase​Service using StoreKit 2
• activate feature tiers based on entitlement
• add minimal paywall and restore flow
• enforce free-tier limits with upgrade paths
• integrate Pro status into settings
• ensure clean and non-intrusive monetization flow
---
 DomainDig/DomainViewModel.swift | 23 +++++++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

(limited to 'DomainDig/DomainViewModel.swift')

diff --git a/DomainDig/DomainViewModel.swift b/DomainDig/DomainViewModel.swift
index 5288eaa..6ec7e64 100644
--- a/DomainDig/DomainViewModel.swift
+++ b/DomainDig/DomainViewModel.swift
@@ -226,6 +226,8 @@ final class DomainViewModel {
     var watchlistSearchText = ""
     var watchlistFilter: WatchlistFilterOption = .all
     var watchlistSortOption: WatchlistSortOption = .pinned
+    var upgradePrompt: UpgradePromptContext?
+    var isPaywallPresented = false
 
     var trimmedDomain: String {
         domain
@@ -578,6 +580,7 @@ final class DomainViewModel {
         }
 
         guard PremiumAccessService.canAddTrackedDomain(currentCount: trackedDomains.count) else {
+            upgradePrompt = FeatureAccessService.upgradePromptForTrackedDomains(currentCount: trackedDomains.count)
             return false
         }
 
@@ -737,12 +740,18 @@ final class DomainViewModel {
     func runBulkLookup() {
         let domains = parsedDomains(from: bulkInput)
         guard !domains.isEmpty else { return }
-        guard FeatureAccessService.canRunBatch(domainCount: domains.count) else { return }
+        guard FeatureAccessService.canRunBatch(domainCount: domains.count) else {
+            upgradePrompt = FeatureAccessService.upgradePromptForBatch(domainCount: domains.count)
+            return
+        }
         startBatchLookup(domains: domains, source: .manual)
     }
 
     func refreshAllTrackedDomains() {
-        guard FeatureAccessService.canRunBatch(domainCount: sortedTrackedDomains.count) else { return }
+        guard FeatureAccessService.canRunBatch(domainCount: sortedTrackedDomains.count) else {
+            upgradePrompt = FeatureAccessService.upgradePromptForBatch(domainCount: sortedTrackedDomains.count)
+            return
+        }
         startBatchLookup(domains: sortedTrackedDomains.map(\.domain), source: .watchlistRefresh)
     }
 
@@ -791,7 +800,10 @@ final class DomainViewModel {
         let normalizedDomains = normalizedDomains(domains)
         let trimmedName = name.trimmingCharacters(in: .whitespacesAndNewlines)
         guard !trimmedName.isEmpty, !normalizedDomains.isEmpty else { return nil }
-        guard FeatureAccessService.canCreateWorkflow(currentCount: workflows.count) else { return nil }
+        guard FeatureAccessService.canCreateWorkflow(currentCount: workflows.count) else {
+            upgradePrompt = FeatureAccessService.upgradePromptForWorkflows(currentCount: workflows.count)
+            return nil
+        }
 
         let workflow = DomainWorkflow(
             name: trimmedName,
@@ -855,7 +867,10 @@ final class DomainViewModel {
 
     func runWorkflow(_ workflow: DomainWorkflow) {
         guard !workflow.domains.isEmpty else { return }
-        guard FeatureAccessService.canRunBatch(domainCount: workflow.domains.count) else { return }
+        guard FeatureAccessService.canRunBatch(domainCount: workflow.domains.count) else {
+            upgradePrompt = FeatureAccessService.upgradePromptForBatch(domainCount: workflow.domains.count)
+            return
+        }
         startBatchLookup(domains: workflow.domains, source: .workflow, workflow: workflow)
     }
 
-- 
cgit v1.2.3