100 lines
3.7 KiB
Swift
100 lines
3.7 KiB
Swift
//
|
|
// https://mczachurski.dev
|
|
// Copyright © 2022 Marcin Czachurski and the repository contributors.
|
|
// Licensed under the MIT License.
|
|
//
|
|
|
|
import Foundation
|
|
import UIKit
|
|
|
|
public extension Dictionary<String, Any> {
|
|
func getExifValue(_ key: String) -> String? {
|
|
if let value = self[key] as? String {
|
|
return value
|
|
}
|
|
|
|
if let dictionary = self[key] as? [String: Any], let value = dictionary[key] {
|
|
return value as? String
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
public extension Data {
|
|
func getExifData() -> [String: Any]? {
|
|
var imageInfo: [String: Any]? = nil
|
|
|
|
guard let imageSource = CGImageSourceCreateWithData(self as CFData, nil),
|
|
let metadata = CGImageSourceCopyMetadataAtIndex(imageSource, 0, nil),
|
|
let tags = CGImageMetadataCopyTags(metadata) else {
|
|
return nil
|
|
}
|
|
|
|
imageInfo = self.readMetadataTagArr(tagArr: tags)
|
|
return imageInfo
|
|
}
|
|
|
|
/// Reads the Arrays of tags and convert them into a dictionary of [String: Any].
|
|
private func readMetadataTagArr(tagArr: CFArray) -> [String: Any]? {
|
|
var result = [String: Any]()
|
|
|
|
for (_, tag) in (tagArr as NSArray).enumerated() {
|
|
let tagMetadata = tag as! CGImageMetadataTag
|
|
if let cfName = CGImageMetadataTagCopyName(tagMetadata) {
|
|
let name = String(cfName)
|
|
result[name] = self.readMetadataTag(metadataTag: tagMetadata)
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
/// Convert CGImageMetadataTag to a dictionary of [String: Any].
|
|
private func readMetadataTag(metadataTag: CGImageMetadataTag) -> [String: Any] {
|
|
var result = [String: Any]()
|
|
guard let cfName = CGImageMetadataTagCopyName(metadataTag) else { return result }
|
|
let name = String(cfName)
|
|
let value = CGImageMetadataTagCopyValue(metadataTag)
|
|
|
|
/// checking the type of `value` object and then performing respective operation on `value`
|
|
if CFGetTypeID(value) == CFStringGetTypeID() {
|
|
let valueStr = String(value as! CFString)
|
|
result[name] = valueStr
|
|
} else if CFGetTypeID(value) == CFDictionaryGetTypeID() {
|
|
let nsDict: NSDictionary = value as! CFDictionary
|
|
result[name] = self.getDictionary(from: nsDict)
|
|
} else if CFGetTypeID(value) == CFArrayGetTypeID() {
|
|
let valueArr: NSArray = value as! CFArray
|
|
for (_, item) in valueArr.enumerated() {
|
|
let tagMetadata = item as! CGImageMetadataTag
|
|
result[name] = self.readMetadataTag(metadataTag: tagMetadata)
|
|
}
|
|
} else {
|
|
// when the data was of some other type
|
|
let descriptionString: CFString = CFCopyDescription(value);
|
|
let str = String(descriptionString)
|
|
result[name] = str
|
|
}
|
|
return result
|
|
}
|
|
|
|
/// Converting CGImage Metadata dictionary to [String: Any]
|
|
private func getDictionary(from nsDict: NSDictionary) -> [String: Any] {
|
|
var subDictionary = [String: Any]()
|
|
for (key, val) in nsDict {
|
|
guard let key = key as? String else { continue }
|
|
let tempDict: [String: Any] = [key: val]
|
|
if JSONSerialization.isValidJSONObject(tempDict) {
|
|
subDictionary[key] = val
|
|
} else {
|
|
let mData = val as! CGImageMetadataTag
|
|
let tempDict: [String: Any] = [key: self.readMetadataTag(metadataTag: mData)]
|
|
if JSONSerialization.isValidJSONObject(tempDict) {
|
|
subDictionary[key] = tempDict
|
|
}
|
|
}
|
|
}
|
|
return subDictionary
|
|
}
|
|
}
|