mirror of
https://github.com/shimunn/aria2xspf.git
synced 2023-11-17 09:27:54 +01:00
compiles
This commit is contained in:
parent
5cc317cd61
commit
33dce1c40a
71
src/main.rs
71
src/main.rs
@ -1,76 +1,89 @@
|
|||||||
#[macro_use]
|
#[macro_use]
|
||||||
extern crate derive_builder;
|
extern crate derive_builder;
|
||||||
|
|
||||||
use std::io::{self, Write};
|
|
||||||
use std::env;
|
use std::env;
|
||||||
use std::fs::File;
|
use std::fs::File;
|
||||||
|
use std::io::{self, Write};
|
||||||
use std::io::{BufRead, BufReader};
|
use std::io::{BufRead, BufReader};
|
||||||
|
|
||||||
use xml::writer::{EventWriter, EmitterConfig, XmlEvent, Result as XResult};
|
use xml::writer::{EmitterConfig, EventWriter, Result as XResult, XmlEvent};
|
||||||
|
|
||||||
#[derive(Builder, Debug)]
|
#[derive(Builder, Debug)]
|
||||||
struct Track {
|
struct Track {
|
||||||
url: String,
|
url: String,
|
||||||
title: String
|
title: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Track {
|
impl Track {
|
||||||
|
|
||||||
fn new(url: String, title: String) -> Track {
|
fn new(url: String, title: String) -> Track {
|
||||||
Track {
|
Track {
|
||||||
url: url,
|
url: url,
|
||||||
title: title
|
title: title,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fn write_xml<W: Write>(self, w: &mut EventWriter<W>) -> XResult<()> {
|
fn write_xml<W: Write>(self, w: &mut EventWriter<W>) -> XResult<()> {
|
||||||
let events: [XmlEvent; 8] = [
|
w.write(XmlEvent::start_element("track"))?;
|
||||||
XmlEvent::start_element("track".into()).into(),
|
w.write(XmlEvent::start_element("location"))?;
|
||||||
XmlEvent::start_element("location".into()).into(),
|
w.write(XmlEvent::characters(&self.url))?;
|
||||||
XmlEvent::characters(&self.url).into(),
|
w.write(XmlEvent::end_element())?;
|
||||||
XmlEvent::end_element().into(),
|
w.write(XmlEvent::start_element("title"))?;
|
||||||
XmlEvent::start_element("title".into()).into(),
|
w.write(XmlEvent::characters(&self.title))?;
|
||||||
XmlEvent::characters(&self.title).into(),
|
w.write(XmlEvent::end_element())?;
|
||||||
XmlEvent::end_element().into(),
|
w.write(XmlEvent::end_element())
|
||||||
XmlEvent::end_element().into()
|
|
||||||
];
|
|
||||||
for event in &events {
|
|
||||||
w.write(event.into()).into()?;
|
|
||||||
}
|
}
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
let file = File::open("VIDs").expect("Failed to open file!");
|
let file = File::open("VIDs").expect("Failed to open file!");
|
||||||
let mut out = File::create("VIDs.xspf").unwrap();
|
let mut out = File::create("VIDs.xspf").unwrap();
|
||||||
let mut writer = EmitterConfig::new().perform_indent(true).create_writer(&mut out);
|
let mut writer = EmitterConfig::new()
|
||||||
convert(tracks(BufReader::new(file).lines().map(|line| line.expect("Failed to read line"))), &mut writer).expect("Failed to parse");
|
.perform_indent(true)
|
||||||
|
.create_writer(&mut out);
|
||||||
|
convert(
|
||||||
|
tracks(
|
||||||
|
BufReader::new(file)
|
||||||
|
.lines()
|
||||||
|
.map(|line| line.expect("Failed to read line")),
|
||||||
|
),
|
||||||
|
&mut writer,
|
||||||
|
)
|
||||||
|
.expect("Failed to parse");
|
||||||
}
|
}
|
||||||
|
|
||||||
fn convert<I: Iterator<Item = Track>, W: Write>(tracks: I, w: &mut EventWriter<W>) -> XResult<()> {
|
fn convert<I: Iterator<Item = Track>, W: Write>(tracks: I, w: &mut EventWriter<W>) -> XResult<()> {
|
||||||
w.write(XmlEvent::start_element("playlist".into()).ns("", "http://xspf.org/ns/0/").ns("vlc", "http://www.videolan.org/vlc/playlist/ns/0/")).into()?;
|
w.write(
|
||||||
w.write(XmlEvent::start_element("trackList".into())).into()?;
|
XmlEvent::start_element("playlist")
|
||||||
|
.ns("", "http://xspf.org/ns/0/")
|
||||||
|
.ns("vlc", "http://www.videolan.org/vlc/playlist/ns/0/"),
|
||||||
|
)?;
|
||||||
|
w.write(XmlEvent::start_element("trackList"))?;
|
||||||
for track in tracks {
|
for track in tracks {
|
||||||
track.write_xml(w)?;
|
track.write_xml(w)?;
|
||||||
}
|
}
|
||||||
w.write(XmlEvent::end_element().into()).into()?;
|
w.write(XmlEvent::end_element())?;
|
||||||
w.write(XmlEvent::end_element().into()).into()?
|
w.write(XmlEvent::end_element())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn tracks<'a, I: Iterator<Item = String> + 'a>(aria: I) -> Box<Iterator<Item = Track> + 'a> {
|
fn tracks<'a, I: Iterator<Item = String> + 'a>(aria: I) -> Box<Iterator<Item = Track> + 'a> {
|
||||||
let tracks = aria.scan(TrackBuilder::default(), |mut builder, line| {
|
let tracks = aria
|
||||||
|
.scan(TrackBuilder::default(), |mut builder, line| {
|
||||||
if !builder.url.is_some() {
|
if !builder.url.is_some() {
|
||||||
builder.url(line);
|
builder.url(line);
|
||||||
} else {
|
} else {
|
||||||
if line.starts_with("\t") {
|
if line.starts_with("\t") {
|
||||||
builder.title(line.chars().skip_while(|c| c != &'=').skip(1).collect::<String>());
|
builder.title(
|
||||||
|
line.chars()
|
||||||
|
.skip_while(|c| c != &'=')
|
||||||
|
.skip(1)
|
||||||
|
.collect::<String>(),
|
||||||
|
);
|
||||||
let track = Some(builder.build().unwrap());
|
let track = Some(builder.build().unwrap());
|
||||||
*builder = TrackBuilder::default();
|
*builder = TrackBuilder::default();
|
||||||
return Some(track);
|
return Some(track);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Some(None)
|
Some(None)
|
||||||
}).filter_map(|track| track);
|
})
|
||||||
|
.filter_map(|track| track);
|
||||||
Box::new(tracks)
|
Box::new(tracks)
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user