2021-06-07 07:20:26 +00:00
|
|
|
//
|
|
|
|
// 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
|
2021-06-14 14:47:59 +00:00
|
|
|
import AVFoundation
|
2021-06-22 10:19:39 +00:00
|
|
|
import DSWaveformImage
|
2021-06-07 07:20:26 +00:00
|
|
|
|
|
|
|
@objc public protocol VoiceMessageControllerDelegate: AnyObject {
|
2021-06-14 14:47:59 +00:00
|
|
|
func voiceMessageControllerDidRequestMicrophonePermission(_ voiceMessageController: VoiceMessageController)
|
2021-06-07 07:20:26 +00:00
|
|
|
func voiceMessageController(_ voiceMessageController: VoiceMessageController, didRequestSendForFileAtURL url: URL, completion: @escaping (Bool) -> Void)
|
|
|
|
}
|
|
|
|
|
2021-06-22 10:19:39 +00:00
|
|
|
public class VoiceMessageController: NSObject, VoiceMessageToolbarViewDelegate, VoiceMessageAudioRecorderDelegate, VoiceMessageAudioPlayerDelegate {
|
2021-06-07 07:20:26 +00:00
|
|
|
|
2021-06-24 08:33:14 +00:00
|
|
|
private static let timeFormatter: DateFormatter = {
|
|
|
|
let dateFormatter = DateFormatter()
|
|
|
|
dateFormatter.dateFormat = "m:ss"
|
|
|
|
return dateFormatter
|
|
|
|
}()
|
|
|
|
|
2021-06-07 07:20:26 +00:00
|
|
|
private let themeService: ThemeService
|
2021-06-23 14:52:08 +00:00
|
|
|
private let mediaServiceProvider: VoiceMessageMediaServiceProvider
|
|
|
|
|
2021-06-07 07:20:26 +00:00
|
|
|
private let _voiceMessageToolbarView: VoiceMessageToolbarView
|
2021-06-08 07:04:44 +00:00
|
|
|
private var displayLink: CADisplayLink!
|
2021-06-07 07:20:26 +00:00
|
|
|
|
2021-06-14 14:47:59 +00:00
|
|
|
private var audioRecorder: VoiceMessageAudioRecorder?
|
2021-06-07 07:20:26 +00:00
|
|
|
|
2021-06-22 10:19:39 +00:00
|
|
|
private var audioPlayer: VoiceMessageAudioPlayer?
|
|
|
|
private var waveformAnalyser: WaveformAnalyzer?
|
|
|
|
|
2021-06-18 12:49:58 +00:00
|
|
|
private var audioSamples: [Float] = []
|
|
|
|
private var isInLockedMode: Bool = false
|
|
|
|
|
2021-06-07 07:20:26 +00:00
|
|
|
@objc public weak var delegate: VoiceMessageControllerDelegate?
|
|
|
|
|
|
|
|
@objc public var voiceMessageToolbarView: UIView {
|
|
|
|
return _voiceMessageToolbarView
|
|
|
|
}
|
|
|
|
|
2021-06-23 14:52:08 +00:00
|
|
|
@objc public init(themeService: ThemeService, mediaServiceProvider: VoiceMessageMediaServiceProvider) {
|
2021-06-07 07:20:26 +00:00
|
|
|
self.themeService = themeService
|
2021-06-23 14:52:08 +00:00
|
|
|
self.mediaServiceProvider = mediaServiceProvider
|
|
|
|
|
|
|
|
_voiceMessageToolbarView = VoiceMessageToolbarView.loadFromNib()
|
2021-06-07 07:20:26 +00:00
|
|
|
|
|
|
|
super.init()
|
|
|
|
|
|
|
|
_voiceMessageToolbarView.delegate = self
|
|
|
|
|
2021-06-22 06:59:14 +00:00
|
|
|
displayLink = CADisplayLink(target: WeakObjectWrapper(self), selector: #selector(handleDisplayLinkTick))
|
2021-06-08 07:04:44 +00:00
|
|
|
displayLink.isPaused = true
|
|
|
|
displayLink.add(to: .current, forMode: .common)
|
|
|
|
|
2021-06-23 12:37:34 +00:00
|
|
|
NotificationCenter.default.addObserver(self, selector: #selector(updateTheme), name: .themeServiceDidChangeTheme, object: nil)
|
|
|
|
updateTheme()
|
2021-06-18 12:49:58 +00:00
|
|
|
|
|
|
|
updateUI()
|
2021-06-07 07:20:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// MARK: - VoiceMessageToolbarViewDelegate
|
|
|
|
|
|
|
|
func voiceMessageToolbarViewDidRequestRecordingStart(_ toolbarView: VoiceMessageToolbarView) {
|
2021-06-14 14:47:59 +00:00
|
|
|
guard AVAudioSession.sharedInstance().recordPermission == .granted else {
|
|
|
|
delegate?.voiceMessageControllerDidRequestMicrophonePermission(self)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
let temporaryDirectoryURL = URL(fileURLWithPath: NSTemporaryDirectory(), isDirectory: true)
|
|
|
|
let temporaryFileURL = temporaryDirectoryURL.appendingPathComponent(ProcessInfo().globallyUniqueString).appendingPathExtension("m4a")
|
|
|
|
|
2021-06-23 14:52:08 +00:00
|
|
|
audioRecorder = mediaServiceProvider.audioRecorder()
|
|
|
|
audioRecorder?.registerDelegate(self)
|
2021-06-22 10:19:39 +00:00
|
|
|
audioRecorder?.recordWithOuputURL(temporaryFileURL)
|
2021-06-07 07:20:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func voiceMessageToolbarViewDidRequestRecordingFinish(_ toolbarView: VoiceMessageToolbarView) {
|
|
|
|
audioRecorder?.stopRecording()
|
|
|
|
|
|
|
|
guard let url = audioRecorder?.url else {
|
|
|
|
MXLog.error("Invalid audio recording URL")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-06-22 10:19:39 +00:00
|
|
|
guard isInLockedMode else {
|
|
|
|
sendRecordingAtURL(url)
|
|
|
|
return
|
2021-06-07 07:20:26 +00:00
|
|
|
}
|
2021-06-22 10:19:39 +00:00
|
|
|
|
2021-06-23 14:52:08 +00:00
|
|
|
audioPlayer = mediaServiceProvider.audioPlayer()
|
|
|
|
audioPlayer?.registerDelegate(self)
|
2021-06-22 10:19:39 +00:00
|
|
|
audioPlayer?.loadContentFromURL(url)
|
|
|
|
audioSamples = []
|
|
|
|
|
|
|
|
updateUI()
|
2021-06-07 07:20:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func voiceMessageToolbarViewDidRequestRecordingCancel(_ toolbarView: VoiceMessageToolbarView) {
|
2021-06-18 12:49:58 +00:00
|
|
|
isInLockedMode = false
|
2021-06-07 07:20:26 +00:00
|
|
|
audioRecorder?.stopRecording()
|
|
|
|
deleteRecordingAtURL(audioRecorder?.url)
|
2021-06-08 07:04:44 +00:00
|
|
|
UINotificationFeedbackGenerator().notificationOccurred(.error)
|
2021-06-22 10:19:39 +00:00
|
|
|
updateUI()
|
2021-06-07 07:20:26 +00:00
|
|
|
}
|
|
|
|
|
2021-06-18 12:49:58 +00:00
|
|
|
func voiceMessageToolbarViewDidRequestLockedModeRecording(_ toolbarView: VoiceMessageToolbarView) {
|
|
|
|
isInLockedMode = true
|
|
|
|
updateUI()
|
|
|
|
}
|
|
|
|
|
2021-06-22 10:19:39 +00:00
|
|
|
func voiceMessageToolbarViewDidRequestPlaybackToggle(_ toolbarView: VoiceMessageToolbarView) {
|
|
|
|
if audioPlayer?.isPlaying ?? false {
|
|
|
|
audioPlayer?.pause()
|
|
|
|
} else {
|
|
|
|
audioPlayer?.play()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func voiceMessageToolbarViewDidRequestSend(_ toolbarView: VoiceMessageToolbarView) {
|
|
|
|
guard let url = audioRecorder?.url else {
|
|
|
|
MXLog.error("Invalid audio recording URL")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-06-22 13:58:36 +00:00
|
|
|
audioPlayer?.stop()
|
|
|
|
audioRecorder?.stopRecording()
|
|
|
|
|
2021-06-22 10:19:39 +00:00
|
|
|
sendRecordingAtURL(url)
|
|
|
|
|
|
|
|
isInLockedMode = false
|
|
|
|
updateUI()
|
|
|
|
}
|
|
|
|
|
2021-06-07 07:20:26 +00:00
|
|
|
// MARK: - AudioRecorderDelegate
|
|
|
|
|
2021-06-14 14:47:59 +00:00
|
|
|
func audioRecorderDidStartRecording(_ audioRecorder: VoiceMessageAudioRecorder) {
|
2021-06-18 12:49:58 +00:00
|
|
|
updateUI()
|
2021-06-07 07:20:26 +00:00
|
|
|
}
|
|
|
|
|
2021-06-14 14:47:59 +00:00
|
|
|
func audioRecorderDidFinishRecording(_ audioRecorder: VoiceMessageAudioRecorder) {
|
2021-06-18 12:49:58 +00:00
|
|
|
updateUI()
|
2021-06-07 07:20:26 +00:00
|
|
|
}
|
|
|
|
|
2021-06-14 14:47:59 +00:00
|
|
|
func audioRecorder(_ audioRecorder: VoiceMessageAudioRecorder, didFailWithError: Error) {
|
2021-06-18 12:49:58 +00:00
|
|
|
isInLockedMode = false
|
|
|
|
updateUI()
|
|
|
|
|
2021-06-07 07:20:26 +00:00
|
|
|
MXLog.error("Failed recording voice message.")
|
|
|
|
}
|
|
|
|
|
2021-06-22 10:19:39 +00:00
|
|
|
// MARK: - VoiceMessageAudioPlayerDelegate
|
|
|
|
|
|
|
|
func audioPlayerDidStartPlaying(_ audioPlayer: VoiceMessageAudioPlayer) {
|
|
|
|
updateUI()
|
|
|
|
}
|
|
|
|
|
2021-06-23 14:52:08 +00:00
|
|
|
func audioPlayerDidPausePlaying(_ audioPlayer: VoiceMessageAudioPlayer) {
|
|
|
|
updateUI()
|
|
|
|
}
|
|
|
|
|
2021-06-22 10:19:39 +00:00
|
|
|
func audioPlayerDidStopPlaying(_ audioPlayer: VoiceMessageAudioPlayer) {
|
|
|
|
updateUI()
|
|
|
|
}
|
|
|
|
|
|
|
|
func audioPlayerDidFinishPlaying(_ audioPlayer: VoiceMessageAudioPlayer) {
|
|
|
|
audioPlayer.seekToTime(0.0)
|
|
|
|
updateUI()
|
|
|
|
}
|
|
|
|
|
|
|
|
func audioPlayer(_ audioPlayer: VoiceMessageAudioPlayer, didFailWithError: Error) {
|
|
|
|
updateUI()
|
|
|
|
|
|
|
|
MXLog.error("Failed playing voice message.")
|
|
|
|
}
|
|
|
|
|
2021-06-07 07:20:26 +00:00
|
|
|
// MARK: - Private
|
|
|
|
|
2021-06-22 10:19:39 +00:00
|
|
|
private func sendRecordingAtURL(_ url: URL) {
|
|
|
|
delegate?.voiceMessageController(self, didRequestSendForFileAtURL: url) { [weak self] success in
|
|
|
|
UINotificationFeedbackGenerator().notificationOccurred( (success ? .success : .error))
|
|
|
|
self?.deleteRecordingAtURL(url)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-06-07 07:20:26 +00:00
|
|
|
private func deleteRecordingAtURL(_ url: URL?) {
|
|
|
|
guard let url = url else {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
do {
|
|
|
|
try FileManager.default.removeItem(at: url)
|
|
|
|
} catch {
|
|
|
|
MXLog.error(error)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-06-23 12:37:34 +00:00
|
|
|
@objc private func updateTheme() {
|
2021-06-22 10:19:39 +00:00
|
|
|
_voiceMessageToolbarView.update(theme: themeService.theme)
|
2021-06-07 07:20:26 +00:00
|
|
|
}
|
2021-06-08 07:04:44 +00:00
|
|
|
|
|
|
|
@objc private func handleDisplayLinkTick() {
|
2021-06-18 12:49:58 +00:00
|
|
|
updateUI()
|
|
|
|
}
|
|
|
|
|
|
|
|
private func updateUI() {
|
2021-06-22 10:19:39 +00:00
|
|
|
|
|
|
|
let shouldUpdateFromAudioPlayer = isInLockedMode && !(audioRecorder?.isRecording ?? false)
|
|
|
|
|
2021-06-22 13:58:36 +00:00
|
|
|
if shouldUpdateFromAudioPlayer {
|
|
|
|
updateUIFromAudioPlayer()
|
|
|
|
} else {
|
2021-06-22 10:19:39 +00:00
|
|
|
updateUIFromAudioRecorder()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private func updateUIFromAudioRecorder() {
|
|
|
|
displayLink.isPaused = !(self.audioRecorder?.isRecording ?? false)
|
2021-06-18 12:49:58 +00:00
|
|
|
|
|
|
|
let requiredNumberOfSamples = _voiceMessageToolbarView.getRequiredNumberOfSamples()
|
|
|
|
|
|
|
|
if audioSamples.count != requiredNumberOfSamples {
|
2021-06-23 12:37:34 +00:00
|
|
|
padSamplesArrayToSize(requiredNumberOfSamples)
|
2021-06-18 12:49:58 +00:00
|
|
|
}
|
|
|
|
|
2021-06-22 10:19:39 +00:00
|
|
|
let sample = audioRecorder?.averagePowerForChannelNumber(0) ?? 0.0
|
|
|
|
audioSamples.append(sample)
|
|
|
|
audioSamples.remove(at: 0)
|
|
|
|
|
|
|
|
var details = VoiceMessageToolbarViewDetails()
|
|
|
|
details.state = (self.audioRecorder?.isRecording ?? false ? (isInLockedMode ? .lockedModeRecord : .record) : (isInLockedMode ? .lockedModePlayback : .idle))
|
2021-06-24 08:33:14 +00:00
|
|
|
details.elapsedTime = VoiceMessageController.timeFormatter.string(from: Date(timeIntervalSinceReferenceDate: self.audioRecorder?.currentTime ?? 0.0))
|
2021-06-22 10:19:39 +00:00
|
|
|
details.audioSamples = audioSamples
|
|
|
|
_voiceMessageToolbarView.configureWithDetails(details)
|
|
|
|
}
|
|
|
|
|
|
|
|
private func updateUIFromAudioPlayer() {
|
|
|
|
guard let audioPlayer = audioPlayer else {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
guard let url = audioPlayer.url else {
|
|
|
|
MXLog.error("Invalid audio player url.")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
displayLink.isPaused = !audioPlayer.isPlaying
|
|
|
|
|
|
|
|
let requiredNumberOfSamples = _voiceMessageToolbarView.getRequiredNumberOfSamples()
|
|
|
|
if audioSamples.count != requiredNumberOfSamples {
|
2021-06-23 12:37:34 +00:00
|
|
|
padSamplesArrayToSize(requiredNumberOfSamples)
|
2021-06-22 10:19:39 +00:00
|
|
|
|
|
|
|
waveformAnalyser = WaveformAnalyzer(audioAssetURL: url)
|
|
|
|
waveformAnalyser?.samples(count: requiredNumberOfSamples, completionHandler: { [weak self] samples in
|
|
|
|
guard let samples = samples else {
|
|
|
|
MXLog.error("Could not sample audio recording.")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
DispatchQueue.main.async {
|
|
|
|
self?.audioSamples = samples
|
|
|
|
self?.updateUIFromAudioPlayer()
|
|
|
|
}
|
|
|
|
})
|
2021-06-08 07:04:44 +00:00
|
|
|
}
|
|
|
|
|
2021-06-18 12:49:58 +00:00
|
|
|
var details = VoiceMessageToolbarViewDetails()
|
|
|
|
details.state = (audioRecorder?.isRecording ?? false ? (isInLockedMode ? .lockedModeRecord : .record) : (isInLockedMode ? .lockedModePlayback : .idle))
|
2021-06-24 08:33:14 +00:00
|
|
|
details.elapsedTime = VoiceMessageController.timeFormatter.string(from: Date(timeIntervalSinceReferenceDate: (audioPlayer.isPlaying ? audioPlayer.currentTime : audioPlayer.duration)))
|
2021-06-18 12:49:58 +00:00
|
|
|
details.audioSamples = audioSamples
|
2021-06-22 10:19:39 +00:00
|
|
|
details.isPlaying = audioPlayer.isPlaying
|
|
|
|
details.progress = (audioPlayer.duration > 0.0 ? audioPlayer.currentTime / audioPlayer.duration : 0.0)
|
2021-06-18 12:49:58 +00:00
|
|
|
_voiceMessageToolbarView.configureWithDetails(details)
|
2021-06-08 07:04:44 +00:00
|
|
|
}
|
2021-06-23 12:37:34 +00:00
|
|
|
|
|
|
|
private func padSamplesArrayToSize(_ size: Int) {
|
|
|
|
let delta = size - audioSamples.count
|
|
|
|
guard delta > 0 else {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
audioSamples = audioSamples + [Float](repeating: 0.0, count: delta)
|
|
|
|
}
|
2021-06-07 07:20:26 +00:00
|
|
|
}
|