import SourceKittenFramework

public struct ModifierOrderConfiguration: RuleConfiguration, Equatable {
    private(set) var severityConfiguration = SeverityConfiguration(.warning)
    private(set) var preferredModifierOrder = [SwiftDeclarationAttributeKind.ModifierGroup]()

    public var consoleDescription: String {
        return severityConfiguration.consoleDescription + ", preferred_modifier_order: \(preferredModifierOrder)"
    }

    public init(preferredModifierOrder: [SwiftDeclarationAttributeKind.ModifierGroup] = []) {
        self.preferredModifierOrder = preferredModifierOrder
    }

    public mutating func apply(configuration: Any) throws {
        guard let configuration = configuration as? [String: Any] else {
            throw ConfigurationError.unknownConfiguration
        }

        if let preferredModifierOrder = configuration["preferred_modifier_order"] as? [String] {
            self.preferredModifierOrder = try preferredModifierOrder.map {
                guard let modifierGroup = SwiftDeclarationAttributeKind.ModifierGroup(rawValue: $0),
                      modifierGroup != .atPrefixed else {
                    throw ConfigurationError.unknownConfiguration
                }

                return modifierGroup
            }
        }

        if let severityString = configuration["severity"] as? String {
            try severityConfiguration.apply(configuration: severityString)
        }
    }

    public static func == (lhs: ModifierOrderConfiguration,
                           rhs: ModifierOrderConfiguration) -> Bool {
        return lhs.preferredModifierOrder == rhs.preferredModifierOrder &&
               lhs.severityConfiguration == rhs.severityConfiguration
    }
}