From 625bbd339de7624c01a905d8173774aab0ca5c62 Mon Sep 17 00:00:00 2001
From: Matt Czech <matt@livefront.com>
Date: Tue, 17 Dec 2024 16:41:04 -0600
Subject: [PATCH] PM-15435: Use TOTP manual entry in extension when camera
 isn't supported (#1202)

---
 .../AddEditItem/AddEditItemProcessor.swift    |  4 +++-
 .../AddEditItemProcessorTests.swift           | 10 +++++++++
 .../AuthenticatorKeyCaptureCoordinator.swift  | 10 ++++++++-
 ...henticatorKeyCaptureCoordinatorTests.swift | 21 +++++++++++++++++--
 .../VaultItem/VaultItemCoordinator.swift      |  2 ++
 5 files changed, 43 insertions(+), 4 deletions(-)

diff --git a/BitwardenShared/UI/Vault/VaultItem/AddEditItem/AddEditItemProcessor.swift b/BitwardenShared/UI/Vault/VaultItem/AddEditItem/AddEditItemProcessor.swift
index 206b49ec3..08d2f2aac 100644
--- a/BitwardenShared/UI/Vault/VaultItem/AddEditItem/AddEditItemProcessor.swift
+++ b/BitwardenShared/UI/Vault/VaultItem/AddEditItem/AddEditItemProcessor.swift
@@ -709,7 +709,9 @@ final class AddEditItemProcessor: StateProcessor<// swiftlint:disable:this type_
     /// Kicks off the TOTP setup flow.
     ///
     private func setupTotp() async {
-        guard services.cameraService.deviceSupportsCamera() else {
+        guard services.cameraService.deviceSupportsCamera(),
+              appExtensionDelegate?.isInAppExtension != true // Extensions don't allow camera access.
+        else {
             coordinator.navigate(to: .setupTotpManual, context: self)
             return
         }
diff --git a/BitwardenShared/UI/Vault/VaultItem/AddEditItem/AddEditItemProcessorTests.swift b/BitwardenShared/UI/Vault/VaultItem/AddEditItem/AddEditItemProcessorTests.swift
index dac3a4d8c..9a7380fee 100644
--- a/BitwardenShared/UI/Vault/VaultItem/AddEditItem/AddEditItemProcessorTests.swift
+++ b/BitwardenShared/UI/Vault/VaultItem/AddEditItem/AddEditItemProcessorTests.swift
@@ -1393,6 +1393,16 @@ class AddEditItemProcessorTests: BitwardenTestCase {
         XCTAssertEqual(coordinator.routes.last, .setupTotpManual)
     }
 
+    /// `perform(_:)` with `.setupTotpPressed` when in the app extension navigates to the
+    /// `.setupTotpManual` route.
+    @MainActor
+    func test_perform_setupTotpPressed_extension() async {
+        appExtensionDelegate.isInAppExtension = true
+        await subject.perform(.setupTotpPressed)
+
+        XCTAssertEqual(coordinator.routes.last, .setupTotpManual)
+    }
+
     /// `receive(_:)` with `authKeyVisibilityTapped` updates the value in the state.
     @MainActor
     func test_receive_authKeyVisibilityTapped() {
diff --git a/BitwardenShared/UI/Vault/VaultItem/AuthenticatorKeyCapture/AuthenticatorKeyCaptureCoordinator.swift b/BitwardenShared/UI/Vault/VaultItem/AuthenticatorKeyCapture/AuthenticatorKeyCaptureCoordinator.swift
index 41f74ea38..a9a2269f9 100644
--- a/BitwardenShared/UI/Vault/VaultItem/AuthenticatorKeyCapture/AuthenticatorKeyCaptureCoordinator.swift
+++ b/BitwardenShared/UI/Vault/VaultItem/AuthenticatorKeyCapture/AuthenticatorKeyCaptureCoordinator.swift
@@ -48,6 +48,9 @@ final class AuthenticatorKeyCaptureCoordinator: Coordinator, HasStackNavigator {
 
     // MARK: Private Properties
 
+    /// A delegate used to communicate with the app extension.
+    private weak var appExtensionDelegate: AppExtensionDelegate?
+
     /// A delegate that responds to events in this coordinator.
     private weak var delegate: AuthenticatorKeyCaptureDelegate?
 
@@ -64,15 +67,18 @@ final class AuthenticatorKeyCaptureCoordinator: Coordinator, HasStackNavigator {
     /// Creates a new `AuthenticatorKeyCaptureCoordinator`.
     ///
     /// - Parameters:
+    ///   - appExtensionDelegate: A delegate used to communicate with the app extension.
     ///   - delegate: An optional delegate that responds to events in this coordinator.
     ///   - services: The services used by this coordinator.
     ///   - stackNavigator: The stack navigator that is managed by this coordinator.
     ///
     init(
+        appExtensionDelegate: AppExtensionDelegate?,
         delegate: AuthenticatorKeyCaptureDelegate?,
         services: Services,
         stackNavigator: StackNavigator
     ) {
+        self.appExtensionDelegate = appExtensionDelegate
         self.delegate = delegate
         self.services = services
         self.stackNavigator = stackNavigator
@@ -168,11 +174,13 @@ final class AuthenticatorKeyCaptureCoordinator: Coordinator, HasStackNavigator {
     /// Shows the totp manual setup screen.
     ///
     private func showManualTotp() {
+        let deviceSupportsCamera = services.cameraService.deviceSupportsCamera() &&
+            appExtensionDelegate?.isInAppExtension != true  // Extensions don't allow camera access.
         let processor = ManualEntryProcessor(
             coordinator: asAnyCoordinator(),
             services: services,
             state: DefaultEntryState(
-                deviceSupportsCamera: services.cameraService.deviceSupportsCamera()
+                deviceSupportsCamera: deviceSupportsCamera
             )
         )
         let view = ManualEntryView(
diff --git a/BitwardenShared/UI/Vault/VaultItem/AuthenticatorKeyCapture/AuthenticatorKeyCaptureCoordinatorTests.swift b/BitwardenShared/UI/Vault/VaultItem/AuthenticatorKeyCapture/AuthenticatorKeyCaptureCoordinatorTests.swift
index 34f7fedc3..d18d6502d 100644
--- a/BitwardenShared/UI/Vault/VaultItem/AuthenticatorKeyCapture/AuthenticatorKeyCaptureCoordinatorTests.swift
+++ b/BitwardenShared/UI/Vault/VaultItem/AuthenticatorKeyCapture/AuthenticatorKeyCaptureCoordinatorTests.swift
@@ -7,6 +7,7 @@ import XCTest
 class AuthenticatorKeyCaptureCoordinatorTests: BitwardenTestCase {
     // MARK: Properties
 
+    var appExtensionDelegate: MockAppExtensionDelegate!
     var cameraService: MockCameraService!
     var delegate: MockAuthenticatorKeyCaptureDelegate!
     var errorReporter: MockErrorReporter!
@@ -18,12 +19,14 @@ class AuthenticatorKeyCaptureCoordinatorTests: BitwardenTestCase {
     override func setUp() {
         super.setUp()
 
+        appExtensionDelegate = MockAppExtensionDelegate()
         cameraService = MockCameraService()
         delegate = MockAuthenticatorKeyCaptureDelegate()
         errorReporter = MockErrorReporter()
         stackNavigator = MockStackNavigator()
 
         subject = AuthenticatorKeyCaptureCoordinator(
+            appExtensionDelegate: appExtensionDelegate,
             delegate: delegate,
             services: ServiceContainer.withMocks(
                 cameraService: cameraService,
@@ -38,6 +41,7 @@ class AuthenticatorKeyCaptureCoordinatorTests: BitwardenTestCase {
     override func tearDown() {
         super.tearDown()
 
+        appExtensionDelegate = nil
         cameraService = nil
         errorReporter = nil
         stackNavigator = nil
@@ -92,8 +96,21 @@ class AuthenticatorKeyCaptureCoordinatorTests: BitwardenTestCase {
 
         let action = try XCTUnwrap(stackNavigator.actions.last)
         XCTAssertEqual(action.type, .replaced)
-        let view = action.view as? (any View)
-        XCTAssertNotNil(try? view?.inspect().find(ManualEntryView.self))
+        let view = try XCTUnwrap(action.view as? ManualEntryView)
+        XCTAssertTrue(view.store.state.deviceSupportsCamera)
+    }
+
+    /// `navigate(to:)` with `.setupTotpManual` presents the manual entry view in the extension.
+    @MainActor
+    func test_navigateTo_setupTotpManual_extension() throws {
+        appExtensionDelegate.isInAppExtension = true
+        cameraService.deviceHasCamera = true
+        subject.navigate(to: .manualKeyEntry)
+
+        let action = try XCTUnwrap(stackNavigator.actions.last)
+        XCTAssertEqual(action.type, .replaced)
+        let view = try XCTUnwrap(action.view as? ManualEntryView)
+        XCTAssertFalse(view.store.state.deviceSupportsCamera)
     }
 
     /// `navigate(to:)` with `.setupTotpManual` presents the manual entry view.
diff --git a/BitwardenShared/UI/Vault/VaultItem/VaultItemCoordinator.swift b/BitwardenShared/UI/Vault/VaultItem/VaultItemCoordinator.swift
index 5bd8f7a00..04bf1f2fe 100644
--- a/BitwardenShared/UI/Vault/VaultItem/VaultItemCoordinator.swift
+++ b/BitwardenShared/UI/Vault/VaultItem/VaultItemCoordinator.swift
@@ -201,6 +201,7 @@ class VaultItemCoordinator: NSObject, Coordinator, HasStackNavigator { // swiftl
     private func showCamera(delegate: AuthenticatorKeyCaptureDelegate) async {
         let navigationController = UINavigationController()
         let coordinator = AuthenticatorKeyCaptureCoordinator(
+            appExtensionDelegate: appExtensionDelegate,
             delegate: delegate,
             services: services,
             stackNavigator: navigationController
@@ -339,6 +340,7 @@ class VaultItemCoordinator: NSObject, Coordinator, HasStackNavigator { // swiftl
     private func showManualTotp(delegate: AuthenticatorKeyCaptureDelegate) {
         let navigationController = UINavigationController()
         let coordinator = AuthenticatorKeyCaptureCoordinator(
+            appExtensionDelegate: appExtensionDelegate,
             delegate: delegate,
             services: services,
             stackNavigator: navigationController