feat: update CameraViewModel and ContentView for sponsored video improvements

Author: Pedroodelvalle

Camera/CameraViewModel.swift

@@ -12,10 +12,12 @@ import UIKit
 import Photos
 import CoreImage
 import CoreImage.CIFilterBuiltins
+import SwiftUI
 
 final class CameraViewModel: NSObject, ObservableObject {
     @Published var isAuthorized = false
     @Published var isSessionRunning = false
+    @Published var isCameraReady = false
     @Published var isRecording = false
     @Published var frameRateLabel: String = "60 fps"
     @Published var quickZoomIndex: Int = 1 // 0 -> 0.5x, 1 -> 1x, 2 -> 2x
@@ -28,6 +30,7 @@ final class CameraViewModel: NSObject, ObservableObject {
     @Published var teleprompterText: String = "Adicione seu roteiro aqui..."
     @Published var teleprompterSpeed: Double = 16.0 // points per second (default slower)
     @Published var teleprompterFontSize: CGFloat = 24.0 // default smaller
+    @Published var isProcessingVideo: Bool = false
 
     let controller = CaptureSessionController()
     private var recorder: SegmentedRecorder?
@@ -38,17 +41,45 @@ final class CameraViewModel: NSObject, ObservableObject {
 
     override init() {
         super.init()
+        loadTeleprompterSettingsFromUserDefaults()
+    }
+    
+    /// Load teleprompter text from UserDefaults (passed from Flutter)
+    /// Auto-enables teleprompter if valid script is present
+    private func loadTeleprompterSettingsFromUserDefaults() {
+        if let savedText = UserDefaults.standard.string(forKey: "teleprompter_initial_text"),
+           !savedText.isEmpty,
+           savedText != "Adicione seu roteiro aqui..." {
+            teleprompterText = savedText
+            // Auto-enable teleprompter when a script is provided
+            isTeleprompterOn = true
+        }
     }
 
     deinit {
-        // Ensure brightness is restored if we used screen torch
-        setScreenTorchEnabled(false)
+        // Stop recording if active
+        if let recorder, recorder.isRecording {
+            recorder.stopCurrentSegment()
+        }
 
-        // Remove orientation observer
-        if let observer = orientationObserver {
-            NotificationCenter.default.removeObserver(observer)
+        // Stop the camera session
+        controller.stopSession()
+        
+        // Restore screen brightness if we modified it
+        if let originalBrightness = savedScreenBrightness {
+            let brightness = originalBrightness
+            DispatchQueue.main.async {
+                UIScreen.main.brightness = brightness
+            }
+        }
+        
+        // Clean up temporary segment files in background
+        let segmentURLs = segments.map { $0.url }
+        DispatchQueue.global(qos: .background).async {
+            for url in segmentURLs {
+                try? FileManager.default.removeItem(at: url)
+            }
         }
-        UIDevice.current.endGeneratingDeviceOrientationNotifications()
     }
 
     // MARK: - Grid
@@ -75,6 +106,12 @@ final class CameraViewModel: NSObject, ObservableObject {
     func start() {
         controller.startSession()
         isSessionRunning = true
+        // Small delay to ensure session is fully running before showing preview
+        DispatchQueue.main.asyncAfter(deadline: .now() + 0.15) { [weak self] in
+            withAnimation(.easeOut(duration: 0.3)) {
+                self?.isCameraReady = true
+            }
+        }
     }
 
     // MARK: - Filter
@@ -104,6 +141,13 @@ final class CameraViewModel: NSObject, ObservableObject {
         setScreenTorchEnabled(false)
     }
 
+    /// Safely dismisses the camera and notifies Flutter of cancellation.
+    func dismissCamera() {
+        // Notify Flutter immediately - let the plugin handle the dismiss
+        // This must happen BEFORE we start any cleanup to avoid race conditions
+        NotificationCenter.default.post(name: Notification.Name("conty.teleprompter.cancelled"), object: nil)
+    }
+
     func attachPreview(_ view: CameraPreviewView) {
         view.attach(session: controller.session)
         view.onTapToFocus = { [weak self] devicePoint in
@@ -285,13 +329,16 @@ extension CameraViewModel {
             try? FileManager.default.removeItem(at: url)
         }
     }
-
+    
     func nextAction() {
         print("Next pressed with \(segments.count) takes")
 
         guard !segments.isEmpty else { return }
+        guard !isProcessingVideo else { return }
+        
+        isProcessingVideo = true
 
-        // Concatenate all segments into a single video and save to gallery
+        // Concatenate all segments into a single video and notify Flutter
         concatenateAndSaveSegments()
     }
 
@@ -353,10 +400,13 @@ extension CameraViewModel {
     private func exportComposition(_ composition: AVMutableComposition) {
         guard let exporter = AVAssetExportSession(asset: composition, presetName: AVAssetExportPresetHighestQuality) else {
             print("Failed to create export session")
+            DispatchQueue.main.async { self.isProcessingVideo = false }
             return
         }
 
-        let outputURL = URL(fileURLWithPath: NSTemporaryDirectory()).appendingPathComponent("final_video_\(UUID().uuidString).mov")
+        // Use app documents directory so file persists for Flutter to use
+        let docsDir = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first!
+        let outputURL = docsDir.appendingPathComponent("final_video_\(UUID().uuidString).mov")
         exporter.outputURL = outputURL
         exporter.outputFileType = .mov
 
@@ -370,71 +420,61 @@ extension CameraViewModel {
                 case .completed:
                     print("Video concatenation completed successfully")
                     print("Output URL: \(outputURL)")
-                    self.saveVideoToPhotos(outputURL)
-                    
-                    // CLEANUP segments ONLY after successful export
-                    for segment in self.segments {
-                        try? FileManager.default.removeItem(at: segment.url)
-                    }
-                    DispatchQueue.main.async {
-                        self.segments.removeAll()
-                    }
-
-                    // Clean up temp file after a delay to ensure save completes
-                    DispatchQueue.main.asyncAfter(deadline: .now() + 2.0) {
-                        try? FileManager.default.removeItem(at: outputURL)
-                    }
+                    self.finishWithVideo(at: outputURL)
                 case .failed, .cancelled:
                     print("Video concatenation failed: \(exporter.error?.localizedDescription ?? "Unknown error")")
                     if let error = exporter.error {
                         print("Export error details: \(error)")
                     }
+                    DispatchQueue.main.async { self.isProcessingVideo = false }
                 default:
                     print("Export in progress or unknown status")
                     break
                 }
             }
         } else {
             print("Applying filter (\(selectedFilter.displayName)) to concatenated video")
-            applyFilter(selectedFilter, to: composition) { result in
+            applyFilter(selectedFilter, to: composition) { [weak self] result in
+                guard let self = self else { return }
                 switch result {
                 case .success(let filteredURL):
-                    print("Filter applied successfully, saving filtered video")
-                    self.saveVideoToPhotos(filteredURL)
-
-                    // CLEANUP segments ONLY after successful filtered export
-                    for segment in self.segments {
-                        try? FileManager.default.removeItem(at: segment.url)
-                    }
-                    DispatchQueue.main.async {
-                        self.segments.removeAll()
-                    }
-
-                    // Clean up temp files after a delay
-                    DispatchQueue.main.asyncAfter(deadline: .now() + 2.0) {
-                        try? FileManager.default.removeItem(at: filteredURL)
-                        try? FileManager.default.removeItem(at: outputURL)
-                    }
+                    print("Filter applied successfully")
+                    self.finishWithVideo(at: filteredURL)
+                    // Clean up unfiltered temp file
+                    try? FileManager.default.removeItem(at: outputURL)
                 case .failure(let error):
                     print("Filter application failed: \(error)")
-                    print("Saving unfiltered video instead")
-                    self.saveVideoToPhotos(outputURL)
-                    
-                    // Even on failure, if we are saving the raw one, we cleanup segments after saving
-                    for segment in self.segments {
-                        try? FileManager.default.removeItem(at: segment.url)
-                    }
-                    DispatchQueue.main.async {
-                        self.segments.removeAll()
-                    }
-
-                    // Clean up temp file after a delay
-                    DispatchQueue.main.asyncAfter(deadline: .now() + 2.0) {
-                        try? FileManager.default.removeItem(at: outputURL)
-                    }
+                    print("Using unfiltered video instead")
+                    self.finishWithVideo(at: outputURL)
                 }
             }
         }
+
+        // Clean up individual segment files
+        for segment in segments {
+            try? FileManager.default.removeItem(at: segment.url)
+        }
+
+        // Clear segments after processing
+        DispatchQueue.main.async {
+            self.segments.removeAll()
+        }
+    }
+    
+    /// Finishes the recording flow by notifying Flutter with the video path
+    private func finishWithVideo(at url: URL) {
+        // Stop the camera session
+        stop()
+        
+        // Notify Flutter with the video path
+        DispatchQueue.main.async {
+            self.isProcessingVideo = false
+            NotificationCenter.default.post(
+                name: Notification.Name("conty.teleprompter.finished"),
+                object: nil,
+                userInfo: ["path": url.path, "savedToGallery": false]
+            )
+        }
     }
 
     private func applyFilter(_ filterType: VideoFilter, to composition: AVMutableComposition, completion: @escaping (Result<URL, Error>) -> Void) {

Camera/ContentView.swift

@@ -36,15 +36,27 @@ struct ContentView: View {
 
                     ZStack {
                         CameraPreviewRepresentable(model: model)
+                            .opacity(model.isCameraReady ? 1 : 0)
+
+                        // Loading overlay while camera initializes
+                        if !model.isCameraReady {
+                            ZStack {
+                                Color.black
+                                ProgressView()
+                                    .progressViewStyle(CircularProgressViewStyle(tint: .white))
+                                    .scaleEffect(1.2)
+                            }
+                            .transition(.opacity)
+                        }
 
                         // Grid overlay (rule of thirds) - clipped to preview
-                        if model.showGrid {
+                        if model.showGrid && model.isCameraReady {
                             GridOverlay()
                                 .allowsHitTesting(false)
                         }
 
                         // Lightweight overlay hint to approximate selected filter visually
-                        if let overlay = previewOverlayColor(for: model.selectedFilter) {
+                        if let overlay = previewOverlayColor(for: model.selectedFilter), model.isCameraReady {
                             overlay
                                 .opacity(0.12)
                                 .allowsHitTesting(false)
@@ -56,6 +68,7 @@ struct ContentView: View {
                     )
                     .clipShape(RoundedRectangle(cornerRadius: 20, style: .continuous))
                     .offset(y: -30)
+                    .animation(.easeOut(duration: 0.3), value: model.isCameraReady)
 
                     Spacer()
                 }
@@ -68,7 +81,7 @@ struct ContentView: View {
                 if !model.isRecording {
                     HStack {
                         // Close button
-                        Button(action: { dismiss() }) {
+                        Button(action: { model.dismissCamera() }) {
                             Image(systemName: "xmark")
                                 .font(.system(size: 14, weight: .semibold))
                                 .foregroundColor(.white)
@@ -318,18 +331,42 @@ struct ContentView: View {
                     HStack {
                         Spacer()
                         Button(action: { model.nextAction() }) {
-                            Image(systemName: "arrow.right.circle.fill")
-                                .font(.system(size: 26, weight: .semibold))
-                                .foregroundColor(.white)
-                                .padding(11)
+                            if model.isProcessingVideo {
+                                ProgressView()
+                                    .progressViewStyle(CircularProgressViewStyle(tint: .white))
+                                    .padding(11)
+                            } else {
+                                Image(systemName: "arrow.right.circle.fill")
+                                    .font(.system(size: 26, weight: .semibold))
+                                    .foregroundColor(.white)
+                                    .padding(11)
+                            }
                         }
+                        .disabled(model.isProcessingVideo)
                         .buttonStyle(GlassCircleButtonStyle(isProminent: true, accentColor: Color(red: 0x29/255.0, green: 0x84/255.0, blue: 0xf6/255.0)))
                         .padding(.trailing, 12)
                         .padding(.bottom, 10)
                         .accessibilityLabel("Avançar")
                     }
                 }
             }
+            
+            // Processing overlay
+            if model.isProcessingVideo {
+                Color.black.opacity(0.4)
+                    .ignoresSafeArea()
+                    .allowsHitTesting(true)
+                
+                VStack(spacing: 16) {
+                    ProgressView()
+                        .progressViewStyle(CircularProgressViewStyle(tint: .white))
+                        .scaleEffect(1.5)
+                    Text("Preparando vídeo...")
+                        .font(.system(size: 16, weight: .medium))
+                        .foregroundColor(.white)
+                }
+                .transition(.opacity)
+            }
         }
         .onAppear { model.requestPermissionsAndConfigure() }
         .onDisappear { model.stop() }