Swiftgram/submodules/TelegramUI/Components/AvatarEditorScreen/Sources/PremiumAlertController.swift

import Foundation
import UIKit
import SwiftSignalKit
import AsyncDisplayKit
import Display
import ComponentFlow
import Postbox
import TelegramCore
import TelegramPresentationData
import TelegramUIPreferences
import AccountContext
import AppBundle
import AlertComponent
import PremiumStarComponent

func premiumAlertController(
    context: AccountContext,
    parentController: ViewController,
    title: String? = nil,
    text: String
) -> ViewController {
    let presentationData = context.sharedContext.currentPresentationData.with { $0 }
    let strings = presentationData.strings
    
    var content: [AnyComponentWithIdentity<AlertComponentEnvironment>] = []
    content.append(AnyComponentWithIdentity(
        id: "header",
        component: AnyComponent(
            AlertPremiumStarComponent()
        )
    ))
    
    let title = title ?? strings.PremiumNeeded_Title
    content.append(AnyComponentWithIdentity(
        id: "title",
        component: AnyComponent(
            AlertTitleComponent(title: title)
        )
    ))
    content.append(AnyComponentWithIdentity(
        id: "text",
        component: AnyComponent(
            AlertTextComponent(content: .plain(text))
        )
    ))
    
    let alertController = AlertScreen(
        context: context,
        content: content,
        actions: [
            .init(title: strings.Common_Cancel),
            .init(title: strings.PremiumNeeded_Subscribe, type: .default, action: { [weak parentController] in
                let controller = context.sharedContext.makePremiumIntroController(context: context, source: .nameColor, forceDark: false, dismissed: nil)
                parentController?.push(controller)
            })
        ]
    )
    return alertController
}

private final class AlertPremiumStarComponent: Component {
    public typealias EnvironmentType = AlertComponentEnvironment
        
    public init() {
    }
    
    public static func ==(lhs: AlertPremiumStarComponent, rhs: AlertPremiumStarComponent) -> Bool {
        return true
    }
    
    public final class View: UIView {
        private let clippingView = UIView()
        private let icon = ComponentView<Empty>()
        
        private var component: AlertPremiumStarComponent?
        private weak var state: EmptyComponentState?
        
        func update(component: AlertPremiumStarComponent, availableSize: CGSize, state: EmptyComponentState, environment: Environment<AlertComponentEnvironment>, transition: ComponentTransition) -> CGSize {
            self.component = component
            self.state = state
            
            let environment = environment[AlertComponentEnvironment.self]
            
            let starHeight: CGFloat = 105.0
            let starSize = self.icon.update(
                transition: .immediate,
                component: AnyComponent(
                    PremiumStarComponent(
                        theme: environment.theme,
                        isIntro: false,
                        isVisible: true,
                        hasIdleAnimations: true,
                        colors: [
                            UIColor(rgb: 0x6a94ff),
                            UIColor(rgb: 0x9472fd),
                            UIColor(rgb: 0xe26bd3)
                        ]
                    )
                ),
                environment: {},
                containerSize: CGSize(width: availableSize.width + 60.0, height: 200.0)
            )
            if let view = self.icon.view {
                if view.superview == nil {
                    self.addSubview(self.clippingView)
                    self.clippingView.addSubview(view)
                }
                view.frame = CGRect(origin: CGPoint(x: 0.0, y: -24.0), size: starSize)
            }
            
            self.clippingView.clipsToBounds = true
            self.clippingView.layer.cornerRadius = 35.0
            self.clippingView.frame = CGRect(origin: CGPoint(x: -30.0, y: -22.0), size: CGSize(width: starSize.width, height: starSize.height))
           
            return CGSize(width: availableSize.width, height: starHeight + 10.0)
        }
    }
    
    public func makeView() -> View {
        return View(frame: CGRect())
    }
    
    public func update(view: View, availableSize: CGSize, state: EmptyComponentState, environment: Environment<AlertComponentEnvironment>, transition: ComponentTransition) -> CGSize {
        return view.update(component: self, availableSize: availableSize, state: state, environment: environment, transition: transition)
    }
}