VideoCompression AVFoundation

gistfile1.txt

import Foundation
import AVFoundation

protocol VideoCompressorProtocol {
    func compress(fileURL: URL, outputURL: URL, completion: @escaping (URL?, Error?) -> Void)
}

class VideoCompressor: VideoCompressorProtocol {
    
    func compress(fileURL: URL, outputURL: URL, completion: @escaping (URL?, Error?) -> Void) {
        if let data = try? Data(contentsOf: fileURL) {
            print("LOG: File size before compression: \(Double(data.count / 1048576)) mb")
        }
        
        compressVideo(inputURL: fileURL, outputURL: outputURL) { compressedURL, error in
            if let compressedURL = compressedURL, let data = try? Data(contentsOf: compressedURL) {
                print("LOG: File size after compression: \(Double(data.count / 1048576)) mb")
            }
            
            completion(compressedURL, error)
        }
        
    }
    
    private func compressVideo(inputURL: URL, outputURL: URL, completion: @escaping (URL?, Error?) -> Void) {
        let urlAsset = AVURLAsset(url: inputURL, options: nil)
        let exportSession = AVAssetExportSession(asset: urlAsset, presetName: AVAssetExportPresetMediumQuality)
        
        exportSession?.outputURL = outputURL
        exportSession?.outputFileType = AVFileType.mov
        exportSession?.shouldOptimizeForNetworkUse = true
        
        exportSession?.exportAsynchronously { () -> Void in
            guard let exportSession = exportSession else { return }
            
            if exportSession.status == .completed {
                completion(outputURL, nil)
            } else if exportSession.status == .failed {
                completion(nil, exportSession.error)
            }
        }
    }
    
}