pub mod deserializer {
    use log::info;
    use crate::artworkdb::aobjects::{ADataSet, ADatabase, AImageItem, AImageName, SecondTypeTag, ThirdTypeTag};
    use crate::artworkdb::objects::{ChunkHeader, ChunkType, DataSet, Database, ImageFile, ImageItem, ImageName, LocationTag};


    const SAMPLE_DB: [u8; 2144] = [0x6D,0x68,0x66,0x64,0x84,0x00,0x00,0x00,0x60,0x08,0x00,0x00,0x00,0x00,0x00,0x00,0x06,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x66,0x00,0x00,0x00,0xE9,0x15,0xB6,0x2C,0x03,0x42,0x6A,0xDE,0x9F,0xBE,0x2C,0x6B,0x2F,0xEE,0x85,0x72,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xC5,0xA6,0xEC,0x35,0xAC,0xEB,0x95,0xC5,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x73,0x64,0x60,0x00,0x00,0x00,0x6C,0x05,0x00,0x00,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6C,0x69,0x5C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x69,0x69,0x98,0x00,0x00,0x00,0x58,0x02,0x00,0x00,0x03,0x00,0x00,0x00,0x64,0x00,0x00,0x00,0xDC,0x41,0xBD,0xF9,0xDB,0x98,0x58,0xFC,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x5D,0x4F,0x11,0x00,0x00,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xF8,0x7F,0x00,0x00,0x00,0x00,0x00,0x00,0xF8,0x7F,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0xA4,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6E,0x69,0x4C,0x00,0x00,0x00,0x8C,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x04,0x04,0x00,0x00,0x00,0x00,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x64,0x00,0x64,0x00,0x00,0x00,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x40,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x1C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x3A,0x00,0x46,0x00,0x31,0x00,0x30,0x00,0x32,0x00,0x38,0x00,0x5F,0x00,0x31,0x00,0x2E,0x00,0x69,0x00,0x74,0x00,0x68,0x00,0x6D,0x00,0x62,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0xA4,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6E,0x69,0x4C,0x00,0x00,0x00,0x8C,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x05,0x04,0x00,0x00,0x00,0x00,0x00,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0xC8,0x00,0xC8,0x00,0x00,0x00,0x00,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x40,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x1C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x3A,0x00,0x46,0x00,0x31,0x00,0x30,0x00,0x32,0x00,0x39,0x00,0x5F,0x00,0x31,0x00,0x2E,0x00,0x69,0x00,0x74,0x00,0x68,0x00,0x6D,0x00,0x62,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x78,0x00,0x00,0x00,0x06,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x61,0x66,0x60,0x00,0x00,0x00,0x3C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x69,0x69,0x98,0x00,0x00,0x00,0x58,0x02,0x00,0x00,0x03,0x00,0x00,0x00,0x65,0x00,0x00,0x00,0x1F,0xBE,0xB3,0xD9,0x94,0x3F,0xE5,0x5D,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xBC,0x89,0x03,0x00,0x00,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xF8,0x7F,0x00,0x00,0x00,0x00,0x00,0x00,0xF8,0x7F,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0xA4,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6E,0x69,0x4C,0x00,0x00,0x00,0x8C,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x04,0x04,0x00,0x00,0x20,0x4E,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x64,0x00,0x64,0x00,0x00,0x00,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x40,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x1C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x3A,0x00,0x46,0x00,0x31,0x00,0x30,0x00,0x32,0x00,0x38,0x00,0x5F,0x00,0x31,0x00,0x2E,0x00,0x69,0x00,0x74,0x00,0x68,0x00,0x6D,0x00,0x62,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0xA4,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6E,0x69,0x4C,0x00,0x00,0x00,0x8C,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x05,0x04,0x00,0x00,0x80,0x38,0x01,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0xC8,0x00,0xC8,0x00,0x00,0x00,0x00,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x40,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x1C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x3A,0x00,0x46,0x00,0x31,0x00,0x30,0x00,0x32,0x00,0x39,0x00,0x5F,0x00,0x31,0x00,0x2E,0x00,0x69,0x00,0x74,0x00,0x68,0x00,0x6D,0x00,0x62,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x78,0x00,0x00,0x00,0x06,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x61,0x66,0x60,0x00,0x00,0x00,0x3C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x73,0x64,0x60,0x00,0x00,0x00,0xBC,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6C,0x61,0x5C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x73,0x64,0x60,0x00,0x00,0x00,0xB4,0x01,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6C,0x66,0x5C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x69,0x66,0x7C,0x00,0x00,0x00,0x7C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x04,0x04,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x69,0x66,0x7C,0x00,0x00,0x00,0x7C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x05,0x04,0x00,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00];

    /*
        File system:
        /iPod_Control/Artwork/ArtworkDB
        /iPod_Control/Artwork/[any].ithmb
     */


    enum ChunkState {
        Header,
        Data
    }

    pub fn new_db() -> ADatabase {
        let mut adb = parse_bytes(&SAMPLE_DB[..]);
        let data_set = adb.find_dataset(3);
        data_set.child = Vec::new();
        let data_set = adb.find_dataset(1);
        data_set.child = Vec::new();
        adb
    }

    pub fn parse_bytes(data: &[u8]) -> ADatabase {
        let mut adb = ADatabase {data: None, header: None, children: Vec::new() };
        let mut state = ChunkState::Header;
        let mut chunk_header: Option<ChunkHeader> = None;
        let mut last_type: u32 = 0;
        let mut i = 0;
        while i < data.len() {
            state = match state {
                ChunkState::Header => {
                    if i + 12 >= data.len() { break; }
                    chunk_header = Some(bincode::deserialize(&data[i..i + 12]).unwrap());
                    i += 12;
                    ChunkState::Data
                },
                ChunkState::Data => {
                    let mut u = 0;
                    let header = chunk_header.unwrap();
                    match ChunkType::from(header.chunk_type) {
                        ChunkType::ArtworkDB => {
                            u = usize::try_from(header.end_of_chunk).unwrap() - 12;
                            let db: Database = bincode::deserialize(&data[i..i+u]).unwrap();
                            info!("val: {:?}", db);
                            adb.data = Some(db);
                            adb.header = Some(header);
                        },
                        ChunkType::DataSet => {
                            u = usize::try_from(header.end_of_chunk).unwrap() - 12;
                            let ds: DataSet = bincode::deserialize(&data[i..i + u]).unwrap();
                            last_type = ds.data_type;
                            adb.children.push(ADataSet { header, data: ds, child: Vec::new()});
                        },
                        ChunkType::ImageList | ChunkType::AlbumList | ChunkType::FileList => {
                            u = usize::try_from(header.end_of_chunk).unwrap() - 12;
                        },
                        ChunkType::ImageItem => {
                            info!("ImageItem tag");
                            u = usize::try_from(header.end_of_chunk).unwrap() - 12;
                            let ai: ImageItem = bincode::deserialize(&data[i..i+u]).unwrap();
                            let images = &mut adb.find_dataset(last_type).child;
                            images.push(AImageItem { data: Some(ai), tag: Vec::new(), file: None });
                        },
                        ChunkType::FileImage => {
                            u = usize::try_from(header.end_of_chunk).unwrap() - 12;
                            let ai: ImageFile = bincode::deserialize(&data[i..i+u]).unwrap();
                            let images = &mut adb.find_dataset(last_type).child;
                            images.push(AImageItem { data: None, tag: Vec::new(), file: Some(ai) });
                        },
                        ChunkType::LocationTag => {
                            u = usize::try_from(header.end_of_chunk).unwrap() - 12;
                            let ds: LocationTag = bincode::deserialize(&data[i..i + u]).unwrap();
                            let images = &mut adb.find_dataset(last_type).child;
                            let mut str = None;
                            match ds.tag_type {
                                3 => {
                                    let mut bytes = Vec::new();
                                    let str_end = u32::from_le_bytes(data[i+12..i+16].try_into().unwrap()) as usize;
                                    let mut h = i+24;
                                    u += 12 + str_end;
                                    while h < i+24+str_end {
                                        bytes.push(u16::from_le_bytes(data[h..h+2].try_into().unwrap()));
                                        h+=2;
                                    }
                                    let g = String::from_utf16(&bytes).unwrap();
                                    info!("str: {}", g);
                                    str = Some(g);
                                    images.last_mut().unwrap().tag.last_mut().unwrap().child.as_mut().unwrap().tag = Some(ThirdTypeTag { data: ds, str });
                                },
                                2 => {
                                    images.last_mut().unwrap().tag.push(SecondTypeTag { data: ds, str, child: None });
                                },
                                _ => { u = usize::try_from(header.children_count).unwrap() - 12; }
                            }
                        },
                        ChunkType::ImageName => {
                            info!("ImageName tag");
                            u = usize::try_from(header.end_of_chunk).unwrap() - 12;
                            let ds: ImageName = bincode::deserialize(&data[i..i + u]).unwrap();
                            let images = &mut adb.find_dataset(last_type).child;
                            images.last_mut().unwrap().tag.last_mut().unwrap().child = Some(AImageName { iname: ds, tag: None });
                        },
                        _ => { u = 1; info!("Unknown stuff happened {:X?}", header.chunk_type.to_vec()); }
                    }
                    i += u;
                    chunk_header = None;
                    ChunkState::Header
                }
            }
        }
        adb
    }
}

pub mod serializer {
    use log::info;
    use crate::artworkdb::aobjects::ADatabase;
    use crate::artworkdb::objects::{ChunkHeader, ChunkType};

    pub fn to_bytes(adb: ADatabase) -> Vec<u8> {
        let mut bytes: Vec<u8> = Vec::new();
        for i in 0..(adb.children.len()) {
            let data_set = adb.children.get(i).unwrap();
            let mut entry_bytes = Vec::new();

            match data_set.data.data_type {
                1 => {
                    entry_bytes.append(&mut generate_header_raw(ChunkType::ImageList, 80, data_set.child.len()));
                    entry_bytes.append(&mut [0; 80].to_vec());
                }, // Image List
                2 => {
                    entry_bytes.append(&mut generate_header_raw(ChunkType::AlbumList, 80, 0));
                    entry_bytes.append(&mut [0; 80].to_vec());
                }, // Album list
                3 => {
                    entry_bytes.append(&mut generate_header_raw(ChunkType::FileList, 80, data_set.child.len()));
                    entry_bytes.append(&mut [0; 80].to_vec());
                }, // File list
                _ => { info!("Unknown data_set type!"); }
            }

            for img in data_set.child.iter() {
                let mut args = Vec::new();
                if img.data.is_some() {
                    for o in &img.tag {
                        let mut inameb = Vec::new();
                        if let Some(name) = &o.child {
                            info!("Serializing: {:#?}", name);
                            let mut nb = Vec::new();

                            let arg = name.tag.as_ref().unwrap();
                            let mut str_b = string_to_ipod16(arg.str.as_ref().unwrap());
                            str_b = [3u32.to_le_bytes().to_vec(), 0u32.to_le_bytes().to_vec(), 0u32.to_le_bytes().to_vec(), (str_b.len() as u32).to_le_bytes().to_vec(), 2u32.to_le_bytes().to_vec(), 0u32.to_le_bytes().to_vec(), str_b ].concat();
                            nb.append(&mut generate_header(ChunkType::LocationTag, 12, str_b.len()));
                            nb.append(&mut str_b);

                            inameb.append(&mut generate_header(ChunkType::ImageName, 64, nb.len()));
                            inameb.append(&mut bincode::serialize(&name.iname).unwrap());
                            inameb.append(&mut [0; 32].to_vec());
                            inameb.append(&mut nb);
                        }

                        let mut data = [2u32.to_le_bytes(), 0u32.to_le_bytes(), 0u32.to_le_bytes()].concat();
                        args.append(&mut generate_header_raw(ChunkType::LocationTag, 12, inameb.len()));
                        args.append(&mut data);
                        args.append(&mut inameb);
                    }
                    let mut item = Vec::new();
                    item.append(&mut generate_header(ChunkType::ImageItem, 140, args.len()));
                    item.append(&mut bincode::serialize(img.data.as_ref().unwrap()).unwrap());
                    item.append(&mut [0; 100].to_vec());
                    entry_bytes.append(&mut item);
                }

                if img.file.is_some() {
                    entry_bytes.append(&mut generate_header(ChunkType::FileImage, 112, 0));
                    entry_bytes.append(&mut bincode::serialize(img.file.as_ref().unwrap()).unwrap());
                    entry_bytes.append(&mut [0; 100].to_vec());
                }

                entry_bytes.append(&mut args);
            }

            bytes.append(&mut generate_header(ChunkType::DataSet, 84, entry_bytes.len()));
            bytes.append(&mut bincode::serialize(&data_set.data).unwrap());
            bytes.append(&mut [0; 80].to_vec());

            bytes.append(&mut entry_bytes);
        }

        let sdb = bincode::serialize(&adb.data.as_ref().unwrap()).unwrap();
        bytes = [generate_header(ChunkType::ArtworkDB, 56 + 64, bytes.len()), sdb, [0; 64].to_vec(), bytes].concat();
        bytes
    }

    fn string_to_ipod16(str: &str) -> Vec<u8> {
        str.encode_utf16().flat_map(|f| [f as u8, (f >> 8) as u8]).collect()
    }

    fn generate_header(ct: ChunkType, header_size: usize, data_len: usize) -> Vec<u8> {
        let header_size = 12 + header_size as u32;
        let header = ChunkHeader{ chunk_type: ct.into(), end_of_chunk: header_size, children_count: header_size + data_len as u32};
        bincode::serialize(&header).unwrap()
    }

    fn generate_header_raw(ct: ChunkType, header_size: usize, child_cnt: usize) -> Vec<u8> {
        let header_size = 12 + header_size as u32;
        let header = ChunkHeader{ chunk_type: ct.into(), end_of_chunk: header_size, children_count: child_cnt as u32};
        bincode::serialize(&header).unwrap()
    }
}

pub mod aobjects {
    use log::info;
    use crate::artworkdb::objects::{ChunkHeader, DataSet, Database, ImageFile, ImageItem, ImageName, LocationTag};

    #[derive(Debug, serde::Serialize)]
    pub struct ADatabase {
        pub header: Option<ChunkHeader>,
        pub data: Option<Database>,
        pub children: Vec<ADataSet>
    }

    impl ADatabase {
        pub(crate) fn find_dataset(&mut self, p0: u32) -> &mut ADataSet {
            self.children.iter_mut().find(|d| d.data.data_type == p0).unwrap()
        }

        fn get_last_cor_id(&mut self) -> u32{
            let files = self.find_dataset(3);
            let image_files = &mut files.child;
            image_files.iter().map(|p| p.file.as_ref().unwrap().correlation_id).max().unwrap_or(1027)
        }

        pub fn add_images(&mut self, small_image_path: &str, large_image_path: &str, song_dbid: u64) -> u32 {
            let last_cor_id = self.get_last_cor_id();
            let ds = self.find_dataset(1);
            let image_items = &mut ds.child;
            let new_id = image_items.last().map_or(100, |p| p.data.as_ref().unwrap().id + 1);
            let image_item = crate::artworkdb::objects::ImageItem {
                number_of_children: 2,
                id: new_id,
                song_dbid,
                unknown4: 0,
                rating: 0,
                unknown6: 0,
                original_date: 0,
                digitized_date: 0,
                source_image_size: 0
            };
            
            let small_tag = SecondTypeTag {
                data: LocationTag {
                    tag_type: 2,
                    unk1: 0,
                    unk2: 0
                },
                str: None,
                child: Some(
                    AImageName {
                        iname: ImageName {
                            number_of_children: 1,
                            correlation_id: last_cor_id + 1,
                            ithmb_offset: 0,
                            image_size: 20000,
                            vertical_padding: 0,
                            horizontal_padding: 0,
                            image_height: 100,
                            image_width: 100,
                            unknown: 0,
                            image_size_n: 20000,
                        },
                        tag: Some(
                            ThirdTypeTag {
                                data: LocationTag {
                                    tag_type: 3,
                                    unk1: 0,
                                    unk2: 0,
                                },
                                str: Some(
                                    [":F", &(last_cor_id + 1).to_string(), "_1.ithmb"].concat(),
                                ),
                            }
                        ),
                    }
                ),
            };

            let large_tag = SecondTypeTag {
                data: LocationTag {
                    tag_type: 2,
                    unk1: 0,
                    unk2: 0
                },
                str: None,
                child: Some(
                    AImageName {
                        iname: ImageName {
                            number_of_children: 1,
                            correlation_id: last_cor_id + 2,
                            ithmb_offset: 0,
                            image_size: 80000,
                            vertical_padding: 0,
                            horizontal_padding: 0,
                            image_height: 200,
                            image_width: 200,
                            unknown: 0,
                            image_size_n: 80000,
                        },
                        tag: Some(
                            ThirdTypeTag {
                                data: LocationTag {
                                    tag_type: 3,
                                    unk1: 0,
                                    unk2: 0,
                                },
                                str: Some(
                                    [":F", &(last_cor_id + 2).to_string(), "_1.ithmb"].concat(),
                                ),
                            }
                        ),
                    }
                ),
            };
            
            
            image_items.push(AImageItem {
                tag: vec![small_tag, large_tag],
                data: Some(image_item),
                file: None,
            });
            
            let files = self.find_dataset(3);
            let image_files = &mut files.child;
            
            image_files.append(&mut vec![
                AImageItem {
                    tag: Vec::new(),
                    data: None,
                    file: Some(
                        ImageFile {
                            unknown1: 0,
                            correlation_id: last_cor_id + 1,
                            image_size: 20000,
                        },
                    ),
                },
                AImageItem {
                    tag: Vec::new(),
                    data: None,
                    file: Some(
                        ImageFile {
                            unknown1: 0,
                            correlation_id: last_cor_id + 2,
                            image_size: 80000,
                        },
                    ),
                },
            ]);
            self.data.as_mut().unwrap().next_id_for_mhii = new_id + 1;
            new_id
        }
    }

    #[derive(Debug, serde::Serialize)]
    pub struct ADataSet {
        pub header: ChunkHeader,
        pub data: DataSet,
        pub child: Vec<AImageItem>
    }

    #[derive(Debug, serde::Serialize)]
    pub struct AImageItem {
        pub tag: Vec<SecondTypeTag>,
        pub data: Option<ImageItem>,
        pub file: Option<ImageFile>,
    }
    
    #[derive(Debug, serde::Serialize)]
    pub struct AImageName {
        pub iname: ImageName,
        pub tag: Option<ThirdTypeTag>,
    }

    #[derive(Debug, serde::Serialize)]
    pub struct SecondTypeTag {
        pub data: LocationTag,
        pub str: Option<String>,
        pub child: Option<AImageName>
    }

    #[derive(Debug, serde::Serialize)]
    pub struct ThirdTypeTag {
        pub data: LocationTag,
        pub str: Option<String>
    }
}

pub mod objects {
    use serde::{Deserialize, Serialize};

    pub enum ChunkType {
        ArtworkDB,
        DataSet,
        ImageList,
        AlbumList,
        FileList,
        ImageItem,
        LocationTag,
        ImageName,
        FileImage,
        Unknown,
    }

    impl From<[u8; 4]> for ChunkType {
        fn from(value: [u8; 4]) -> Self {
            match value {
                [0x6D, 0x68, 0x66, 0x64] => ChunkType::ArtworkDB,
                [0x6D, 0x68, 0x73, 0x64] => ChunkType::DataSet,
                [0x6D, 0x68, 0x6C, 0x69] => ChunkType::ImageList,
                [0x6D, 0x68, 0x69, 0x69] => ChunkType::ImageItem,
                [0x6D, 0x68, 0x6F, 0x64] => ChunkType::LocationTag,
                [0x6D, 0x68, 0x6E, 0x69] => ChunkType::ImageName,
                [0x6D, 0x68, 0x6C, 0x61] => ChunkType::AlbumList,
                [0x6D, 0x68, 0x6C, 0x66] => ChunkType::FileList,
                [0x6D, 0x68, 0x69, 0x66] => ChunkType::FileImage,
                _ => ChunkType::Unknown,
            }
        }
    }

    impl From<ChunkType> for [u8; 4] {
        fn from(value: ChunkType) -> Self {
            match value {
                ChunkType::ArtworkDB => [0x6D, 0x68, 0x66, 0x64],
                ChunkType::DataSet => [0x6D, 0x68, 0x73, 0x64],
                ChunkType::Unknown => [0x00, 0x00, 0x00, 0x00],
                ChunkType::ImageList => [0x6D, 0x68, 0x6C, 0x69],
                ChunkType::ImageItem => [0x6D, 0x68, 0x69, 0x69],
                ChunkType::LocationTag => [0x6D, 0x68, 0x6F, 0x64],
                ChunkType::ImageName => [0x6D, 0x68, 0x6E, 0x69],
                ChunkType::AlbumList => [0x6D, 0x68, 0x6C, 0x61],
                ChunkType::FileList => [0x6D, 0x68, 0x6C, 0x66],
                ChunkType::FileImage => [0x6D, 0x68, 0x69, 0x66],
            }
        }
    }

    #[derive(Serialize, Deserialize, PartialEq, Debug, Clone, Copy)]
    pub struct ChunkHeader {
        pub chunk_type: [u8; 4],
        pub end_of_chunk: u32,
        pub children_count: u32,
    }

    #[derive(Serialize, Deserialize, PartialEq, Debug)]
    pub struct Database {
        unknown1: u32,
        unknown2: u32,
        number_of_children: u32,
        unknown3: u32,
        pub next_id_for_mhii: u32,
        unknown5: u64,
        unknown6: u64,
        unknown7: u32,
        unknown8: u32,
        unknown9: u32,
        unknown10: u64
    }

    #[derive(Serialize, Deserialize, PartialEq, Debug, Clone, Copy)]
    pub struct DataSet {
        pub data_type: u32,
    }

    #[derive(Serialize, Deserialize, PartialEq, Debug, Clone, Copy)]
    pub struct LocationTag {
        pub tag_type: u32,
        pub unk1: u32,
        pub unk2: u32,
    }

    #[derive(Serialize, Deserialize, PartialEq, Debug)]
    pub struct ImageItem {
        pub number_of_children: u32,
        pub id: u32,
        pub song_dbid: u64,
        pub unknown4: u32,
        pub rating: u32,
        pub unknown6: u32,
        pub original_date: u32,
        pub digitized_date: u32,
        pub source_image_size: u32
    }

    #[derive(Serialize, Deserialize, PartialEq, Debug)]
    pub struct ImageFile {
        pub(crate) unknown1: u32,
        pub(crate) correlation_id: u32,
        pub(crate) image_size: u32
    }

    #[derive(Serialize, Deserialize, PartialEq, Debug)]
    pub struct ImageName {
        pub(crate) number_of_children: u32,
        pub(crate) correlation_id: u32,
        pub(crate) ithmb_offset: u32,
        pub(crate) image_size: u32,
        pub(crate) vertical_padding: u16,
        pub(crate) horizontal_padding: u16,
        pub(crate) image_height: u16,
        pub(crate) image_width: u16,
        pub(crate) unknown: u32,
        pub(crate) image_size_n: u32,
    }
}