From cb105af17594cac5487d994dbb70ee205953ab56 Mon Sep 17 00:00:00 2001
From: Ali <>
Date: Fri, 14 Jul 2023 01:01:10 +0400
Subject: [PATCH] Use shared VolumeButtons monitoring

---
 .../Sources/ChatMessageReplyInfoNode.swift    |   2 +-
 .../VolumeButtons/Sources/VolumeButtons.swift | 155 ++++++++++++++++--
 2 files changed, 141 insertions(+), 16 deletions(-)

diff --git a/submodules/TelegramUI/Sources/ChatMessageReplyInfoNode.swift b/submodules/TelegramUI/Sources/ChatMessageReplyInfoNode.swift
index ed5d1ec224..cc0b7751b3 100644
--- a/submodules/TelegramUI/Sources/ChatMessageReplyInfoNode.swift
+++ b/submodules/TelegramUI/Sources/ChatMessageReplyInfoNode.swift
@@ -309,7 +309,7 @@ public class ChatMessageReplyInfoNode: ASDisplayNode {
             let (textLayout, textApply) = textNodeLayout(TextNodeLayoutArguments(attributedString: messageText, backgroundColor: nil, maximumNumberOfLines: 1, truncationType: .end, constrainedSize: contrainedTextSize, alignment: .natural, cutout: nil, insets: textInsets))
             
             let imageSide: CGFloat
-            imageSide = titleLayout.size.height + textLayout.size.height - 13.0
+            imageSide = titleLayout.size.height + textLayout.size.height - 12.0
             
             var applyImage: (() -> TransformImageNode)?
             if let imageDimensions = imageDimensions {
diff --git a/submodules/Utils/VolumeButtons/Sources/VolumeButtons.swift b/submodules/Utils/VolumeButtons/Sources/VolumeButtons.swift
index 58e5d866a8..846d9d0de4 100644
--- a/submodules/Utils/VolumeButtons/Sources/VolumeButtons.swift
+++ b/submodules/Utils/VolumeButtons/Sources/VolumeButtons.swift
@@ -5,9 +5,135 @@ import MediaPlayer
 
 import LegacyComponents
 
-public class VolumeButtonsListener: NSObject {
-    private let handler: PGCameraVolumeButtonHandler
+
+public class VolumeButtonsListener {
+    private final class ListenerReference {
+        let id: Int
+        weak var listener: VolumeButtonsListener?
+        
+        init(id: Int, listener: VolumeButtonsListener) {
+            self.id = id
+            self.listener = listener
+        }
+    }
     
+    private enum Action {
+        case up
+        case upRelease
+        case down
+        case downRelease
+    }
+    
+    private final class SharedContext: NSObject {
+        private var handler: PGCameraVolumeButtonHandler?
+        
+        private var nextListenerId: Int = 0
+        private var listeners: [ListenerReference] = []
+        
+        override init() {
+            super.init()
+                    
+            /*self.disposable = (shouldBeActive
+            |> deliverOnMainQueue).start(next: { [weak self] value in
+                guard let strongSelf = self else {
+                    return
+                }
+                strongSelf.handler.enabled = value
+            })*/
+        }
+        
+        deinit {
+            if let handler = self.handler {
+                handler.enabled = false
+            }
+        }
+        
+        func add(listener: VolumeButtonsListener) -> Int {
+            let id = self.nextListenerId
+            self.nextListenerId += 1
+            
+            self.listeners.append(ListenerReference(id: id, listener: listener))
+            self.updateListeners()
+            
+            return id
+        }
+        
+        func update(id: Int) {
+            self.updateListeners()
+        }
+        
+        func remove(id: Int) {
+            if let index = self.listeners.firstIndex(where: { $0.id == id }) {
+                self.listeners.remove(at: index)
+                self.updateListeners()
+            }
+        }
+        
+        private func performAction(action: Action) {
+            for i in (0 ..< self.listeners.count).reversed() {
+                if let listener = self.listeners[i].listener, listener.isActive {
+                    switch action {
+                    case .up:
+                        listener.upPressed()
+                    case .upRelease:
+                        listener.upReleased()
+                    case .down:
+                        listener.downPressed()
+                    case .downRelease:
+                        listener.downReleased()
+                    }
+                }
+            }
+        }
+        
+        private func updateListeners() {
+            var isActive = false
+            
+            for i in (0 ..< self.listeners.count).reversed() {
+                if let listener = self.listeners[i].listener {
+                    if listener.isActive {
+                        isActive = true
+                    }
+                } else {
+                    self.listeners.remove(at: i)
+                }
+            }
+            
+            if isActive {
+                if self.handler == nil {
+                    self.handler = PGCameraVolumeButtonHandler(upButtonPressedBlock: { [weak self] in
+                        self?.performAction(action: .up)
+                    }, upButtonReleasedBlock: { [weak self] in
+                        self?.performAction(action: .upRelease)
+                    }, downButtonPressedBlock: { [weak self] in
+                        self?.performAction(action: .down)
+                    }, downButtonReleasedBlock: { [weak self] in
+                        self?.performAction(action: .downRelease)
+                    })
+                }
+                self.handler?.enabled = true
+            } else {
+                if let handler = self.handler {
+                    self.handler = nil
+                    
+                    handler.enabled = false
+                }
+            }
+        }
+    }
+    
+    private static var sharedContext: SharedContext = {
+        return SharedContext()
+    }()
+    
+    fileprivate let upPressed: () -> Void
+    fileprivate let upReleased: () -> Void
+    fileprivate let downPressed: () -> Void
+    fileprivate let downReleased: () -> Void
+    
+    private var index: Int?
+    
+    fileprivate var isActive: Bool = false
     private var disposable: Disposable?
     
     public init(
@@ -17,29 +143,28 @@ public class VolumeButtonsListener: NSObject {
         downPressed: @escaping () -> Void,
         downReleased: @escaping () -> Void = {}
     ) {
-        self.handler = PGCameraVolumeButtonHandler(upButtonPressedBlock: {
-            upPressed()
-        }, upButtonReleasedBlock: {
-            upReleased()
-        }, downButtonPressedBlock: {
-            downPressed()
-        }, downButtonReleasedBlock: {
-            downReleased()
-        })
+        self.upPressed = upPressed
+        self.upReleased = upReleased
+        self.downPressed = downPressed
+        self.downReleased = downReleased
         
-        super.init()
+        self.index = VolumeButtonsListener.sharedContext.add(listener: self)
                 
         self.disposable = (shouldBeActive
+        |> distinctUntilChanged
         |> deliverOnMainQueue).start(next: { [weak self] value in
-            guard let strongSelf = self else {
+            guard let self, let index = self.index else {
                 return
             }
-            strongSelf.handler.enabled = value
+            self.isActive = value
+            VolumeButtonsListener.sharedContext.update(id: index)
         })
     }
     
     deinit {
-        self.handler.enabled = false
+        if let index = self.index {
+            VolumeButtonsListener.sharedContext.remove(id: index)
+        }
         self.disposable?.dispose()
     }
 }