// // PasswordsViewController.swift // pass // // Created by Mingshen Sun on 3/2/2017. // Copyright © 2017 Bob Sun. All rights reserved. // import UIKit import Result import SVProgressHUD class PasswordsViewController: UIViewController, UITableViewDataSource, UITableViewDelegate { private var passwordEntities: [PasswordEntity]? var filteredPasswordEntities = [PasswordEntity]() var sections : [(index: Int, length :Int, title: String)] = Array() var searchActive : Bool = false let searchController = UISearchController(searchResultsController: nil) lazy var refreshControl: UIRefreshControl = { let refreshControl = UIRefreshControl() refreshControl.addTarget(self, action: #selector(PasswordsViewController.handleRefresh(_:)), for: UIControlEvents.valueChanged) refreshControl.attributedTitle = NSAttributedString(string: "Sync Password Store") return refreshControl }() let searchBarView = UIView(frame: CGRect(x: 0, y: 64, width: UIScreen.main.bounds.width, height: 44)) @IBOutlet weak var tableView: UITableView! @IBAction func refreshPasswords(_ sender: UIBarButtonItem) { syncPasswords() } func syncPasswords() { SVProgressHUD.setDefaultMaskType(.black) SVProgressHUD.show(withStatus: "Sync Passwords") DispatchQueue.global(qos: .userInitiated).async { [unowned self] in do { try PasswordStore.shared.pullRepository(transferProgressBlock: {(git_transfer_progress, stop) in DispatchQueue.main.async { SVProgressHUD.showProgress(Float(git_transfer_progress.pointee.received_objects)/Float(git_transfer_progress.pointee.total_objects), status: "Pull Remote Repository") } }) DispatchQueue.main.async { SVProgressHUD.showSuccess(withStatus: "Done") SVProgressHUD.dismiss(withDelay: 1) } print("pull success") self.passwordEntities = PasswordStore.shared.fetchPasswordEntityCoreData() self.reloadTableView(data: self.passwordEntities!) } catch { DispatchQueue.main.async { SVProgressHUD.showError(withStatus: error.localizedDescription) SVProgressHUD.dismiss(withDelay: 3) } } } } override func viewDidLoad() { super.viewDidLoad() passwordEntities = PasswordStore.shared.fetchPasswordEntityCoreData() NotificationCenter.default.addObserver(self, selector: #selector(PasswordsViewController.actOnPasswordUpdatedNotification), name: NSNotification.Name(rawValue: "passwordUpdated"), object: nil) generateSections(item: passwordEntities!) tableView.delegate = self tableView.dataSource = self searchController.searchResultsUpdater = self searchController.dimsBackgroundDuringPresentation = false searchController.searchBar.sizeToFit() definesPresentationContext = true searchBarView.addSubview(searchController.searchBar) view.addSubview(searchBarView) tableView.insertSubview(refreshControl, at: 0) } override func viewWillAppear(_ animated: Bool) { super.viewWillAppear(animated) if let path = tableView.indexPathForSelectedRow { tableView.deselectRow(at: path, animated: false) } } func numberOfSections(in tableView: UITableView) -> Int { return sections.count } func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int { return sections[section].length } func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell { let cell = tableView.dequeueReusableCell(withIdentifier: "passwordTableViewCell", for: indexPath) var password: PasswordEntity let index = sections[indexPath.section].index + indexPath.row if searchController.isActive && searchController.searchBar.text != "" { password = filteredPasswordEntities[index] } else { password = passwordEntities![index] } cell.textLabel?.text = password.name return cell } func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int) -> String? { return sections[section].title } func sectionIndexTitles(for tableView: UITableView) -> [String]? { return sections.map { $0.title } } func tableView(_ tableView: UITableView, sectionForSectionIndexTitle title: String, at index: Int) -> Int { return index } func tableView(_ tableView: UITableView, accessoryButtonTappedForRowWith indexPath: IndexPath) { let index = sections[indexPath.section].index + indexPath.row let password: PasswordEntity if searchController.isActive && searchController.searchBar.text != "" { password = filteredPasswordEntities[index] } else { password = passwordEntities![index] } let decryptedPassword = password.decrypt()! UIPasteboard.general.string = decryptedPassword.password } func generateSections(item: [PasswordEntity]) { sections.removeAll() if item.count == 0 { return } var index = 0 for i in 0 ..< item.count { let name = item[index].name!.uppercased() let commonPrefix = item[i].name!.commonPrefix(with: name, options: .caseInsensitive) if commonPrefix.characters.count == 0 { let firstCharacter = name[name.startIndex] let newSection = (index: index, length: i - index, title: "\(firstCharacter)") sections.append(newSection) index = i } } let name = item[index].name!.uppercased() let firstCharacter = name[name.startIndex] let newSection = (index: index, length: item.count - index, title: "\(firstCharacter)") sections.append(newSection) } func actOnPasswordUpdatedNotification() { passwordEntities = PasswordStore.shared.fetchPasswordEntityCoreData() reloadTableView(data: passwordEntities!) print("actOnPasswordUpdatedNotification") } override func prepare(for segue: UIStoryboardSegue, sender: Any?) { if segue.identifier == "showPasswordDetail" { if let viewController = segue.destination as? PasswordDetailTableViewController { let selectedIndexPath = self.tableView.indexPath(for: sender as! UITableViewCell)! let index = sections[selectedIndexPath.section].index + selectedIndexPath.row let password: PasswordEntity if searchController.isActive && searchController.searchBar.text != "" { password = filteredPasswordEntities[index] } else { password = passwordEntities![index] } viewController.passwordEntity = password viewController.navigationItem.title = password.name } } } func filterContentForSearchText(searchText: String, scope: String = "All") { filteredPasswordEntities = passwordEntities!.filter { password in return password.name!.lowercased().contains(searchText.lowercased()) } if searchController.isActive && searchController.searchBar.text != "" { reloadTableView(data: filteredPasswordEntities) } else { reloadTableView(data: passwordEntities!) } } func reloadTableView (data: [PasswordEntity]) { generateSections(item: data) tableView.reloadData() } func handleRefresh(_ refreshControl: UIRefreshControl) { syncPasswords() refreshControl.endRefreshing() } } extension PasswordsViewController: UISearchResultsUpdating { func updateSearchResults(for searchController: UISearchController) { filterContentForSearchText(searchText: searchController.searchBar.text!) } }