1556 Refresh user access token on creating new account and coming bac…

Source/UI/IdentityUI.swift

@@ -777,11 +777,14 @@ extension IdentityUI {
                     self?.complete(with: output, presentingViewController: presentingViewController)
                 }
             }
-        case let .validateAuthCode(code, shouldPersistUser, codeVerifier):
+        case let .validateAuthCode(code, shouldPersistUser, codeVerifier, codeAfterSignup):
             // Let's check if the code validates.
             authenticationCodeInteractor.validate(authCode: code, codeVerifier: codeVerifier) { [weak self] result in
                 switch result {
                 case let .success(user):
+                    if codeAfterSignup {
+                        self?._identityManager.currentUser.refresh(completion: { _ in })
+                    }
                     self?.configuration.tracker?.loginID = self?._identityManager.currentUser.legacyID
                     // User has validated the identifier and the code matches, nothing else to do.
                     self?.complete(with: .success(user: user, persistUser: shouldPersistUser), presentingViewController: presentingViewController)

Source/UI/IdentityUIRoute.swift

@@ -17,7 +17,7 @@ public extension IdentityUI {
         case enterPassword(for: EmailAddress, scopes: [String])
 
         /// Route to validate an authcode
-        case validateAuthCode(code: String, shouldPersistUser: Bool?, codeVerifier: String? = nil)
+        case validateAuthCode(code: String, shouldPersistUser: Bool?, codeVerifier: String? = nil, codeAfterSignup: Bool = false)
 
         var loginMethod: LoginMethod {
             switch self {
@@ -68,7 +68,7 @@ public extension IdentityUI.Route {
             }
             self = .enterPassword(for: email, scopes: scopes)
         case let .codeAfterSignup(code, shouldPersistUser):
-            self = .validateAuthCode(code: code, shouldPersistUser: shouldPersistUser)
+            self = .validateAuthCode(code: code, shouldPersistUser: shouldPersistUser, codeAfterSignup: true)
         case let .codeAfterUnvalidatedLogin(code):
             // We have no way to retrieve the `shouldPersistUser` flag from the redirect URL in this case, so we just fall back to `false`.
             self = .validateAuthCode(code: code, shouldPersistUser: false)