//
//  UIView.swift
//  VDS
//
//  Created by Matt Bruce on 11/17/22.
//

import Foundation
import UIKit
import VDSFormControlsTokens

extension UIView {
    public enum ConstraintIdentifier: CustomStringConvertible {
        case leading, trailing, top, bottom, height, width
        case custom(String)
        
        public var description: String {
            switch self {
            case .leading: return "leading"
            case .trailing: return "trailing"
            case .top: return "top"
            case .bottom: return "bottom"
            case .height: return "height"
            case .width: return "width"
            case .custom(let identifier): return identifier
            }
        }
    }
    
    public enum ConstraintType {
        case equal, lessThanOrEqual, greaterThanOrEqual
    }
    
    public var _topConstraint: NSLayoutConstraint? { constraint(with: .top)}
    public var _leadingConstraint: NSLayoutConstraint? { constraint(with: .leading)}
    public var _trailingConstraint: NSLayoutConstraint? { constraint(with: .trailing)}
    public var _bottomConstraint: NSLayoutConstraint? { constraint(with: .bottom)}
    public var _widthConstraint: NSLayoutConstraint? { constraint(with: .width)}
    public var _heightConstraint: NSLayoutConstraint? { constraint(with: .height)}

    public func constraint(with identifier: ConstraintIdentifier) -> NSLayoutConstraint? {
        return constraint(with: identifier.description)
    }
    
    public func constraint(with identifier: String) -> NSLayoutConstraint? {
        return constraints.first { $0.identifier == identifier }
    }
    
    public func removeConstraint(edges: [UIRectEdge]) {
        let topConstraint: NSLayoutConstraint? = constraint(with: .top)
        let leadingConstraint: NSLayoutConstraint? = constraint(with: .leading)
        let trailingConstraint: NSLayoutConstraint? = constraint(with: .trailing)
        let bottomConstraint: NSLayoutConstraint? = constraint(with: .bottom)

        edges.forEach { edge in
            switch edge {
            case .all:
                if let leadingConstraint {
                    removeConstraint(leadingConstraint)
                }
                if let trailingConstraint {
                    removeConstraint(trailingConstraint)
                }
                if let topConstraint {
                    removeConstraint(topConstraint)
                }
                if let bottomConstraint {
                    removeConstraint(bottomConstraint)
                }
            case .left:
                if let leadingConstraint {
                    removeConstraint(leadingConstraint)
                }
            case .right:
                if let trailingConstraint {
                    removeConstraint(trailingConstraint)
                }
            case .top:
                if let topConstraint {
                    removeConstraint(topConstraint)
                }
            case .bottom:
                if let bottomConstraint {
                    removeConstraint(bottomConstraint)
                }
            default:
                break
            }
        }
    }
}

//--------------------------------------------------
// MARK: - Pinning
//--------------------------------------------------
extension UIView {
    public func pin(_ view: UIView, with edges: UIEdgeInsets = UIEdgeInsets.zero) {
        leadingAnchor.constraint(equalTo: view.leadingAnchor, constant: edges.left, identifier: ConstraintIdentifier.leading.description).isActive = true
        trailingAnchor.constraint(equalTo: view.trailingAnchor, constant: -edges.right, identifier: ConstraintIdentifier.trailing.description).isActive = true
        topAnchor.constraint(equalTo: view.topAnchor, constant: edges.top, identifier: ConstraintIdentifier.top.description).isActive = true
        bottomAnchor.constraint(equalTo: view.bottomAnchor, constant: -edges.bottom, identifier: ConstraintIdentifier.bottom.description).isActive = true
    }
    
    public func pinToSuperView(_ edges: UIEdgeInsets = UIEdgeInsets.zero) {
        if let superview {
            pin(superview, with: edges)
        }
    }
}

//--------------------------------------------------
// MARK: - HeightAnchor
//--------------------------------------------------
extension UIView {

    @discardableResult
    public func height(_ constant: CGFloat, type: ConstraintType = .equal) -> Self {
        switch type {
        case .equal:
            height(constant)
            
        case .lessThanOrEqual:
            heightLessThanEqualTo(constant)
            
        case .greaterThanOrEqual:
            heightGreaterThanEqualTo(constant)
        }
        return self
    }

    @discardableResult
    public func height(_ constant: CGFloat) -> Self {
        heightAnchor.constraint(equalToConstant: constant, identifier: ConstraintIdentifier.height.description).isActive = true
        return self
    }
    
    @discardableResult
    public func heightGreaterThanEqualTo(_ constant: CGFloat) -> Self {
        heightAnchor.constraint(greaterThanOrEqualToConstant: constant, identifier: ConstraintIdentifier.height.description).isActive = true
        return self
    }
    
    @discardableResult
    public func heightLessThanEqualTo(_ constant: CGFloat) -> Self {
        heightAnchor.constraint(lessThanOrEqualToConstant: constant, identifier: ConstraintIdentifier.height.description).isActive = true
        return self
    }
    
}

//--------------------------------------------------
// MARK: - WidthAnchor
//--------------------------------------------------
extension UIView {
    
    @discardableResult
    public func width(_ constant: CGFloat, type: ConstraintType = .equal) -> Self {
        switch type {
        case .equal:
            width(constant)
            
        case .lessThanOrEqual:
            widthLessThanEqualTo(constant)
            
        case .greaterThanOrEqual:
            widthGreaterThanEqualTo(constant)
        }
        return self
    }
    
    @discardableResult
    public func width(_ constant: CGFloat) -> Self {
        widthAnchor.constraint(equalToConstant: constant, identifier: ConstraintIdentifier.width.description).isActive = true
        return self
    }
        
    @discardableResult
    public func widthGreaterThanEqualTo(_ constant: CGFloat) -> Self {
        widthAnchor.constraint(greaterThanOrEqualToConstant: constant, identifier: ConstraintIdentifier.width.description).isActive = true
        return self
    }
    
    @discardableResult
    public func widthLessThanEqualTo(_ constant: CGFloat) -> Self {
        widthAnchor.constraint(lessThanOrEqualToConstant: constant, identifier: ConstraintIdentifier.width.description).isActive = true
        return self
    }
}

//--------------------------------------------------
// MARK: - TopAnchor
//--------------------------------------------------
extension UIView {
    
    @discardableResult
    public func pinTop(_ anchor: NSLayoutYAxisAnchor? = nil, _ constant: CGFloat = 0.0, _ type: ConstraintType = .equal) -> Self {
        switch type {
        case .equal:
            pinTop(anchor, constant)
            
        case .lessThanOrEqual:
            pinTopLessThanOrEqualTo(anchor, constant)
            
        case .greaterThanOrEqual:
            pinTopGreaterThanOrEqualTo(anchor, constant)
        }
        return self
    }
    
    @discardableResult
    public func pinTop(_ constant: CGFloat = 0.0) -> Self {
        return pinTop(nil, constant)
    }
    
    @discardableResult
    public func pinTop(_ anchor: NSLayoutYAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutYAxisAnchor? = anchor ?? superview?.topAnchor
        if let found {
            topAnchor.constraint(equalTo: found, constant: constant, identifier: ConstraintIdentifier.top.description).isActive = true
        }
        return self
    }
    
    @discardableResult
    public func pinTopLessThanOrEqualTo(_ anchor: NSLayoutYAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutYAxisAnchor? = anchor ?? superview?.topAnchor
        if let found {
            topAnchor.constraint(lessThanOrEqualTo: found, constant: constant, identifier: ConstraintIdentifier.top.description).isActive = true
        }
        return self
    }
    
    @discardableResult
    public func pinTopGreaterThanOrEqualTo(_ anchor: NSLayoutYAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutYAxisAnchor? = anchor ?? superview?.topAnchor
        if let found {
            topAnchor.constraint(greaterThanOrEqualTo: found, constant: constant, identifier: ConstraintIdentifier.top.description).isActive = true
        }
        return self
    }
}

//--------------------------------------------------
// MARK: - BottomAnchor
//--------------------------------------------------
extension UIView {
    @discardableResult
    public func pinBottom(_ anchor: NSLayoutYAxisAnchor? = nil, _ constant: CGFloat = 0.0, _ type: ConstraintType = .equal) -> Self {
        switch type {
        case .equal:
            pinBottom(anchor, constant)
            
        case .lessThanOrEqual:
            pinBottomLessThanOrEqualTo(anchor, constant)
            
        case .greaterThanOrEqual:
            pinBottomGreaterThanOrEqualTo(anchor, constant)
        }
        return self
    }
    
    @discardableResult
    public func pinBottom(_ constant: CGFloat = 0.0) -> Self {
        return pinBottom(nil, constant)
    }
    
    @discardableResult
    public func pinBottom(_ anchor: NSLayoutYAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutYAxisAnchor? = anchor ?? superview?.bottomAnchor
        if let found {
            bottomAnchor.constraint(equalTo: found, constant: -constant, identifier: ConstraintIdentifier.bottom.description).isActive = true
        }
        return self
    }
    
    @discardableResult
    public func pinBottomLessThanOrEqualTo(_ anchor: NSLayoutYAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutYAxisAnchor? = anchor ?? superview?.bottomAnchor
        if let found {
            bottomAnchor.constraint(lessThanOrEqualTo: found, constant: -constant, identifier: ConstraintIdentifier.bottom.description).isActive = true
        }
        return self
    }
    
    @discardableResult
    public func pinBottomGreaterThanOrEqualTo(_ anchor: NSLayoutYAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutYAxisAnchor? = anchor ?? superview?.bottomAnchor
        if let found {
            bottomAnchor.constraint(greaterThanOrEqualTo: found, constant: -constant, identifier: ConstraintIdentifier.bottom.description).isActive = true
        }
        return self
    }
}

//--------------------------------------------------
// MARK: - LeadingAnchor
//--------------------------------------------------
extension UIView {
    @discardableResult
    public func pinLeading(_ anchor: NSLayoutXAxisAnchor? = nil, _ constant: CGFloat = 0.0, _ type: ConstraintType = .equal) -> Self {
        switch type {
        case .equal:
            pinLeading(anchor, constant)
            
        case .lessThanOrEqual:
            pinLeadingLessThanOrEqualTo(anchor, constant)
            
        case .greaterThanOrEqual:
            pinLeadingGreaterThanOrEqualTo(anchor, constant)
        }
        return self
    }
    
    @discardableResult
    public func pinLeading(_ constant: CGFloat = 0.0) -> Self {
        return pinLeading(nil, constant)
    }
    
    @discardableResult
    public func pinLeading(_ anchor: NSLayoutXAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutXAxisAnchor? = anchor ?? superview?.leadingAnchor
        if let found {
            leadingAnchor.constraint(equalTo: found, constant: constant, identifier: ConstraintIdentifier.leading.description).isActive = true
        }
        return self
    }
    
    @discardableResult
    public func pinLeadingLessThanOrEqualTo(_ anchor: NSLayoutXAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutXAxisAnchor? = anchor ?? superview?.leadingAnchor
        if let found {
            leadingAnchor.constraint(lessThanOrEqualTo: found, constant: constant, identifier: ConstraintIdentifier.leading.description).isActive = true
        }
        return self
    }
    
    @discardableResult
    public func pinLeadingGreaterThanOrEqualTo(_ anchor: NSLayoutXAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutXAxisAnchor? = anchor ?? superview?.leadingAnchor
        if let found {
            leadingAnchor.constraint(greaterThanOrEqualTo: found, constant: constant, identifier: ConstraintIdentifier.leading.description).isActive = true
        }
        return self
    }
}

//--------------------------------------------------
// MARK: - TrailingAnchor
//--------------------------------------------------
extension UIView {
    
    @discardableResult
    public func pinTrailing(_ anchor: NSLayoutXAxisAnchor? = nil, _ constant: CGFloat = 0.0, _ type: ConstraintType = .equal) -> Self {
        switch type {
        case .equal:
            pinTrailing(anchor, constant)
            
        case .lessThanOrEqual:
            pinTrailingLessThanOrEqualTo(anchor, constant)
            
        case .greaterThanOrEqual:
            pinLeadingGreaterThanOrEqualTo(anchor, constant)
        }
        return self
    }
    
    @discardableResult
    public func pinTrailing(_ constant: CGFloat = 0.0) -> Self {
        return pinTrailing(nil, constant)
    }

    @discardableResult
    public func pinTrailing(_ anchor: NSLayoutXAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutXAxisAnchor? = anchor ?? superview?.trailingAnchor
        if let found {
            trailingAnchor.constraint(equalTo: found, constant: -constant, identifier: ConstraintIdentifier.trailing.description).isActive = true
        }
        return self
    }
    
    @discardableResult
    public func pinTrailingLessThanOrEqualTo(_ anchor: NSLayoutXAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutXAxisAnchor? = anchor ?? superview?.trailingAnchor
        if let found {
            trailingAnchor.constraint(lessThanOrEqualTo: found, constant: -constant, identifier: ConstraintIdentifier.trailing.description).isActive = true
        }
        return self
    }

    @discardableResult
    public func pinTrailingGreaterThanOrEqualTo(_ anchor: NSLayoutXAxisAnchor? = nil, _ constant: CGFloat = 0.0) -> Self {
        let found: NSLayoutXAxisAnchor? = anchor ?? superview?.trailingAnchor
        if let found {
            trailingAnchor.constraint(greaterThanOrEqualTo: found, constant: -constant, identifier: ConstraintIdentifier.trailing.description).isActive = true
        }
        return self
    }
}