From c9b9e1d654ea697afad9f6427d94623bfdf55cce Mon Sep 17 00:00:00 2001
From: Ruben Beltran del Rio <ruben@unlimited.pizza>
Date: Fri, 28 Jul 2023 13:48:53 +0200
Subject: Add clipboard

---
 Captura/Domain/CapturaCaptureSession.swift | 60 ++++++++++++++++++++++++++++++
 Captura/Domain/CaptureState.swift          |  8 ++++
 2 files changed, 68 insertions(+)
 create mode 100644 Captura/Domain/CapturaCaptureSession.swift
 create mode 100644 Captura/Domain/CaptureState.swift

(limited to 'Captura/Domain')

diff --git a/Captura/Domain/CapturaCaptureSession.swift b/Captura/Domain/CapturaCaptureSession.swift
new file mode 100644
index 0000000..80240e6
--- /dev/null
+++ b/Captura/Domain/CapturaCaptureSession.swift
@@ -0,0 +1,60 @@
+import AppKit
+import AVFoundation
+
+class CapturaCaptureSession: AVCaptureSession, AVCaptureFileOutputRecordingDelegate, AVCaptureVideoDataOutputSampleBufferDelegate {
+
+  let videoOutput = AVCaptureVideoDataOutput()
+  let movieFileOutput = AVCaptureMovieFileOutput()
+  var receivedFrames = false
+  
+  init(_ screen: NSScreen, box: NSRect) {
+    super.init()
+    
+    let displayId = screen.deviceDescription[NSDeviceDescriptionKey("NSScreenNumber")] as! CGDirectDisplayID
+    let screenInput = AVCaptureScreenInput(displayID: displayId)
+    screenInput?.cropRect = box.insetBy(dx: 1, dy: 1)
+    
+    if self.canAddInput(screenInput!) {
+      self.addInput(screenInput!)
+    }
+    
+    videoOutput.setSampleBufferDelegate(self, queue: Dispatch.DispatchQueue(label: "sample buffer delegate", attributes: []))
+    
+    if self.canAddOutput(videoOutput) {
+      self.addOutput(videoOutput)
+    }
+    
+    if self.canAddOutput(movieFileOutput) {
+      self.addOutput(movieFileOutput)
+    }
+  }
+  
+  func startRecording() {
+    receivedFrames = false
+    self.startRunning()
+    
+    DispatchQueue.main.asyncAfter(deadline: .now() + 1) {
+      if !self.receivedFrames {
+        NotificationCenter.default.post(name: .failedToStart, object: nil, userInfo: nil)
+      }
+    }
+  }
+  
+  func startRecording(to url: URL) {
+    self.startRecording()
+    movieFileOutput.startRecording(to: url, recordingDelegate: self)
+  }
+  
+  // MARK: - AVCaptureVideoDataOutputSampleBufferDelegate Implementation
+  
+  func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
+    receivedFrames = true
+    
+    guard let imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return }
+    NotificationCenter.default.post(name: .receivedFrame, object: nil, userInfo: ["frame": imageBuffer])
+  }
+  
+  // MARK: - AVCaptureFileOutputRecordingDelegate Implementation
+  
+  func fileOutput(_ output: AVCaptureFileOutput, didFinishRecordingTo outputFileURL: URL, from connections: [AVCaptureConnection], error: Error?) {}
+}
diff --git a/Captura/Domain/CaptureState.swift b/Captura/Domain/CaptureState.swift
new file mode 100644
index 0000000..0761b88
--- /dev/null
+++ b/Captura/Domain/CaptureState.swift
@@ -0,0 +1,8 @@
+enum CaptureState {
+  case idle
+  case selectingArea
+  case recording
+  case uploading
+  case uploaded
+  case error
+}
-- 
cgit