use std::env;
use std::error::Error;
use std::fs::File;
use std::io::Read;
use std::io::stdin;
use std::io::stdout;
use std::process::exit;
use pickled as pickle;
use serde_json as json;
fn main() -> Result<(), Box<dyn Error>> {
let args = env::args().collect::<Vec<_>>();
if args.len() < 2 {
println!("Usage: pickle (decode | transcode | to_json | from_json) [filename]");
println!();
println!("Input is either given file or stdin.");
println!("decode: decode and display pickle");
println!("transcode: decode and re-encode pickle");
println!("to_json: decode and jsonify pickle");
println!("from_json: encode pickle from json");
exit(1);
}
let reader: Box<dyn Read> = if args.len() == 3 {
Box::new(File::open(&args[2])?)
} else {
Box::new(stdin())
};
match &*args[1] {
"decode" => {
let decoded: pickle::Value = pickle::value_from_reader(reader, Default::default())?;
println!("{decoded:#?}");
}
"transcode" => {
let decoded: pickle::Value = pickle::value_from_reader(reader, Default::default())?;
pickle::value_to_writer(&mut stdout(), &decoded, Default::default())?;
}
"to_json" => {
let decoded: json::Value = pickle::from_reader(reader, Default::default())?;
println!("{decoded:#?}");
}
"from_json" => {
let decoded: json::Value = json::from_reader(reader)?;
pickle::to_writer(
&mut stdout(),
&decoded,
pickle::SerOptions::new().proto_v2(),
)?;
}
_ => {
println!("No such subcommand.");
exit(1);
}
}
Ok(())
}