]> git.r.bdr.sh - rbdr/captura/blobdiff - Captura/CapturaApp.swift
Add URL parsing for record action
[rbdr/captura] / Captura / CapturaApp.swift
index e1c4e7e87b78c3017690d473aa89e4f45216a1b8..eb6f02ecb264a24147165745a7944435a176ca9b 100644 (file)
@@ -34,10 +34,10 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
   var images: [CGImage] = []
   var outputFile: CapturaFile? = nil
   var gifCallbackTimer = ContinuousClock.now
-  var fps = CapturaSettings.frameRate
   var pixelDensity: CGFloat = 1.0
   var stopTimer: DispatchWorkItem?
   var remoteFiles: [CapturaRemoteFile] = []
+  var captureSessionConfiguration: CaptureSessionConfiguration = CaptureSessionConfiguration()
   
   func applicationDidFinishLaunching(_ notification: Notification) {
     setupStatusBar()
@@ -98,6 +98,31 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
     }
   }
   
+  // MARK: - URL Event Handler
+  
+  func application(_ application: NSApplication, open urls: [URL]) {
+    if (CapturaSettings.shouldAllowURLAutomation) {
+      for url in urls {
+        if let action = CapturaURLDecoder.decodeParams(url: url) {
+          switch action {
+            case let .configure(config):
+              CapturaSettings.apply(config)
+            case let .record(config):
+              captureSessionConfiguration = CaptureSessionConfiguration(from: config)
+              NotificationCenter.default.post(name: .startAreaSelection, object: nil, userInfo: nil)
+          }
+        }
+      }
+    } else {
+      let alert = NSAlert()
+      alert.messageText = "URL Automation Prevented"
+              alert.informativeText = "A website or application attempted to record your screen using URL Automation. If you want to allow this, enable it in Preferences."
+              alert.alertStyle = .warning
+              alert.addButton(withTitle: "OK")
+              alert.runModal()
+    }
+  }
+  
   // MARK: - UI Event Handlers
   
   func menuWillOpen(_ menu: NSMenu) {
@@ -190,7 +215,7 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
         let rectInWindow = button.convert(button.bounds, to: nil)
         let rectInScreen = button.window?.convertToScreen(rectInWindow)
         NSApp.activate(ignoringOtherApps: true)
-        recordingWindow = RecordingWindow(rectInScreen)
+        recordingWindow = RecordingWindow(captureSessionConfiguration, rectInScreen)
         recordingWindow?.makeKeyAndOrderFront(nil)
         recordingWindow?.orderFrontRegardless()
         boxListener = recordingWindow?.recordingContentView.$box
@@ -211,7 +236,6 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
   func startRecording() {
     captureState = .recording
     updateImage()
-    fps = CapturaSettings.frameRate
     outputFile = nil
     images = [];
     pixelDensity = recordingWindow?.pixelDensity ?? 1.0
@@ -225,10 +249,10 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
           stopTimer = DispatchWorkItem {
             self.stopRecording()
           }
-          DispatchQueue.main.asyncAfter(deadline: .now() + 300, execute: stopTimer!)
+          DispatchQueue.main.asyncAfter(deadline: .now() + Double(captureSessionConfiguration.maxLength), execute: stopTimer!)
           
           outputFile = CapturaFile()
-          if CapturaSettings.shouldSaveMp4 {
+          if captureSessionConfiguration.shouldSaveMp4 {
             captureSession.startRecording(to: outputFile!.mp4URL)
           } else {
             captureSession.startRunning()
@@ -246,9 +270,9 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
     stop()
     
     Task.detached {
-      if CapturaSettings.shouldSaveGif {
+      if self.captureSessionConfiguration.shouldSaveGif {
         if let outputFile = self.outputFile {
-          await GifRenderer.render(self.images, at: self.fps, to: outputFile.gifURL)
+          await GifRenderer.render(self.images, at: self.captureSessionConfiguration.frameRate, to: outputFile.gifURL)
         }
       }
       let wasSuccessful = await self.uploadOrCopy()
@@ -271,13 +295,14 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
   func reset() {
     captureState = .idle
     updateImage()
+    captureSessionConfiguration = CaptureSessionConfiguration()
     stop()
   }
   
   func receivedFrame(_ frame: CVImageBuffer) {
     let now = ContinuousClock.now
     
-    if now - gifCallbackTimer > .nanoseconds(1_000_000_000 / UInt64(fps)) {
+    if now - gifCallbackTimer > .nanoseconds(1_000_000_000 / UInt64(captureSessionConfiguration.frameRate)) {
       gifCallbackTimer = now
       DispatchQueue.main.async {
         if let cgImage = frame.cgImage?.resize(by: self.pixelDensity) {
@@ -361,9 +386,9 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
   }
   
   private func uploadOrCopy() async -> Bool {
-    if CapturaSettings.shouldUseBackend {
+    if captureSessionConfiguration.shouldUseBackend {
       let result = await uploadToBackend()
-      if result && !CapturaSettings.shouldKeepLocalFiles {
+      if result && !captureSessionConfiguration.shouldKeepLocalFiles {
         deleteLocalFiles()
       }
       return result
@@ -374,8 +399,8 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
   }
   
   private func copyLocalToClipboard() {
-    let fileType: NSPasteboard.PasteboardType = .init(rawValue: CapturaSettings.shouldSaveGif ? "com.compuserve.gif" : "public.mpeg-4")
-    if let url = CapturaSettings.shouldSaveGif ? outputFile?.gifURL : outputFile?.mp4URL {
+    let fileType: NSPasteboard.PasteboardType = .init(rawValue: captureSessionConfiguration.shouldSaveGif ? "com.compuserve.gif" : "public.mpeg-4")
+    if let url = captureSessionConfiguration.shouldSaveGif ? outputFile?.gifURL : outputFile?.mp4URL {
       if let data = try? Data(contentsOf: url) {
         let pasteboard = NSPasteboard.general
         pasteboard.declareTypes([fileType], owner: nil)
@@ -385,10 +410,10 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
   }
   
   private func uploadToBackend() async -> Bool {
-    let contentType = CapturaSettings.shouldUploadGif ? "image/gif" : "video/mp4"
-    if let url = CapturaSettings.shouldUploadGif ? outputFile?.gifURL : outputFile?.mp4URL {
+    let contentType = captureSessionConfiguration.shouldUploadGif ? "image/gif" : "video/mp4"
+    if let url = captureSessionConfiguration.shouldUploadGif ? outputFile?.gifURL : outputFile?.mp4URL {
       if let data = try? Data(contentsOf: url) {
-        if let remoteUrl = CapturaSettings.backend {
+        if let remoteUrl = captureSessionConfiguration.backend {
           var request = URLRequest(url: remoteUrl)
           request.httpMethod = "POST"
           request.httpBody = data
@@ -424,12 +449,12 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
   }
   
   private func deleteLocalFiles() {
-    if CapturaSettings.shouldSaveGif {
+    if captureSessionConfiguration.shouldSaveGif {
       if let url = outputFile?.gifURL {
           try? FileManager.default.removeItem(at: url)
       }
     }
-    if CapturaSettings.shouldSaveMp4 {
+    if captureSessionConfiguration.shouldSaveMp4 {
       if let url = outputFile?.mp4URL {
         try? FileManager.default.removeItem(at: url)
       }