使用 Swift 脚本获取照片 EXIF 信息

#!/usr/bin/env swift

import AppKit
import CoreGraphics
import CoreServices
import Foundation
import ImageIO

func checkImageIsHDR(for fileURL: URL) -> Bool {
    guard let imageSource = CGImageSourceCreateWithURL(fileURL as CFURL, nil) else {
        print("无法读取图片")
        return false
    }

    guard
        let imageProperties = CGImageSourceCopyPropertiesAtIndex(imageSource, 0, nil)
            as? [CFString: Any]
    else {
        print("无法获取图片属性")
        return false
    }

    if let profileName = imageProperties[kCGImagePropertyProfileName] as? String {
        print("ICC Profile 名称: \(profileName)")
        if profileName.contains("PQ") || profileName.contains("BT.2100") {
            print("色彩空间是 PQ")
            return true
        }
    }

    // 查看是否有 GainMap
    if let auxData = CGImageSourceCopyAuxiliaryDataInfoAtIndex(
        imageSource, 0, kCGImageAuxiliaryDataTypeHDRGainMap)
    {
        print("检测到 HDR Gain Map 数据: \(auxData)")
        return true
    }
    return false
}

func readExifDescription(from fileURL: URL) -> String? {
    guard let imageSource = CGImageSourceCreateWithURL(fileURL as CFURL, nil) else {
        print("无法读取文件: \(fileURL.path)")
        return nil
    }

    guard
        let properties = CGImageSourceCopyPropertiesAtIndex(imageSource, 0, nil) as? [CFString: Any]
    else {
        print("无法提取元数据")
        return nil
    }

    let tiff = properties[kCGImagePropertyTIFFDictionary] as? [CFString: Any]
    let exif = properties[kCGImagePropertyExifDictionary] as? [CFString: Any]
    // Get the Apple-specific metadata dictionary

    // 使用官方支持的字段
    let originalLens = exif?[kCGImagePropertyExifLensModel] as? String

    let make = (tiff?[kCGImagePropertyTIFFMake] as? String ?? "").trimmingCharacters(
        in: .whitespaces)
    let model = (tiff?[kCGImagePropertyTIFFModel] as? String ?? "").trimmingCharacters(
        in: .whitespaces)

    let iso = exif?[kCGImagePropertyExifISOSpeedRatings] as? [Int] ?? []
    let focalLength = exif?[kCGImagePropertyExifFocalLength] as? Double
    let aperture = exif?[kCGImagePropertyExifFNumber] as? Double
    let shutterSpeedValue = exif?[kCGImagePropertyExifExposureTime] as? Double
    let exposureBias = exif?[kCGImagePropertyExifExposureBiasValue] as? Double

    // Define lens replacement map
    let lensReplacementMap: [String: String] = [
        "iPhone 15 Pro Max back triple camera 6.765mm f/1.78": "Main Camera — 24 mm ƒ1.78",
        // Add more replacements here:
        // "Original Lens Model String": "Desired Replacement String",
        "iPhone 15 Pro Max back triple camera 9.03mm f/2.8": "Telephoto Camera — 77 mm ƒ2.8",
        "iPhone 15 Pro Max back triple camera 2.22mm f/2.2": "Ultra Wide Camera — 13 mm ƒ2.2",
    ]

    // Apply replacement if found, otherwise use the original lens string
    let lens = originalLens.flatMap { lensReplacementMap[$0] } ?? originalLens

    // Check for HDR
    let isHDR: Bool = checkImageIsHDR(for: fileURL)

    // 构造输出格式
    var parts: [String] = []
    if !make.isEmpty || !model.isEmpty {
        parts.append("\(make) \(model)")
    }
    if let isoValue = iso.first {
        parts.append("ISO \(isoValue)")
    }
    if let fl = focalLength {
        parts.append("\(Int(round(fl)))mm")
    }
    if let ap = aperture {
        parts.append("ƒ\(String(format: "%.1f", ap))")
    }
    if let ss = shutterSpeedValue {
        if ss >= 1.0 {
            parts.append("\(Int(ss)) s")
        } else {
            let denominator = Int(round(1.0 / ss))
            parts.append("1/\(denominator) s")
        }
    }
    if let ev = exposureBias {
        if abs(ev) < 0.001 {
            parts.append("")
        } else {
            parts.append("\(String(format: "%+.1f", ev))ev")
        }
    }
    if let lens = lens {
        parts.append("\(lens)")
    }
    // Append HDR tag if the image is HDR
    if isHDR {
        parts.append("HDR")
    }

    return parts.map { "{\($0)}" }.joined()
}

// 主程序入口
let args = CommandLine.arguments
guard args.count >= 2 else {
    print("用法: readExif.swift <图片路径>")
    exit(1)
}

let filePath = args[1]
let fileURL = URL(fileURLWithPath: filePath)

if let description = readExifDescription(from: fileURL) {
    print(description)
    // copy to clipboard
    let pasteboard = NSPasteboard.general
    pasteboard.clearContents()
    pasteboard.setString(description, forType: .string)
} else {
    print("读取失败")
}