Refactoring

submodules/TelegramUI/Components/Chat/ChatInstantVideoMessageDurationNode/BUILD

@@ -0,0 +1,21 @@
+load("@build_bazel_rules_swift//swift:swift.bzl", "swift_library")
+
+swift_library(
+    name = "ChatInstantVideoMessageDurationNode",
+    module_name = "ChatInstantVideoMessageDurationNode",
+    srcs = glob([
+        "Sources/**/*.swift",
+    ]),
+    copts = [
+        "-warnings-as-errors",
+    ],
+    deps = [
+        "//submodules/AsyncDisplayKit",
+        "//submodules/SSignalKit/SwiftSignalKit",
+        "//submodules/Display",
+        "//submodules/MediaPlayer:UniversalMediaPlayer",
+    ],
+    visibility = [
+        "//visibility:public",
+    ],
+)

submodules/TelegramUI/Components/Chat/ChatInstantVideoMessageDurationNode/Sources/ChatInstantVideoMessageDurationNode.swift

@@ -0,0 +1,220 @@
+import Foundation
+import UIKit
+import AsyncDisplayKit
+import SwiftSignalKit
+import Display
+import UniversalMediaPlayer
+
+private let textFont = Font.with(size: 11.0, design: .regular, weight: .regular, traits: [.monospacedNumbers])
+
+private struct ChatInstantVideoMessageDurationNodeState: Equatable {
+    let hours: Int32?
+    let minutes: Int32?
+    let seconds: Int32?
+    
+    init() {
+        self.hours = nil
+        self.minutes = nil
+        self.seconds = nil
+    }
+    
+    init(hours: Int32, minutes: Int32, seconds: Int32) {
+        self.hours = hours
+        self.minutes = minutes
+        self.seconds = seconds
+    }
+    
+    static func ==(lhs: ChatInstantVideoMessageDurationNodeState, rhs: ChatInstantVideoMessageDurationNodeState) -> Bool {
+        if lhs.hours != rhs.hours || lhs.minutes != rhs.minutes || lhs.seconds != rhs.seconds {
+            return false
+        }
+        return true
+    }
+}
+
+private final class ChatInstantVideoMessageDurationNodeParameters: NSObject {
+    let state: ChatInstantVideoMessageDurationNodeState
+    let isSeen: Bool
+    let textColor: UIColor
+    
+    init(state: ChatInstantVideoMessageDurationNodeState, isSeen: Bool, textColor: UIColor) {
+        self.state = state
+        self.isSeen = isSeen
+        self.textColor = textColor
+        
+        super.init()
+    }
+}
+
+public final class ChatInstantVideoMessageDurationNode: ASImageNode {
+    private var textColor: UIColor
+    
+    public var defaultDuration: Double? {
+        didSet {
+            if self.defaultDuration != oldValue {
+                self.updateTimestamp()
+                self.updateContents()
+            }
+        }
+    }
+    
+    public var isSeen: Bool = false {
+        didSet {
+            if self.isSeen != oldValue {
+                self.updateContents()
+            }
+        }
+    }
+    
+    private var updateTimer: SwiftSignalKit.Timer?
+    
+    private var statusValue: MediaPlayerStatus? {
+        didSet {
+            if self.statusValue != oldValue {
+                if let statusValue = statusValue, case .playing = statusValue.status {
+                    self.ensureHasTimer()
+                } else {
+                    self.stopTimer()
+                }
+                self.updateTimestamp()
+            }
+        }
+    }
+    
+    private var state = ChatInstantVideoMessageDurationNodeState() {
+        didSet {
+            if self.state != oldValue {
+                self.updateContents()
+            }
+        }
+    }
+    
+    private var statusDisposable: Disposable?
+    private var statusValuePromise = Promise<MediaPlayerStatus?>()
+    
+    public var status: Signal<MediaPlayerStatus?, NoError>? {
+        didSet {
+            if let status = self.status {
+                self.statusValuePromise.set(status)
+            } else {
+                self.statusValuePromise.set(.never())
+            }
+        }
+    }
+
+    public var size: CGSize = CGSize()
+    public var sizeUpdated: ((CGSize) -> Void)?
+    
+    public init(textColor: UIColor) {
+        self.textColor = textColor
+        
+        super.init()
+        
+        self.isOpaque = false
+        self.contentsScale = UIScreenScale
+        self.contentMode = .topRight
+        
+        self.statusDisposable = (self.statusValuePromise.get()
+        |> deliverOnMainQueue).startStrict(next: { [weak self] status in
+            if let strongSelf = self {
+                strongSelf.statusValue = status
+            }
+        })
+    }
+    
+    deinit {
+        self.statusDisposable?.dispose()
+        self.updateTimer?.invalidate()
+    }
+    
+    public func updateTheme(textColor: UIColor) {
+        if !self.textColor.isEqual(textColor) {
+            self.textColor = textColor
+            self.updateContents()
+        }
+    }
+    
+    private func ensureHasTimer() {
+        if self.updateTimer == nil {
+            let timer = SwiftSignalKit.Timer(timeout: 0.5, repeat: true, completion: { [weak self] in
+                self?.updateTimestamp()
+            }, queue: Queue.mainQueue())
+            self.updateTimer = timer
+            timer.start()
+        }
+    }
+    
+    private func stopTimer() {
+        self.updateTimer?.invalidate()
+        self.updateTimer = nil
+    }
+    
+    public func updateTimestamp() {
+        if let statusValue = self.statusValue, Double(0.0).isLess(than: statusValue.duration) {
+            let timestampSeconds: Double
+            if !statusValue.generationTimestamp.isZero {
+                timestampSeconds = statusValue.timestamp + (CACurrentMediaTime() - statusValue.generationTimestamp)
+            } else {
+                timestampSeconds = statusValue.timestamp
+            }
+            let timestamp = Int32(timestampSeconds)
+            self.state = ChatInstantVideoMessageDurationNodeState(hours: timestamp / (60 * 60), minutes: timestamp % (60 * 60) / 60, seconds: timestamp % 60)
+        } else if let defaultDuration = self.defaultDuration {
+            let timestamp = Int32(defaultDuration)
+            self.state = ChatInstantVideoMessageDurationNodeState(hours: timestamp / (60 * 60), minutes: timestamp % (60 * 60) / 60, seconds: timestamp % 60)
+        } else {
+            self.state = ChatInstantVideoMessageDurationNodeState()
+        }
+    }
+
+    private func updateContents() {
+        let image = self.generateContents(withParameters: self.getParameters(), isCancelled: { return false })
+        let previousSize = self.image?.size
+        self.image = image
+        if let image = image, previousSize != image.size {
+            self.size = image.size
+            self.sizeUpdated?(image.size)
+        }
+    }
+    
+    private func getParameters() -> NSObjectProtocol? {
+        return ChatInstantVideoMessageDurationNodeParameters(state: self.state, isSeen: self.isSeen, textColor: self.textColor)
+    }
+    
+    private func generateContents(withParameters: Any?, isCancelled: () -> Bool) -> UIImage? {
+        guard let parameters = withParameters as? ChatInstantVideoMessageDurationNodeParameters else {
+            return nil
+        }
+        
+        let text: String
+        if let hours = parameters.state.hours, let minutes = parameters.state.minutes, let seconds = parameters.state.seconds {
+            if hours != 0 {
+                text = String(format: "%d:%02d:%02d", hours, minutes, seconds)
+            } else {
+                text = String(format: "%d:%02d", minutes, seconds)
+            }
+        } else {
+            text = "-:--"
+        }
+        let string = NSAttributedString(string: text, font: textFont, textColor: parameters.textColor)
+        let textRect = string.boundingRect(with: CGSize(width: 200.0, height: 100.0), options: NSStringDrawingOptions.usesLineFragmentOrigin, context: nil)
+        
+        let unseenInset: CGFloat = (parameters.isSeen ? 0.0 : 10.0)
+        let imageSize = CGSize(width: ceil(textRect.width) + 10.0 + unseenInset, height: 18.0)
+        
+        return generateImage(imageSize, rotatedContext: { size, context in
+            context.clear(CGRect(origin: CGPoint(x: 0.0, y: 0.0), size: size))
+            context.setBlendMode(.normal)
+            
+            if !parameters.isSeen {
+                context.setFillColor(parameters.textColor.cgColor)
+                let diameter: CGFloat = 4.0
+                context.fillEllipse(in: CGRect(origin: CGPoint(x: size.width - size.height + floor((size.height - diameter) / 2.0), y: floor((size.height - diameter) / 2.0)), size: CGSize(width: diameter, height: diameter)))
+            }
+
+            UIGraphicsPushContext(context)
+            string.draw(at: CGPoint(x: floor((size.width - unseenInset - textRect.size.width) / 2.0) + textRect.origin.x, y: 2.0 + textRect.origin.y + UIScreenPixel))
+            UIGraphicsPopContext()
+        })
+    }
+}