use crate::{FileDataIdx, FileInfoIdx, FileInfoIndiceIdx, FilePathIdx, Hash40, InfoToDataIdx};
use modular_bitfield::prelude::*;
use std::convert::TryFrom;

use binrw::{binread, io::*, BinRead, BinResult, Endian};

#[derive(BinRead, Debug, Clone, Copy)]
#[br(magic = 0x10_u32)]
pub struct CompTableHeader {
    pub decomp_size: u32,
    pub comp_size: u32,
    pub section_size: u32,
}

pub(crate) struct CompressedFileSystem(pub FileSystem);

impl BinRead for CompressedFileSystem {
    type Args<'a> = ();

    fn read_options<R>(reader: &mut R, endian: Endian, args: Self::Args<'_>) -> BinResult<Self>
    where
        R: Read + Seek,
    {
        let current_offset = reader.stream_position().unwrap();
        let header = CompTableHeader::read_options(reader, endian, args)?;

        let mut compressed = vec![0; header.comp_size as usize];
        reader.read_exact(&mut compressed)?;

        reader.seek(SeekFrom::Start(current_offset + header.section_size as u64))?; // seek to end of section to continue reading properly

        let compressed = Cursor::new(compressed);
        let mut decompressed = Cursor::new(crate::zstd_backend::decode_all(compressed)?);

        FileSystem::read_options(&mut decompressed, endian, ()).map(CompressedFileSystem)
    }
}

pub(crate) struct CompressedSearchFileSystem(pub SearchFileSystem);

impl BinRead for CompressedSearchFileSystem {
    type Args<'a> = ();

    fn read_options<R>(reader: &mut R, endian: Endian, args: Self::Args<'_>) -> BinResult<Self>
    where
        R: Read + Seek,
    {
        let current_offset = reader.stream_position().unwrap();
        let header = CompTableHeader::read_options(reader, endian, args)?;

        let mut compressed = vec![0; header.comp_size as usize];
        reader.read_exact(&mut compressed)?;

        reader.seek(SeekFrom::Start(current_offset + header.section_size as u64))?; // seek to end of section to continue reading properly

        let compressed = Cursor::new(compressed);
        let mut decompressed = Cursor::new(crate::zstd_backend::decode_all(compressed)?);

        SearchFileSystem::read_options(&mut decompressed, endian, ())
            .map(CompressedSearchFileSystem)
    }

}

/// The filesystem itself. Includes all the linking between paths, file data, directories, and
/// mass-loading groups.
#[binread]
#[derive(Debug)]
pub struct FileSystem {
    pub fs_header: FileSystemHeader,

    #[br(align_before = 0x100)]
    pub stream_header: StreamHeader,

    #[br(count = stream_header.quick_dir_count)]
    pub quick_dirs: Vec<QuickDir>,

    #[br(count = stream_header.stream_hash_count)]
    pub stream_hash_to_entries: Vec<HashToIndex>,

    #[br(count = stream_header.stream_hash_count)]
    pub stream_entries: Vec<StreamEntry>,

    #[br(count = stream_header.stream_file_index_count)]
    pub stream_file_indices: Vec<u32>,

    #[br(count = stream_header.stream_offset_entry_count)]
    pub stream_datas: Vec<StreamData>,

    #[br(temp)]
    pub hash_index_group_count: u32,

    #[br(temp)]
    pub bucket_count: u32,

    #[br(count = bucket_count)]
    pub file_info_buckets: Vec<FileInfoBucket>,

    #[br(count = hash_index_group_count)]
    pub file_hash_to_path_index: Vec<HashToIndex>,

    #[br(count = fs_header.file_info_path_count)]
    pub file_paths: Vec<FilePath>,

    #[br(count = fs_header.file_info_index_count)]
    pub file_info_indices: Vec<FileInfoIndex>,

    #[br(count = fs_header.folder_count)]
    pub dir_hash_to_info_index: Vec<HashToIndex>,

    #[br(count = fs_header.folder_count)]
    pub dir_infos: Vec<DirInfo>,

    #[br(count = fs_header.folder_offset_count_1 + fs_header.folder_offset_count_2 + fs_header.extra_folder)]
    pub folder_offsets: Vec<DirectoryOffset>,

    #[br(count = fs_header.hash_folder_count)]
    pub folder_child_hashes: Vec<HashToIndex>,

    #[br(count = fs_header.file_info_count + fs_header.file_data_count_2 + fs_header.extra_count)]
    pub file_infos: Vec<FileInfo>,

    #[br(count = fs_header.file_info_sub_index_count + fs_header.file_data_count_2 + fs_header.extra_count_2)]
    pub file_info_to_datas: Vec<FileInfoToFileData>,

    #[br(count = fs_header.file_data_count + fs_header.file_data_count_2 + fs_header.extra_sub_count)]
    pub file_datas: Vec<FileData>,
}

#[binread]
#[derive(Debug)]
pub struct SearchFileSystem {
    pub header: SearchFileSystemHeader,

    #[br(count = header.folder_count)]
    pub folder_lookup: Vec<HashToIndex>,

    #[br(count = header.folder_count)]
    pub folders: Vec<FolderPathListEntry>,

    #[br(count = header.path_index_count)]
    pub path_index_lookup: Vec<HashToIndex>,

    #[br(count = header.path_index_count)]
    pub path_indices: Vec<u32>,

    #[br(count = header.path_count)]
    pub paths: Vec<PathListEntry>,
}

#[repr(C)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct FileSystemHeader {
    pub table_filesize: u32,
    pub file_info_path_count: u32,
    pub file_info_index_count: u32,
    pub folder_count: u32,

    pub folder_offset_count_1: u32,

    pub hash_folder_count: u32,
    pub file_info_count: u32,
    pub file_info_sub_index_count: u32,
    pub file_data_count: u32,

    pub folder_offset_count_2: u32,
    pub file_data_count_2: u32,
    pub padding: u32,

    pub unk1_10: u32, // always 0x10
    pub unk2_10: u32, // always 0x10

    pub regional_count_1: u8,
    pub regional_count_2: u8,
    pub padding2: u16,

    pub version: u32,
    pub extra_folder: u32,
    pub extra_count: u32,

    pub unk: [u32; 2], // in the loaded arc this is a pointer to the parsed search data

    pub extra_count_2: u32,
    pub extra_sub_count: u32,
}

#[repr(C)]
#[derive(BinRead, Debug, Copy, Clone)]
pub struct SearchFileSystemHeader {
    pub size: u64,
    pub folder_count: u32,
    pub path_index_count: u32,
    pub path_count: u32,
}

#[repr(C)]
#[derive(BinRead, Debug)]
pub struct StreamHeader {
    pub quick_dir_count: u32,
    pub stream_hash_count: u32,
    pub stream_file_index_count: u32,
    pub stream_offset_entry_count: u32,
}

#[bitfield]
#[derive(BinRead, Debug, Clone, Copy)]
#[br(map = Self::from_bytes)]
pub struct QuickDir {
    pub hash: u32,
    pub name_length: u8,
    pub count: B24,
    pub index: u32,
}

#[repr(C)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct StreamEntry {
    pub path: HashToIndex,
    pub flags: StreamEntryFlags,
}

#[bitfield]
#[derive(BinRead, Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[br(map = Self::from_bytes)]
pub struct StreamEntryFlags {
    pub is_regional: bool,
    pub is_localized: bool,
    pub unused: B30,
}

#[bitfield]
#[derive(BinRead, Debug, Clone, Copy, PartialEq, Eq, Hash, Default)]
#[br(map = Self::from_bytes)]
pub struct HashToIndex {
    pub hash: u32,
    pub length: u8,
    pub index: B24,
}

impl PartialEq<Hash40> for HashToIndex {
    fn eq(&self, other: &Hash40) -> bool {
        self.hash40() == *other
    }
}

#[repr(C)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct FileInfoBucket {
    pub start: u32,
    pub count: u32,
}
#[repr(C)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct FilePath {
    pub path: HashToIndex,
    pub ext: HashToIndex,
    pub parent: HashToIndex,
    pub file_name: HashToIndex,
}
#[repr(C)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct FileInfoIndex {
    pub dir_offset_index: u32,
    pub file_info_index: FileInfoIdx,
}

/// Also known as MassLoadingGroup
#[repr(C)]
#[cfg_attr(feature = "smash-runtime", repr(packed))]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct DirInfo {
    pub path: HashToIndex,
    pub name: Hash40,
    pub parent: Hash40,
    pub extra_dis_re: u32,
    pub extra_dis_re_length: u32,
    pub file_info_start_index: u32,
    pub file_count: u32,
    pub child_dir_start_index: u32,
    pub child_dir_count: u32,
    pub flags: DirInfoFlags,
}

#[repr(C)]
#[derive(Debug, Clone)]
pub enum RedirectionType {
    Symlink(DirInfo),
    Shared(DirectoryOffset),
}

#[bitfield]
#[derive(BinRead, Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[br(map = Self::from_bytes)]
pub struct DirInfoFlags {
    pub unk1: B26,
    pub redirected: bool,
    pub unk2: bool,
    pub is_symlink: bool,
    pub unk3: B3,
}

#[repr(C)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct StreamData {
    pub size: u64,
    pub offset: u64,
}

/// Also known as MassLoadingData
#[repr(packed)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct DirectoryOffset {
    pub offset: u64,
    pub decomp_size: u32,
    pub size: u32,
    /// FileData index if using DirInfo.path.index(), FileInfo if redirected from a DirectoryOffset.directory_index
    pub file_start_index: u32,
    pub file_count: u32,
    /// This can be a DirInfo OR a DirectoryOffset index, depending on the flags of the matching DirInfo. Considering checking the tests in lookups.rs for an example.
    pub directory_index: u32,
}

#[repr(C)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct FileInfo {
    pub file_path_index: FilePathIdx,
    pub file_info_indice_index: FileInfoIndiceIdx,
    // SubIndexIndex
    pub info_to_data_index: InfoToDataIdx,
    // Flags
    pub flags: FileInfoFlags,
}

#[bitfield]
#[derive(BinRead, Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[br(map = Self::from_bytes)]
pub struct FileInfoFlags {
    pub unused: B4,
    pub is_redirect: bool,
    pub unused2: B7,
    pub unknown1: bool,
    pub padding3: B2,
    pub is_regional: bool,
    pub is_localized: bool,
    pub unused3: B3,
    pub unknown2: bool,
    pub unknown3: bool,
    pub unused4: B10,
}

#[repr(C)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct FileInfoToFileData {
    pub folder_offset_index: u32,
    pub file_data_index: FileDataIdx,
    pub file_info_index_and_load_type: FileInfoToFileDataBitfield,
}

#[bitfield]
#[derive(BinRead, Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[br(map = Self::from_bytes)]
pub struct FileInfoToFileDataBitfield {
    pub file_info_idx: B24,
    pub load_type: u8,
}

#[repr(C)]
#[derive(BinRead, Debug, Clone, Copy)]
pub struct FileData {
    pub offset_in_folder: u32,
    pub comp_size: u32,
    pub decomp_size: u32,
    pub flags: FileDataFlags,
}

#[bitfield]
#[derive(BinRead, Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[br(map = Self::from_bytes)]
pub struct FileDataFlags {
    pub compressed: bool,
    pub use_zstd: bool,
    pub unk: B30,
}

#[repr(C)]
#[derive(BinRead, Debug, Copy, Clone)]
pub struct SearchListEntry {
    pub path: HashToIndex,
    pub parent: HashToIndex,
    pub file_name: HashToIndex,
    pub ext: HashToIndex,
}

#[repr(transparent)]
#[derive(BinRead, Debug, Copy, Clone)]
pub struct PathListEntry(pub SearchListEntry);

#[repr(transparent)]
#[derive(BinRead, Debug, Copy, Clone)]
pub struct FolderPathListEntry(pub SearchListEntry);

macro_rules! impl_fs_index {
    ($to_index:ty, $index_with:ty) => {
        impl std::ops::Index<$index_with> for [$to_index] {
            type Output = $to_index;
            fn index(&self, index: $index_with) -> &Self::Output {
                self.get(usize::from(index)).unwrap()
            }
        }

        impl std::ops::IndexMut<$index_with> for [$to_index] {
            fn index_mut(&mut self, index: $index_with) -> &mut Self::Output {
                self.get_mut(usize::from(index)).unwrap()
            }
        }
    };
}

impl_fs_index!(FilePath, FilePathIdx);
impl_fs_index!(FileInfoIndex, FileInfoIndiceIdx);
impl_fs_index!(FileInfo, FileInfoIdx);
impl_fs_index!(FileInfoToFileData, InfoToDataIdx);
impl_fs_index!(FileData, FileDataIdx);

use std::ops::{Deref, DerefMut};

impl Deref for PathListEntry {
    type Target = SearchListEntry;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl DerefMut for PathListEntry {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.0
    }
}

impl Deref for FolderPathListEntry {
    type Target = SearchListEntry;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl DerefMut for FolderPathListEntry {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.0
    }
}

impl PathListEntry {
    pub fn is_directory(&self) -> bool {
        self.parent.index() & 0x40_0000 != 0
    }

    pub fn as_folder_entry(&self) -> FolderPathListEntry {
        let mut file_path = self.path;
        file_path.set_index(0xFF_FFFF);
        let mut parent = self.parent;
        parent.set_index(0x40_0000);
        let mut ext = self.ext;
        ext.set_hash(0xFFFF_FFFF);
        ext.set_length(0x00);
        ext.set_index(0x00);

        FolderPathListEntry(SearchListEntry {
            path: file_path,
            parent,
            ext,
            file_name: self.file_name,
        })
    }
}

impl FolderPathListEntry {
    pub fn get_child_count(&self) -> usize {
        self.parent.index() as usize
    }

    pub fn get_first_child_index(&self) -> usize {
        (self.ext.hash40().as_u64() & 0xFF_FFFF) as usize
    }

    pub fn as_path_entry(&self) -> PathListEntry {
        let mut file_path = self.path;
        file_path.set_index(0xFF_FFFF);
        let mut parent = self.parent;
        parent.set_index(0x40_0000);
        let mut ext = self.ext;
        ext.set_hash(0x0);
        ext.set_length(0x0);
        ext.set_index(0x0);
        PathListEntry(SearchListEntry {
            path: file_path,
            parent,
            ext,
            file_name: self.file_name,
        })
    }

    pub fn set_first_child_index(&mut self, idx: u32) {
        self.ext.set_hash(idx & 0xFF_FFFF)
    }
}