feat/#40 :: 비디오 캡쳐 기능 추가

- 시즌 1235125호 비디오 캡쳐 기능 추가
- 이번엔 연결 안된다고 안터져요

Co-Authored-By: Youngkyu Song <[email protected]>

Author: Kiyoung-Kim-57

PhotoGether/DataLayer/PhotoGetherData/PhotoGetherData/CapturableVideoView.swift

@@ -0,0 +1,32 @@
+import Foundation
+import WebRTC
+
+public class CapturableVideoView: RTCMTLVideoView {
+    public var capturedImage: UIImage?
+
+    public override func setSize(_ size: CGSize) {
+        super.setSize(size)
+    }
+
+    public override func renderFrame(_ frame: RTCVideoFrame?) {
+        guard let frame = frame else { return }
+        
+        super.renderFrame(frame)
+        capturedImage = convertFrameToImage(frame)
+    }
+
+    private func convertFrameToImage(_ frame: RTCVideoFrame) -> UIImage? {
+        // frame의 버퍼를 CVPixelBuffer로 가져옴
+        guard let pixelBuffer = (frame.buffer as? RTCCVPixelBuffer)?.pixelBuffer else { return nil }
+        
+        // CVPixelBuffer를 CIImage로 변환
+        let ciImage = CIImage(cvPixelBuffer: pixelBuffer).oriented(.right)
+        
+        // CIImage를 UIImage로 변환
+        let context = CIContext(options: nil)
+        if let cgImage = context.createCGImage(ciImage, from: ciImage.extent) {
+            return UIImage(cgImage: cgImage)
+        }
+        return nil
+    }
+}

PhotoGether/DataLayer/PhotoGetherData/PhotoGetherData/ConnectionClientImpl.swift

@@ -9,8 +9,8 @@ public final class ConnectionClientImpl: ConnectionClient {
 
     public var receivedDataPublisher = PassthroughSubject<Data, Never>()
 
-    public var remoteVideoView: UIView = RTCMTLVideoView()
-    public var localVideoView: UIView = RTCMTLVideoView()
+    public var remoteVideoView: UIView = CapturableVideoView()
+    public var localVideoView: UIView = CapturableVideoView()
 
     public var peerID: String = ""
     public var roomID: String = ""
@@ -40,6 +40,27 @@ public final class ConnectionClientImpl: ConnectionClient {
         self.webRTCService.sendData(data)
     }
 
+    public func captureVideos() -> [UIImage] {
+        let localCaptureImage = getCapturedVideos(isLocal: true)
+        let remoteCaptureImage = getCapturedVideos(isLocal: false)
+        
+        return [localCaptureImage, remoteCaptureImage]
+    }
+    
+    private func getCapturedVideos(isLocal: Bool) -> UIImage {
+        let targetVideo = isLocal ? self.localVideoView : self.remoteVideoView
+        
+        guard let videoView = targetVideo as? CapturableVideoView else {
+            return UIImage()
+        }
+        
+        guard let capturedImage = videoView.capturedImage else {
+            return UIImage()
+        }
+        
+        return capturedImage
+    }
+    
     private func connect() {
         self.signalingService.connect()
     }
@@ -112,7 +133,7 @@ extension ConnectionClientImpl: WebRTCServiceDelegate {
     ) {
         // TODO: 피어커넥션 연결 상태 변경에 따른 처리
     }
-
+    
     /// peerConnection의 remoteDataChannel 에 데이터가 수신되면 호출됨
     public func webRTCService(
         _ service: WebRTCService,

PhotoGether/DomainLayer/PhotoGetherDomain/PhotoGetherDomain/UseCaseImpl/CaptureVideosUseCaseImpl.swift

@@ -0,0 +1,17 @@
+import UIKit
+import PhotoGetherDomainInterface
+
+public final class CaptureVideosUseCaseImpl: CaptureVideosUseCase {
+    public func execute() -> [UIImage] {
+        let localImage = [connectionRepository.clients[0].captureVideos()[0]]
+        let remoteImages = connectionRepository.clients.map { $0.captureVideos()[1] }
+        
+        return localImage + remoteImages
+    }
+    
+    private let connectionRepository: ConnectionRepository
+    
+    public init(connectionRepository: ConnectionRepository) {
+        self.connectionRepository = connectionRepository
+    }
+}

PhotoGether/DomainLayer/PhotoGetherDomain/PhotoGetherDomainInterface/ConnectionClient.swift

@@ -10,4 +10,5 @@ public protocol ConnectionClient {
 
     func sendOffer()
     func sendData(data: Data)
+    func captureVideos() -> [UIImage]
 }

PhotoGether/DomainLayer/PhotoGetherDomain/PhotoGetherDomainInterface/UseCase/CaptureVideosUseCase.swift

@@ -0,0 +1,5 @@
+import UIKit
+
+public protocol CaptureVideosUseCase {
+    func execute() -> [UIImage]
+}