From 192c9b2894a4bb116bfd8f186086e34d18912497 Mon Sep 17 00:00:00 2001
From: nielsandriesse <andriesseniels@gmail.com>
Date: Mon, 5 Oct 2020 10:54:27 +1100
Subject: [PATCH] Clean

---
 Signal/src/Loki/Components/VoiceMessageView.swift | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/Signal/src/Loki/Components/VoiceMessageView.swift b/Signal/src/Loki/Components/VoiceMessageView.swift
index 56a3784d1..5dde35138 100644
--- a/Signal/src/Loki/Components/VoiceMessageView.swift
+++ b/Signal/src/Loki/Components/VoiceMessageView.swift
@@ -5,6 +5,7 @@ final class VoiceMessageView : UIView {
     private let voiceMessage: TSAttachment
     private let isOutgoing: Bool
     private var isLoading = false
+    private var isForcedAnimation = false
     private var volumeSamples: [Float] = [] { didSet { updateShapeLayers() } }
     @objc var progress: CGFloat = 0 { didSet { updateShapeLayers() } }
     @objc var duration: Int = 0 { didSet { updateDurationLabel() } }
@@ -71,6 +72,7 @@ final class VoiceMessageView : UIView {
                 AudioUtilities.getVolumeSamples(for: url, targetSampleCount: targetSampleCount).done(on: DispatchQueue.main) { [weak self] volumeSamples in
                     guard let self = self else { return }
                     self.hideLoader()
+                    self.isForcedAnimation = true
                     self.volumeSamples = volumeSamples
                     Storage.write { transaction in
                         Storage.setVolumeSamples(for: voiceMessageID, to: volumeSamples, using: transaction)
@@ -160,7 +162,7 @@ final class VoiceMessageView : UIView {
         }
         backgroundPath.close()
         foregroundPath.close()
-        if isLoading {
+        if isLoading || isForcedAnimation {
             let animation = CABasicAnimation(keyPath: "path")
             animation.duration = 0.25
             animation.toValue = backgroundPath
@@ -171,6 +173,7 @@ final class VoiceMessageView : UIView {
             backgroundShapeLayer.path = backgroundPath.cgPath
         }
         foregroundShapeLayer.path = foregroundPath.cgPath
+        isForcedAnimation = false
     }
 
     private func updateDurationLabel() {