ios-mail/Modules/App/Sources/UI/Screens/AppSettings/AppProtection/AppProtectionSelectionScreen.swift

// Copyright (c) 2025 Proton Technologies AG
//
// This file is part of Proton Mail.
//
// Proton Mail is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Proton Mail is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Proton Mail. If not, see https://www.gnu.org/licenses/.

import InboxCore
import InboxCoreUI
import InboxDesignSystem
import ProtonUIFoundations
import SwiftUI
import proton_app_uniffi

struct AppProtectionSelectionScreen: View {
    private let state: AppProtectionSelectionState
    private let appSettingsRepository: AppSettingsRepository
    private let appProtectionConfigurator: AppProtectionConfigurator
    @EnvironmentObject var settingsRouter: Router<SettingsRoute>

    init(
        state: AppProtectionSelectionState = .initial,
        appSettingsRepository: AppSettingsRepository = AppContext.shared.mailSession,
        appProtectionConfigurator: AppProtectionConfigurator = AppContext.shared.mailSession
    ) {
        self.state = state
        self.appSettingsRepository = appSettingsRepository
        self.appProtectionConfigurator = appProtectionConfigurator
    }

    var body: some View {
        StoreView(
            store: AppProtectionSelectionStore(
                state: state,
                router: settingsRouter,
                appSettingsRepository: appSettingsRepository,
                appProtectionConfigurator: appProtectionConfigurator
            )
        ) { state, store in
            ScrollView {
                VStack(spacing: .zero) {
                    FormSection(footer: L10n.Settings.App.protectionSelectionListFooterInformation) {
                        FormList(
                            collection: state.availableAppProtectionMethods,
                            style: .separator(.normLeftPadding)
                        ) { viewModel in
                            FormSmallButton(
                                title: viewModel.type.name,
                                rightSymbol: viewModel.isSelected ? .checkmark : nil
                            ) {
                                store.handle(action: .selected(viewModel.type))
                            }
                        }
                    }
                    if state.shouldShowChangePINButton {
                        FormSection {
                            FormSmallButton(title: L10n.Settings.App.changePINcode, rightSymbol: .chevronRight) {
                                store.handle(action: .changePINTapped)
                            }
                            .roundedRectangleStyle()
                        }
                        .animation(.easeInOut, value: state.shouldShowChangePINButton)
                        .padding(.top, DS.Spacing.standard)
                    }
                    if state.shouldShowAutoLockButton {
                        FormSection {
                            FormBigButton(
                                title: L10n.Settings.App.autoLock,
                                symbol: .chevronRight,
                                value: state.autoLock?.humanReadable.string ?? .empty
                            ) {
                                store.handle(action: .autoLockTapped)
                            }
                        }
                        .animation(.easeInOut, value: state.shouldShowAutoLockButton)
                    }
                    Spacer()
                }
            }
            .padding(.horizontal, DS.Spacing.large)
            .background(DS.Color.BackgroundInverted.norm)
            .navigationTitle(L10n.Settings.App.appLock.string)
            .navigationBarTitleDisplayMode(.inline)
            .onAppear { store.handle(action: .onAppear) }
            .sheet(item: presentPINScreen(state: state, store: store)) { pinScreenType in
                PINRouterView(type: pinScreenType)
            }
        }
    }

    private func presentPINScreen(
        state: AppProtectionSelectionState,
        store: AppProtectionSelectionStore
    ) -> Binding<PINScreenType?> {
        .init(
            get: { state.presentedPINScreen },
            set: { newValue in
                if let newValue {
                    store.handle(action: .pinScreenPresented(newValue))
                } else {
                    store.handle(action: .pinScreenDismissed)
                }
            }
        )
    }
}

#Preview {
    NavigationStack {
        AppProtectionSelectionScreen(
            state: .init(
                currentProtection: .biometrics,
                availableAppProtectionMethods: [
                    .init(type: .none, isSelected: false),
                    .init(type: .pin, isSelected: false),
                    .init(type: .faceID, isSelected: true),
                ]
            ),
            appSettingsRepository: MailSession(noPointer: .init()),
            appProtectionConfigurator: MailSession(noPointer: .init())
        )
    }
}

private extension AppProtectionSelectionState {
    var shouldShowAutoLockButton: Bool {
        currentProtection != .none
    }

    var shouldShowChangePINButton: Bool {
        currentProtection == .pin
    }
}