Allow URL configuration

[rbdr/captura] / Captura / CapturaApp.swift

diff --git a/Captura/CapturaApp.swift b/Captura/CapturaApp.swift
index e1c4e7e87b78c3017690d473aa89e4f45216a1b8..d9c3ff0bcfffd18786ae622bda386b149bf34733 100644 (file)
--- a/Captura/CapturaApp.swift
+++ b/Captura/CapturaApp.swift
@@ -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,32 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
     }
   }
   
+  // MARK: - URL Event Handler
+  
+  func application(_ application: NSApplication, open urls: [URL]) {
+    print("AAAH OPENING")
+    if (CapturaSettings.shouldAllowURLAutomation) {
+      for url in urls {
+        if let action = CapturaURLDecoder.decodeParams(url: url) {
+          switch action {
+            case let .configure(config):
+            print("AAAH CONFIGURING \(config)")
+              CapturaSettings.apply(config)
+            case let .record(config):
+              print(config)
+          }
+        }
+      }
+    } 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) {
@@ -211,7 +237,6 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
   func startRecording() {
     captureState = .recording
     updateImage()
-    fps = CapturaSettings.frameRate
     outputFile = nil
     images = [];
     pixelDensity = recordingWindow?.pixelDensity ?? 1.0
@@ -228,7 +253,7 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
           DispatchQueue.main.asyncAfter(deadline: .now() + 300, execute: stopTimer!)
           
           outputFile = CapturaFile()
-          if CapturaSettings.shouldSaveMp4 {
+          if captureSessionConfiguration.shouldSaveMp4 {
             captureSession.startRecording(to: outputFile!.mp4URL)
           } else {
             captureSession.startRunning()
@@ -246,9 +271,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()
@@ -277,7 +302,7 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
   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) {
@@ -358,12 +383,13 @@ class CapturaAppDelegate: NSObject, NSApplicationDelegate, NSMenuDelegate {
     boxListener?.cancel()
     recordingWindow?.close()
     recordingWindow = nil
+    captureSessionConfiguration = CaptureSessionConfiguration()
   }
   
   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 +400,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 +411,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 +450,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)
       }