Improve refresh token

duckduckgo · Apr 8, 2024 · 926e38a · 926e38a
1 parent 1c61816
commit 926e38a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/DuckDuckGo/Subscription/ViewModel/SubscriptionRestoreViewModel.swift b/DuckDuckGo/Subscription/ViewModel/SubscriptionRestoreViewModel.swift
@@ -79,6 +79,7 @@ final class SubscriptionRestoreViewModel: ObservableObject {
     func onFirstAppear() async {
         Pixel.fire(pixel: .privacyProSettingsAddDevice)
         await setupTransactionObserver()
+        await refreshToken()
     }
 
     private func cleanUp() {
@@ -87,14 +88,14 @@ final class SubscriptionRestoreViewModel: ObservableObject {
     }
 
     private func refreshToken() async {
-        await AppStoreAccountManagementFlow.refreshAuthTokenIfNeeded(subscriptionAppGroup: Bundle.main.appGroup(bundle: .subs))
+        if state.isAddingDevice {
+            await AppStoreAccountManagementFlow.refreshAuthTokenIfNeeded(subscriptionAppGroup: Bundle.main.appGroup(bundle: .subs))
+        }
     }
 
     private func setupContent() async {
         if state.isAddingDevice {
 
-            await refreshToken()
-
             DispatchQueue.main.async {
                 self.state.isLoading = true
             }