SimpleX-Chat/apps/ios/Shared/Views/Chat/VerifyCodeView.swift

//
//  VerifyCodeView.swift
//  SimpleX (iOS)
//
//  Created by Evgeny on 10/12/2022.
//  Copyright © 2022 SimpleX Chat. All rights reserved.
//

import SwiftUI

struct VerifyCodeView: View {
    @Environment(\.dismiss) var dismiss: DismissAction
    @EnvironmentObject var theme: AppTheme
    var displayName: String
    @State var connectionCode: String?
    @State var connectionVerified: Bool
    var verify: (String?) -> (Bool, String)?
    @State private var showCodeError = false

    var body: some View {
        if let code = connectionCode {
            verifyCodeView(code)
        }
    }

    private func verifyCodeView(_ code: String) -> some View {
        let splitCode = splitToParts(code, length: 24)
        return List {
            Section {
                QRCode(uri: code, small: true)

                Text(splitCode)
                    .multilineTextAlignment(.leading)
                    .font(.body.monospaced())
                    .lineLimit(20)
                    .frame(maxWidth: .infinity, alignment: .center)
            } header: {
                if connectionVerified {
                    HStack {
                        Image(systemName: "checkmark.shield").foregroundColor(theme.colors.secondary)
                        Text("\(displayName) is verified").textCase(.none)
                    }
                } else {
                    Text("\(displayName) is not verified").textCase(.none)
                }
            } footer: {
                Text("To verify end-to-end encryption with your contact compare (or scan) the code on your devices.")
            }

            Section {
                if connectionVerified {
                    Button {
                        verifyCode(nil)
                    } label: {
                        Label("Clear verification", systemImage: "shield")
                    }
                } else {
                    NavigationLink {
                        ScanCodeView(connectionVerified: $connectionVerified, verify: verify)
                            .navigationBarTitleDisplayMode(.large)
                            .navigationTitle("Scan code")
                            .modifier(ThemedBackground())
                    } label: {
                        Label("Scan code", systemImage: "qrcode")
                    }
                    Button {
                        verifyCode(code) { verified in
                            if !verified { showCodeError = true }
                        }
                    } label: {
                        Label("Mark verified", systemImage: "checkmark.shield")
                    }
                    .alert(isPresented: $showCodeError) {
                        Alert(title: Text("Incorrect security code!"))
                    }
                }
            }
        }
        .toolbar {
            ToolbarItem(placement: .navigationBarTrailing) {
                Button {
                    showShareSheet(items: [splitCode])
                } label: {
                    Image(systemName: "square.and.arrow.up")
                }
            }
        }
        .onChange(of: connectionVerified) { _ in
            if connectionVerified { dismiss() }
        }
    }

    private func verifyCode(_ code: String?, _ cb: ((Bool) -> Void)? = nil) {
        if let (verified, existingCode) = verify(code) {
            connectionVerified = verified
            connectionCode = existingCode
            cb?(verified)
        }
    }

    private func splitToParts(_ s: String, length: Int) -> String {
        if length >= s.count { return s }
        return (0 ... (s.count - 1) / length)
            .map { s.dropFirst($0 * length).prefix(length) }
            .joined(separator: "\n")
    }
}

struct VerifyCodeView_Previews: PreviewProvider {
    static var previews: some View {
        VerifyCodeView(displayName: "alice", connectionCode: "12345 67890 12345 67890", connectionVerified: false, verify: {_ in nil})
            .environmentObject(CurrentColors.toAppTheme())
    }
}