xref: /xiu/protocol/rtmp/src/netstream/errors.rs (revision 69de9bbd)

use {
    crate::{amf0::errors::Amf0WriteError, chunk::errors::PackError},
    failure::{Backtrace, Fail},
    std::fmt,
};

#[derive(Debug)]
pub struct NetStreamError {
    pub value: NetStreamErrorValue,
}

#[derive(Debug, Fail)]
pub enum NetStreamErrorValue {
    #[fail(display = "amf0 write error: {}", _0)]
    Amf0WriteError(Amf0WriteError),
    #[fail(display = "invalid max chunk size")]
    InvalidMaxChunkSize { chunk_size: usize },
    #[fail(display = "pack error")]
    PackError(PackError),
}

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

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

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

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

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