Verify Element X with an existing Element Classic account. (#5374)

* Read and import the secrets from ClassicAppAccounts. * Record snapshots. * Add some documentation, tidy up tests and fix the dismissal of the backup instructions. * Workaround flakey tests (the fulfilments weren't always firing). * Allow a custom Classic App deep link URL to be configured.
2026-11-05 11:58:32 +00:00 · 2026-04-13 15:30:09 +01:00
parent 001a389f51
commit 7c812201cb
40 changed files with 947 additions and 103 deletions
@@ -13,6 +13,10 @@ struct AuthenticationClassicAppAccountView: View {
    
    let classicAppAccount: ClassicAppAccount
    
+    var isLoadingAccount: Bool {
+        classicAppAccount.state.isServerSupported == nil || classicAppAccount.state.availableSecrets == nil
+    }
+    
    var body: some View {
        FullscreenDialog(topPadding: 25, background: .gradient) {
            VStack(spacing: 38) {
@@ -28,6 +32,11 @@ struct AuthenticationClassicAppAccountView: View {
        }
        .navigationBarTitleDisplayMode(.inline)
        .alert(item: $context.alertInfo)
+        .sheet(isPresented: $context.showClassicAppBackupInstructions) {
+            AuthenticationClassicAppBackupInstructionsView(classicAppAccount: classicAppAccount) {
+                context.send(viewAction: .openClassicApp)
+            }
+        }
        .introspect(.window, on: .supportedVersions) { window in
            context.send(viewAction: .updateWindow(window))
        }
@@ -73,15 +82,30 @@ struct AuthenticationClassicAppAccountView: View {
    
    var buttons: some View {
        VStack(spacing: 16) {
-            Button(L10n.actionContinue) {
-                context.send(viewAction: .continueWithClassic(classicAppAccount))
+            if isLoadingAccount {
+                Button {
+                    context.send(viewAction: .continueWithClassic(classicAppAccount))
+                } label: {
+                    Label {
+                        Text(L10n.screenOnboardingCheckingAccount)
+                    } icon: {
+                        ProgressView()
+                            .tint(.compound.iconOnSolidPrimary)
+                    }
+                }
+                .buttonStyle(.compound(.primary))
+                .disabled(true)
+            } else {
+                Button(L10n.actionContinue) {
+                    context.send(viewAction: .continueWithClassic(classicAppAccount))
+                }
+                .buttonStyle(.compound(.primary))
+                
+                Button(L10n.commonOtherOptions) {
+                    context.send(viewAction: .otherOptions(classicAppAccount))
+                }
+                .buttonStyle(.compound(.secondary))
            }
-            .buttonStyle(.compound(.primary))
-            
-            Button(L10n.commonOtherOptions) {
-                context.send(viewAction: .otherOptions(classicAppAccount))
-            }
-            .buttonStyle(.compound(.secondary))
        }
    }
 }
@@ -102,17 +126,30 @@ private extension ClassicAppAccount {

 struct AuthenticationClassicAppAccountView_Previews: PreviewProvider { // Not Testable – snapshots generated by main screen.
    static let viewModel = makeViewModel()
+    static let classicAppAccount = {
+        let account = ClassicAppAccount.mockDan
+        account.state.isServerSupported = true
+        account.state.availableSecrets = .complete
+        return account
+    }()
    
    static var previews: some View {
+        ElementNavigationStack {
+            AuthenticationClassicAppAccountView(context: viewModel.context, classicAppAccount: classicAppAccount)
+        }
+        .previewDisplayName("Ready")
+        
        ElementNavigationStack {
            AuthenticationClassicAppAccountView(context: viewModel.context, classicAppAccount: .mockDan)
        }
+        .previewDisplayName("Loading")
    }
    
    static func makeViewModel() -> AuthenticationStartScreenViewModel {
        AuthenticationStartScreenViewModel(authenticationService: AuthenticationService.mock,
                                           provisioningParameters: nil,
                                           isBugReportServiceEnabled: false,
+                                           appMediator: AppMediatorMock(),
                                           appSettings: ServiceLocator.shared.settings,
                                           mediaProvider: MediaProviderMock(configuration: .init()),
                                           userIndicatorController: UserIndicatorControllerMock())
@@ -9,10 +9,19 @@ import Compound
 import SwiftUI

 struct AuthenticationClassicAppBackupInstructionsView: View {
-    let context: AuthenticationStartScreenViewModel.Context
-    
    @Environment(\.dismiss) private var dismiss
    
+    let classicAppAccount: ClassicAppAccount
+    let openClassicAppAction: () -> Void
+    
+    private var isRefreshingSecrets: Bool {
+        classicAppAccount.state.availableSecrets == nil
+    }
+    
+    private var buttonTitle: String {
+        isRefreshingSecrets ? L10n.screenOnboardingCheckingAccount : L10n.screenMissingKeyBackupOpenElementClassic
+    }
+    
    var body: some View {
        ElementNavigationStack {
            FullscreenDialog(topPadding: 24, horizontalPadding: 24) {
@@ -32,6 +41,7 @@ struct AuthenticationClassicAppBackupInstructionsView: View {
            TitleAndIcon(title: L10n.screenMissingKeyBackupTitle(InfoPlistReader.main.bundleDisplayName),
                         icon: \.keySolid,
                         iconStyle: .default)
+                .frame(maxWidth: .infinity)
            
            SFNumberedListView(items: [
                AttributedString(L10n.screenMissingKeyBackupStep1),
@@ -44,10 +54,18 @@ struct AuthenticationClassicAppBackupInstructionsView: View {
    }
    
    var buttons: some View {
-        Button(L10n.screenMissingKeyBackupOpenElementClassic) {
-            UIApplication.shared.open("element://open")
+        Button(action: openClassicAppAction) {
+            Label {
+                Text(buttonTitle)
+            } icon: {
+                if isRefreshingSecrets {
+                    ProgressView()
+                        .tint(.compound.iconOnSolidPrimary)
+                }
+            }
        }
        .buttonStyle(.compound(.primary))
+        .disabled(isRefreshingSecrets)
    }
    
    var toolbar: some ToolbarContent {
@@ -58,18 +76,17 @@ struct AuthenticationClassicAppBackupInstructionsView: View {
 }

 struct AuthenticationClassicAppBackupInstructionsView_Previews: PreviewProvider, TestablePreview {
-    static let viewModel = makeViewModel()
+    static let loadedAccount = {
+        let account = ClassicAppAccount.mockDan
+        account.state.availableSecrets = .requiresBackup
+        return account
+    }()
    
    static var previews: some View {
-        AuthenticationClassicAppBackupInstructionsView(context: viewModel.context)
-    }
-    
-    static func makeViewModel() -> AuthenticationStartScreenViewModel {
-        AuthenticationStartScreenViewModel(authenticationService: AuthenticationService.mock,
-                                           provisioningParameters: nil,
-                                           isBugReportServiceEnabled: false,
-                                           appSettings: ServiceLocator.shared.settings,
-                                           mediaProvider: MediaProviderMock(configuration: .init()),
-                                           userIndicatorController: UserIndicatorControllerMock())
+        AuthenticationClassicAppBackupInstructionsView(classicAppAccount: loadedAccount) { }
+            .previewDisplayName("Initial")
+        
+        AuthenticationClassicAppBackupInstructionsView(classicAppAccount: .mockAlice) { }
+            .previewDisplayName("Refreshing")
    }
 }
@@ -16,6 +16,15 @@ struct AuthenticationStartScreen: View {
    @Bindable var context: AuthenticationStartScreenViewModel.Context
    
    var body: some View {
+        if case let .welcomeBack(classicAppAccount) = context.viewState.classicAppMode,
+           classicAppAccount.state.isServerSupported != false {
+            AuthenticationClassicAppAccountView(context: context, classicAppAccount: classicAppAccount)
+        } else {
+            standardContent
+        }
+    }
+    
+    var standardContent: some View {
        // This view uses a GeometryReader instead of FullscreenDialog so its content takes the full
        // height available (after taking the buttons out of the equation) in order for the logo
        // and title to appear vertically centred and equally spaced within this content area.
@@ -41,14 +50,12 @@ struct AuthenticationStartScreen: View {
            }
            .scrollBounceBehavior(.basedOnSize)
        }
-        .navigationBarHidden(true)
        .background {
            AuthenticationStartScreenBackgroundImage()
        }
+        .navigationBarHidden(context.viewState.classicAppMode == nil)
+        .toolbar { toolbar }
        .alert(item: $context.alertInfo)
-        .sheet(isPresented: $context.showClassicAppBackupInstructions) {
-            AuthenticationClassicAppBackupInstructionsView(context: context)
-        }
        .introspect(.window, on: .supportedVersions) { window in
            context.send(viewAction: .updateWindow(window))
        }
@@ -132,6 +139,17 @@ struct AuthenticationStartScreen: View {
        let shortVersionString = ProcessInfo.isRunningTests ? "0.0.0" : InfoPlistReader.main.bundleShortVersionString
        return Text(L10n.screenOnboardingAppVersion(shortVersionString))
    }
+    
+    @ToolbarContentBuilder
+    var toolbar: some ToolbarContent {
+        ToolbarItem(placement: .primaryAction) {
+            if case let .otherOptions(classicAppAccount) = context.viewState.classicAppMode {
+                ToolbarButton(role: .close) {
+                    context.send(viewAction: .closeOtherOptions(classicAppAccount))
+                }
+            }
+        }
+    }
 }

 // MARK: - Previews
@@ -139,20 +157,32 @@ struct AuthenticationStartScreen: View {
 struct AuthenticationStartScreen_Previews: PreviewProvider, TestablePreview {
    static let viewModel = makeViewModel()
    static let provisionedViewModel = makeViewModel(provisionedServerName: "example.com")
+    static let classicAppViewModel = makeViewModel(hasClassicAppAccount: true)
    
    static var previews: some View {
        AuthenticationStartScreen(context: viewModel.context)
            .previewDisplayName("Default")
        AuthenticationStartScreen(context: provisionedViewModel.context)
            .previewDisplayName("Provisioned")
+        
+        ElementNavigationStack {
+            AuthenticationStartScreen(context: classicAppViewModel.context)
+        }
+        .previewDisplayName("Classic App")
    }
    
-    static func makeViewModel(provisionedServerName: String? = nil) -> AuthenticationStartScreenViewModel {
-        AuthenticationStartScreenViewModel(authenticationService: AuthenticationService.mock,
-                                           provisioningParameters: provisionedServerName.map { .init(accountProvider: $0, loginHint: nil) },
-                                           isBugReportServiceEnabled: true,
-                                           appSettings: ServiceLocator.shared.settings,
-                                           mediaProvider: MediaProviderMock(configuration: .init()),
-                                           userIndicatorController: UserIndicatorControllerMock())
+    static func makeViewModel(provisionedServerName: String? = nil, hasClassicAppAccount: Bool = false) -> AuthenticationStartScreenViewModel {
+        let classicAppAccount = ClassicAppAccount.mockDan
+        classicAppAccount.state.isServerSupported = true
+        classicAppAccount.state.availableSecrets = .complete
+        let classicAppManager: ClassicAppManagerMock? = hasClassicAppAccount ? .init(.init(accounts: [classicAppAccount])) : nil
+        
+        return AuthenticationStartScreenViewModel(authenticationService: AuthenticationService.mock(classicAppManager: classicAppManager),
+                                                  provisioningParameters: provisionedServerName.map { .init(accountProvider: $0, loginHint: nil) },
+                                                  isBugReportServiceEnabled: true,
+                                                  appMediator: AppMediatorMock(),
+                                                  appSettings: ServiceLocator.shared.settings,
+                                                  mediaProvider: MediaProviderMock(configuration: .init()),
+                                                  userIndicatorController: UserIndicatorControllerMock())
    }
 }