//! Minimal CLI tool to generate a m3u8 playlist from a MPS file. //! This does not support playback, so it can run on any platform with a filesystem. //! Use `mps-m3u8 --help` for usage instructions. //! mod cli; use std::fs::File; use std::path::Path; use std::io::{BufReader, BufWriter, Cursor}; use m3u8_rs::{MediaPlaylist, MediaSegment}; use mps_interpreter::MpsRunner; fn main() { let args = cli::parse(); let out_path = Path::new(&args.playlist); let mut out_file = BufWriter::new(File::create(out_path).expect("Invalid output file")); let mut playlist = MediaPlaylist { version: 6, ..Default::default() }; let mut skipped_count = 0; if args.raw { println!("Executing: {}", &args.input); let in_file = Cursor::new(&args.input); let runner = MpsRunner::with_stream(in_file); for item in runner { match item { Ok(music) => { if let Some(filename) = music.field("filename").and_then(|x| x.to_owned().to_str()) { playlist.segments.push(MediaSegment { uri: filename, title: music.field("title").and_then(|x| x.to_owned().to_str()), ..Default::default() }); } else { skipped_count += 1; } }, Err(e) => eprintln!("{}", e), } } } else { let in_path = Path::new(&args.input); let in_file = BufReader::new(File::open(in_path).expect("Invalid/missing input file")); let runner = MpsRunner::with_stream(in_file); for item in runner { match item { Ok(music) => { if let Some(filename) = music.field("filename").and_then(|x| x.to_owned().to_str()) { playlist.segments.push(MediaSegment { uri: filename, title: music.field("title").and_then(|x| x.to_owned().to_str()), ..Default::default() }); } else { skipped_count += 1; } }, Err(e) => eprintln!("{}", e), } } } if skipped_count != 0 { eprintln!("Skipped {} items due to missing `filename` field", skipped_count); } if let Err(e) = playlist.write_to(&mut out_file) { eprintln!("Playlist save error: {}", e); } }