passforios/passKit/Helpers/YubiKeyConnection.swift

//
//  YubiKeyConnection.swift
//  passKit
//
//  Copyright © 2022 Bob Sun. All rights reserved.
//

import Foundation
import YubiKit

public class YubiKeyConnection: NSObject {
    public static let shared = YubiKeyConnection()

    var accessoryConnection: YKFAccessoryConnection?
    var nfcConnection: YKFNFCConnection?
    var connectionCallback: ((_ connection: YKFConnectionProtocol) -> Void)?
    var cancellationCallback: ((_ error: Error) -> Void)?

    override init() {
        super.init()
        if YubiKitDeviceCapabilities.supportsISO7816NFCTags {
            YubiKitManager.shared.delegate = self
            YubiKitManager.shared.startAccessoryConnection()
        }
    }

    public func connection(cancellation: @escaping (_ error: Error) -> Void, completion: @escaping (_ connection: YKFConnectionProtocol) -> Void) {
        if let connection = accessoryConnection {
            completion(connection)
        } else {
            connectionCallback = completion
            YubiKitManager.shared.startNFCConnection()
        }
        cancellationCallback = cancellation
    }
}

extension YubiKeyConnection: YKFManagerDelegate {
    public func didConnectNFC(_ connection: YKFNFCConnection) {
        nfcConnection = connection
        if let callback = connectionCallback {
            callback(connection)
        }
    }

    public func didDisconnectNFC(_: YKFNFCConnection, error _: Error?) {
        nfcConnection = nil
    }

    public func didConnectAccessory(_ connection: YKFAccessoryConnection) {
        accessoryConnection = connection
    }

    public func didDisconnectAccessory(_: YKFAccessoryConnection, error _: Error?) {
        accessoryConnection = nil
    }

    public func didFailConnectingNFC(_ error: Error) {
        if let callback = cancellationCallback {
            callback(error)
        }
    }
}