//
//  Copyright (c) 2021 Open Whisper Systems. All rights reserved.
//

import Foundation

@objc
public enum ConversationStyleType: UInt {
    // The style used from initialization until presentation begins.
    // It does not have valid values and should not be rendered.
    case initial
    // The style used until the presentation has configured the view(s).
    // It has values inferred from the navigationController.
    case placeholder
    // The style once presentation has configured the view(s).
    // It has values from the CVC view state.
    case `default`
    // The style used in the message detail view
    case messageDetails

    fileprivate var isValid: Bool { self != .initial }
}

// MARK: -

// An immutable snapshot of the core styling
// state used by CVC for a given load/render cycle.
@objc
public class ConversationStyle: NSObject {

    @objc
    public let type: ConversationStyleType
    @objc
    public var isValidStyle: Bool { type.isValid }

    // The width of the collection view.
    @objc
    public let viewWidth: CGFloat

    @objc
    public let isDarkThemeEnabled: Bool

    @objc
    public let hasWallpaper: Bool

    private let dynamicBodyTypePointSize: CGFloat
    private let primaryTextColor: UIColor

    @objc
    public let contentMarginTop: CGFloat = 24
    @objc
    public let contentMarginBottom: CGFloat = 24

    @objc
    public let gutterLeading: CGFloat
    @objc
    public let gutterTrailing: CGFloat

    @objc
    public let headerGutterLeading: CGFloat = 28
    @objc
    public let headerGutterTrailing: CGFloat = 28

    // These are the gutters used by "full width" views
    // like "contact offer" and "info message".
    @objc
    public let fullWidthGutterLeading: CGFloat
    @objc
    public let fullWidthGutterTrailing: CGFloat

    @objc
    static public let groupMessageAvatarDiameter: CGFloat = 28
    @objc
    static public let selectionViewWidth: CGFloat = 24
    @objc
    static public let messageStackSpacing: CGFloat = 8
    @objc
    static public let defaultMessageSpacing: CGFloat = 12
    @objc
    static public let compactMessageSpacing: CGFloat = 2
    @objc
    static public let systemMessageSpacing: CGFloat = 20

    @objc
    public let contentWidth: CGFloat

    @objc
    public var headerViewContentWidth: CGFloat {
        viewWidth - (headerGutterLeading + headerGutterTrailing)
    }

    @objc
    public let maxMessageWidth: CGFloat
    @objc
    public let maxMediaMessageWidth: CGFloat
    @objc
    public let maxAudioMessageWidth: CGFloat

    @objc
    public let textInsetTop: CGFloat
    @objc
    public let textInsetBottom: CGFloat
    @objc
    public let textInsetHorizontal: CGFloat
    @objc
    public var textInsets: UIEdgeInsets {
        UIEdgeInsets(top: textInsetTop,
                     leading: textInsetHorizontal,
                     bottom: textInsetBottom,
                     trailing: textInsetHorizontal)
    }

    // We want to align "group sender" avatars with the v-center of the
    // "last line" of the message body text - or where it would be for
    // non-text content.
    //
    // This is the distance from that v-center to the bottom of the
    // message bubble.
    @objc
    public let lastTextLineAxis: CGFloat

    // Incoming and outgoing messages are visually distinguished
    // by leading and trailing alignment, respectively.
    //
    // Reserve a space so that in the most compressed layouts
    // (small form factor, group avatar, multi-select, etc.)
    // there is space for them to align in these directions.
    @objc
    public static let messageDirectionSpacing: CGFloat = 12

    @objc
    public required init(type: ConversationStyleType,
                         thread: TSThread,
                         viewWidth: CGFloat,
                         hasWallpaper: Bool) {

        self.type = type
        self.conversationColor = ConversationStyle.conversationColor(thread: thread)
        self.viewWidth = viewWidth
        self.isDarkThemeEnabled = Theme.isDarkThemeEnabled
        self.primaryTextColor = Theme.primaryTextColor
        self.hasWallpaper = hasWallpaper

        if type == .messageDetails {
            gutterLeading = 0
            gutterTrailing = 0
            fullWidthGutterLeading = 0
            fullWidthGutterTrailing = 0
        } else {
            gutterLeading = thread.isGroupThread ? 12 : 16
            gutterTrailing = 16

            fullWidthGutterLeading = thread.isGroupThread ? 12 : 16
            fullWidthGutterTrailing = thread.isGroupThread ? 12 : 16
        }

        let messageTextFont = UIFont.ows_dynamicTypeBody

        dynamicBodyTypePointSize = messageTextFont.pointSize

        let baseFontOffset: CGFloat = 11

        // Don't include the distance from the "cap height" to the top of the UILabel
        // in the top margin.
        textInsetTop = max(0, round(baseFontOffset - (messageTextFont.ascender - messageTextFont.capHeight)))
        // Don't include the distance from the "baseline" to the bottom of the UILabel
        // (e.g. the descender) in the top margin. Note that UIFont.descender is a
        // negative value.
        textInsetBottom = max(0, round(baseFontOffset - abs(messageTextFont.descender)))

        textInsetHorizontal = 12

        lastTextLineAxis = CGFloat(round(baseFontOffset + messageTextFont.capHeight * 0.5))

        contentWidth = viewWidth - (gutterLeading + gutterTrailing)

        var maxMessageWidth = contentWidth - (Self.selectionViewWidth + Self.messageStackSpacing)

        maxMessageWidth -= Self.messageDirectionSpacing

        if thread.isGroupThread {
            maxMessageWidth -= (Self.groupMessageAvatarDiameter + Self.messageStackSpacing)
        }
        self.maxMessageWidth = maxMessageWidth

        // This upper bound should have no effect in portrait orientation.
        // It limits body media size in landscape.
        let kMaxBodyMediaSize: CGFloat = 350
        maxMediaMessageWidth = floor(min(maxMessageWidth, kMaxBodyMediaSize))

        let kMaxAudioMessageWidth: CGFloat = 244
        maxAudioMessageWidth = floor(min(maxMessageWidth, kMaxAudioMessageWidth))

        super.init()
    }

    // MARK: Colors

    @objc
    public let conversationColor: OWSConversationColor

    private class func conversationColor(thread: TSThread) -> OWSConversationColor {
        let colorName = thread.conversationColorName

        return OWSConversationColor.conversationColorOrDefault(colorName: colorName)
    }

    @objc
    private var defaultBubbleColorIncoming: UIColor {
        if hasWallpaper {
            return isDarkThemeEnabled ? .ows_gray95 : .white
        } else {
            return isDarkThemeEnabled ? UIColor.ows_gray80 : UIColor.ows_gray05
        }
    }

    @objc
    public let bubbleColorOutgoingFailed = UIColor.ows_accentBlue

    @objc
    public let bubbleColorOutgoingSending = UIColor.ows_accentBlue

    @objc
    public let bubbleColorOutgoingSent = UIColor.ows_accentBlue

    @objc
    public let dateBreakTextColor = UIColor.ows_gray60

    @objc
    public func bubbleColor(message: TSMessage) -> UIColor {
        if message.wasRemotelyDeleted {
            return Theme.backgroundColor
        } else if message is TSIncomingMessage {
            return defaultBubbleColorIncoming
        } else if let outgoingMessage = message as? TSOutgoingMessage {
            switch outgoingMessage.messageState {
            case .failed:
                return bubbleColorOutgoingFailed
            case .sending, .pending:
                return bubbleColorOutgoingSending
            default:
                return bubbleColorOutgoingSent
            }
        } else {
            owsFailDebug("Unexpected message type: \(message)")
            return bubbleColorOutgoingSent
        }
    }

    @objc
    public func bubbleColor(isIncoming: Bool) -> UIColor {
        isIncoming ? defaultBubbleColorIncoming : bubbleColorOutgoingSent
    }

    @objc
    public static var bubbleTextColorIncoming: UIColor {
        Theme.isDarkThemeEnabled ? UIColor.ows_gray05 : UIColor.ows_gray90
    }

    @objc
    public static var bubbleTextColorOutgoing: UIColor {
        Theme.isDarkThemeEnabled ? UIColor.ows_gray05 : UIColor.ows_white
    }

    @objc
    public var bubbleTextColorIncoming: UIColor {
        isDarkThemeEnabled ? UIColor.ows_gray05 : UIColor.ows_gray90
    }

    @objc
    public var bubbleSecondaryTextColorIncoming: UIColor {
        isDarkThemeEnabled ? .ows_gray25 : .ows_gray60
    }

    @objc
    public var bubbleTextColorOutgoing: UIColor {
        isDarkThemeEnabled ? UIColor.ows_gray05 : UIColor.ows_white
    }

    @objc
    public var bubbleSecondaryTextColorOutgoing: UIColor {
        isDarkThemeEnabled ? .ows_whiteAlpha60 : .ows_whiteAlpha80
    }

    @objc
    public func bubbleTextColor(message: TSMessage) -> UIColor {
        if message.wasRemotelyDeleted && !hasWallpaper {
            return primaryTextColor
        } else if message is TSIncomingMessage {
            return bubbleTextColorIncoming
        } else if message is TSOutgoingMessage {
            return bubbleTextColorOutgoing
        } else {
            owsFailDebug("Unexpected message type: \(message)")
            return bubbleTextColorOutgoing
        }
    }

    @objc
    public func bubbleTextColor(isIncoming: Bool) -> UIColor {
        if isIncoming {
            return bubbleTextColorIncoming
        } else {
            return bubbleTextColorOutgoing
        }
    }

    @objc
    public func bubbleReadMoreTextColor(message: TSMessage) -> UIColor {
        if message is TSIncomingMessage {
            return isDarkThemeEnabled ? .ows_whiteAlpha90 : .ows_accentBlue
        } else if message is TSOutgoingMessage {
            return isDarkThemeEnabled ? .ows_whiteAlpha90 : .white
        } else {
            owsFailDebug("Unexpected message type: \(message)")
            return bubbleTextColorOutgoing
        }
    }

    @objc
    public func bubbleSecondaryTextColor(isIncoming: Bool) -> UIColor {
        if isIncoming {
            return bubbleSecondaryTextColorIncoming
        } else {
            return bubbleSecondaryTextColorOutgoing
        }
    }

    @objc
    public var quotedReplyBubbleColor: UIColor {
        isDarkThemeEnabled ? .ows_signalBlueDark : .ows_accentBlueTint
    }

    @objc
    public func quotedReplyStripeColor(isIncoming: Bool) -> UIColor {
        if isDarkThemeEnabled {
            if isIncoming {
                return .ows_accentBlueTint
            } else {
                return .ows_black
            }
        } else if isIncoming {
            return .ows_accentBlue
        } else {
            return .ows_white
        }
    }

    @objc
    public func quotingSelfHighlightColor() -> UIColor {
        // TODO:
        UIColor.init(rgbHex: 0xB5B5B5)
    }

    @objc
    public func quotedReplyAuthorColor() -> UIColor {
        quotedReplyTextColor()
    }

    @objc
    public func quotedReplyTextColor() -> UIColor {
        isDarkThemeEnabled ? .ows_gray05 : .ows_gray90
    }

    @objc
    public func quotedReplyAttachmentColor() -> UIColor {
        isDarkThemeEnabled ? .ows_gray05 : UIColor.ows_gray90
    }

    @objc
    public func isEqualForCellRendering(_ other: ConversationStyle) -> Bool {
        // We need to compare any state that could affect
        // how we render view appearance.
        (type.isValid == other.type.isValid &&
            viewWidth == other.viewWidth &&
            dynamicBodyTypePointSize == other.dynamicBodyTypePointSize &&
            conversationColor == other.conversationColor &&
            isDarkThemeEnabled == other.isDarkThemeEnabled &&
            hasWallpaper == other.hasWallpaper &&
            maxMessageWidth == other.maxMessageWidth &&
            maxMediaMessageWidth == other.maxMediaMessageWidth &&
            textInsets == other.textInsets &&
            gutterLeading == other.gutterLeading &&
            gutterTrailing == other.gutterTrailing &&
            fullWidthGutterLeading == other.fullWidthGutterLeading &&
            fullWidthGutterTrailing == other.fullWidthGutterTrailing &&
            textInsets == other.textInsets &&
            lastTextLineAxis == other.lastTextLineAxis)
    }

    @objc
    public override var debugDescription: String {
        "[" +
            "type.isValid: \(type.isValid), " +
            "viewWidth: \(viewWidth), " +
            "dynamicBodyTypePointSize: \(dynamicBodyTypePointSize), " +
            "conversationColor: \(conversationColor), " +
            "isDarkThemeEnabled: \(isDarkThemeEnabled), " +
            "maxMessageWidth: \(maxMessageWidth), " +
            "maxMediaMessageWidth: \(maxMediaMessageWidth), " +
            "textInsets: \(textInsets), " +
            "gutterLeading: \(gutterLeading), " +
            "gutterTrailing: \(gutterTrailing), " +
            "fullWidthGutterLeading: \(fullWidthGutterLeading), " +
            "fullWidthGutterTrailing: \(fullWidthGutterTrailing), " +
            "textInsets: \(textInsets), " +
            "lastTextLineAxis: \(lastTextLineAxis), " +
            "]"
    }
}