// Button Style class in the CustomStyles target as framework
import Foundation
import SwiftUI
import SwiftExtension

public enum ButtonState {
    case normal
    case highlighted
}

public class RoundedButtonStyleBuilder<Background: View & ShapeStyle> {
    var buttonProperties = [ButtonState: Background]()

    public init() {}

    public func color(
        _ color: Background,
        for buttonState: ButtonState
    ) -> RoundedButtonStyleBuilder {
        self.buttonProperties[buttonState] = color

        return self
    }

    public func build() -> RoundedButtonStyle<Background> {
        var roundedButtonStyle = RoundedButtonStyle<Background>()
        roundedButtonStyle.buttonProperties = buttonProperties
        return roundedButtonStyle
    }
}

public struct RoundedButtonStyle<Background: View & ShapeStyle>: ButtonStyle {

    var buttonProperties = [ButtonState: Background]()
    var propertyBuilder = RoundedButtonStyleBuilder<Background>()

    public init() {}

    public func makeBody(configuration: Configuration) -> some View {
        configuration.label
            .frame(maxWidth: .infinity, maxHeight: .infinity)
            .font(.system(size: 32.0, weight: .medium))
            .foregroundColor(.white)
            .background(configuration.isPressed ? buttonProperties[.highlighted] : buttonProperties[.normal])
            .cornerRadius(16.0)
    }
}

// Custom View using the custom button style with the Builder
import Foundation
import SwiftUI
import SwiftExtension
import CustomStyles


public struct ReceiptView: View {
    public var body: some View {
        VStack(alignment: .center) {
            Spacer().frame(minHeight: 10.0)
            VStack(alignment: .center, spacing: 16.0) {
                Image("clinicLogo")
                Text("Clinic Name")
                    .font(.system(size: 40.0, weight: .medium))
            }
            Spacer().frame(maxHeight: 112.0)
            Button(
                action: {
                    debugPrint(#function)
                },
                label: {
                    VStack(alignment: .center, spacing: 0.0) {
                        Spacer().frame(height: 64.0)
                        Image("icon")
                        Spacer().frame(height: 16.0)
                        Text("Receipt")
                        Spacer().frame(height: 72.0)
                    }
            })
            .frame(width: 440.0, height: 440.0)
            .buttonStyle(
                RoundedButtonStyleBuilder()
                    .color(
                        LinearGradient(
                            colors: [.lightTheme, .defaultTheme],
                            startPoint: .top,
                            endPoint: .bottom
                        ),
                        for: .normal
                    )
                    .color(Color.highlighted, for: .highlighted) // This case causes the compile error!!!!!
//                    .color(
//                        LinearGradient(
//                            colors: [.defaultTheme],
//                            startPoint: .top,
//                            endPoint: .bottom
//                        ),
//                        for: .highlighted
//                    ) // If the same type is passed, it's not occurred the compile error!!
                    .build()
            )
            Spacer().frame(height: 24.0)
        }
        .navigationBarTitleDisplayMode(.inline)
        .navigationBarHidden(true)
    }
}