import UIKit
import ExpandablePicker

struct Project {
    let id: String
    let title: String
    let parentId: String?
}

class ViewController: UIViewController {
    
    var myPretendData = [Project]()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // my pretend api call
        // populate my data
        let deathStar = Project(id: "1", title: "Death Star", parentId: nil)
        let detentionBlock = Project(id: "10", title: "Detention Block", parentId: "1")
        let cellaa23 = Project(id: "100", title: "Cell AA-23", parentId: "10")
        let emperorsThroneRoom = Project(id: "11", title: "Emperor's Throne Room", parentId: "1")
        let tractorBeamControl = Project(id: "12", title: "Tractor Beam Control", parentId: "1")
        let hangar272 = Project(id: "13", title: "Hangar 272", parentId: "1")
        
        // if you want barcode / qr code scanning, add `scanMatchable:` paramter.  (any string you pass in that array will match barcode/qrcode)
        
        // does not matter what order items are placed
        myPretendData.append(contentsOf: [hangar272, cellaa23, detentionBlock, tractorBeamControl, deathStar, emperorsThroneRoom])
    }
    
    @IBAction func lineVersionPressed(_ sender: Any) {
        
        // convert data to picker data
        let pickerData = myPretendData.map { (project) -> PickerData in
            PickerData(id: project.id, title: project.title, parentId: project.parentId)
        }
        
        let vc = PickerViewController()
        vc.set(title: "My Storm Trooper", subtitle: "Death to all rebels")
        vc.datasource.data = pickerData
        vc.delegate = self
        let nc = UINavigationController(rootViewController: vc)
        present(nc, animated: true, completion: nil)
    }
}

extension ViewController: PickerProtocol {
    
    func didSelectPickerData(with id: String, pvc: PickerViewController) {
        
        guard let d = myPretendData.filter({ $0.id == id }).first else { return }
        
        pvc.dismiss(animated: true) { [weak self] in
            guard let s = self else { return }
            let alert = UIAlertController(title: "Selected", message: "\(d.title) was selected.", preferredStyle: .alert)
            s.present(alert, animated: true, completion: nil)
            DispatchQueue.main.asyncAfter(deadline: .now() + 1.0) {
                alert.dismiss(animated: true, completion: nil)
            }
        }
    }
}