VB: Move VoiceBroadcastRecorderService into a MatrixSDK

to make pure SwiftUI projects build
2026-04-20 16:42:44 +02:00 · 2022-10-20 20:10:41 +02:00
parent 0aa333038e 0aa333038e
commit f26ee722f9
2 changed files with 4 additions and 4 deletions
@@ -0,0 +1,274 @@
+// 
+// Copyright 2022 New Vector Ltd
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+
+import Foundation
+
+class VoiceBroadcastRecorderService: VoiceBroadcastRecorderServiceProtocol {
+    
+    // MARK: - Properties
+    
+    // MARK: Private
+    
+    private let roomId: String
+    private let session: MXSession
+    private var voiceBroadcastService: VoiceBroadcastService? {
+        session.voiceBroadcastService
+    }
+    
+    private let audioEngine = AVAudioEngine()
+    private let audioNodeBus = AVAudioNodeBus(0)
+    
+    private var chunkFile: AVAudioFile! = nil
+    private var chunkFrames: AVAudioFrameCount = 0
+    private var chunkFileNumber: Int = 1
+        
+    // MARK: Public
+    
+    weak var serviceDelegate: VoiceBroadcastRecorderServiceDelegate?
+
+    // MARK: - Setup
+    
+    init(session: MXSession, roomId: String) {
+        self.session = session
+        self.roomId = roomId
+    }
+    
+    // MARK: - VoiceBroadcastRecorderServiceProtocol
+    
+    func startRecordingVoiceBroadcast() {
+        let inputNode = audioEngine.inputNode
+
+        let inputFormat = inputNode.inputFormat(forBus: audioNodeBus)
+        MXLog.debug("[VoiceBroadcastRecorderService] Start recording voice broadcast for bus name : \(String(describing: inputNode.name(forInputBus: audioNodeBus)))")
+
+        inputNode.installTap(onBus: audioNodeBus,
+                             bufferSize: 512,
+                             format: inputFormat) { (buffer, time) -> Void in
+            DispatchQueue.main.async {
+                self.writeBuffer(buffer)
+            }
+        }
+
+        try? audioEngine.start()
+    }
+    
+    func stopRecordingVoiceBroadcast() {
+        MXLog.debug("[VoiceBroadcastRecorderService] Stop recording voice broadcast")
+        audioEngine.stop()
+        audioEngine.inputNode.removeTap(onBus: audioNodeBus)
+
+        resetValues()
+
+        voiceBroadcastService?.stopVoiceBroadcast(success: { [weak self] _ in
+            MXLog.debug("[VoiceBroadcastRecorderService] Stopped")
+            
+            guard let self = self else { return }
+            
+            // Update state
+            self.serviceDelegate?.voiceBroadcastRecorderService(self, didUpdateState: .stopped)
+            
+            // Send current chunk
+            if self.chunkFile != nil {
+                self.sendChunkFile(at: self.chunkFile.url, sequence: self.chunkFileNumber)
+            }
+            
+            self.session.tearDownVoiceBroadcastService()
+        }, failure: { error in
+            MXLog.error("[VoiceBroadcastRecorderService] Failed to stop voice broadcast", context: error)
+        })
+    }
+    
+    func pauseRecordingVoiceBroadcast() {
+        audioEngine.pause()
+        
+        voiceBroadcastService?.pauseVoiceBroadcast(success: { [weak self] _ in
+            guard let self = self else { return }
+            
+            // Send current chunk
+            self.sendChunkFile(at: self.chunkFile.url, sequence: self.chunkFileNumber)
+            self.chunkFile = nil
+            
+        }, failure: { error in
+            MXLog.error("[VoiceBroadcastRecorderService] Failed to pause voice broadcast", context: error)
+        })
+    }
+    
+    func resumeRecordingVoiceBroadcast() {
+        try? audioEngine.start()
+        
+        voiceBroadcastService?.resumeVoiceBroadcast(success: { [weak self] _ in
+            guard let self = self else { return }
+            
+            // Update state
+            self.serviceDelegate?.voiceBroadcastRecorderService(self, didUpdateState: .started)
+        }, failure: { error in
+            MXLog.error("[VoiceBroadcastRecorderService] Failed to resume voice broadcast", context: error)
+        })
+    }
+    
+    // MARK: - Private
+    /// Reset chunk values.
+    private func resetValues() {
+        chunkFrames = 0
+        chunkFileNumber = 1
+    }
+    
+    /// Write audio buffer to chunk file.
+    private func writeBuffer(_ buffer: AVAudioPCMBuffer) {
+        let sampleRate = buffer.format.sampleRate
+        
+        if chunkFile == nil {
+            createNewChunkFile(channelsCount: buffer.format.channelCount, sampleRate: sampleRate)
+        }
+        try? chunkFile.write(from: buffer)
+        
+        chunkFrames += buffer.frameLength
+        
+        if chunkFrames > AVAudioFrameCount(Double(BuildSettings.voiceBroadcastChunkLength) * sampleRate) {
+            sendChunkFile(at: chunkFile.url, sequence: self.chunkFileNumber)
+            // Reset chunkFile
+            chunkFile = nil
+        }
+    }
+    
+    /// Create new chunk file with sample rate.
+    private func createNewChunkFile(channelsCount: AVAudioChannelCount, sampleRate: Float64) {
+        guard let directory = FileManager.default.urls(for: .cachesDirectory, in: .userDomainMask).first else {
+            // FIXME: Manage error
+            return
+        }
+        let temporaryFileName = "VoiceBroadcastChunk-\(roomId)-\(chunkFileNumber)"
+        let fileUrl = directory
+            .appendingPathComponent(temporaryFileName)
+            .appendingPathExtension("aac")
+        MXLog.debug("[VoiceBroadcastRecorderService] Create chunk file to \(fileUrl)")
+        
+        let settings: [String: Any] = [AVFormatIDKey: Int(kAudioFormatMPEG4AAC),
+                                     AVSampleRateKey: sampleRate,
+                                     AVEncoderBitRateKey: 128000,
+                                     AVNumberOfChannelsKey: channelsCount,
+                                     AVEncoderAudioQualityKey: AVAudioQuality.high.rawValue]
+        
+        chunkFile = try? AVAudioFile(forWriting: fileUrl, settings: settings)
+        
+        if chunkFile != nil {
+            chunkFileNumber += 1
+            chunkFrames = 0
+        } else {
+            stopRecordingVoiceBroadcast()
+            // FIXME: Manage error ?
+        }
+    }
+    
+    /// Send chunk file to the server.
+    private func sendChunkFile(at url: URL, sequence: Int) {
+        guard let voiceBroadcastService = voiceBroadcastService else {
+            // FIXME: Manage error
+            return
+        }
+        
+        let dispatchGroup = DispatchGroup()
+        var duration = 0.0
+        
+        dispatchGroup.enter()
+        VoiceMessageAudioConverter.mediaDurationAt(url) { result in
+            switch result {
+            case .success:
+                if let someDuration = try? result.get() {
+                    duration = someDuration
+                } else {
+                    MXLog.error("[VoiceBroadcastRecorderService] Failed to retrieve media duration")
+                }
+            case .failure(let error):
+                MXLog.error("[VoiceBroadcastRecorderService] Failed to get audio duration", context: error)
+            }
+            
+            dispatchGroup.leave()
+        }
+        
+        convertAACToM4A(at: url) { [weak self] convertedUrl in
+            guard let self = self else { return }
+            
+            if let convertedUrl = convertedUrl {
+                dispatchGroup.notify(queue: .main) {
+                    self.voiceBroadcastService?.sendChunkOfVoiceBroadcast(audioFileLocalURL: convertedUrl,
+                                                                          mimeType: "audio/mp4",
+                                                                          duration: UInt(duration * 1000),
+                                                                          samples: nil,
+                                                                          sequence: UInt(sequence)) { eventId in
+                        MXLog.debug("[VoiceBroadcastRecorderService] Send voice broadcast chunk with success.")
+                        if eventId != nil {
+                            self.deleteRecording(at: url)
+                        }
+                    } failure: { error in
+                        MXLog.error("[VoiceBroadcastRecorderService] Failed to send voice broadcast chunk.", context: error)
+                    }
+                }
+            }
+        }
+    }
+    
+    /// Delete voice broadcast chunk at URL.
+    private func deleteRecording(at url: URL?) {
+        guard let url = url else {
+            return
+        }
+        
+        do {
+            try FileManager.default.removeItem(at: url)
+        } catch {
+            MXLog.error("[VoiceBroadcastRecorderService] Delete chunk file error.", context: error)
+        }
+    }
+    
+    /// Convert AAC file into m4a one.
+    private func convertAACToM4A(at url: URL, completion: @escaping (URL?) -> Void) {
+        // FIXME: Manage errors at completion
+        let asset = AVURLAsset(url: url)
+        let updatedPath = url.path.replacingOccurrences(of: ".aac", with: ".m4a")
+        let outputUrl = URL(string: "file://" + updatedPath)
+        MXLog.debug("[VoiceBroadcastRecorderService] convertAACToM4A updatedPath : \(updatedPath).")
+        
+        if FileManager.default.fileExists(atPath: updatedPath) {
+            try? FileManager.default.removeItem(atPath: updatedPath)
+        }
+        
+        guard let exportSession = AVAssetExportSession(asset: asset,
+                                                       presetName: AVAssetExportPresetPassthrough) else {
+            completion(nil)
+            return
+        }
+        
+        exportSession.outputURL = outputUrl
+        exportSession.outputFileType = AVFileType.m4a
+        let start = CMTimeMakeWithSeconds(0.0, preferredTimescale: 0)
+        let range = CMTimeRangeMake(start: start, duration: asset.duration)
+        exportSession.timeRange = range
+        exportSession.exportAsynchronously() {
+            switch exportSession.status {
+            case .failed:
+                MXLog.error("[VoiceBroadcastRecorderService] convertAACToM4A error", context: exportSession.error)
+                completion(nil)
+            case .completed:
+                MXLog.debug("[VoiceBroadcastRecorderService] convertAACToM4A success.")
+                completion(outputUrl)
+            default:
+                MXLog.debug("[VoiceBroadcastRecorderService] convertAACToM4A other cases.")
+                completion(nil)
+            }
+        }
+    }
+}