iOS - Masking a UIImage with color

UIImage+Masking.swift

// Swift 4
extension UIImage {
    func maskWith(color: UIColor) -> UIImage {
        UIGraphicsBeginImageContextWithOptions(size, false, 0)

        let rect = CGRect(x: 0, y: 0, width: size.width, height: size.height)

        color.setFill()

        draw(in: rect)

        UIBezierPath(rect: rect).fill(with: .sourceIn, alpha: 1.0)

        let image = UIGraphicsGetImageFromCurrentImageContext()!

        UIGraphicsEndImageContext()

        return image
    }
}

// iOS 10+
extension UIImage {
    func maskWith(color: UIColor) -> UIImage {
        return UIGraphicsImageRenderer(size: size).image { (context) in
            let rect = CGRect(x: 0, y: 0, width: size.width, height: size.height)
            color.setFill()
            draw(in: rect)
            UIBezierPath(rect: rect).fill(with: .sourceIn, alpha: 1.0)
        }
    }
}