xref: /xiu/protocol/rtmp/src/session/errors.rs (revision c2f3fbfa)

use crate::amf0::errors::Amf0WriteError;
use crate::chunk::errors::PackError;
use crate::chunk::errors::UnpackError;
use crate::handshake::errors::HandshakeError;
use crate::messages::errors::MessageError;
use crate::netconnection::errors::NetConnectionError;
use crate::netstream::errors::NetStreamError;
use crate::protocol_control_messages::errors::ControlMessagesError;
use crate::user_control_messages::errors::EventMessagesError;

use netio::bytes_errors::BytesWriteError;
use netio::netio_errors::NetIOError;

use tokio::time::Elapsed;

use failure::{Backtrace, Fail};
use std::fmt;

#[derive(Debug)]
pub struct SessionError {
    pub value: SessionErrorValue,
}

#[derive(Debug, Fail)]
pub enum SessionErrorValue {
    #[fail(display = "amf0 write error: {}", _0)]
    Amf0WriteError(#[cause] Amf0WriteError),
    #[fail(display = "bytes write error: {}", _0)]
    BytesWriteError(#[cause] BytesWriteError),
    #[fail(display = "timeout error: {}", _0)]
    TimeoutError(#[cause] Elapsed),
    #[fail(display = "unpack error: {}", _0)]
    UnPackError(#[cause] UnpackError),


    #[fail(display = "message error: {}", _0)]
    MessageError(#[cause] MessageError),
    #[fail(display = "control message error: {}", _0)]
    ControlMessagesError(#[cause] ControlMessagesError),
    #[fail(display = "net connection error: {}", _0)]
    NetConnectionError(#[cause] NetConnectionError),
    #[fail(display = "net stream error: {}", _0)]
    NetStreamError(#[cause] NetStreamError),

    #[fail(display = "event messages error: {}", _0)]
    EventMessagesError(#[cause] EventMessagesError),
    #[fail(display = "net io error: {}", _0)]
    NetIOError(#[cause] NetIOError),
    #[fail(display = "pack error: {}", _0)]
    PackError(#[cause] PackError),
    #[fail(display = "handshake error: {}", _0)]
    HandshakeError(#[cause] HandshakeError),

    #[fail(display = "amf0 count not correct error")]
    Amf0ValueCountNotCorrect,
    #[fail(display = "amf0 value type not correct error")]
    Amf0ValueTypeNotCorrect,
    #[fail(display = "channel event send error")]
    ChannelEventSendErr,
    #[fail(display = "none channel data sender error")]
    NoneChannelDataSender,
    #[fail(display = "none channel data receiver error")]
    NoneChannelDataReceiver,
    #[fail(display = "send channel data error")]
    SendChannelDataErr,

    #[fail(display = "no app name error")]
    NoAppName,
}

impl From<Amf0WriteError> for SessionError {
    fn from(error: Amf0WriteError) -> Self {
        SessionError {
            value: SessionErrorValue::Amf0WriteError(error),
        }
    }
}

impl From<BytesWriteError> for SessionError {
    fn from(error: BytesWriteError) -> Self {
        SessionError {
            value: SessionErrorValue::BytesWriteError(error),
        }
    }
}

impl From<Elapsed> for SessionError {
    fn from(error: Elapsed) -> Self {
        SessionError {
            value: SessionErrorValue::TimeoutError(error),
        }
    }
}

impl From<UnpackError> for SessionError {
    fn from(error: UnpackError) -> Self {
        SessionError {
            value: SessionErrorValue::UnPackError(error),
        }
    }
}

impl From<MessageError> for SessionError {
    fn from(error: MessageError) -> Self {
        SessionError {
            value: SessionErrorValue::MessageError(error),
        }
    }
}

impl From<ControlMessagesError> for SessionError {
    fn from(error: ControlMessagesError) -> Self {
        SessionError {
            value: SessionErrorValue::ControlMessagesError(error),
        }
    }
}

impl From<NetConnectionError> for SessionError {
    fn from(error: NetConnectionError) -> Self {
        SessionError {
            value: SessionErrorValue::NetConnectionError(error),
        }
    }
}

impl From<NetStreamError> for SessionError {
    fn from(error: NetStreamError) -> Self {
        SessionError {
            value: SessionErrorValue::NetStreamError(error),
        }
    }
}

impl From<EventMessagesError> for SessionError {
    fn from(error: EventMessagesError) -> Self {
        SessionError {
            value: SessionErrorValue::EventMessagesError(error),
        }
    }
}

impl From<NetIOError> for SessionError {
    fn from(error: NetIOError) -> Self {
        SessionError {
            value: SessionErrorValue::NetIOError(error),
        }
    }
}

impl From<PackError> for SessionError {
    fn from(error: PackError) -> Self {
        SessionError {
            value: SessionErrorValue::PackError(error),
        }
    }
}

impl From<HandshakeError> for SessionError {
    fn from(error: HandshakeError) -> Self {
        SessionError {
            value: SessionErrorValue::HandshakeError(error),
        }
    }
}

impl fmt::Display for SessionError {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        fmt::Display::fmt(&self.value, f)
    }
}

impl Fail for SessionError {
    fn cause(&self) -> Option<&dyn Fail> {
        self.value.cause()
    }

    fn backtrace(&self) -> Option<&Backtrace> {
        self.value.backtrace()
    }
}