[WIP] Video chats

submodules/ComponentFlow/Source/Components/RoundedRectangle.swift

@@ -1,5 +1,6 @@
 import Foundation
 import UIKit
+import Display
 
 public final class RoundedRectangle: Component {
     public enum GradientDirection: Equatable {
@@ -117,3 +118,136 @@ public final class RoundedRectangle: Component {
         return view.update(component: self, availableSize: availableSize, transition: transition)
     }
 }
+
+public final class FilledRoundedRectangleComponent: Component {
+    public let color: UIColor
+    public let cornerRadius: CGFloat
+    public let smoothCorners: Bool
+    
+    public init(
+        color: UIColor,
+        cornerRadius: CGFloat,
+        smoothCorners: Bool
+    ) {
+        self.color = color
+        self.cornerRadius = cornerRadius
+        self.smoothCorners = smoothCorners
+    }
+    
+    public static func ==(lhs: FilledRoundedRectangleComponent, rhs: FilledRoundedRectangleComponent) -> Bool {
+        if lhs === rhs {
+            return true
+        }
+        if lhs.color != rhs.color {
+            return false
+        }
+        if lhs.cornerRadius != rhs.cornerRadius {
+            return false
+        }
+        if lhs.smoothCorners != rhs.smoothCorners {
+            return false
+        }
+        return true
+    }
+    
+    public final class View: UIImageView {
+        private var component: FilledRoundedRectangleComponent?
+        
+        private var currentCornerRadius: CGFloat?
+        private var cornerImage: UIImage?
+        
+        override init(frame: CGRect) {
+            super.init(frame: frame)
+        }
+        
+        required init?(coder: NSCoder) {
+            fatalError("init(coder:) has not been implemented")
+        }
+        
+        private func applyStaticCornerRadius() {
+            guard let component = self.component else {
+                return
+            }
+            guard let cornerRadius = self.currentCornerRadius else {
+                return
+            }
+            if cornerRadius == 0.0 {
+                if let cornerImage = self.cornerImage, cornerImage.size.width == 1.0 {
+                } else {
+                    self.cornerImage = generateImage(CGSize(width: 1.0, height: 1.0), rotatedContext: { size, context in
+                        context.setFillColor(UIColor.white.cgColor)
+                        context.fill(CGRect(origin: CGPoint(), size: size))
+                    })?.stretchableImage(withLeftCapWidth: Int(cornerRadius) + 5, topCapHeight: Int(cornerRadius) + 5).withRenderingMode(.alwaysTemplate)
+                }
+            } else {
+                if component.smoothCorners {
+                    let size = CGSize(width: cornerRadius * 2.0 + 10.0, height: cornerRadius * 2.0 + 10.0)
+                    if let cornerImage = self.cornerImage, cornerImage.size == size {
+                    } else {
+                        self.cornerImage = generateImage(size, rotatedContext: { size, context in
+                            context.clear(CGRect(origin: CGPoint(), size: size))
+                            context.addPath(UIBezierPath(roundedRect: CGRect(origin: CGPoint(), size: size), cornerRadius: cornerRadius).cgPath)
+                            context.setFillColor(UIColor.white.cgColor)
+                            context.fillPath()
+                        })?.stretchableImage(withLeftCapWidth: Int(cornerRadius) + 5, topCapHeight: Int(cornerRadius) + 5).withRenderingMode(.alwaysTemplate)
+                    }
+                } else {
+                    let size = CGSize(width: cornerRadius * 2.0, height: cornerRadius * 2.0)
+                    if let cornerImage = self.cornerImage, cornerImage.size == size {
+                    } else {
+                        self.cornerImage = generateStretchableFilledCircleImage(diameter: size.width, color: UIColor.white)?.withRenderingMode(.alwaysTemplate)
+                    }
+                }
+            }
+            self.image = self.cornerImage
+            self.clipsToBounds = false
+            self.backgroundColor = nil
+            self.layer.cornerRadius = 0.0
+        }
+        
+        func update(component: FilledRoundedRectangleComponent, availableSize: CGSize, transition: ComponentTransition) -> CGSize {
+            self.component = component
+            
+            transition.setTintColor(view: self, color: component.color)
+            
+            if self.currentCornerRadius != component.cornerRadius {
+                let previousCornerRadius = self.currentCornerRadius
+                self.currentCornerRadius = component.cornerRadius
+                if transition.animation.isImmediate {
+                    self.applyStaticCornerRadius()
+                } else {
+                    self.image = nil
+                    self.clipsToBounds = true
+                    self.backgroundColor = component.color
+                    if let previousCornerRadius, self.layer.animation(forKey: "cornerRadius") == nil {
+                        self.layer.cornerRadius = previousCornerRadius
+                    }
+                    if #available(iOS 13.0, *) {
+                        if component.smoothCorners {
+                            self.layer.cornerCurve = .continuous
+                        } else {
+                            self.layer.cornerCurve = .circular
+                        }
+                        
+                    }
+                    transition.setCornerRadius(layer: self.layer, cornerRadius: component.cornerRadius, completion: { [weak self] completed in
+                        guard let self, completed else {
+                            return
+                        }
+                        self.applyStaticCornerRadius()
+                    })
+                }
+            }
+            
+            return availableSize
+        }
+    }
+    
+    public func makeView() -> View {
+        return View(frame: CGRect())
+    }
+    
+    public func update(view: View, availableSize: CGSize, state: EmptyComponentState, environment: Environment<Empty>, transition: ComponentTransition) -> CGSize {
+        return view.update(component: self, availableSize: availableSize, transition: transition)
+    }
+}