Swiftgram/TelegramCore/SecureIdValue.swift

import Foundation

public enum SecureIdValueKey: Int32 {
    case personalDetails
    case passport
    case driversLicense
    case idCard
    case address
    case utilityBill
    case bankStatement
    case rentalAgreement
    case phone
    case email
}

public enum SecureIdValue: Equatable {
    case personalDetails(SecureIdPersonalDetailsValue)
    case passport(SecureIdPassportValue)
    case driversLicense(SecureIdDriversLicenseValue)
    case idCard(SecureIdIDCardValue)
    case address(SecureIdAddressValue)
    case utilityBill(SecureIdUtilityBillValue)
    case bankStatement(SecureIdBankStatementValue)
    case rentalAgreement(SecureIdRentalAgreementValue)
    case phone(SecureIdPhoneValue)
    case email(SecureIdEmailValue)
    
    var fileReferences: [SecureIdVerificationDocumentReference] {
        switch self {
            case let .passport(passport):
                var result = passport.verificationDocuments
                if let selfie = passport.selfieDocument {
                    result.append(selfie)
                }
                return result
            case let .driversLicense(driversLicense):
                var result = driversLicense.verificationDocuments
                if let selfie = driversLicense.selfieDocument {
                    result.append(selfie)
                }
                return result
            case let .idCard(idCard):
                var result = idCard.verificationDocuments
                if let selfie = idCard.selfieDocument {
                    result.append(selfie)
                }
                return result
            case let .bankStatement(bankStatement):
                return bankStatement.verificationDocuments
            case let .utilityBill(utilityBill):
                return utilityBill.verificationDocuments
            case let .rentalAgreement(rentalAgreement):
                return rentalAgreement.verificationDocuments
            default:
                return []
        }
    }
    
    public var key: SecureIdValueKey {
        switch self {
            case .personalDetails:
                return .personalDetails
            case .passport:
                return .passport
            case .driversLicense:
                return .driversLicense
            case .idCard:
                return .idCard
            case .address:
                return .address
            case .utilityBill:
                return .utilityBill
            case .bankStatement:
                return .bankStatement
            case .rentalAgreement:
                return .rentalAgreement
            case .phone:
                return .phone
            case .email:
                return .email
        }
    }
    
    func serialize() -> ([String: Any], [SecureIdVerificationDocumentReference], SecureIdVerificationDocumentReference?)? {
        switch self {
            case let .personalDetails(personalDetails):
                let (dict, files) = personalDetails.serialize()
                return (dict, files, nil)
            case let .passport(passport):
                return passport.serialize()
            case let .driversLicense(driversLicense):
                return driversLicense.serialize()
            case let .idCard(idCard):
                return idCard.serialize()
            case let .address(address):
                let (dict, files) = address.serialize()
                return (dict, files, nil)
            case let .utilityBill(utilityBill):
                let (dict, files) = utilityBill.serialize()
                return (dict, files, nil)
            case let .bankStatement(bankStatement):
                let (dict, files) = bankStatement.serialize()
                return (dict, files, nil)
            case let .rentalAgreement(rentalAgreement):
                let (dict, files) = rentalAgreement.serialize()
                return (dict, files, nil)
            case .phone:
                return nil
            case .email:
                return nil
        }
    }
}

struct SecureIdEncryptedValueFileMetadata: Equatable {
    let hash: Data
    let secret: Data
}

struct SecureIdEncryptedValueMetadata: Equatable {
    let valueDataHash: Data
    let decryptedSecret: Data
    let files: [SecureIdEncryptedValueFileMetadata]
}

public struct SecureIdValueWithContext: Equatable {
    public let value: SecureIdValue
    let encryptedMetadata: SecureIdEncryptedValueMetadata?
    let opaqueHash: Data
    
    init(value: SecureIdValue, encryptedMetadata: SecureIdEncryptedValueMetadata?, opaqueHash: Data) {
        self.value = value
        self.encryptedMetadata = encryptedMetadata
        self.opaqueHash = opaqueHash
    }
}