vade-didcomm 0.3.0

use serde::{Deserialize, Serialize};

use super::protocol::{
    generate_receive_step,
    generate_send_step,
    generate_step_output,
    Protocol,
    StepResult,
};
use crate::datatypes::MessageWithBody;

pub const PING_PONG_PROTOCOL_URL: &str = "https://didcomm.org/trust_ping/1.0";

/// Struct for parsing incoming ping messages.
#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct PingBody {
    response_requested: Option<bool>,
}

/// Creates a new ping_pong protocol and maps the specific step handler functions.
///
/// # Returns
/// * `Protocol` - the new ping pong protocol handler
pub fn generate_ping_pong_protocol() -> Protocol {
    Protocol {
        name: String::from(PING_PONG_PROTOCOL_URL),
        steps: vec![
            generate_send_step("ping", send_ping),
            generate_send_step("ping_response", send_pong),
            generate_receive_step("ping", receive_ping),
            generate_receive_step("ping_response", receive_pong),
        ],
    }
}

/// Protocol handler for direction: `send`, type: `trust_ping/ping`
pub fn send_ping(_options: &str, message: &str) -> StepResult {
    let mut parsed_message: MessageWithBody<PingBody> = serde_json::from_str(message)?;
    parsed_message.body = Some(PingBody {
        response_requested: Some(true),
    });
    generate_step_output(&serde_json::to_string(&parsed_message)?, "{}")
}

/// Protocol handler for direction: `send`, type: `trust_ping/pong`
pub fn send_pong(_options: &str, message: &str) -> StepResult {
    generate_step_output(message, "{}")
}

/// Protocol handler for direction: `receive`, type: `trust_ping/ping`
pub fn receive_ping(_options: &str, message: &str) -> StepResult {
    generate_step_output(message, "{}")
}

/// Protocol handler for direction: `receive`, type: `trust_ping/pong`
pub fn receive_pong(_options: &str, message: &str) -> StepResult {
    generate_step_output(message, "{}")
}