2016-06-03 18:56:45 +00:00
|
|
|
//! Contains all the structs used for parsing.
|
|
|
|
//!
|
|
|
|
//! The main type here is the `Playlist` enum.
|
|
|
|
//! Which is either a `MasterPlaylist` or a `MediaPlaylist`.
|
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
use std::io::Write;
|
2016-06-03 18:56:45 +00:00
|
|
|
use std::collections::HashMap;
|
|
|
|
use std::str::FromStr;
|
|
|
|
use std::fmt;
|
|
|
|
use std::f32;
|
2021-10-14 19:21:03 +00:00
|
|
|
use std::fmt::Display;
|
2016-06-03 18:56:45 +00:00
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
macro_rules! write_some_attribute_quoted {
|
|
|
|
($w:expr, $tag:expr, $o:expr) => (
|
|
|
|
if let &Some(ref v) = $o { write!($w, "{}=\"{}\"", $tag, v) } else { Ok(()) }
|
|
|
|
);
|
2020-02-27 01:17:13 +00:00
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
macro_rules! write_some_attribute {
|
|
|
|
($w:expr, $tag:expr, $o:expr) => (
|
|
|
|
if let &Some(ref v) = $o { write!($w, "{}={}", $tag, v) } else { Ok(()) }
|
|
|
|
);
|
2020-02-27 01:17:13 +00:00
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
macro_rules! bool_default_false {
|
|
|
|
($optional:expr) => (
|
|
|
|
match $optional {
|
|
|
|
Some(ref s) if s == "YES" => true,
|
|
|
|
Some(_) | None => false,
|
|
|
|
}
|
|
|
|
);
|
2020-02-27 01:17:13 +00:00
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
2016-06-03 18:56:45 +00:00
|
|
|
/// [Playlist](https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.1),
|
|
|
|
/// can either be a `MasterPlaylist` or a `MediaPlaylist`.
|
|
|
|
///
|
|
|
|
/// A Playlist is a Media Playlist if all URI lines in the Playlist
|
|
|
|
/// identify Media Segments. A Playlist is a Master Playlist if all URI
|
|
|
|
/// lines in the Playlist identify Media Playlists. A Playlist MUST be
|
|
|
|
/// either a Media Playlist or a Master Playlist; all other Playlists are invalid.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub enum Playlist {
|
|
|
|
MasterPlaylist(MasterPlaylist),
|
|
|
|
MediaPlaylist(MediaPlaylist),
|
|
|
|
}
|
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
impl Playlist {
|
|
|
|
pub fn write_to<T: Write>(&self, writer: &mut T) -> std::io::Result<()> {
|
|
|
|
match self {
|
|
|
|
&Playlist::MasterPlaylist(ref pl) => pl.write_to(writer),
|
|
|
|
&Playlist::MediaPlaylist(ref pl) => pl.write_to(writer),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-06-03 18:56:45 +00:00
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
// Master Playlist
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
/// A [Master Playlist]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4)
|
|
|
|
/// provides a set of Variant Streams, each of which
|
|
|
|
/// describes a different version of the same content.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct MasterPlaylist {
|
|
|
|
pub version: usize,
|
|
|
|
pub variants: Vec<VariantStream>,
|
2021-04-19 15:55:02 +00:00
|
|
|
pub session_data: Vec<SessionData>,
|
|
|
|
pub session_key: Vec<SessionKey>,
|
2016-06-03 18:56:45 +00:00
|
|
|
pub start: Option<Start>,
|
|
|
|
pub independent_segments: bool,
|
2021-03-11 20:15:22 +00:00
|
|
|
pub alternatives: Vec<AlternativeMedia>, // EXT-X-MEDIA tags
|
2021-10-14 19:21:03 +00:00
|
|
|
pub unknown_tags: Vec<ExtTag>,
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
impl MasterPlaylist {
|
|
|
|
|
2021-04-18 16:51:29 +00:00
|
|
|
pub fn get_newest_variant(&mut self) -> Option<&mut VariantStream> {
|
2016-06-03 18:56:45 +00:00
|
|
|
self.variants.iter_mut().rev().find(|v| !v.is_i_frame)
|
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
writeln!(w, "{}" ,"#EXTM3U")?;
|
|
|
|
writeln!(w, "#EXT-X-VERSION:{}", self.version)?;
|
|
|
|
|
2021-04-16 19:04:03 +00:00
|
|
|
for alternative in &self.alternatives {
|
|
|
|
alternative.write_to(w)?;
|
|
|
|
}
|
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
for variant in &self.variants {
|
|
|
|
variant.write_to(w)?;
|
|
|
|
}
|
2021-04-19 15:55:02 +00:00
|
|
|
for session_data in &self.session_data {
|
2017-02-17 13:50:50 +00:00
|
|
|
session_data.write_to(w)?;
|
|
|
|
}
|
2021-04-19 15:55:02 +00:00
|
|
|
for session_key in &self.session_key {
|
2017-02-17 13:50:50 +00:00
|
|
|
session_key.write_to(w)?;
|
|
|
|
}
|
|
|
|
if let Some(ref start) = self.start {
|
|
|
|
start.write_to(w)?;
|
|
|
|
}
|
2020-02-27 01:17:13 +00:00
|
|
|
if self.independent_segments {
|
2017-02-17 13:50:50 +00:00
|
|
|
writeln!(w, "#EXT-X-INDEPENDENT-SEGMENTS")?;
|
|
|
|
}
|
2021-10-14 19:21:03 +00:00
|
|
|
for unknown_tag in &self.unknown_tags {
|
|
|
|
writeln!(w, "{}", unknown_tag)?;
|
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
/// [`#EXT-X-STREAM-INF:<attribute-list>
|
|
|
|
/// <URI>`]
|
2016-06-03 18:56:45 +00:00
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4.2)
|
|
|
|
/// [`#EXT-X-I-FRAME-STREAM-INF:<attribute-list>`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4.3)
|
|
|
|
///
|
|
|
|
/// A Variant Stream includes a Media Playlist that specifies media
|
|
|
|
/// encoded at a particular bit rate, in a particular format, and at a
|
|
|
|
/// particular resolution for media containing video.
|
|
|
|
///
|
|
|
|
/// A Variant Stream can also specify a set of Renditions. Renditions
|
|
|
|
/// are alternate versions of the content, such as audio produced in
|
|
|
|
/// different languages or video recorded from different camera angles.
|
|
|
|
///
|
|
|
|
/// Clients should switch between different Variant Streams to adapt to
|
|
|
|
/// network conditions. Clients should choose Renditions based on user
|
|
|
|
/// preferences.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct VariantStream {
|
|
|
|
pub is_i_frame: bool,
|
|
|
|
pub uri: String,
|
|
|
|
|
|
|
|
// <attribute-list>
|
|
|
|
pub bandwidth: String,
|
|
|
|
pub average_bandwidth: Option<String>,
|
2020-02-27 01:17:13 +00:00
|
|
|
pub codecs: Option<String>,
|
2016-06-03 18:56:45 +00:00
|
|
|
pub resolution: Option<String>,
|
|
|
|
pub frame_rate: Option<String>,
|
2021-01-20 10:25:52 +00:00
|
|
|
pub hdcp_level: Option<String>,
|
2016-06-03 18:56:45 +00:00
|
|
|
pub audio: Option<String>,
|
|
|
|
pub video: Option<String>,
|
|
|
|
pub subtitles: Option<String>,
|
|
|
|
pub closed_captions: Option<String>,
|
|
|
|
// PROGRAM-ID tag was removed in protocol version 6
|
|
|
|
}
|
|
|
|
|
|
|
|
impl VariantStream {
|
2017-02-17 13:50:50 +00:00
|
|
|
|
2016-06-03 18:56:45 +00:00
|
|
|
pub fn from_hashmap(mut attrs: HashMap<String, String>, is_i_frame: bool) -> VariantStream {
|
|
|
|
VariantStream {
|
|
|
|
is_i_frame: is_i_frame,
|
|
|
|
uri: attrs.remove("URI").unwrap_or_else(String::new),
|
|
|
|
bandwidth: attrs.remove("BANDWIDTH").unwrap_or_else(String::new),
|
|
|
|
average_bandwidth: attrs.remove("AVERAGE-BANDWIDTH"),
|
2020-02-27 01:17:13 +00:00
|
|
|
codecs: attrs.remove("CODECS"),
|
2016-06-03 18:56:45 +00:00
|
|
|
resolution: attrs.remove("RESOLUTION"),
|
|
|
|
frame_rate: attrs.remove("FRAME-RATE"),
|
2021-01-20 10:25:52 +00:00
|
|
|
hdcp_level: attrs.remove("HDCP-LEVEL"),
|
2016-06-03 18:56:45 +00:00
|
|
|
audio: attrs.remove("AUDIO"),
|
|
|
|
video: attrs.remove("VIDEO"),
|
|
|
|
subtitles: attrs.remove("SUBTITLES"),
|
|
|
|
closed_captions: attrs.remove("CLOSED-CAPTIONS"),
|
|
|
|
}
|
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
|
|
|
|
|
|
|
|
if self.is_i_frame {
|
|
|
|
write!(w, "#EXT-X-I-FRAME-STREAM-INF:")?;
|
|
|
|
self.write_stream_inf_common_attributes(w)?;
|
2020-02-11 20:43:06 +00:00
|
|
|
writeln!(w, ",URI=\"{}\"", self.uri)
|
2017-02-17 13:50:50 +00:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
write!(w, "#EXT-X-STREAM-INF:")?;
|
|
|
|
self.write_stream_inf_common_attributes(w)?;
|
|
|
|
write_some_attribute_quoted!(w, ",AUDIO", &self.audio)?;
|
|
|
|
write_some_attribute_quoted!(w, ",SUBTITLES", &self.subtitles)?;
|
|
|
|
write_some_attribute_quoted!(w, ",CLOSED-CAPTIONS", &self.closed_captions)?;
|
|
|
|
write!(w, "\n")?;
|
|
|
|
writeln!(w, "{}", self.uri)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn write_stream_inf_common_attributes<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
write!(w, "BANDWIDTH={}", &self.bandwidth)?;
|
|
|
|
write_some_attribute!(w, ",AVERAGE-BANDWIDTH", &self.average_bandwidth)?;
|
2020-02-27 01:17:13 +00:00
|
|
|
write_some_attribute_quoted!(w, ",CODECS", &self.codecs)?;
|
2017-02-17 13:50:50 +00:00
|
|
|
write_some_attribute!(w, ",RESOLUTION", &self.resolution)?;
|
|
|
|
write_some_attribute!(w, ",FRAME-RATE", &self.frame_rate)?;
|
2021-01-20 10:25:52 +00:00
|
|
|
write_some_attribute!(w, ",HDCP-LEVEL", &self.hdcp_level)?;
|
2017-02-17 13:50:50 +00:00
|
|
|
write_some_attribute_quoted!(w, ",VIDEO", &self.video)
|
|
|
|
}
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// [`#EXT-X-MEDIA:<attribute-list>`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4.1)
|
|
|
|
///
|
|
|
|
/// The EXT-X-MEDIA tag is used to relate Media Playlists that contain
|
|
|
|
/// alternative Renditions (Section 4.3.4.2.1) of the same content. For
|
|
|
|
/// example, three EXT-X-MEDIA tags can be used to identify audio-only
|
|
|
|
/// Media Playlists that contain English, French and Spanish Renditions
|
|
|
|
/// of the same presentation. Or two EXT-X-MEDIA tags can be used to
|
|
|
|
/// identify video-only Media Playlists that show two different camera angles.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct AlternativeMedia {
|
|
|
|
// <attribute-list>
|
|
|
|
pub media_type: AlternativeMediaType,
|
|
|
|
pub uri: Option<String>,
|
|
|
|
pub group_id: String,
|
|
|
|
pub language: Option<String>,
|
|
|
|
pub assoc_language: Option<String>,
|
|
|
|
pub name: String, // All EXT-X-MEDIA tags in the same Group MUST have different NAME attributes.
|
|
|
|
pub default: bool, // Its absence indicates an implicit value of NO
|
|
|
|
pub autoselect: bool, // Its absence indicates an implicit value of NO
|
|
|
|
pub forced: bool, // Its absence indicates an implicit value of NO
|
|
|
|
pub instream_id: Option<String>,
|
|
|
|
pub characteristics: Option<String>,
|
2021-01-20 10:25:52 +00:00
|
|
|
pub channels: Option<String>,
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
impl AlternativeMedia {
|
2017-02-17 13:50:50 +00:00
|
|
|
|
2016-06-03 18:56:45 +00:00
|
|
|
pub fn from_hashmap(mut attrs: HashMap<String, String>) -> AlternativeMedia {
|
|
|
|
AlternativeMedia {
|
|
|
|
media_type: attrs.get("TYPE")
|
|
|
|
.and_then(|s| AlternativeMediaType::from_str(s).ok())
|
|
|
|
.unwrap_or_else(Default::default),
|
|
|
|
uri: attrs.remove("URI"),
|
|
|
|
group_id: attrs.remove("GROUP-ID").unwrap_or_else(String::new),
|
|
|
|
language: attrs.remove("LANGUAGE"),
|
|
|
|
assoc_language: attrs.remove("ASSOC-LANGUAGE"),
|
|
|
|
name: attrs.remove("NAME").unwrap_or(String::new()),
|
2017-02-17 13:50:50 +00:00
|
|
|
default: bool_default_false!(attrs.remove("DEFAULT")),
|
|
|
|
autoselect: bool_default_false!(attrs.remove("AUTOSELECT")),
|
|
|
|
forced: bool_default_false!(attrs.remove("FORCED")),
|
2016-06-03 18:56:45 +00:00
|
|
|
instream_id: attrs.remove("INSTREAM-ID"),
|
|
|
|
characteristics: attrs.remove("CHARACTERISTICS"),
|
2021-01-20 10:25:52 +00:00
|
|
|
channels: attrs.remove("CHANNELS"),
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
write!(w, "#EXT-X-MEDIA:")?;
|
|
|
|
write!(w, "TYPE={}", self.media_type)?;
|
|
|
|
write_some_attribute_quoted!(w, ",URI", &self.uri)?;
|
|
|
|
write!(w, ",GROUP-ID=\"{}\"", self.group_id)?;
|
|
|
|
write_some_attribute_quoted!(w, ",LANGUAGE", &self.language)?;
|
|
|
|
write_some_attribute_quoted!(w, ",ASSOC-LANGUAGE", &self.assoc_language)?;
|
|
|
|
write!(w, ",NAME=\"{}\"", self.name)?;
|
|
|
|
if self.default { write!(w, ",DEFAULT=YES")?; }
|
|
|
|
if self.autoselect { write!(w, ",AUTOSELECT=YES")?; }
|
|
|
|
if self.forced { write!(w, ",FORCED=YES")?; }
|
|
|
|
write_some_attribute_quoted!(w, ",INSTREAM-ID", &self.instream_id)?;
|
|
|
|
write_some_attribute_quoted!(w, ",CHARACTERISTICS", &self.characteristics)?;
|
2021-01-20 10:25:52 +00:00
|
|
|
write_some_attribute_quoted!(w, ",CHANNELS", &self.channels)?;
|
2017-02-17 13:50:50 +00:00
|
|
|
write!(w, "\n")
|
|
|
|
}
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub enum AlternativeMediaType {
|
|
|
|
Audio,
|
|
|
|
Video,
|
|
|
|
Subtitles,
|
|
|
|
ClosedCaptions,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl FromStr for AlternativeMediaType {
|
|
|
|
type Err = String;
|
|
|
|
|
|
|
|
fn from_str(s: &str) -> Result<AlternativeMediaType, String> {
|
|
|
|
match s {
|
|
|
|
"AUDIO" => Ok(AlternativeMediaType::Audio),
|
|
|
|
"VIDEO" => Ok(AlternativeMediaType::Video),
|
|
|
|
"SUBTITLES" => Ok(AlternativeMediaType::Subtitles),
|
|
|
|
"CLOSEDCAPTIONS" => Ok(AlternativeMediaType::ClosedCaptions),
|
|
|
|
_ => Err(format!("Unable to create AlternativeMediaType from {:?}", s)),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Default for AlternativeMediaType {
|
|
|
|
fn default() -> AlternativeMediaType {
|
|
|
|
AlternativeMediaType::Video
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
impl fmt::Display for AlternativeMediaType {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
|
|
write!(f, "{}", match self {
|
|
|
|
&AlternativeMediaType::Audio => "AUDIO",
|
|
|
|
&AlternativeMediaType::Video => "VIDEO",
|
|
|
|
&AlternativeMediaType::Subtitles => "SUBTITLES",
|
|
|
|
&AlternativeMediaType::ClosedCaptions => "CLOSEDCAPTIONS",
|
|
|
|
})
|
2020-02-27 01:17:13 +00:00
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-06-03 18:56:45 +00:00
|
|
|
/// [`#EXT-X-SESSION-KEY:<attribute-list>`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4.5)
|
2021-04-19 16:35:45 +00:00
|
|
|
/// The EXT-X-SESSION-KEY tag allows encryption keys from Media Playlists
|
|
|
|
/// to be specified in a Master Playlist. This allows the client to
|
|
|
|
/// preload these keys without having to read the Media Playlist(s) first.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct SessionKey(pub Key);
|
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
impl SessionKey {
|
|
|
|
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
write!(w, "#EXT-X-SESSION-KEY:")?;
|
|
|
|
self.0.write_attributes_to(w)?;
|
|
|
|
write!(w, "\n")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-04-19 16:35:45 +00:00
|
|
|
#[derive(Debug, PartialEq, Clone)]
|
|
|
|
pub enum SessionDataField {
|
|
|
|
Value(String),
|
|
|
|
Uri(String)
|
|
|
|
}
|
|
|
|
|
2016-06-03 18:56:45 +00:00
|
|
|
/// [`#EXT-X-SESSION-DATA:<attribute-list>`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4.4)
|
2021-04-19 16:35:45 +00:00
|
|
|
/// The EXT-X-SESSION-DATA tag allows arbitrary session data to be carried
|
|
|
|
/// in a Master Playlist.
|
|
|
|
#[derive(Debug, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct SessionData {
|
|
|
|
pub data_id: String,
|
2021-04-19 16:35:45 +00:00
|
|
|
pub field: SessionDataField,
|
2016-06-03 18:56:45 +00:00
|
|
|
pub language: Option<String>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl SessionData {
|
2021-04-19 16:35:45 +00:00
|
|
|
pub fn from_hashmap(mut attrs: HashMap<String, String>) -> Result<SessionData, String> {
|
|
|
|
let data_id = match attrs.remove("DATA-ID") {
|
|
|
|
Some(data_id) => data_id,
|
|
|
|
None => return Err("EXT-X-SESSION-DATA field without DATA-ID".to_string())
|
|
|
|
};
|
|
|
|
|
|
|
|
let value = attrs.remove("VALUE");
|
|
|
|
let uri = attrs.remove("URI");
|
|
|
|
|
|
|
|
// SessionData must contain either a VALUE or a URI,
|
|
|
|
// but not both https://tools.ietf.org/html/rfc8216#section-4.3.4.4
|
|
|
|
let field = match (value, uri) {
|
|
|
|
(Some(value), None) => SessionDataField::Value(value),
|
|
|
|
(None, Some(uri)) => SessionDataField::Uri(uri),
|
|
|
|
(Some(_), Some(_)) => return Err(format!["EXT-X-SESSION-DATA tag {} contains both a value and a uri", data_id]),
|
|
|
|
(None, None) => return Err(format!["EXT-X-SESSION-DATA tag {} must contain either a value or a uri", data_id]),
|
|
|
|
};
|
|
|
|
|
|
|
|
Ok(SessionData {
|
|
|
|
data_id,
|
|
|
|
field,
|
2017-02-17 13:50:50 +00:00
|
|
|
language: attrs.remove("LANGUAGE"),
|
2021-04-19 16:35:45 +00:00
|
|
|
})
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
write!(w, "#EXT-X-SESSION-DATA:")?;
|
|
|
|
write!(w, "DATA-ID=\"{}\"", self.data_id)?;
|
2021-04-19 16:35:45 +00:00
|
|
|
match &self.field {
|
|
|
|
SessionDataField::Value(value) => write!(w, ",VALUE=\"{}\"", value)?,
|
|
|
|
SessionDataField::Uri(uri) => write!(w, ",URI=\"{}\"", uri)?,
|
|
|
|
};
|
2017-02-17 13:50:50 +00:00
|
|
|
write_some_attribute_quoted!(w, ",LANGUAGE", &self.language)?;
|
|
|
|
write!(w, "\n")
|
|
|
|
}
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
// Media Playlist
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
/// A [Media Playlist]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.3)
|
|
|
|
/// contains a list of Media Segments, which when played
|
|
|
|
/// sequentially will play the multimedia presentation.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct MediaPlaylist {
|
|
|
|
pub version: usize,
|
|
|
|
/// `#EXT-X-TARGETDURATION:<s>`
|
|
|
|
pub target_duration: f32,
|
|
|
|
/// `#EXT-X-MEDIA-SEQUENCE:<number>`
|
|
|
|
pub media_sequence: i32,
|
|
|
|
pub segments: Vec<MediaSegment>,
|
|
|
|
/// `#EXT-X-DISCONTINUITY-SEQUENCE:<number>`
|
|
|
|
pub discontinuity_sequence: i32,
|
|
|
|
/// `#EXT-X-ENDLIST`
|
|
|
|
pub end_list: bool,
|
|
|
|
/// `#EXT-X-PLAYLIST-TYPE`
|
2017-02-17 13:50:50 +00:00
|
|
|
pub playlist_type: Option<MediaPlaylistType>,
|
2016-06-03 18:56:45 +00:00
|
|
|
/// `#EXT-X-I-FRAMES-ONLY`
|
|
|
|
pub i_frames_only: bool,
|
|
|
|
/// `#EXT-X-START`
|
|
|
|
pub start: Option<Start>,
|
|
|
|
/// `#EXT-X-INDEPENDENT-SEGMENTS`
|
|
|
|
pub independent_segments: bool,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl MediaPlaylist {
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
writeln!(w, "{}" ,"#EXTM3U")?;
|
|
|
|
writeln!(w, "#EXT-X-VERSION:{}", self.version)?;
|
|
|
|
writeln!(w, "#EXT-X-TARGETDURATION:{}", self.target_duration)?;
|
|
|
|
|
2020-02-27 01:17:13 +00:00
|
|
|
if self.media_sequence != 0 {
|
2017-02-17 13:50:50 +00:00
|
|
|
writeln!(w, "#EXT-X-MEDIA-SEQUENCE:{}", self.media_sequence)?;
|
|
|
|
}
|
|
|
|
if self.discontinuity_sequence != 0 {
|
|
|
|
writeln!(w, "#EXT-X-DISCONTINUITY-SEQUENCE:{}", self.discontinuity_sequence)?;
|
|
|
|
}
|
2020-02-27 01:17:13 +00:00
|
|
|
if self.end_list {
|
|
|
|
writeln!(w, "#EXT-X-ENDLIST")?;
|
2017-02-17 13:50:50 +00:00
|
|
|
}
|
|
|
|
if let Some(ref v) = self.playlist_type {
|
|
|
|
writeln!(w, "#EXT-X-PLAYLIST-TYPE:{}", v)?;
|
|
|
|
}
|
2020-02-27 01:17:13 +00:00
|
|
|
if self.i_frames_only {
|
2017-02-17 13:50:50 +00:00
|
|
|
writeln!(w, "#EXT-X-I-FRAMES-ONLY")?;
|
|
|
|
}
|
|
|
|
if let Some(ref start) = self.start {
|
|
|
|
start.write_to(w)?;
|
|
|
|
}
|
2020-02-27 01:17:13 +00:00
|
|
|
if self.independent_segments {
|
2017-02-17 13:50:50 +00:00
|
|
|
writeln!(w, "#EXT-X-INDEPENDENT-SEGMENTS")?;
|
|
|
|
}
|
|
|
|
for segment in &self.segments {
|
|
|
|
segment.write_to(w)?;
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// [`#EXT-X-PLAYLIST-TYPE:<EVENT|VOD>`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.3.5)
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub enum MediaPlaylistType {
|
|
|
|
Event,
|
|
|
|
Vod,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl FromStr for MediaPlaylistType {
|
|
|
|
type Err = String;
|
|
|
|
|
|
|
|
fn from_str(s: &str) -> Result<MediaPlaylistType, String> {
|
|
|
|
match s {
|
|
|
|
"EVENT" => Ok(MediaPlaylistType::Event),
|
|
|
|
"VOD" => Ok(MediaPlaylistType::Vod),
|
|
|
|
_ => Err(format!("Unable to create MediaPlaylistType from {:?}", s)),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
impl fmt::Display for MediaPlaylistType {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
|
|
write!(f, "{}", match self {
|
|
|
|
&MediaPlaylistType::Event => "EVENT",
|
|
|
|
&MediaPlaylistType::Vod => "VOD",
|
|
|
|
})
|
2020-02-27 01:17:13 +00:00
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
}
|
|
|
|
|
2016-06-03 18:56:45 +00:00
|
|
|
impl Default for MediaPlaylistType {
|
|
|
|
fn default() -> MediaPlaylistType {
|
|
|
|
MediaPlaylistType::Event
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
// Media Segment
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
/// A [Media Segment](https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-3)
|
|
|
|
/// is specified by a URI and optionally a byte range.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct MediaSegment {
|
|
|
|
pub uri: String,
|
|
|
|
/// `#EXTINF:<duration>,[<title>]`
|
|
|
|
pub duration: f32,
|
|
|
|
/// `#EXTINF:<duration>,[<title>]`
|
|
|
|
pub title: Option<String>,
|
|
|
|
/// `#EXT-X-BYTERANGE:<n>[@<o>]`
|
|
|
|
pub byte_range: Option<ByteRange>,
|
|
|
|
/// `#EXT-X-DISCONTINUITY`
|
|
|
|
pub discontinuity: bool,
|
|
|
|
/// `#EXT-X-KEY:<attribute-list>`
|
|
|
|
pub key: Option<Key>,
|
|
|
|
/// `#EXT-X-MAP:<attribute-list>`
|
|
|
|
pub map: Option<Map>,
|
|
|
|
/// `#EXT-X-PROGRAM-DATE-TIME:<YYYY-MM-DDThh:mm:ssZ>`
|
|
|
|
pub program_date_time: Option<String>,
|
|
|
|
/// `#EXT-X-DATERANGE:<attribute-list>`
|
|
|
|
pub daterange: Option<String>,
|
2021-10-12 21:06:47 +00:00
|
|
|
/// `#EXT-`
|
|
|
|
pub unknown_tags: Vec<ExtTag>,
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
impl MediaSegment {
|
2017-02-17 13:50:50 +00:00
|
|
|
pub fn empty() -> MediaSegment {
|
2016-06-03 18:56:45 +00:00
|
|
|
Default::default()
|
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
|
|
|
|
if let Some(ref byte_range) = self.byte_range {
|
|
|
|
write!(w, "#EXT-X-BYTERANGE:")?;
|
|
|
|
byte_range.write_value_to(w)?;
|
|
|
|
write!(w, "\n")?;
|
|
|
|
}
|
2020-02-27 01:17:13 +00:00
|
|
|
if self.discontinuity {
|
|
|
|
writeln!(w, "{}", "#EXT-X-DISCONTINUITY")?;
|
2017-02-17 13:50:50 +00:00
|
|
|
}
|
|
|
|
if let Some(ref key) = self.key {
|
|
|
|
write!(w, "#EXT-X-KEY:")?;
|
|
|
|
key.write_attributes_to(w)?;
|
|
|
|
write!(w, "\n")?;
|
|
|
|
}
|
|
|
|
if let Some(ref map) = self.map {
|
|
|
|
write!(w, "#EXT-X-MAP:")?;
|
|
|
|
map.write_attributes_to(w)?;
|
|
|
|
write!(w, "\n")?;
|
|
|
|
}
|
|
|
|
if let Some(ref v) = self.program_date_time {
|
|
|
|
writeln!(w, "#EXT-X-PROGRAM-DATE-TIME:{}", v)?;
|
|
|
|
}
|
|
|
|
if let Some(ref v) = self.daterange {
|
|
|
|
writeln!(w, "#EXT-X-DATERANGE:{}", v)?;
|
2020-02-27 01:17:13 +00:00
|
|
|
}
|
2021-10-12 21:06:47 +00:00
|
|
|
for unknown_tag in &self.unknown_tags {
|
2021-10-14 19:21:03 +00:00
|
|
|
writeln!(w, "{}", unknown_tag)?;
|
2021-10-12 21:06:47 +00:00
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
write!(w, "#EXTINF:{},", self.duration)?;
|
|
|
|
|
|
|
|
if let Some(ref v) = self.title {
|
|
|
|
writeln!(w, "{}", v)?;
|
2020-03-07 05:52:22 +00:00
|
|
|
} else {
|
2020-03-22 04:23:03 +00:00
|
|
|
write!(w, "\n")?;
|
2017-02-17 13:50:50 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
writeln!(w, "{}", self.uri)
|
|
|
|
}
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// [`#EXT-X-KEY:<attribute-list>`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.2.4)
|
|
|
|
///
|
|
|
|
/// Media Segments MAY be encrypted. The EXT-X-KEY tag specifies how to
|
|
|
|
/// decrypt them. It applies to every Media Segment that appears between
|
|
|
|
/// it and the next EXT-X-KEY tag in the Playlist file with the same
|
|
|
|
/// KEYFORMAT attribute (or the end of the Playlist file). Two or more
|
|
|
|
/// EXT-X-KEY tags with different KEYFORMAT attributes MAY apply to the
|
|
|
|
/// same Media Segment if they ultimately produce the same decryption key.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct Key {
|
|
|
|
pub method: String,
|
|
|
|
pub uri: Option<String>,
|
|
|
|
pub iv: Option<String>,
|
|
|
|
pub keyformat: Option<String>,
|
|
|
|
pub keyformatversions: Option<String>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Key {
|
|
|
|
pub fn from_hashmap(mut attrs: HashMap<String, String>) -> Key {
|
|
|
|
Key {
|
|
|
|
method: attrs.remove("METHOD").unwrap_or_else(String::new),
|
|
|
|
uri: attrs.remove("URI"),
|
|
|
|
iv: attrs.remove("IV"),
|
|
|
|
keyformat: attrs.remove("KEYFORMAT"),
|
|
|
|
keyformatversions: attrs.remove("KEYFORMATVERSIONS"),
|
|
|
|
}
|
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
2020-02-27 01:17:13 +00:00
|
|
|
pub fn write_attributes_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
2017-02-17 13:50:50 +00:00
|
|
|
write!(w, "METHOD={}", self.method)?;
|
|
|
|
write_some_attribute_quoted!(w, ",URI", &self.uri)?;
|
|
|
|
write_some_attribute!(w, ",IV", &self.iv)?;
|
|
|
|
write_some_attribute!(w, ",KEYFORMAT", &self.keyformat)?;
|
|
|
|
write_some_attribute!(w, ",KEYFORMATVERSIONS", &self.keyformatversions)
|
|
|
|
}
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// [`#EXT-X-MAP:<attribute-list>`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.2.5)
|
|
|
|
///
|
|
|
|
/// The EXT-X-MAP tag specifies how to obtain the Media Initialization Section
|
|
|
|
/// [(Section 3)]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-3)
|
|
|
|
/// required to parse the applicable Media Segments.
|
|
|
|
/// It applies to every Media Segment that appears after it in the
|
|
|
|
/// Playlist until the next EXT-X-MAP tag or until the end of the
|
|
|
|
/// playlist.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct Map {
|
|
|
|
pub uri: String,
|
2017-02-17 13:50:50 +00:00
|
|
|
pub byte_range: Option<ByteRange>,
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
impl Map {
|
|
|
|
pub fn write_attributes_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
write!(w, "URI=\"{}\"", self.uri)?;
|
|
|
|
if let Some(ref byte_range) = self.byte_range {
|
|
|
|
write!(w, ",BYTERANGE=")?;
|
|
|
|
byte_range.write_value_to(w)?;
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-06-03 18:56:45 +00:00
|
|
|
/// [`#EXT-X-BYTERANGE:<n>[@<o>]`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.2.2)
|
|
|
|
///
|
|
|
|
/// The EXT-X-BYTERANGE tag indicates that a Media Segment is a sub-range
|
|
|
|
/// of the resource identified by its URI. It applies only to the next
|
|
|
|
/// URI line that follows it in the Playlist.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct ByteRange {
|
|
|
|
pub length: i32,
|
|
|
|
pub offset: Option<i32>,
|
|
|
|
}
|
|
|
|
|
2017-02-17 13:50:50 +00:00
|
|
|
impl ByteRange {
|
|
|
|
pub fn write_value_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
write!(w, "{}", self.length)?;
|
|
|
|
if let Some(offset) = self.offset {
|
|
|
|
write!(w, "@{}", offset)?;
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-06-03 18:56:45 +00:00
|
|
|
/// [`#EXT-X-DATERANGE:<attribute-list>`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.2.7)
|
|
|
|
///
|
|
|
|
/// The EXT-X-DATERANGE tag associates a Date Range (i.e. a range of time
|
|
|
|
/// defined by a starting and ending date) with a set of attribute /
|
|
|
|
/// value pairs.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct DateRange {
|
|
|
|
pub id: String,
|
|
|
|
pub class: Option<String>,
|
|
|
|
pub start_date: String,
|
|
|
|
pub end_date: Option<String>,
|
|
|
|
pub duration: Option<String>,
|
|
|
|
pub planned_duration: Option<String>,
|
|
|
|
pub x_prefixed: Option<String>, // X-<client-attribute>
|
|
|
|
pub end_on_next: bool,
|
|
|
|
}
|
|
|
|
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
// Rest
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
/// [`#EXT-X-START:<attribute-list>`]
|
|
|
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.5.2)
|
|
|
|
///
|
|
|
|
/// The EXT-X-START tag indicates a preferred point at which to start
|
|
|
|
/// playing a Playlist. By default, clients SHOULD start playback at
|
|
|
|
/// this point when beginning a playback session.
|
2018-11-26 01:43:42 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct Start {
|
|
|
|
pub time_offset: String,
|
|
|
|
pub precise: Option<String>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Start {
|
|
|
|
pub fn from_hashmap(mut attrs: HashMap<String, String>) -> Start {
|
|
|
|
Start {
|
|
|
|
time_offset: attrs.remove("TIME-OFFSET").unwrap_or_else(String::new),
|
|
|
|
precise: attrs.remove("PRECISE").or(Some("NO".to_string())),
|
|
|
|
}
|
|
|
|
}
|
2017-02-17 13:50:50 +00:00
|
|
|
|
|
|
|
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
|
|
|
write!(w, "#EXT-X-START:TIME-OFFSET={}", self.time_offset)?;
|
|
|
|
write_some_attribute!(w, ",PRECISE", &self.precise)?;
|
|
|
|
write!(w, "\n")
|
|
|
|
}
|
2016-06-03 18:56:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// A simple `#EXT-` tag
|
2021-03-11 20:15:22 +00:00
|
|
|
#[derive(Debug, Default, PartialEq, Clone)]
|
2016-06-03 18:56:45 +00:00
|
|
|
pub struct ExtTag {
|
|
|
|
pub tag: String,
|
2021-10-12 21:06:47 +00:00
|
|
|
pub rest: Option<String>,
|
2020-02-11 20:43:06 +00:00
|
|
|
}
|
2021-03-11 20:15:22 +00:00
|
|
|
|
2021-10-14 19:21:03 +00:00
|
|
|
impl Display for ExtTag {
|
|
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
|
|
write!(f, "#EXT-{}", self.tag)?;
|
|
|
|
if let Some(v) = &self.rest {
|
|
|
|
write!(f, ":{}", v)?;
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod test {
|
|
|
|
use super::*;
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn ext_tag_with_value_is_printable() {
|
|
|
|
let cue_out_tag = ExtTag {
|
|
|
|
tag: "X-CUE-OUT".into(),
|
|
|
|
rest: Some("DURATION=30".into()),
|
|
|
|
};
|
|
|
|
|
|
|
|
let mut output = Vec::new();
|
2021-10-14 19:35:35 +00:00
|
|
|
write!(output, "{}", cue_out_tag).unwrap();
|
2021-10-14 19:21:03 +00:00
|
|
|
|
|
|
|
assert_eq!(std::str::from_utf8(output.as_slice()).unwrap(), "#EXT-X-CUE-OUT:DURATION=30")
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn ext_tag_without_value_is_printable() {
|
|
|
|
let cue_in_tag = ExtTag {
|
|
|
|
tag: "X-CUE-IN".into(),
|
|
|
|
rest: None,
|
|
|
|
};
|
|
|
|
|
|
|
|
let mut output = Vec::new();
|
2021-10-14 19:35:35 +00:00
|
|
|
write!(output, "{}", cue_in_tag).unwrap();
|
2021-10-14 19:21:03 +00:00
|
|
|
|
|
|
|
assert_eq!(std::str::from_utf8(output.as_slice()).unwrap(), "#EXT-X-CUE-IN")
|
|
|
|
}
|
|
|
|
}
|