#4096 - Extracted logic from the VoiceMessagePlaybackView. Exposed power levels from audio recorder.

2026-04-19 08:03:50 +02:00 · 2021-06-17 13:33:37 +03:00
parent 699fe622ea
commit 28131c8828
4 changed files with 274 additions and 176 deletions
@@ -0,0 +1,198 @@
+// 
+// Copyright 2021 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
+import DSWaveformImage
+
+enum VoiceMessagePlaybackControllerState {
+    case stopped
+    case playing
+    case paused
+    case error
+}
+
+class VoiceMessagePlaybackController: VoiceMessageAudioPlayerDelegate, VoiceMessagePlaybackViewDelegate {
+    private let audioPlayer: VoiceMessageAudioPlayer
+    private let timeFormatter: DateFormatter
+    private var displayLink: CADisplayLink!
+    private var samples: [Float] = []
+    
+    private var state: VoiceMessagePlaybackControllerState = .stopped {
+        didSet {
+            updateUI()
+            displayLink.isPaused = (state != .playing)
+        }
+    }
+    
+    let playbackView: VoiceMessagePlaybackView
+    
+    init() {
+        playbackView = VoiceMessagePlaybackView.instanceFromNib()
+        audioPlayer = VoiceMessageAudioPlayer()
+
+        timeFormatter = DateFormatter()
+        timeFormatter.dateFormat = "m:ss"
+        
+        audioPlayer.delegate = self
+        playbackView.delegate = self
+        
+        displayLink = CADisplayLink(target: self, selector: #selector(handleDisplayLinkTick))
+        displayLink.isPaused = true
+        displayLink.add(to: .current, forMode: .common)
+    }
+ 
+    var attachment: MXKAttachment? {
+        didSet {
+            if oldValue?.contentURL == attachment?.contentURL &&
+                oldValue?.eventSentState == attachment?.eventSentState {
+                return
+            }
+            
+            switch attachment?.eventSentState {
+            case MXEventSentStateFailed:
+                state = .error
+            default:
+                state = .stopped
+                loadAttachmentData()
+            }
+        }
+    }
+    
+    // MARK: - VoiceMessagePlaybackViewDelegate
+    
+    func voiceMessagePlaybackViewDidRequestToggle() {
+        if audioPlayer.isPlaying {
+            audioPlayer.pause()
+        } else {
+            audioPlayer.play()
+        }
+    }
+    
+    // MARK: - VoiceMessageAudioPlayerDelegate
+    
+    func audioPlayerDidFinishLoading(_ audioPlayer: VoiceMessageAudioPlayer) {
+        updateUI()
+    }
+    
+    func audioPlayerDidStartPlaying(_ audioPlayer: VoiceMessageAudioPlayer) {
+        state = .playing
+    }
+    
+    func audioPlayerDidStopPlaying(_ audioPlayer: VoiceMessageAudioPlayer) {
+        state = .paused
+    }
+    
+    func audioPlayer(_ audioPlayer: VoiceMessageAudioPlayer, didFailWithError error: Error) {
+        state = .error
+        MXLog.error("Failed playing voice message with error: \(error)")
+    }
+    
+    func audioPlayerDidFinishPlaying(_ audioPlayer: VoiceMessageAudioPlayer) {
+        audioPlayer.seekToTime(0.0)
+        state = .stopped
+    }
+    
+    // MARK: - Private
+    
+    @objc private func handleDisplayLinkTick() {
+        updateUI()
+    }
+    
+    private func updateUI() {
+        var details = VoiceMessagePlaybackViewDetails()
+        
+        details.playbackEnabled = (state != .error)
+        details.playing = (state == .playing)
+        details.samples = samples
+        
+        switch state {
+        case .stopped:
+            details.currentTime = timeFormatter.string(from: Date(timeIntervalSinceReferenceDate: audioPlayer.duration))
+            details.progress = 0.0
+        default:
+            details.currentTime = timeFormatter.string(from: Date(timeIntervalSinceReferenceDate: audioPlayer.currentTime))
+            details.progress = (audioPlayer.duration > 0.0 ? audioPlayer.currentTime / audioPlayer.duration : 0.0)
+        }
+        
+        playbackView.configureWithDetails(details)
+    }
+    
+    private func loadAttachmentData() {
+        guard let attachment = attachment else {
+            return
+        }
+        
+        if attachment.isEncrypted {
+            attachment.decrypt(toTempFile: { [weak self] filePath in
+                self?.loadFileAtPath(filePath)
+            }, failure: { [weak self] error in
+                // A nil error in this case is a cancellation on the MXMediaLoader
+                if let error = error {
+                    MXLog.error("Failed decrypting attachment with error: \(String(describing: error))")
+                    self?.state = .error
+                }
+            })
+        } else {
+            attachment.prepare({ [weak self] in
+                self?.loadFileAtPath(attachment.cacheFilePath)
+            }, failure: { [weak self] error in
+                MXLog.error("Failed preparing attachment with error: \(String(describing: error))")
+                self?.state = .error
+            })
+        }
+    }
+    
+    private func loadFileAtPath(_ path: String?) {
+        guard let filePath = path else {
+            return
+        }
+        
+        let url = URL(fileURLWithPath: filePath)
+        
+        // AVPlayer doesn't want to play it otherwise. https://stackoverflow.com/a/9350824
+        let newURL = url.appendingPathExtension("m4a")
+        
+        do {
+            try? FileManager.default.removeItem(at: newURL)
+            try FileManager.default.moveItem(at: url, to: newURL)
+        } catch {
+            self.state = .error
+            MXLog.error("Failed appending voice message extension.")
+            return
+        }
+        
+        audioPlayer.loadContentFromURL(newURL)
+        
+        let requiredNumberOfSamples = playbackView.getRequiredNumberOfSamples()
+        
+        if requiredNumberOfSamples == 0 {
+            return
+        }
+        
+        let analyser = WaveformAnalyzer(audioAssetURL: newURL)
+        analyser?.samples(count: requiredNumberOfSamples, completionHandler: { [weak self] samples in
+            guard let samples = samples else {
+                self?.state = .error
+                return
+            }
+            
+            DispatchQueue.main.async {
+                self?.samples = samples
+                self?.updateUI()
+            }
+        })
+    }
+}