twitch-irc 6.0.0

use crate::message::IRCMessage;
use crate::message::commands::ServerMessageParseError;
use std::convert::TryFrom;

#[cfg(feature = "with-serde")]
use {serde::Deserialize, serde::Serialize};

/// A `PONG` connection-control message.
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "with-serde", derive(Serialize, Deserialize))]
pub struct PongMessage {
    /// The message that this `PongMessage` was parsed from.
    pub source: IRCMessage,
}

impl TryFrom<IRCMessage> for PongMessage {
    type Error = ServerMessageParseError;

    fn try_from(source: IRCMessage) -> Result<PongMessage, ServerMessageParseError> {
        if source.command != "PONG" {
            return Err(ServerMessageParseError::MismatchedCommand(Box::new(source)));
        }

        Ok(PongMessage { source })
    }
}

impl From<PongMessage> for IRCMessage {
    fn from(msg: PongMessage) -> IRCMessage {
        msg.source
    }
}

#[cfg(test)]
mod tests {
    use crate::message::{IRCMessage, PongMessage};
    use std::convert::TryFrom;

    #[test]
    pub fn test_basic() {
        // this is what the Twitch servers answers "PING" with
        let src = "PONG :tmi.twitch.tv";
        let irc_message = IRCMessage::parse(src).unwrap();
        let msg = PongMessage::try_from(irc_message.clone()).unwrap();

        assert_eq!(
            msg,
            PongMessage {
                source: irc_message
            }
        );
    }

    #[test]
    pub fn test_with_argument() {
        // this is the answer to "PING test"
        let src = ":tmi.twitch.tv PONG tmi.twitch.tv :test";
        let irc_message = IRCMessage::parse(src).unwrap();
        let msg = PongMessage::try_from(irc_message.clone()).unwrap();

        assert_eq!(
            msg,
            PongMessage {
                source: irc_message
            }
        );
    }
}