passforios/pass/Controllers/PGPKeyUrlImportTableViewController.swift

//
//  PGPKeyUrlImportTableViewController.swift
//  pass
//
//  Created by Mingshen Sun on 21/1/2017.
//  Copyright © 2017 Bob Sun. All rights reserved.
//

import UIKit
import passKit

class PGPKeyUrlImportTableViewController: AutoCellHeightUITableViewController {

    @IBOutlet weak var pgpPublicKeyURLTextField: UITextField!
    @IBOutlet weak var pgpPrivateKeyURLTextField: UITextField!

    var pgpPrivateKeyURL: URL?
    var pgpPublicKeyURL: URL?

    override func viewDidLoad() {
        super.viewDidLoad()
        pgpPublicKeyURLTextField.text = Defaults.pgpPublicKeyURL?.absoluteString
        pgpPrivateKeyURLTextField.text = Defaults.pgpPrivateKeyURL?.absoluteString
    }
    
    @IBAction func save(_ sender: Any) {
        guard let publicKeyURLText = pgpPublicKeyURLTextField.text,
            let publicKeyURL = URL(string: publicKeyURLText),
            let privateKeyURLText = pgpPrivateKeyURLTextField.text,
            let privateKeyURL = URL(string: privateKeyURLText) else {
                Utils.alert(title: "CannotSavePgpKey".localize(), message: "SetPgpKeyUrlsFirst.".localize(), controller: self)
                return
        }
        if privateKeyURL.scheme?.lowercased() == "http" || publicKeyURL.scheme?.lowercased() == "http" {
            Utils.alert(title: "HttpNotSecure".localize(), message: "ReallyUseHttp.".localize(), controller: self)
        }
        pgpPrivateKeyURL = privateKeyURL
        pgpPublicKeyURL = publicKeyURL
        self.saveImportedKeys()
    }
}

extension PGPKeyUrlImportTableViewController: PGPKeyImporter {

    static let keySource = KeySource.url
    static let label = "DownloadFromUrl".localize()

    func isReadyToUse() -> Bool {
        return validate(pgpKeyUrl: pgpPublicKeyURLTextField.text ?? "")
            && validate(pgpKeyUrl: pgpPrivateKeyURLTextField.text ?? "")
    }

    func importKeys() throws {
        Defaults.pgpPrivateKeyURL = pgpPrivateKeyURL
        Defaults.pgpPublicKeyURL = pgpPublicKeyURL

        try KeyFileManager.PublicPgp.importKey(from: Defaults.pgpPublicKeyURL!)
        try KeyFileManager.PrivatePgp.importKey(from: Defaults.pgpPrivateKeyURL!)
    }

    func doAfterImport() {
        Utils.alert(title: "RememberToRemoveKey".localize(), message: "RememberToRemoveKeyFromServer.".localize(), controller: self)
    }

    func saveImportedKeys() {
        performSegue(withIdentifier: "savePGPKeySegue", sender: self)
    }

    private func validate(pgpKeyUrl: String) -> Bool {
        guard let url = URL(string: pgpKeyUrl) else {
            Utils.alert(title: "CannotSavePgpKey".localize(), message: "SetPgpKeyUrlsFirst.".localize(), controller: self)
            return false
        }
        guard url.scheme == "https" || url.scheme == "http" else {
            Utils.alert(title: "CannotSavePgpKey".localize(), message: "UseEitherHttpsOrHttp.".localize(), controller: self)
            return false
        }
        return true
    }
}