iOS Button ImageView and titleLabel align vertically center

UIButton+VerticalCenter.swift

import UIKit

extension UIButton {
 
    func alignVerticalCenter(padding: CGFloat = 7.0) {
        guard let imageSize = imageView?.frame.size, let titleText = titleLabel?.text, let titleFont = titleLabel?.font else {
            return
        }
        
        let titleSize = (titleText as NSString).size(withAttributes: [.font: titleFont])
        print(titleSize)
        let total = imageSize.height + titleSize.height + padding
        imageEdgeInsets = UIEdgeInsets(top: -(total - imageSize.height), left: 0, bottom: 0, right: -titleSize.width)
        titleEdgeInsets = UIEdgeInsets(top: 0, left: -imageSize.width, bottom: -(total - titleSize.height), right: 0)
    }
}

How to use it.

let button = UIButton(type: .custom)
button.backgroundColor = .red
button.frame = CGRect(x: 30, y: 30, width: 80, height: 80)
button.setTitle("{Your_Title}", for: .normal)
button.titleLabel?.font = UIFont.systemFont(ofSize: 14, weight: .semibold)
button.setImage(UIImage(named: "{YOUR_IMAHE_NAME}"), for: .normal)
button.setTitleColor(.white, for: .normal)

button. alignVerticalCenter()

it is not perfect