use blend::Blend;
use libflate::gzip::Decoder;
use std::{
    env,
    fs::File,
    io::{self, BufWriter, Read, Write},
    path::{self, PathBuf},
};

fn print_blend(file_name: impl AsRef<str>) -> Result<(), io::Error> {
    let file_name = file_name.as_ref();
    let base_path = path::PathBuf::from(
        env::var_os("CARGO_MANIFEST_DIR").expect("could not find cargo manifest dir"),
    );

    let blend_path = base_path.join(format!("examples/blend_files/{}", file_name));
    let output_path = base_path.join(format!("examples/print_blend/output_{}.txt", file_name));

    println!("{}", blend_path.display());
    let mut file = File::open(blend_path)?;
    let mut data = Vec::new();
    file.read_to_end(&mut data)?;

    if data[0..7] != *b"BLENDER" {
        let mut decoder = Decoder::new(&data[..])?;
        let mut gzip_data = Vec::new();
        decoder.read_to_end(&mut gzip_data)?;

        data = gzip_data;
    }

    let blend = Blend::new(&data[..]).expect("error loading blend file");
    let mut output_path_without_file = PathBuf::from(&output_path);
    output_path_without_file.pop();
    std::fs::create_dir_all(&output_path_without_file)?;
    let mut buffer = BufWriter::new(File::create(output_path)?);

    for o in blend.root_instances() {
        write!(buffer, "{}", o)?;
    }

    writeln!(buffer)?;
    buffer.flush()?;

    println!("done: {}", file_name);

    Ok(())
}

pub fn main() -> Result<(), io::Error> {
    print_blend("2_80.blend")?;
    print_blend("2_90.blend")?;
    print_blend("3_0.blend")?;
    print_blend("3_5.blend")?;
    Ok(())
}