use crate::message::IRCMessage;
use crate::message::commands::ServerMessageParseError;
use crate::message::commands::ServerMessageParseError::MismatchedCommand;
use std::convert::TryFrom;
#[cfg(feature = "with-serde")]
use {serde::Deserialize, serde::Serialize};
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "with-serde", derive(Serialize, Deserialize))]
pub struct ReconnectMessage {
pub source: IRCMessage,
}
impl TryFrom<IRCMessage> for ReconnectMessage {
type Error = ServerMessageParseError;
fn try_from(source: IRCMessage) -> Result<ReconnectMessage, ServerMessageParseError> {
if source.command == "RECONNECT" {
Ok(ReconnectMessage { source })
} else {
Err(MismatchedCommand(Box::new(source)))
}
}
}
impl From<ReconnectMessage> for IRCMessage {
fn from(msg: ReconnectMessage) -> IRCMessage {
msg.source
}
}
#[cfg(test)]
mod tests {
use crate::message::{IRCMessage, ReconnectMessage};
use std::convert::TryFrom;
#[test]
pub fn test_basic() {
let src = ":tmi.twitch.tv RECONNECT";
let irc_message = IRCMessage::parse(src).unwrap();
let msg = ReconnectMessage::try_from(irc_message.clone()).unwrap();
assert_eq!(
msg,
ReconnectMessage {
source: irc_message
}
);
}
}