2019-09-06 10:55:00 +00:00
|
|
|
use std::fmt;
|
2019-09-10 09:05:20 +00:00
|
|
|
use std::ops::Deref;
|
2019-09-06 10:55:00 +00:00
|
|
|
use std::str::FromStr;
|
2019-09-10 09:05:20 +00:00
|
|
|
|
2019-09-22 08:57:28 +00:00
|
|
|
use crate::types::{ByteRange, ProtocolVersion, RequiredVersion};
|
2019-09-13 14:06:52 +00:00
|
|
|
use crate::utils::tag;
|
|
|
|
use crate::Error;
|
2019-09-10 09:05:20 +00:00
|
|
|
|
2019-09-06 10:55:00 +00:00
|
|
|
/// [4.3.2.2. EXT-X-BYTERANGE]
|
|
|
|
///
|
|
|
|
/// [4.3.2.2. EXT-X-BYTERANGE]: https://tools.ietf.org/html/rfc8216#section-4.3.2.2
|
|
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
2019-09-10 09:05:20 +00:00
|
|
|
pub struct ExtXByteRange(ByteRange);
|
2019-09-06 10:55:00 +00:00
|
|
|
|
|
|
|
impl ExtXByteRange {
|
|
|
|
pub(crate) const PREFIX: &'static str = "#EXT-X-BYTERANGE:";
|
|
|
|
|
|
|
|
/// Makes a new `ExtXByteRange` tag.
|
2019-09-22 08:57:28 +00:00
|
|
|
///
|
2019-09-10 09:05:20 +00:00
|
|
|
/// # Example
|
|
|
|
/// ```
|
|
|
|
/// use hls_m3u8::tags::ExtXByteRange;
|
|
|
|
///
|
|
|
|
/// let byte_range = ExtXByteRange::new(20, Some(5));
|
|
|
|
/// ```
|
|
|
|
pub const fn new(length: usize, start: Option<usize>) -> Self {
|
|
|
|
Self(ByteRange::new(length, start))
|
2019-09-06 10:55:00 +00:00
|
|
|
}
|
|
|
|
|
2019-09-10 09:05:20 +00:00
|
|
|
/// Converts the [ExtXByteRange] to a [ByteRange].
|
2019-09-22 08:57:28 +00:00
|
|
|
///
|
2019-09-10 09:05:20 +00:00
|
|
|
/// # Example
|
|
|
|
/// ```
|
|
|
|
/// use hls_m3u8::tags::ExtXByteRange;
|
|
|
|
/// use hls_m3u8::types::ByteRange;
|
|
|
|
///
|
|
|
|
/// let byte_range = ExtXByteRange::new(20, Some(5));
|
|
|
|
/// let range: ByteRange = byte_range.to_range();
|
|
|
|
/// ```
|
|
|
|
pub const fn to_range(&self) -> ByteRange {
|
|
|
|
self.0
|
2019-09-06 10:55:00 +00:00
|
|
|
}
|
2019-09-22 08:57:28 +00:00
|
|
|
}
|
2019-09-06 10:55:00 +00:00
|
|
|
|
2019-09-22 08:57:28 +00:00
|
|
|
impl RequiredVersion for ExtXByteRange {
|
|
|
|
fn required_version(&self) -> ProtocolVersion {
|
2019-09-06 10:55:00 +00:00
|
|
|
ProtocolVersion::V4
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-09-10 09:05:20 +00:00
|
|
|
impl Deref for ExtXByteRange {
|
|
|
|
type Target = ByteRange;
|
|
|
|
|
|
|
|
fn deref(&self) -> &Self::Target {
|
|
|
|
&self.0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-09-06 10:55:00 +00:00
|
|
|
impl fmt::Display for ExtXByteRange {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
2019-09-10 09:05:20 +00:00
|
|
|
write!(f, "{}", Self::PREFIX)?;
|
|
|
|
write!(f, "{}", self.0)?;
|
|
|
|
Ok(())
|
2019-09-06 10:55:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl FromStr for ExtXByteRange {
|
|
|
|
type Err = Error;
|
2019-09-08 10:23:33 +00:00
|
|
|
|
2019-09-13 14:06:52 +00:00
|
|
|
fn from_str(input: &str) -> Result<Self, Self::Err> {
|
|
|
|
let input = tag(input, Self::PREFIX)?;
|
|
|
|
|
|
|
|
let tokens = input.splitn(2, '@').collect::<Vec<_>>();
|
2019-09-22 08:57:28 +00:00
|
|
|
|
2019-09-10 09:05:20 +00:00
|
|
|
if tokens.is_empty() {
|
2019-09-13 14:06:52 +00:00
|
|
|
return Err(Error::invalid_input());
|
2019-09-10 09:05:20 +00:00
|
|
|
}
|
|
|
|
|
2019-09-13 14:06:52 +00:00
|
|
|
let length = tokens[0].parse()?;
|
|
|
|
|
2019-09-10 09:05:20 +00:00
|
|
|
let start = {
|
|
|
|
let mut result = None;
|
|
|
|
if tokens.len() == 2 {
|
2019-09-13 14:06:52 +00:00
|
|
|
result = Some(tokens[1].parse()?);
|
2019-09-10 09:05:20 +00:00
|
|
|
}
|
|
|
|
result
|
|
|
|
};
|
|
|
|
|
|
|
|
Ok(ExtXByteRange::new(length, start))
|
2019-09-06 10:55:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod test {
|
|
|
|
use super::*;
|
|
|
|
|
|
|
|
#[test]
|
2019-09-10 09:05:20 +00:00
|
|
|
fn test_display() {
|
|
|
|
let byte_range = ExtXByteRange::new(0, Some(5));
|
|
|
|
assert_eq!(byte_range.to_string(), "#EXT-X-BYTERANGE:0@5".to_string());
|
|
|
|
|
|
|
|
let byte_range = ExtXByteRange::new(99999, Some(2));
|
|
|
|
assert_eq!(
|
|
|
|
byte_range.to_string(),
|
|
|
|
"#EXT-X-BYTERANGE:99999@2".to_string()
|
|
|
|
);
|
|
|
|
|
|
|
|
let byte_range = ExtXByteRange::new(99999, None);
|
|
|
|
assert_eq!(byte_range.to_string(), "#EXT-X-BYTERANGE:99999".to_string());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2019-09-21 09:53:34 +00:00
|
|
|
fn test_parser() {
|
2019-09-10 09:05:20 +00:00
|
|
|
let byte_range = ExtXByteRange::new(99999, Some(2));
|
|
|
|
assert_eq!(
|
|
|
|
byte_range,
|
|
|
|
"#EXT-X-BYTERANGE:99999@2".parse::<ExtXByteRange>().unwrap()
|
|
|
|
);
|
|
|
|
|
|
|
|
let byte_range = ExtXByteRange::new(99999, None);
|
|
|
|
assert_eq!(
|
|
|
|
byte_range,
|
|
|
|
"#EXT-X-BYTERANGE:99999".parse::<ExtXByteRange>().unwrap()
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_deref() {
|
|
|
|
let byte_range = ExtXByteRange::new(0, Some(22));
|
|
|
|
|
2019-09-21 09:53:34 +00:00
|
|
|
assert_eq!(byte_range.length(), 0);
|
|
|
|
assert_eq!(byte_range.start(), Some(22));
|
2019-09-06 10:55:00 +00:00
|
|
|
}
|
2019-09-22 08:57:28 +00:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_required_version() {
|
|
|
|
assert_eq!(
|
|
|
|
ExtXByteRange::new(20, Some(5)).required_version(),
|
|
|
|
ProtocolVersion::V4
|
|
|
|
);
|
|
|
|
}
|
2019-09-06 10:55:00 +00:00
|
|
|
}
|