127 lines
3.6 KiB
Swift
127 lines
3.6 KiB
Swift
//
|
|
// Copyright 2024 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Foundation
|
|
|
|
public enum ImageFormat: CustomStringConvertible {
|
|
case png
|
|
case gif
|
|
case tiff
|
|
case jpeg
|
|
case bmp
|
|
case webp
|
|
case heic
|
|
case heif
|
|
|
|
public var description: String {
|
|
switch self {
|
|
case .png:
|
|
"ImageFormat_Png"
|
|
case .gif:
|
|
"ImageFormat_Gif"
|
|
case .tiff:
|
|
"ImageFormat_Tiff"
|
|
case .jpeg:
|
|
"ImageFormat_Jpeg"
|
|
case .bmp:
|
|
"ImageFormat_Bmp"
|
|
case .webp:
|
|
"ImageFormat_Webp"
|
|
case .heic:
|
|
"ImageFormat_Heic"
|
|
case .heif:
|
|
"ImageFormat_Heif"
|
|
}
|
|
}
|
|
|
|
public var mimeType: MimeType {
|
|
return self.mimeTypes.preferredMimeType
|
|
}
|
|
|
|
private var mimeTypes: (preferredMimeType: MimeType, alternativeMimeTypes: [MimeType]) {
|
|
switch self {
|
|
case .png: (.imagePng, [.imageApng, .imageVndMozillaApng])
|
|
case .gif: (.imageGif, [])
|
|
case .tiff: (.imageTiff, [.imageXTiff])
|
|
case .jpeg: (.imageJpeg, [])
|
|
case .bmp: (.imageBmp, [.imageXWindowsBmp])
|
|
case .webp: (.imageWebp, [])
|
|
case .heic: (.imageHeic, [])
|
|
case .heif: (.imageHeif, [])
|
|
}
|
|
}
|
|
|
|
public var fileExtension: String {
|
|
// All known ImageFormats must have a corresponding extension.
|
|
return MimeTypeUtil.fileExtensionForMimeType(mimeType.rawValue)!
|
|
}
|
|
|
|
internal func isValid(source: OWSImageSource) -> Bool {
|
|
switch self {
|
|
case .png, .tiff, .jpeg, .bmp, .webp, .heic, .heif:
|
|
return true
|
|
case .gif:
|
|
return source.ows_hasValidGifSize
|
|
}
|
|
}
|
|
|
|
internal func isValid(mimeType: String) -> Bool {
|
|
owsAssertDebug(!mimeType.isEmpty)
|
|
let mimeTypes = self.mimeTypes
|
|
return ([mimeTypes.preferredMimeType] + mimeTypes.alternativeMimeTypes).contains(where: {
|
|
return mimeType.caseInsensitiveCompare($0.rawValue) == .orderedSame
|
|
})
|
|
}
|
|
}
|
|
|
|
public struct ImageMetadata {
|
|
public let isValid: Bool
|
|
public let imageFormat: ImageFormat?
|
|
public let pixelSize: CGSize
|
|
public let hasAlpha: Bool
|
|
let isAnimated: Bool
|
|
|
|
internal init(isValid: Bool, imageFormat: ImageFormat?, pixelSize: CGSize, hasAlpha: Bool, isAnimated: Bool) {
|
|
self.isValid = isValid
|
|
self.imageFormat = imageFormat
|
|
self.pixelSize = pixelSize
|
|
self.hasAlpha = hasAlpha
|
|
self.isAnimated = isAnimated
|
|
}
|
|
|
|
internal static func invalid() -> ImageMetadata {
|
|
.init(isValid: false, imageFormat: nil, pixelSize: .zero, hasAlpha: false, isAnimated: false)
|
|
}
|
|
}
|
|
|
|
internal struct WebpMetadata {
|
|
let isValid: Bool
|
|
let canvasWidth: UInt32
|
|
let canvasHeight: UInt32
|
|
let frameCount: UInt32
|
|
}
|
|
|
|
public enum ImageMetadataResult {
|
|
/// Source data exceeded size limit for all attachments;
|
|
/// as a precaution no validation was performed.
|
|
case genericSizeLimitExceeded
|
|
|
|
/// Exceeded the file size limit for the inferred type of image.
|
|
/// Smaller than the generic size limit.
|
|
case imageTypeSizeLimitExceeded
|
|
|
|
case invalid
|
|
|
|
case valid(ImageMetadata)
|
|
|
|
/// A mime type was provided, and it did not match the contents.
|
|
/// Metadata is still valid and the error can be safely ignored.
|
|
case mimeTypeMismatch(ImageMetadata)
|
|
|
|
/// A file extension was provided, and it did not match the contents.
|
|
/// Metadata is still valid and the error can be safely ignored.
|
|
case fileExtensionMismatch(ImageMetadata)
|
|
}
|