mobile-ios/CovidSafe/SelectCountryViewController.swift

//
//  SelectCountryViewController.swift
//  CovidSafe
//
//  Copyright © 2020 Australian Government. All rights reserved.
//

import UIKit
import FlagKit
import SafariServices

class Country {
    @objc var name: String!
    var isoCode: String!
    var phoneCode: String!
    var flag: Flag?
    
    init(name: String, isoCode: String, phoneCode: String, flag: Flag?) {
        self.name = name
        self.isoCode = isoCode
        self.phoneCode = phoneCode
        self.flag = flag
    }
}

class SelectCountryViewController: UITableViewController, UISearchResultsUpdating {
    
    let collation = UILocalizedIndexedCollation.current()
    let searchController = UISearchController(searchResultsController: nil)
    let AU_OPTIONS_KEY = "OptionsForAustralia"

    var countriesTableData: Dictionary<String, [Country]> = [:]
    lazy var countriesSortedByName: [Country] =
        collation.sortedArray(from: CountriesData.countries, collationStringSelector: #selector(getter: Country.name)) as! [Country]
    
    var filteredCountriesTableData: [Country] = []
    var countrySelectionDelegate: CountrySelectionDelegate?
    
    var countriesSectionTitles: [String] = []
    var isSearchBarEmpty = true
    var isFiltering: Bool {
      return searchController.isActive && !isSearchBarEmpty
    }
    
    override func viewDidLoad() {
        super.viewDidLoad()
        let label = UILabel()
        label.text = "Select_country_or_region_headline".localizedString()
        label.font = UIFont.preferredFont(forTextStyle: .headline)
        self.navigationItem.titleView = label
        
        searchController.searchResultsUpdater = self
        searchController.obscuresBackgroundDuringPresentation = false
        definesPresentationContext = true
        
        navigationController?.navigationBar.isTranslucent = false
        navigationController?.navigationBar.barTintColor = UIColor.white
        //Set search bar
        if #available(iOS 11.0, *) {
            navigationItem.searchController = searchController
        } else {
            tableView.tableHeaderView = searchController.searchBar
            searchController.hidesNavigationBarDuringPresentation = false
            searchController.dimsBackgroundDuringPresentation = false
            automaticallyAdjustsScrollViewInsets = false
        }
        
        // Create the data source of the countries table
        let norfolkIslandCountry = Country(name: "country_region_name_au2".localizedString(), isoCode: "AU2", phoneCode: "672", flag: Flag(countryCode: "AU"))
        let australiaCountry = Country(name: "country_region_name_au".localizedString(), isoCode: "AU", phoneCode: "61", flag: Flag(countryCode: "AU"))
        countriesTableData.updateValue([australiaCountry, norfolkIslandCountry], forKey: AU_OPTIONS_KEY)
        countriesSectionTitles.append(" ")
        
        let sectionsTitles = collation.sectionTitles
        
        for country in countriesSortedByName {
            let sectionIndexNumber = collation.section(for: country, collationStringSelector: #selector(getter: Country.name))
            let sectionIndexInitial = sectionsTitles[sectionIndexNumber]
            var sectionCountries: [Country] = countriesTableData[sectionIndexInitial] ?? []
            if sectionCountries.count == 0 {
                let sectionTitle = sectionsTitles[sectionIndexNumber]
                countriesSectionTitles.append(sectionTitle)
            }
            sectionCountries.append(country)
            countriesTableData.updateValue(sectionCountries, forKey: sectionIndexInitial)
        }
    }
    
    override func viewWillAppear(_ animated: Bool) {
        super.viewWillAppear(animated)
        //this is to show the search bar initially
        if #available(iOS 11.0, *) {
            navigationItem.hidesSearchBarWhenScrolling = false
        }        
    }
    
    override func viewDidAppear(_ animated: Bool) {
        super.viewDidAppear(animated)
        if #available(iOS 11.0, *) {
            if !UIAccessibility.isVoiceOverRunning {
                navigationItem.hidesSearchBarWhenScrolling = true
            }
        }
        if UIAccessibility.isVoiceOverRunning {
            UIAccessibility.post(notification: .layoutChanged, argument: self.navigationItem.titleView )
        }
    }
    
    override func viewWillDisappear(_ animated: Bool) {
        super.viewWillDisappear(animated)
        navigationController?.setNavigationBarHidden(true, animated: false)
    }
    
    func updateSearchResults(for searchController: UISearchController) {
        let searchBar = searchController.searchBar
        let searchBarText = searchBar.text?.lowercased() ?? ""
        if (searchBarText == "") {
            isSearchBarEmpty = true
            filteredCountriesTableData = []
            tableView.reloadData()
            return
        }
        isSearchBarEmpty = false
        filteredCountriesTableData = countriesSortedByName.filter({ (country) -> Bool in
            let countryName = country.name.lowercased()
            let searchCriteria = searchBarText.lowercased()
            if countryName.contains(searchCriteria) {
                return true
            }
            return false
        })
        tableView.reloadData()
    }
    
    override func sectionIndexTitles(for tableView: UITableView) -> [String]? {
        if !isFiltering {
            return countriesSectionTitles
        }
        return nil
    }
    
    override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
        var country: Country
        if !isFiltering {
            var sectionIndex = AU_OPTIONS_KEY
            if indexPath.section > 0 {
                sectionIndex = countriesSectionTitles[indexPath.section]
            }
            guard let sectionCountries = countriesTableData[sectionIndex] else {
                return
            }
            country = sectionCountries[indexPath.row]
        } else {
            country = filteredCountriesTableData[indexPath.row]
        }
        countrySelectionDelegate?.setCountry(country: country)
        navigationController?.setNavigationBarHidden(true, animated: false)
        self.navigationController?.popViewController(animated: true)
    }
    
    override func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int) -> String? {
        if !isFiltering {
            if section == 0 {
                return "options_for_australia".localizedString()
            }

            return countriesSectionTitles[section]
        } else {
            return nil
        }
    }
    
    override func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        if !isFiltering {
            var sectionIndex = AU_OPTIONS_KEY
            if section > 0 {
                sectionIndex = countriesSectionTitles[section]
            }
            return countriesTableData[sectionIndex]?.count ?? 0
        } else {
            return filteredCountriesTableData.count
        }
    }
    
    override func numberOfSections(in tableView: UITableView) -> Int {
        if !isFiltering {
            return countriesSectionTitles.count
        } else {
            return 1
        }
    }
    
    override func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "CountryCell", for: indexPath) as! CountryCellViewCell
        var country: Country
        
        if !isFiltering {
            var sectionIndex = AU_OPTIONS_KEY
            if indexPath.section > 0 {
                sectionIndex = countriesSectionTitles[indexPath.section]
            }
            guard let sectionCountries = countriesTableData[sectionIndex] else {
                return cell
            }
            
            country = sectionCountries[indexPath.row]
        } else {
            country = filteredCountriesTableData[indexPath.row]
        }
        
        cell.countryTitleLabel.text = country.name
        cell.countryPhoneLabel.text = "+\(country.phoneCode ?? "")"
        cell.countryFlagImageView.image = country.flag?.originalImage

        return cell
    }
    
}

class CountryCellViewCell: UITableViewCell {
    
    @IBOutlet weak var countryTitleLabel: UILabel!
    @IBOutlet weak var countryPhoneLabel: UILabel!
    @IBOutlet weak var countryFlagImageView: UIImageView!
}