//
// Copyright 2021 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//

import SignalServiceKit
public import SignalUI

// Compare with ColorOrGradientSwatchView:
//
// * CVColorOrGradientView is intended to be used in CVC cells.
//   It does not assume the gradient bounds corresponds to the
//   view bounds.
// * ColorOrGradientSwatchView is for use elsewhere.
//   It can pin gradient bounds to the edges of a circle for previews.
//   It can be used to render wallpapers.
//
// Although we could combine these two views, these two scenarios are
// just different enough that its convenient to have two separate views.
public class CVColorOrGradientView: ManualLayoutViewWithLayer {

    private weak var referenceView: UIView?
    private var value: ColorOrGradientValue?

    public struct StrokeConfig {
        let color: UIColor
        let width: CGFloat
    }
    public struct BubbleConfig {
        let sharpCorners: OWSDirectionalRectCorner
        let sharpCornerRadius: CGFloat
        let wideCornerRadius: CGFloat
        let strokeConfig: StrokeConfig?
    }
    private var bubbleConfig: BubbleConfig?
    private var hasPillRounding = false

    private let gradientLayer = CAGradientLayer()
    private let shapeLayer = CAShapeLayer()
    private let maskLayer = CAShapeLayer()
    private var dimmerLayer: CALayer?

    public var ensureSubviewsFillBounds = false
    public var animationsEnabled = false

    public init() {
        super.init(name: "CVColorOrGradientView")

        gradientLayer.disableAnimationsWithDelegate()
        shapeLayer.disableAnimationsWithDelegate()
        maskLayer.disableAnimationsWithDelegate()
    }

    private func addDefaultLayoutBlock() {
        addLayoutBlock { view in
            guard let view = view as? CVColorOrGradientView else { return }
            view.updateAppearance()
        }
    }

    private func ensureSubviewLayout() {
        guard ensureSubviewsFillBounds else { return }
        for subview in subviews {
            ManualLayoutView.setSubviewFrame(subview: subview, frame: bounds)
        }
    }

    public func configure(value: ColorOrGradientValue,
                          referenceView: UIView,
                          bubbleConfig: BubbleConfig? = nil,
                          hasPillRounding: Bool = false) {
        // At most one of these parameters should be set.
        owsAssertDebug(!hasPillRounding || bubbleConfig == nil)

        self.value = value
        self.referenceView = referenceView
        self.bubbleConfig = bubbleConfig
        self.hasPillRounding = hasPillRounding

        addDefaultLayoutBlock()

        updateAppearance()
    }

    public static func build(conversationStyle: ConversationStyle, referenceView: UIView) -> CVColorOrGradientView {
        let view = CVColorOrGradientView()
        view.configure(value: conversationStyle.bubbleChatColorOutgoing,
                       referenceView: referenceView)
        return view
    }

    public func updateAppearance() {

        guard let value = self.value,
              let referenceView = self.referenceView else {
            self.backgroundColor = nil
            gradientLayer.removeFromSuperlayer()
            dimmerLayer?.removeFromSuperlayer()
            return
        }

        switch value {
        case .transparent:
            backgroundColor = nil
            gradientLayer.removeFromSuperlayer()
            dimmerLayer?.removeFromSuperlayer()
        case .solidColor(let color):
            backgroundColor = color
            gradientLayer.removeFromSuperlayer()
        case .gradient(let color1, let color2, let angleRadians):

            if gradientLayer.superlayer != self.layer {
                gradientLayer.removeFromSuperlayer()
                layer.insertSublayer(gradientLayer, at: 0)
            }

            gradientLayer.frame = self.bounds

            gradientLayer.colors = [
                color1.cgColor,
                color2.cgColor
            ]

            /* The start and end points of the gradient when drawn into the layer's
             * coordinate space. The start point corresponds to the first gradient
             * stop, the end point to the last gradient stop. Both points are
             * defined in a unit coordinate space that is then mapped to the
             * layer's bounds rectangle when drawn. (I.e. [0,0] is the bottom-left
             * corner of the layer, [1,1] is the top-right corner.) The default values
             * are [.5,0] and [.5,1] respectively. Both are animatable. */
            let unitCenter = CGPoint(x: 0.5, y: 0.5)
            // Note the signs.
            let startVector = CGPoint(x: +sin(angleRadians), y: -cos(angleRadians))
            let startScale: CGFloat
            // In rectangle mode, we want the startPoint and endPoint to reside
            // on the edge of the unit square, and thus edge of the rectangle.
            // We therefore scale such that longer axis is a half unit.
            let startSquareScale: CGFloat = max(abs(startVector.x), abs(startVector.y))
            startScale = 0.5 / startSquareScale

            // Control points within the bounding box of the entire gradient.
            // Expressed as unit values with upper-left origin.
            //
            // 0,0
            // ********************** C1 **
            // *                          *
            // *                          *
            // *                          *
            // *                          *
            // *                          *
            // *                          *
            // ** C2 ********************** 1,1
            //
            let startPointGradientUnitsUL = unitCenter + startVector * +startScale
            // The endpoint should be "opposite" the start point,
            // on the opposite edge of the gradient.
            let endPointGradientUnitsUL = unitCenter + startVector * -startScale

            // Each message bubble renders a subsection of the gradient.
            // We need to convert the control points from the bounding box
            // of the gradient to the local unit coordinate space of this view.
            // The reference frame (bounding box of the entire gradient)
            // in local points.
            let referenceFrameLocalPoints = self.convert(referenceView.bounds, from: referenceView)
            // The reference frame (bounding box of the entire gradient)
            // in local unit coordinates.
            let referenceFrameLocalUnits = CGRect(x: referenceFrameLocalPoints.x.inverseLerp(bounds.minX, bounds.maxX),
                                                  y: referenceFrameLocalPoints.y.inverseLerp(bounds.minY, bounds.maxY),
                                                  width: referenceFrameLocalPoints.width / bounds.width,
                                                  height: referenceFrameLocalPoints.height / bounds.height)
            func convertFromGradientToLocal(_ point: CGPoint) -> CGPoint {
                CGPoint(x: point.x.lerp(referenceFrameLocalUnits.minX, referenceFrameLocalUnits.maxX),
                        y: point.y.lerp(referenceFrameLocalUnits.minY, referenceFrameLocalUnits.maxY))
            }
            // Control points within the local UIView viewport.
            // Expressed as unit values with upper-left origin.
            //
            // ********************** C1 **
            // *                          *
            // *            0,0           *
            // *            ********      *
            // *            *      *      *
            // *            ******** 1,1  *
            // *                          *
            // ** C2 **********************
            //
            let startPointViewportUnitsUL = convertFromGradientToLocal(startPointGradientUnitsUL)
            let endPointViewportUnitsUL = convertFromGradientToLocal(endPointGradientUnitsUL)

            // UIKit/UIView uses an upper-left origin.
            // Core Graphics/CALayer uses a lower-left origin.
            func convertToLayerUnit(_ point: CGPoint) -> CGPoint {
                // TODO: The documentation clearly indicates that
                // CAGradientLayer.startPoint and endPoint use the layer's
                // coordinate space with lower-left origin.  But the
                // observed behavior is that they use an upper-left origin.
                // I can't figure out why.
                // return CGPoint(x: point.x, y: (1 - point.y))
                return point
            }
            // Control points within the local CALayer viewport.
            // Expressed as unit values with lower-left origin.
            //
            // ********************** C1 **
            // *                          *
            // *                          *
            // *            ******** 1,1  *
            // *            *      *      *
            // *        0,0 ********      *
            // *                          *
            // ** C2 **********************
            //
            let startPointLayerUnitsLL = convertToLayerUnit(startPointViewportUnitsUL)
            let endPointLayerUnitsLL = convertToLayerUnit(endPointViewportUnitsUL)

            gradientLayer.startPoint = startPointLayerUnitsLL
            gradientLayer.endPoint = endPointLayerUnitsLL
        }

        if let bubbleConfig = self.bubbleConfig {
            let sharpCorners = UIView.uiRectCorner(forOWSDirectionalRectCorner: bubbleConfig.sharpCorners)
            let bubblePath = UIBezierPath.roundedRect(self.bounds,
                                                      sharpCorners: sharpCorners,
                                                      sharpCornerRadius: bubbleConfig.sharpCornerRadius,
                                                      wideCornerRadius: bubbleConfig.wideCornerRadius)

            if sharpCorners == .allCorners || sharpCorners == [] {
                // If all of the corners have the same radius, don't
                // bother using a mask layer.
                layer.cornerRadius = (sharpCorners == []
                                        ? bubbleConfig.wideCornerRadius
                                        : bubbleConfig.sharpCornerRadius)
                layer.mask = nil
                layer.masksToBounds = true
            } else {
                maskLayer.path = bubblePath.cgPath
                layer.mask = maskLayer
                layer.masksToBounds = false
                layer.cornerRadius = 0
            }

            if let strokeConfig = bubbleConfig.strokeConfig {
                shapeLayer.lineWidth = strokeConfig.width
                shapeLayer.strokeColor = strokeConfig.color.cgColor
                shapeLayer.fillColor = nil
                shapeLayer.path = bubblePath.cgPath
                layer.addSublayer(shapeLayer)
            } else if shapeLayer.superlayer != nil {
                shapeLayer.removeFromSuperlayer()
            }
        } else {
            // If this view isn't being used as a CVC message bubble,
            // don't bother masking or using shapeLayer to render the stroke.
            layer.mask = nil

            if hasPillRounding {
                layer.masksToBounds = true
                layer.cornerRadius = bounds.size.smallerAxis / 2
            } else {
                layer.masksToBounds = false
                layer.cornerRadius = 0
            }
            if shapeLayer.superlayer != nil {
                shapeLayer.removeFromSuperlayer()
            }
        }

        ensureSubviewLayout()
    }

    public override func reset() {
        super.reset()

        self.referenceView = nil
        self.value = nil
        self.backgroundColor = nil
        self.bubbleConfig = nil
        self.hasPillRounding = false
        shapeLayer.removeFromSuperlayer()
        gradientLayer.removeFromSuperlayer()
        dimmerLayer?.removeFromSuperlayer()
    }

    // MARK: - CALayerDelegate

    public override func action(for layer: CALayer, forKey event: String) -> CAAction? {
        // Disable all implicit CALayer animations if needed
        if animationsEnabled {
            return super.action(for: layer, forKey: event)
        } else {
            return NSNull()
        }

    }

    // MARK: - DimmableView

    // Layer is not created until it's needed so storing color in a variable is necessary.
    var dimmingColor: UIColor? {
        didSet {
            dimmerLayer?.backgroundColor = dimmingColor?.cgColor
        }
    }

    var dimmerDimsBackgroundOnly = true

    private var sublayerIndexForDimmerLayer: UInt32 {
        if dimmerDimsBackgroundOnly {
            // As deep as possible but above the gradient layer.
            if gradientLayer.superlayer == self.layer {
                return 1
            } else {
                return 0
            }
        } else {
            return UInt32(layer.sublayers?.count ?? 0)
        }
    }

    func performDimmingAnimation(stepDuration: TimeInterval) {
        guard let dimmingColor else { return }

        var dimmerLayer: CALayer
        if let existingDimmerLayer = self.dimmerLayer {
            dimmerLayer = existingDimmerLayer
        } else {
            dimmerLayer = CALayer()
            dimmerLayer.opacity = 0
            dimmerLayer.backgroundColor = dimmingColor.cgColor
            self.dimmerLayer = dimmerLayer
        }

        // Move dimmer layer to a correct z-index.
        dimmerLayer.removeFromSuperlayer()
        dimmerLayer.frame = layer.bounds
        layer.insertSublayer(dimmerLayer, at: sublayerIndexForDimmerLayer)

        dimmerLayer.removeAllAnimations()

        // Animate fade-in.
        let fadeIn = CABasicAnimation(keyPath: #keyPath(CALayer.opacity))
        fadeIn.fromValue = 0
        fadeIn.toValue = 1
        fadeIn.duration = stepDuration
        fadeIn.fillMode = .forwards
        fadeIn.isRemovedOnCompletion = false
        dimmerLayer.add(fadeIn, forKey: "fadeIn")

        // Schedule fade-out after delay.
        DispatchQueue.main.asyncAfter(deadline: .now() + 2 * stepDuration) {
            let fadeOut = CABasicAnimation(keyPath: #keyPath(CALayer.opacity))
            fadeOut.fromValue = 1
            fadeOut.toValue = 0
            fadeOut.duration = stepDuration
            fadeOut.fillMode = .forwards
            fadeOut.isRemovedOnCompletion = false
            dimmerLayer.add(fadeOut, forKey: "fadeOut")
        }
    }
}

// MARK: -

extension CVColorOrGradientView: OWSBubbleViewHost {
    public var maskPath: UIBezierPath {
        guard let bubbleConfig = self.bubbleConfig else {
            owsFailDebug("Missing bubbleConfig.")
            return UIBezierPath()
        }
        let sharpCorners = UIView.uiRectCorner(forOWSDirectionalRectCorner: bubbleConfig.sharpCorners)
        let bubblePath = UIBezierPath.roundedRect(self.bounds,
                                                  sharpCorners: sharpCorners,
                                                  sharpCornerRadius: bubbleConfig.sharpCornerRadius,
                                                  wideCornerRadius: bubbleConfig.wideCornerRadius)
        return bubblePath
    }

    public var bubbleReferenceView: UIView { self }
}