iOS-CustomButton.swift

import UIKit

@IBDesignable class CustomButton: UIButton {
    
    @IBInspectable var borderWidth: CGFloat = 0.0 {
        didSet {
            self.layer.borderWidth = borderWidth
        }
    }
    
    @IBInspectable var borderColor: UIColor = UIColor.white{
        didSet{
            self.layer.borderColor = borderColor.cgColor
        }
    }
    
    @IBInspectable var isRoundedButton : Bool = false{
        didSet{
            setUpView()
        }
    }
    
    @IBInspectable var cornerRadius: CGFloat = 0.0 {
        didSet {
            setUpView()
        }
    }
    
    override func awakeFromNib() {
        super.awakeFromNib()
        setUpView()
    }
    
    override func prepareForInterfaceBuilder() {
        super.prepareForInterfaceBuilder()
        setUpView()
    }
    
    override func layoutSubviews(){
        super.layoutSubviews()
        setUpView()
    }
    
    internal func setUpView() {
        if isRoundedButton {
            self.layer.cornerRadius = self.bounds.height/2;
            self.clipsToBounds = true
        }else{
            self.layer.cornerRadius = self.cornerRadius;
            self.clipsToBounds = true
        }
    }
}