passforios/passKit/Helpers/KeyFileManager.swift

//
//  KeyFileManager.swift
//  passKit
//
//  Created by Danny Moesch on 29.06.19.
//  Copyright © 2019 Bob Sun. All rights reserved.
//

public class KeyFileManager {
    public typealias KeyHandler = (Data, String) -> ()

    public static let PublicPgp = KeyFileManager(keyType: PgpKey.PUBLIC)
    public static let PrivatePgp = KeyFileManager(keyType: PgpKey.PRIVATE)
    public static let PrivateSsh = KeyFileManager(keyType: SshKey.PRIVATE)

    private let keyType: CryptographicKey
    private let keyPath: String

    private convenience init(keyType: CryptographicKey) {
        self.init(keyType: keyType, keyPath: keyType.getFileSharingPath())
    }

    public init(keyType: CryptographicKey, keyPath: String) {
        self.keyType = keyType
        self.keyPath = keyPath
    }

    public func importKeyAndDeleteFile(keyHandler: KeyHandler = AppKeychain.shared.add) throws {
        guard let keyFileContent = FileManager.default.contents(atPath: keyPath) else {
            throw AppError.ReadingFile(URL(fileURLWithPath: keyPath).lastPathComponent)
        }
        keyHandler(keyFileContent, keyType.getKeychainKey())
        try FileManager.default.removeItem(atPath: keyPath)
    }
    
    public func doesKeyFileExist() -> Bool {
        return FileManager.default.fileExists(atPath: keyPath)
    }
}