gremlin-client 0.8.10

use serde::{Deserialize as SerdeDeserialize, Deserializer};
use serde_derive::{Deserialize, Serialize};
use serde_json::Value;
use uuid::Uuid;

#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct RequestIdV2 {
    #[serde(rename = "@type")]
    id_type: String,

    #[serde(rename = "@value")]
    value: Uuid,
}

#[derive(Serialize)]
#[serde(rename_all = "camelCase", untagged)]
pub enum Message<T> {
    #[serde(rename_all = "camelCase")]
    V1 {
        request_id: Uuid,
        op: String,
        processor: String,
        args: T,
    },
    #[serde(rename_all = "camelCase")]
    V2 {
        request_id: RequestIdV2,
        op: String,
        processor: String,
        args: T,
    },
    #[serde(rename_all = "camelCase")]
    V3 {
        request_id: Uuid,
        op: String,
        processor: String,
        args: T,
    },
}

impl<T> Message<T> {
    #[allow(dead_code)]
    pub fn id(&self) -> &Uuid {
        match self {
            Message::V1 { request_id, .. } => request_id,
            Message::V2 { request_id, .. } => &request_id.value,
            Message::V3 { request_id, .. } => request_id,
        }
    }
}
#[derive(Debug, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Response {
    pub request_id: Uuid,
    pub result: ResponseResult,
    pub status: ReponseStatus,
}

#[derive(Debug, Deserialize)]
pub struct ResponseResult {
    pub data: Value,
}

#[derive(Debug, Deserialize)]
pub struct ReponseStatus {
    pub code: i16,
    //Sometimes the message is omitted, default to empty string rather than panic
    #[serde(default, deserialize_with = "map_null_to_default")]
    pub message: String,
}

fn map_null_to_default<'de, D, T>(de: D) -> Result<T, D::Error>
where
    D: Deserializer<'de>,
    T: Default + SerdeDeserialize<'de>,
{
    Option::<T>::deserialize(de).map(Option::unwrap_or_default)
}

pub fn message_with_args_v2<T>(op: String, processor: String, args: T) -> Message<T> {
    message_with_args_and_uuid_v2(op, processor, Uuid::new_v4(), args)
}

pub fn message_with_args_and_uuid_v2<T>(
    op: String,
    processor: String,
    id: Uuid,
    args: T,
) -> Message<T> {
    Message::V2 {
        request_id: RequestIdV2 {
            id_type: "g:UUID".to_string(),
            value: id,
        },
        op,
        processor,
        args,
    }
}

pub fn message_with_args<T>(op: String, processor: String, args: T) -> Message<T> {
    message_with_args_and_uuid(op, processor, Uuid::new_v4(), args)
}

pub fn message_with_args_and_uuid<T>(
    op: String,
    processor: String,
    id: Uuid,
    args: T,
) -> Message<T> {
    Message::V3 {
        request_id: id,
        op,
        processor,
        args,
    }
}

#[cfg(test)]
mod tests {
    use crate::message::ReponseStatus;

    #[test]
    fn handle_no_response_status_message() {
        let parsed: ReponseStatus =
            serde_json::from_str(r#"{"code": 123}"#).expect("Failed to parse test message");
        assert_eq!(123, parsed.code);
        assert_eq!("", parsed.message);
    }

    #[test]
    fn handle_null_response_status_message() {
        let parsed: ReponseStatus = serde_json::from_str(r#"{"code": 123, "message": null}"#)
            .expect("Failed to parse test message");
        assert_eq!(123, parsed.code);
        assert_eq!("", parsed.message);
    }

    #[test]
    fn handle_response_status_message() {
        let parsed: ReponseStatus =
            serde_json::from_str(r#"{"code": 123, "message": "Hello World"}"#)
                .expect("Failed to parse test message");
        assert_eq!(123, parsed.code);
        assert_eq!("Hello World", parsed.message);
    }
}