1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137

//! This is a very simple url extractor for different kinds of playlist formats: M3U, PLS, ASX, XSPF
//!
//! It is not optimized yet and does create a lot of strings on the way.

extern crate quick_xml;

mod pls;
mod m3u;
mod asx;
mod xspf;

use std::collections::HashSet;
use std::error::Error;

/// Decode playlist content string. It checks for M3U, PLS, XSPF and ASX content in the string.
/// # Example
/// ```rust
/// let list = playlist_decoder::decode(r##"<?xml version="1.0" encoding="UTF-8"?>
///    <playlist version="1" xmlns="http://xspf.org/ns/0/">
///      <trackList>
///        <track>
///          <title>Nobody Move, Nobody Get Hurt</title>
///          <creator>We Are Scientists</creator>
///          <location>file:///mp3s/titel_1.mp3</location>
///        </track>
///        <track>
///          <title>See The World</title>
///          <creator>The Kooks</creator>
///          <location>http://www.example.org/musik/world.ogg</location>
///        </track>
///      </trackList>
///    </playlist>"##).unwrap();
/// assert!(list.len() == 2, "Did not find 2 urls in example");
/// for item in list {
///     println!("{:?}", item);
/// }
/// ```
/// # Arguments
/// * `content` - A string slice containing a playlist
pub fn decode(content: &str) -> Result<Vec<String>, Box<dyn Error>> {
    let mut set = HashSet::new();
    let content_small = content.to_lowercase();
    match content_small.find("<playlist"){
        Some(_)=>{
            let xspf_items = xspf::decode(content)?;
            for item in xspf_items {
                if item.url != "" {
                    set.insert(item.url);
                }
                if item.identifier != "" {
                    set.insert(item.identifier);
                }
            }
        }
        None =>{
            match content_small.find("<asx"){
                Some(_)=>{
                    let pls_items = asx::decode(content)?;
                    for item in pls_items {
                        set.insert(item.url);
                    }
                }
                None =>{
                    match content_small.find("[playlist]"){
                        Some(_) => {
                            let pls_items = pls::decode(content);
                            for item in pls_items {
                                set.insert(item.url);
                            }
                        }
                        None => {
                            let m3u_items = m3u::decode(content);
                            for item in m3u_items {
                                set.insert(item.url);
                            }
                        }
                    }
                }
            }
        }
    }
    let v: Vec<String> = set.into_iter().collect();
    Ok(v)
}

pub fn is_content_hls(content: &str) -> bool {
    if content.contains("EXT-X-STREAM-INF"){
        return true;
    }
    if content.contains("EXT-X-TARGETDURATION"){
        return true;
    }
    return false;
}

#[cfg(test)]
mod tests {
    #[test]
    fn m3u() {
        let items = crate::m3u::decode("http://this.is.an.example");
        assert!(items.len() == 1);
        assert!(items[0].url == "http://this.is.an.example");
    }

    #[test]
    fn pls() {
        let items = crate::pls::decode("[playlist]
File1=http://this.is.an.example
Title1=mytitle
        ");
        assert!(items.len() == 1);
        assert!(items[0].url == "http://this.is.an.example");
        assert!(items[0].title == "mytitle");
    }

    #[test]
    fn pls2() {
        let items = crate::pls::decode("[playlist]
File1=http://this.is.an.example
Title=mytitle
        ");
        assert!(items.len() == 1);
        assert!(items[0].url == "http://this.is.an.example");
        assert!(items[0].title == "mytitle");
    }

    #[test]
    fn pls3() {
        let items = crate::pls::decode("[Playlist]
File1=http://this.is.an.example
Title=mytitle
        ");
        assert!(items.len() == 1);
        assert!(items[0].url == "http://this.is.an.example");
        assert!(items[0].title == "mytitle");
    }
}