// Unless explicitly stated otherwise all files in this repository are licensed under the Apache-2.0 License.
// This product includes software developed at Datadog (https://www.datadoghq.com/).
// Copyright 2019-Present Datadog, Inc.

use serde::{Deserialize, Deserializer, Serialize, Serializer};

#[non_exhaustive]
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum SyntheticsTestSubType {
    DNS,
    GRPC,
    HTTP,
    ICMP,
    MCP,
    MULTI,
    SSL,
    TCP,
    UDP,
    WEBSOCKET,
    UnparsedObject(crate::datadog::UnparsedObject),
}

impl ToString for SyntheticsTestSubType {
    fn to_string(&self) -> String {
        match self {
            Self::DNS => String::from("dns"),
            Self::GRPC => String::from("grpc"),
            Self::HTTP => String::from("http"),
            Self::ICMP => String::from("icmp"),
            Self::MCP => String::from("mcp"),
            Self::MULTI => String::from("multi"),
            Self::SSL => String::from("ssl"),
            Self::TCP => String::from("tcp"),
            Self::UDP => String::from("udp"),
            Self::WEBSOCKET => String::from("websocket"),
            Self::UnparsedObject(v) => v.value.to_string(),
        }
    }
}

impl Serialize for SyntheticsTestSubType {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        match self {
            Self::UnparsedObject(v) => v.serialize(serializer),
            _ => serializer.serialize_str(self.to_string().as_str()),
        }
    }
}

impl<'de> Deserialize<'de> for SyntheticsTestSubType {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        let s: String = String::deserialize(deserializer)?;
        Ok(match s.as_str() {
            "dns" => Self::DNS,
            "grpc" => Self::GRPC,
            "http" => Self::HTTP,
            "icmp" => Self::ICMP,
            "mcp" => Self::MCP,
            "multi" => Self::MULTI,
            "ssl" => Self::SSL,
            "tcp" => Self::TCP,
            "udp" => Self::UDP,
            "websocket" => Self::WEBSOCKET,
            _ => Self::UnparsedObject(crate::datadog::UnparsedObject {
                value: serde_json::Value::String(s.into()),
            }),
        })
    }
}
