mastodon-ios/Mastodon/Diffiable/Section/Status/PollSection.swift

//
//  PollSection.swift
//  Mastodon
//
//  Created by MainasuK Cirno on 2021-3-2.
//

import UIKit
import CoreData
import CoreDataStack

import MastodonSDK

extension Mastodon.Entity.Attachment: Hashable {
    public static func == (lhs: Mastodon.Entity.Attachment, rhs: Mastodon.Entity.Attachment) -> Bool {
        return lhs.id == rhs.id
    }
    
    public func hash(into hasher: inout Hasher) {
        hasher.combine(id)
    }
}

enum PollSection: Equatable, Hashable {
    case main
}

extension PollSection {
    static func tableViewDiffableDataSource(
        for tableView: UITableView,
        managedObjectContext: NSManagedObjectContext
    ) -> UITableViewDiffableDataSource<PollSection, PollItem> {
        return UITableViewDiffableDataSource<PollSection, PollItem>(tableView: tableView) { tableView, indexPath, item -> UITableViewCell? in
            switch item {
            case .option(let objectID, let attribute):
                let cell = tableView.dequeueReusableCell(withIdentifier: String(describing: PollOptionTableViewCell.self), for: indexPath) as! PollOptionTableViewCell
                managedObjectContext.performAndWait {
                    let option = managedObjectContext.object(with: objectID) as! PollOption
                    PollSection.configure(cell: cell, pollOption: option, pollItemAttribute: attribute)
                    
                    cell.isAccessibilityElement = true
                    cell.accessibilityLabel = {
                        var labels: [String] = [option.title]
                        if let percentage = cell.pollOptionView.optionPercentageLabel.text {
                            labels.append(percentage)
                        }
                        return labels.joined(separator: ",")
                    }()
                }
                return cell
            }
       }
    }
}

extension PollSection {
    static func configure(
        cell: PollOptionTableViewCell,
        pollOption option: PollOption,
        pollItemAttribute attribute: PollItem.Attribute
    ) {
        cell.pollOptionView.optionTextField.text = option.title
        configure(cell: cell, selectState: attribute.selectState)
        configure(cell: cell, voteState: attribute.voteState)
        cell.attribute = attribute
        cell.layoutIfNeeded()
        cell.updateTextAppearance()
    }
}

extension PollSection {
    
    static func configure(cell: PollOptionTableViewCell, selectState state: PollItem.Attribute.SelectState) {
        switch state {
        case .none:
            cell.pollOptionView.checkmarkBackgroundView.isHidden = true
            cell.pollOptionView.checkmarkImageView.isHidden = true
        case .off:
            ThemeService.shared.currentTheme
                .receive(on: DispatchQueue.main)
                .sink { [weak cell] theme in
                    guard let cell = cell else { return }
                    cell.pollOptionView.checkmarkBackgroundView.backgroundColor = theme.tertiarySystemBackgroundColor
                    cell.pollOptionView.checkmarkBackgroundView.layer.borderColor = theme.tableViewCellSelectionBackgroundColor.withAlphaComponent(0.3).cgColor
                }
                .store(in: &cell.disposeBag)
            cell.pollOptionView.checkmarkBackgroundView.layer.borderWidth = 1
            cell.pollOptionView.checkmarkBackgroundView.isHidden = false
            cell.pollOptionView.checkmarkImageView.isHidden = true
        case .on:
            ThemeService.shared.currentTheme
                .receive(on: DispatchQueue.main)
                .sink { [weak cell] theme in
                    guard let cell = cell else { return }
                    cell.pollOptionView.checkmarkBackgroundView.backgroundColor = theme.tertiarySystemBackgroundColor
                }
                .store(in: &cell.disposeBag)
            cell.pollOptionView.checkmarkBackgroundView.layer.borderColor = UIColor.clear.cgColor
            cell.pollOptionView.checkmarkBackgroundView.layer.borderWidth = 0
            cell.pollOptionView.checkmarkBackgroundView.isHidden = false
            cell.pollOptionView.checkmarkImageView.isHidden = false
        }
    }

    static func configure(cell: PollOptionTableViewCell, voteState state: PollItem.Attribute.VoteState) {
        switch state {
        case .hidden:
            cell.pollOptionView.optionPercentageLabel.isHidden = true
            cell.pollOptionView.voteProgressStripView.isHidden = true
            cell.pollOptionView.voteProgressStripView.setProgress(0.0, animated: false)
        case .reveal(let voted, let percentage, let animated):
            cell.pollOptionView.optionPercentageLabel.isHidden = false
            cell.pollOptionView.optionPercentageLabel.text = String(Int(100 * percentage)) + "%"
            cell.pollOptionView.voteProgressStripView.isHidden = false
            cell.pollOptionView.voteProgressStripView.tintColor = voted ? Asset.Colors.brandBlue.color : Asset.Colors.Poll.disabled.color
            cell.pollOptionView.voteProgressStripView.setProgress(CGFloat(percentage), animated: animated)
        }
    }
    
}