MediaMessageSizeCalculator.swift

/// A little pieve of genius by me that culculates cell size base on the image given while maintaining aspect ratio
open class MediaMessageSizeCalculator {
    open func messageContainerSize(for image: UIImage) -> CGSize {
        let maxWidth = UIScreen.main.bounds.width - 16
        let sizeForMediaItem = { (maxWidth: CGFloat, item: UIImage) -> CGSize in
            if maxWidth < item.size.width {
                // Maintain the ratio if width is too great
                let height = maxWidth * item.size.height / item.size.width
                return CGSize(width: maxWidth, height: height)
            }
            return item.size
        }
        
         return sizeForMediaItem(maxWidth, image)
    }
}