passforios/pass/Views/LabelTableViewCell.swift

//
//  LabelTableViewCell.swift
//  pass
//
//  Created by Mingshen Sun on 2/2/2017.
//  Copyright © 2017 Bob Sun. All rights reserved.
//

import UIKit


struct LabelTableViewCellData {
    var title: String
    var content: String
}

class LabelTableViewCell: UITableViewCell {

    @IBOutlet weak var contentLabel: UILabel!
    @IBOutlet weak var titleLabel: UILabel!

    var isPasswordCell = false
    var isURLCell = false
    var isReveal = false
    var password: Password?
    let passwordDots = "••••••••••••"
    
    var cellData: LabelTableViewCellData? {
        didSet {
            titleLabel.text = cellData?.title
            if isPasswordCell {
                contentLabel.text = passwordDots
                contentLabel.font = UIFont(name: "Menlo", size: contentLabel.font.pointSize)
            } else {
                contentLabel.text = cellData?.content
            }
        }
    }
    
    override var canBecomeFirstResponder: Bool {
        get {
            return true
        }
    }
    
    override func awakeFromNib() {
        super.awakeFromNib()
    }

    override func setSelected(_ selected: Bool, animated: Bool) {
        super.setSelected(selected, animated: animated)
    }

    override func canPerformAction(_ action: Selector, withSender sender: Any?) -> Bool {
        if isPasswordCell {
            if isReveal {
                return action == #selector(copy(_:)) || action == #selector(LabelTableViewCell.concealPassword(_:))
            } else {
                return action == #selector(copy(_:)) || action == #selector(LabelTableViewCell.revealPassword(_:))
            }
        }
        if isURLCell {
            return action == #selector(copy(_:)) || action == #selector(LabelTableViewCell.openLink(_:))
        }
        return action == #selector(copy(_:))
    }
    
    override func copy(_ sender: Any?) {
        Utils.copyToPasteboard(textToCopy: cellData?.content)
    }
        
    func revealPassword(_ sender: Any?) {
        if let plainPassword = cellData?.content {
            let attributePassword = NSMutableAttributedString.init(string: plainPassword)
            // draw all digits in the password into red
            // draw all punctuation characters in the password into blue
            for (index, element) in plainPassword.unicodeScalars.enumerated() {
                if NSCharacterSet.decimalDigits.contains(element) {
                    attributePassword.addAttribute(NSForegroundColorAttributeName, value: UIColor.red , range: NSRange(location: index, length: 1))
                } else if NSCharacterSet.punctuationCharacters.contains(element) {
                    attributePassword.addAttribute(NSForegroundColorAttributeName, value: UIColor.blue , range: NSRange(location: index, length: 1))
                }
            }
            // set contentLabel
            contentLabel.attributedText = attributePassword
        } else {
            contentLabel.text = ""
        }
        isReveal = true
    }
    
    func concealPassword(_ sender: Any?) {
        contentLabel.text = passwordDots
        isReveal = false
    }
    
    func openLink(_ sender: Any?) {
        Utils.copyToPasteboard(textToCopy: password?.password)
        UIApplication.shared.open(URL(string: cellData!.content)!, options: [:], completionHandler: nil)
    }
}