#![allow(non_camel_case_types)]
#![allow(unused_imports)]
use std::fs::File;
use std::io::BufReader;
use std::path::Path;
use std::io::Read;
use std::io::Write;
use std::env::args;
use mkv::elements::parser::Parser;
use mkv::elements::parser::EventsHandler;
extern crate mkv;
extern crate log;
extern crate env_logger;
const BSIZE : usize = 4096;
fn main() {
env_logger::init().unwrap();
let reader : Box<dyn Read> = match args().len() {
1 => Box::new(std::io::stdin()),
2 => Box::new(File::open(Path::new(args().nth(1).unwrap().as_str())).expect("Failed to open the file")),
_ => panic!("Usage: dom_dumper [filename.mkv]")
};
let mut f = BufReader::new(reader);
let _element_logger = mkv::elements::parser::debug::DebugPrint::new(log::LogLevel::Info);
let mut dom_builder : mkv::elements::builder::Builder = Default::default();
{
let mut m = mkv::elements::parser::new();
loop {
let mut b = [0; BSIZE];
match f.read(&mut b) {
Ok(x) => match x {
0 => break,
x => m.feed_bytes(b.split_at(x).0, &mut dom_builder),
},
Err(e) => { println!("error reading: {}", e); break; },
}
}
}
println!("{:#?}", dom_builder.captured_elements());
}