1
0
Fork 0
mirror of https://github.com/sile/hls_m3u8.git synced 2024-06-16 03:50:34 +00:00
hls_m3u8/src/types/byte_range.rs

109 lines
2.5 KiB
Rust
Raw Normal View History

2019-09-06 11:20:40 +00:00
use std::fmt;
2019-09-13 14:06:52 +00:00
use std::str::FromStr;
2019-09-10 09:05:20 +00:00
use getset::{Getters, MutGetters, Setters};
2019-09-06 11:20:40 +00:00
2019-09-13 14:06:52 +00:00
use crate::Error;
2019-09-10 09:05:20 +00:00
2019-09-06 11:20:40 +00:00
/// Byte range.
///
/// See: [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
2019-09-10 09:05:20 +00:00
#[derive(Getters, Setters, MutGetters, Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[get = "pub"]
#[set = "pub"]
#[get_mut = "pub"]
2019-09-06 11:20:40 +00:00
pub struct ByteRange {
2019-09-10 09:05:20 +00:00
/// The length of the range.
length: usize,
/// The start of the range.
start: Option<usize>,
}
impl ByteRange {
/// Creates a new [ByteRange].
pub const fn new(length: usize, start: Option<usize>) -> Self {
Self { length, start }
}
2019-09-06 11:20:40 +00:00
}
impl fmt::Display for ByteRange {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{}", self.length)?;
if let Some(x) = self.start {
write!(f, "@{}", x)?;
}
Ok(())
}
}
impl FromStr for ByteRange {
type Err = Error;
2019-09-10 09:05:20 +00:00
2019-09-13 14:06:52 +00:00
fn from_str(s: &str) -> Result<Self, Self::Err> {
2019-09-10 09:05:20 +00:00
let tokens = s.splitn(2, '@').collect::<Vec<_>>();
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
2019-09-06 11:20:40 +00:00
};
2019-09-10 09:05:20 +00:00
Ok(ByteRange::new(length, start))
2019-09-06 11:20:40 +00:00
}
}
2019-09-06 11:46:21 +00:00
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_display() {
let byte_range = ByteRange {
length: 0,
start: Some(5),
};
assert_eq!(byte_range.to_string(), "0@5".to_string());
let byte_range = ByteRange {
length: 99999,
start: Some(2),
};
assert_eq!(byte_range.to_string(), "99999@2".to_string());
let byte_range = ByteRange {
length: 99999,
start: None,
};
assert_eq!(byte_range.to_string(), "99999".to_string());
}
#[test]
fn test_parse() {
let byte_range = ByteRange {
length: 99999,
start: Some(2),
};
assert_eq!(byte_range, "99999@2".parse::<ByteRange>().unwrap());
let byte_range = ByteRange {
length: 99999,
start: Some(2),
};
assert_eq!(byte_range, "99999@2".parse::<ByteRange>().unwrap());
let byte_range = ByteRange {
length: 99999,
start: None,
};
assert_eq!(byte_range, "99999".parse::<ByteRange>().unwrap());
}
}