Extension for insert separators in UIStackView

UIStackViewExtension.swift

import UIKit

extension UIStackView {
    func addSeparator(color: UIColor, withThickness thickness: CGFloat = 1.0 ) {
        var index = self.arrangedSubviews.count
        while index > 1 {
            let separator = createSeparator(color: color, withThickness: thickness)
            insertArrangedSubview(separator, at: index - 1)   // (index-1) for centers only
            if self.axis == .horizontal {
                separator.heightAnchor.constraint(equalTo: self.heightAnchor, multiplier: 1).isActive = true
            } else {
                separator.widthAnchor.constraint(equalTo: self.widthAnchor, multiplier: 1).isActive = true
            }
            index -= 1
        }
    }

    private func createSeparator(color : UIColor, withThickness thickness: CGFloat) -> UIView {
        let separator = UIView()
        if self.axis == .horizontal {
            separator.widthAnchor.constraint(equalToConstant: thickness).isActive = true
        } else {
            separator.heightAnchor.constraint(equalToConstant: thickness).isActive = true
        }
        separator.backgroundColor = color
        return separator
    }
}