Generate an UIImage from a NSAttributedString

NSAttributedString+Image.swift

import UIKit

extension NSAttributedString {
    
    func image(size: CGSize) -> UIImage? {
        UIGraphicsBeginImageContext(size)
        self.draw(at: CGPoint.zero)
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return image
    }
}