ColorButton

ColorButton.swift

import UIKit

class ColorButton: UIButton {
    func setBackgroundColor(color: UIColor, forState state: UIControlState) {
        self.setBackgroundImage(self.colorImage(color), forState: state)
    }
    
    func colorImage(color: UIColor)-> UIImage {
        let size = CGSizeMake(1.0, 1.0)
        let rect = CGRect(origin: CGPointZero, size: size)
        UIGraphicsBeginImageContext(size)
        let context = UIGraphicsGetCurrentContext()
        CGContextSetFillColorWithColor(context, color.CGColor)
        CGContextFillRect(context, rect)
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        
        return image
    }
}