hls_m3u8/src/tags/media_playlist/target_duration.rs

use std::fmt;
use std::str::FromStr;
use std::time::Duration;

use derive_more::Deref;

use crate::types::ProtocolVersion;
use crate::utils::tag;
use crate::{Error, RequiredVersion};

/// # [4.3.3.1. EXT-X-TARGETDURATION]
///
/// The [`ExtXTargetDuration`] tag specifies the maximum [`MediaSegment`]
/// duration.
///
/// [`MediaSegment`]: crate::MediaSegment
/// [4.3.3.1. EXT-X-TARGETDURATION]:
/// https://tools.ietf.org/html/rfc8216#section-4.3.3.1
#[derive(Deref, Debug, Clone, Copy, PartialEq, Eq, Hash, Default, PartialOrd, Ord)]
pub struct ExtXTargetDuration(Duration);

impl ExtXTargetDuration {
    pub(crate) const PREFIX: &'static str = "#EXT-X-TARGETDURATION:";

    /// Makes a new [`ExtXTargetDuration`] tag.
    ///
    /// # Example
    ///
    /// ```
    /// # use hls_m3u8::tags::ExtXTargetDuration;
    /// use std::time::Duration;
    ///
    /// let target_duration = ExtXTargetDuration::new(Duration::from_secs(20));
    /// ```
    ///
    /// # Note
    ///
    /// The nanoseconds part of the [`Duration`] will be discarded.
    #[must_use]
    pub const fn new(duration: Duration) -> Self { Self(Duration::from_secs(duration.as_secs())) }

    /// Returns the maximum media segment duration.
    ///
    /// # Example
    ///
    /// ```
    /// # use hls_m3u8::tags::ExtXTargetDuration;
    /// use std::time::Duration;
    ///
    /// let target_duration = ExtXTargetDuration::new(Duration::from_nanos(2_000_000_000));
    ///
    /// assert_eq!(target_duration.duration(), Duration::from_secs(2));
    /// ```
    #[must_use]
    pub const fn duration(&self) -> Duration { self.0 }
}

/// This tag requires [`ProtocolVersion::V1`].
impl RequiredVersion for ExtXTargetDuration {
    fn required_version(&self) -> ProtocolVersion { ProtocolVersion::V1 }
}

impl fmt::Display for ExtXTargetDuration {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{}{}", Self::PREFIX, self.0.as_secs())
    }
}

impl FromStr for ExtXTargetDuration {
    type Err = Error;

    fn from_str(input: &str) -> Result<Self, Self::Err> {
        let input = tag(input, Self::PREFIX)?;
        let input = input.parse().map_err(|e| Error::parse_int(input, e))?;

        Ok(Self::new(Duration::from_secs(input)))
    }
}

impl From<Duration> for ExtXTargetDuration {
    fn from(value: Duration) -> Self { Self::new(value) }
}

#[cfg(test)]
mod test {
    use super::*;
    use pretty_assertions::assert_eq;

    #[test]
    fn test_display() {
        assert_eq!(
            ExtXTargetDuration::new(Duration::from_secs(5)).to_string(),
            "#EXT-X-TARGETDURATION:5".to_string()
        );
    }

    #[test]
    fn test_required_version() {
        assert_eq!(
            ExtXTargetDuration::new(Duration::from_secs(5)).required_version(),
            ProtocolVersion::V1
        );
    }

    #[test]
    fn test_parser() {
        assert_eq!(
            ExtXTargetDuration::new(Duration::from_secs(5)),
            "#EXT-X-TARGETDURATION:5".parse().unwrap()
        );
    }

    #[test]
    fn test_deref() {
        assert_eq!(ExtXTargetDuration::new(Duration::from_secs(5)).as_secs(), 5);
    }
}