#![feature(array_chunks)]
use anyhow::anyhow;
use std::{
    env::args,
    fs::{File, create_dir_all},
    io::{BufReader, BufWriter, Seek, SeekFrom, Write},
};
use unity_tools::{
    classes::mesh::{Mesh, VertexDataChannel},
    object::{parser::FromValue, read::read_value},
    serialized_file::read_serialized_file,
    unityfs::UnityFS,
};

fn main() -> anyhow::Result<()> {
    env_logger::init_from_env("LOG");
    let file = || BufReader::new(File::open(args().nth(1).unwrap()).unwrap());
    let mut fs = UnityFS::open(file())?;
    // let mut fs2 = UnityFS::open(file())?;

    let mut i = 0;
    create_dir_all("/tmp/a").unwrap();

    let cabfile = fs
        .find_main_file()
        .ok_or(anyhow!("no CAB file found"))?
        .to_owned();

    let mut cab = fs.read(&cabfile)?;
    let file = read_serialized_file(&mut cab)?;
    for ob in file.objects {
        cab.seek(SeekFrom::Start(ob.data_offset))?;
        let typetree = if ob.type_id < 0 {
            unimplemented!()
        } else {
            &file.types[ob.type_id as usize]
        };
        if let Some(typetree) = &typetree.type_tree {
            if typetree.type_string != "Mesh" {
                continue;
            }
            let value = read_value(typetree, file.endianness, &mut cab)?;
            let mesh = Mesh::from_value(value).unwrap();
            // if mesh.stream_data.len() == 0 {
            //     let ress = ress.as_mut().unwrap();
            //     ress.seek(SeekFrom::Start(mesh.stream_data.offset))?;
            //     ress.by_ref()
            //         .take(mesh.stream_data.size as u64)
            //         .read_to_end(&mut mesh.)?;
            // }
            let mut obj = BufWriter::new(File::create(format!(
                "/tmp/a/{}_{i}.obj",
                mesh.name.replace("/", "-").replace(".", "-")
            ))?);

            let (pos_dims, positions) = mesh
                .vertex_data
                .read_channel(VertexDataChannel::Position)
                .unwrap();
            assert_eq!(pos_dims, 3);

            for [x, y, z] in positions.array_chunks() {
                writeln!(obj, "v {x} {y} {z}")?;
            }
            for [a, b, c] in mesh.read_indecies() {
                writeln!(obj, "f {} {} {}", a + 1, b + 1, c + 1)?;
            }
            i += 1;
        }
    }

    Ok(())
}