.onPurchaseComplete

Hi. In my app I am calling the paywall when the user reaches more than a certain number of rows. Let's say 2. Then I present the paywall. After a successful purchase I have to close and re-open the app in order to trigger the valid subscription to allow more entries. How do I fix this inline at the time of the purchase. 

1.sheet(isPresented: $isShowingPaywall) {
2
3                            PaywallView()
4
5                        }

Here is my save code where I trigger this.

1func save() {
2
3        let bgcolor = selectedColor.toHex() ?? "#FFFFFF"
4
5        print("Before SubscriptionActive: \(userViewModel.isSubscriptionActive)")
6
7        let isSubscriptionActive = userViewModel.isSubscriptionActive
8
9       // print("AFter SubscriptionActive: \(userViewModel.isSubscriptionActive)")
10
11      
12
13        // Retrieve the total count from UserDefaults
14
15        var totalCount = UserDefaults.standard.integer(forKey: "TotalCount")
16
17      //  print("total count in beginning of save \(totalCount)")
18
19        // Check if it's the first entry and there's no total count saved
20
21        if data.isEmpty && totalCount == 0 {
22
23            // Allow the save
24
25        } else if !isSubscriptionActive && totalCount >= 3 && entryToEdit == nil {
26
27            // Show the paywall
28
29            isShowingPaywall = true
30
31            return
32
33        }
34
35        
36
37        if entryToEdit != nil {
38
39            entryToEdit?.title = title
40
41            entryToEdit?.reset = selectedResetOption.rawValue
42
43            entryToEdit?.defaultcount = defaultcount
44
45            entryToEdit?.locked = locked
46
47            entryToEdit?.bgcolor = bgcolor
48
49        } else {
50
51            let data = dataclass(title: title,
52
53                                 reset: selectedResetOption.rawValue,
54
55                                 defaultcount: defaultcount,
56
57                                 bgcolor: bgcolor,
58
59                                 date: Date(),
60
61                                 locked: locked
62
63            )
64
65            context.insert(data)
66
67            totalCount += 1
68
69            UserDefaults.standard.set(totalCount, forKey: "TotalCount")
70
71        }
72
73        dismiss()
74
75    }