UIView+Extension

extension UIView {
    
    class func gradientViewWithText(text: String, size: CGSize, firstColor: UIColor, secondColor: UIColor, font: UIFont) -> UIView {
        let view = UIView()
        let layer = CAGradientLayer()
        layer.colors = [firstColor.cgColor, secondColor.cgColor]
        layer.frame = CGRect(x: 0, y: 0, width: size.width, height: size.height)
        view.layer.addSublayer(layer)
        
        let textLabel = UILabel()
        textLabel.font = font
        textLabel.textAlignment = .center
        textLabel.text = text
        textLabel.frame = layer.frame
        view.addSubview(textLabel)
        view.mask = textLabel
        return view
    }
}