/*
 * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file is distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

package com.amplifyframework.auth.cognito.featuretest

import kotlinx.serialization.SerialName
import kotlinx.serialization.Serializable
import kotlinx.serialization.json.JsonElement

/**
 * FeatureSpecification for auth feature tests.
 * To generate schema: see [https://github.com/Ricky12Awesome/json-schema-serialization]
 */
@Serializable
data class FeatureTestCase(
    val description: String,
    val preConditions: PreConditions,
    val api: API,
    val validations: List<ExpectationShapes>
) {
    override fun toString(): String {
        return "${api.name} : $description"
    }
}

@Serializable
data class PreConditions(
    val `amplify-configuration`: String,
    val state: String,
    val mockedResponses: List<MockResponse>
)

@Serializable
data class API(
    val name: AuthAPI,
    val params: JsonElement,
    val options: JsonElement
)

@Serializable
data class MockResponse(
    val type: CognitoType,
    val apiName: String,
    val responseType: ResponseType,
    val response: JsonElement
)

@Serializable
enum class ResponseType {
    @SerialName("success")
    Success,
    @SerialName("complete")
    Complete,
    @SerialName("failure")
    Failure
}

@Serializable
enum class CognitoType {
    @SerialName("cognitoIdentity")
    CognitoIdentity,
    @SerialName("cognitoIdentityProvider")
    CognitoIdentityProvider
}

@Serializable
sealed class ExpectationShapes {
    @Serializable
    sealed class Cognito : ExpectationShapes() {
        abstract val apiName: String
        abstract val request: JsonElement

        @Serializable @SerialName("cognitoIdentity")
        data class CognitoIdentity(override val apiName: String, override val request: JsonElement) : Cognito()

        @Serializable @SerialName("cognitoIdentityProvider")
        data class CognitoIdentityProvider(override val apiName: String, override val request: JsonElement) : Cognito()
    }

    @Serializable
    @SerialName("amplify")
    data class Amplify(
        val apiName: AuthAPI,
        val responseType: ResponseType,
        val response: JsonElement
    ) : ExpectationShapes()

    @Serializable
    @SerialName("state")
    data class State(
        val expectedState: String
    ) : ExpectationShapes()
}