UIStackView custom spacing extension

UIStackViewExtensions.swift

import UIKit

extension UIStackView {
    
    struct Spacer {
        var space: CGFloat = 0
        
        init(_ space:CGFloat) {
            self.space = space
        }
    }
    
    var count: Int {
        return arrangedSubviews.count
    }
    
    var lastArrangedSubview: UIView? {
        if count > 0 {
            return arrangedSubviews[count-1]
        } else {
            return nil
        }
    }
    
    func setArrangedSubviews(_ subviews: [Any]) {
        removeAllArrangedSubviews()
        
        var index = 0
        for view in subviews {
            if let view = view as? UIView {
                insertArrangedSubview(view, at: index)
                index += 1
            } else {
                if let spacer = view as? Spacer, let last = lastArrangedSubview {
                    setCustomSpacing(spacer.space, after: last)
                }
            }
        }
    }
    
    func removeAllArrangedSubviews() {
        for subview in arrangedSubviews {
            subview.removeFromSuperview()
        }
    }
}

Usage:

stackView.setArrangedSubviews([view1, Spacer(20), view2, Spacer(10), view3])