Replace react-native-compressor video path with native expo-bluesky-video-compress module [APP-2428] (#10954)

Co-authored-by: Eric Bailey <git@esb.lol>
2026-06-30 11:59:59 -04:00
parent 43d2f939c2
commit 95726c6c6b
24 changed files with 2097 additions and 7 deletions
@@ -0,0 +1,105 @@
+import AVFoundation
+import UniformTypeIdentifiers
+
+struct VideoProber {
+  static func probe(url: URL) async throws -> [String: Any] {
+    let asset = AVURLAsset(url: url)
+
+    let duration = try await asset.load(.duration)
+    let tracks = try await asset.loadTracks(withMediaType: .video)
+
+    guard let videoTrack = tracks.first else {
+      throw NSError(
+        domain: "ExpoBlueskyVideoCompress",
+        code: 1,
+        userInfo: [NSLocalizedDescriptionKey: "No video track found"]
+      )
+    }
+
+    let naturalSize = try await videoTrack.load(.naturalSize)
+    let preferredTransform = try await videoTrack.load(.preferredTransform)
+    let estimatedDataRate = try await videoTrack.load(.estimatedDataRate)
+    let nominalFrameRate = try await videoTrack.load(.nominalFrameRate)
+    let formatDescriptions = try await videoTrack.load(.formatDescriptions)
+
+    var codec = "unknown"
+    var isHDR = false
+    if let formatDescription = formatDescriptions.first {
+      let subType = CMFormatDescriptionGetMediaSubType(formatDescription)
+      codec = fourCCToString(subType)
+      // Dolby Vision codecs are HDR by definition.
+      let dolbyVisionSubtypes: Set<String> = ["dvhe", "dvh1", "dvav", "dva1"]
+      if dolbyVisionSubtypes.contains(codec) {
+        isHDR = true
+      } else if let extensions = CMFormatDescriptionGetExtensions(formatDescription)
+        as? [String: Any]
+      {
+        let transferKey = kCMFormatDescriptionExtension_TransferFunction as String
+        if let transfer = extensions[transferKey] as? String {
+          let hlg = kCMFormatDescriptionTransferFunction_ITU_R_2100_HLG as String
+          let pq = kCMFormatDescriptionTransferFunction_SMPTE_ST_2084_PQ as String
+          isHDR = transfer == hlg || transfer == pq
+        }
+      }
+    }
+
+    let rotation = rotationFromTransform(preferredTransform)
+    let isRotated = rotation == 90 || rotation == 270
+    let width = isRotated ? Int(naturalSize.height) : Int(naturalSize.width)
+    let height = isRotated ? Int(naturalSize.width) : Int(naturalSize.height)
+
+    let audioTracks = try await asset.loadTracks(withMediaType: .audio)
+    let hasAudio = !audioTracks.isEmpty
+
+    let fileSize: Int
+    if let attributes = try? FileManager.default.attributesOfItem(atPath: url.path),
+       let size = attributes[.size] as? Int {
+      fileSize = size
+    } else {
+      fileSize = 0
+    }
+
+    let mimeType: String
+    if let utType = UTType(filenameExtension: url.pathExtension) {
+      mimeType = utType.preferredMIMEType ?? "video/mp4"
+    } else {
+      mimeType = "video/mp4"
+    }
+
+    let durationSeconds = CMTimeGetSeconds(duration)
+    var bitrate = Int(estimatedDataRate)
+    if bitrate == 0 && durationSeconds > 0 {
+      bitrate = Int(Double(fileSize * 8) / durationSeconds)
+    }
+
+    return [
+      "width": width,
+      "height": height,
+      "duration": durationSeconds,
+      "bitrate": bitrate,
+      "fileSize": fileSize,
+      "mimeType": mimeType,
+      "codec": codec,
+      "hasAudio": hasAudio,
+      "frameRate": nominalFrameRate,
+      "rotation": rotation,
+      "isHDR": isHDR
+    ]
+  }
+
+  private static func rotationFromTransform(_ transform: CGAffineTransform) -> Int {
+    let angle = atan2(transform.b, transform.a)
+    let degrees = Int(round(angle * 180.0 / .pi))
+    return ((degrees % 360) + 360) % 360
+  }
+
+  private static func fourCCToString(_ code: FourCharCode) -> String {
+    let chars: [Character] = [
+      Character(UnicodeScalar((code >> 24) & 0xFF)!),
+      Character(UnicodeScalar((code >> 16) & 0xFF)!),
+      Character(UnicodeScalar((code >> 8) & 0xFF)!),
+      Character(UnicodeScalar(code & 0xFF)!)
+    ]
+    return String(chars).trimmingCharacters(in: .whitespaces)
+  }
+}