/*
 * Copyright OpenSearch Contributors
 * SPDX-License-Identifier: Apache-2.0
 */
package org.opensearch.commons.notifications.model.config

import org.opensearch.commons.notifications.model.BaseConfigData
import org.opensearch.commons.notifications.model.Chime
import org.opensearch.commons.notifications.model.ConfigType
import org.opensearch.commons.notifications.model.Email
import org.opensearch.commons.notifications.model.EmailGroup
import org.opensearch.commons.notifications.model.SesAccount
import org.opensearch.commons.notifications.model.Slack
import org.opensearch.commons.notifications.model.SmtpAccount
import org.opensearch.commons.notifications.model.Sns
import org.opensearch.commons.notifications.model.Webhook
import org.opensearch.commons.notifications.model.XParser
import org.opensearch.core.common.io.stream.Writeable.Reader
import org.opensearch.core.xcontent.XContentParser

internal object ConfigDataProperties {
    /**
     * Properties for ConfigTypes.
     * This data class is used to provide contract across configTypes without reading into config data classes.
     */
    private data class ConfigProperty(
        val configDataReader: Reader<out BaseConfigData>,
        val configDataParser: XParser<out BaseConfigData>
    )

    private val CONFIG_PROPERTIES_MAP = mapOf(
        Pair(ConfigType.SLACK, ConfigProperty(Slack.reader, Slack.xParser)),
        Pair(ConfigType.CHIME, ConfigProperty(Chime.reader, Chime.xParser)),
        Pair(ConfigType.WEBHOOK, ConfigProperty(Webhook.reader, Webhook.xParser)),
        Pair(ConfigType.EMAIL, ConfigProperty(Email.reader, Email.xParser)),
        Pair(ConfigType.SNS, ConfigProperty(Sns.reader, Sns.xParser)),
        Pair(ConfigType.SES_ACCOUNT, ConfigProperty(SesAccount.reader, SesAccount.xParser)),
        Pair(ConfigType.EMAIL_GROUP, ConfigProperty(EmailGroup.reader, EmailGroup.xParser)),
        Pair(ConfigType.SMTP_ACCOUNT, ConfigProperty(SmtpAccount.reader, SmtpAccount.xParser))
    )

    /**
     * Get Reader for provided config type
     * @param @ConfigType
     * @return Reader
     */
    fun getReaderForConfigType(configType: ConfigType): Reader<out BaseConfigData> {
        return CONFIG_PROPERTIES_MAP[configType]?.configDataReader
            ?: throw IllegalArgumentException("Transport action used with unknown ConfigType:$configType")
    }

    /**
     * Validate config data is of ConfigType
     */
    fun validateConfigData(configType: ConfigType, configData: BaseConfigData?): Boolean {
        return when (configType) {
            ConfigType.SLACK -> configData is Slack
            ConfigType.WEBHOOK -> configData is Webhook
            ConfigType.EMAIL -> configData is Email
            ConfigType.EMAIL_GROUP -> configData is EmailGroup
            ConfigType.SMTP_ACCOUNT -> configData is SmtpAccount
            ConfigType.CHIME -> configData is Chime
            ConfigType.SNS -> configData is Sns
            ConfigType.SES_ACCOUNT -> configData is SesAccount
            ConfigType.NONE -> true
        }
    }

    /**
     * Creates config data from parser for given configType
     * @param configType the ConfigType
     * @param parser parser for ConfigType
     * @return created BaseConfigData on success. null if configType is not recognized
     *
     */
    fun createConfigData(configType: ConfigType, parser: XContentParser): BaseConfigData? {
        return CONFIG_PROPERTIES_MAP[configType]?.configDataParser?.parse(parser)
    }
}