passforios-mirror/passAutoFillExtension/Services/PasswordsTableDataSource.swift

//
//  PasswordsTableDataSource.swift
//  passAutoFillExtension
//
//  Created by Sun, Mingshen on 12/31/20.
//  Copyright © 2020 Bob Sun. All rights reserved.
//

import passKit
import UIKit

class PasswordsTableDataSource: NSObject, UITableViewDataSource {
    var passwordTableEntries: [PasswordTableEntry]
    var filteredPasswordsTableEntries: [PasswordTableEntry]
    var suggestedPasswordsTableEntries: [PasswordTableEntry]
    var otherPasswordsTableEntries: [PasswordTableEntry]

    var showSuggestion = false

    init(entries: [PasswordTableEntry] = []) {
        self.passwordTableEntries = entries
        self.filteredPasswordsTableEntries = passwordTableEntries
        self.suggestedPasswordsTableEntries = []
        self.otherPasswordsTableEntries = []
    }

    func numberOfSections(in _: UITableView) -> Int {
        !showSuggestion ? 1 : 2
    }

    func tableView(_: UITableView, numberOfRowsInSection section: Int) -> Int {
        tableEntries(at: section).count
    }

    func tableView(_: UITableView, titleForHeaderInSection section: Int) -> String? {
        if suggestedPasswordsTableEntries.isEmpty {
            return nil
        }
        if !showSuggestion {
            return "All Passwords"
        }
        if section == 0 {
            return "Suggested Passwords"
        }
        return "Other Passwords"
    }

    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "passwordTableViewCell", for: indexPath) as! PasswordTableViewCell

        let entry = tableEntry(at: indexPath)
        cell.configure(with: entry)

        return cell
    }

    func showTableEntries(matching text: String) {
        guard !text.isEmpty else {
            filteredPasswordsTableEntries = passwordTableEntries
            showSuggestion = !suggestedPasswordsTableEntries.isEmpty
            return
        }

        filteredPasswordsTableEntries = passwordTableEntries.filter { $0.matches(text) }
        showSuggestion = false
    }

    func showTableEntriesWithSuggestion(matching text: String) {
        guard !text.isEmpty else {
            filteredPasswordsTableEntries = passwordTableEntries
            showSuggestion = false
            return
        }

        guard suggestedPasswordsTableEntries.isEmpty, otherPasswordsTableEntries.isEmpty else {
            return
        }

        for entry in passwordTableEntries {
            if entry.matches(text) {
                suggestedPasswordsTableEntries.append(entry)
            } else {
                otherPasswordsTableEntries.append(entry)
            }
        }
        showSuggestion = !suggestedPasswordsTableEntries.isEmpty
    }

    func tableEntry(at indexPath: IndexPath) -> PasswordTableEntry {
        tableEntries(at: indexPath.section)[indexPath.row]
    }

    func tableEntries(at section: Int) -> [PasswordTableEntry] {
        if showSuggestion {
            if section == 0 {
                return suggestedPasswordsTableEntries
            }
            return otherPasswordsTableEntries
        }
        return filteredPasswordsTableEntries
    }
}