weareopensource/Swift

/**
 * Dependencies
 */

import UIKit

/**
 * extension
 */

extension UIButton {
    /**
     * @desc setBackgroundColor
     * @param {UIColor} UIControl,
     * @param {UIControl.state} forState,
     */
    func setBackgroundColor(color: UIColor, forState: UIControl.State) {
        self.clipsToBounds = true  // add this to maintain corner radius
        UIGraphicsBeginImageContext(CGSize(width: 1, height: 1))
        if let context = UIGraphicsGetCurrentContext() {
            context.setFillColor(color.cgColor)
            context.fill(CGRect(x: 0, y: 0, width: 1, height: 1))
            let colorImage = UIGraphicsGetImageFromCurrentImageContext()
            UIGraphicsEndImageContext()
            self.setBackgroundImage(colorImage, for: forState)
        }
    }

}