use std::io;
use std::io::prelude::*;

use ape_fatfs::{
    fs::{self, FatType, FileSystem, FormatVolumeOptions, FsOptions, LossyOemCpConverter},
    io::*,
    time::DefaultTimeProvider,
};
use fscommon::BufStream;

const KB: u64 = 1024;
const MB: u64 = KB * 1024;
const TEST_STR: &str = "Hi there Rust programmer!\n";

type TestFileSystem = FileSystem<
    StdIoWrapper<BufStream<io::Cursor<Vec<u8>>>>,
    DefaultTimeProvider,
    LossyOemCpConverter,
>;

fn basic_fs_test(fs: &TestFileSystem) {
    let stats = fs.stats().expect("stats");
    if fs.fat_type() == FatType::Fat32 {
        // On FAT32 one cluster is allocated for root directory
        assert_eq!(stats.total_clusters(), stats.free_clusters() + 1);
    } else {
        assert_eq!(stats.total_clusters(), stats.free_clusters());
    }

    let root_dir = fs.root_dir();
    let entries = root_dir.iter().map(|r| r.unwrap()).collect::<Vec<_>>();
    assert_eq!(entries.len(), 0);

    let subdir1 = root_dir.create_dir("subdir1").expect("create_dir subdir1");
    let subdir2 = root_dir
        .create_dir("subdir1/subdir2 with long name")
        .expect("create_dir subdir2");

    let test_str = TEST_STR.repeat(1000);
    {
        let mut file = subdir2
            .create_file("test file name.txt")
            .expect("create file");
        file.truncate().expect("truncate file");
        file.write_all(test_str.as_bytes()).expect("write file");
    }

    let mut file = root_dir
        .open_file("subdir1/subdir2 with long name/test file name.txt")
        .unwrap();
    let mut content = String::new();
    file.read_to_string(&mut content).expect("read_to_string");
    assert_eq!(content, test_str);

    let filenames = root_dir
        .iter()
        .map(|r| r.unwrap().file_name())
        .collect::<Vec<String>>();
    assert_eq!(filenames, ["subdir1"]);

    let filenames = subdir2
        .iter()
        .map(|r| r.unwrap().file_name())
        .collect::<Vec<String>>();
    assert_eq!(filenames, [".", "..", "test file name.txt"]);

    subdir1
        .rename(
            "subdir2 with long name/test file name.txt",
            &root_dir,
            "new-name.txt",
        )
        .expect("rename");

    let filenames = subdir2
        .iter()
        .map(|r| r.unwrap().file_name())
        .collect::<Vec<String>>();
    assert_eq!(filenames, [".", ".."]);

    let filenames = root_dir
        .iter()
        .map(|r| r.unwrap().file_name())
        .collect::<Vec<String>>();
    assert_eq!(filenames, ["subdir1", "new-name.txt"]);
}

fn test_format_fs(opts: FormatVolumeOptions, total_bytes: u64) -> TestFileSystem {
    let _ = env_logger::builder().is_test(true).try_init();
    // Init storage to 0xD1 bytes (value has been choosen to be parsed as normal file)
    let storage_vec: Vec<u8> = vec![0xD1_u8; total_bytes as usize];
    let storage_cur = io::Cursor::new(storage_vec);
    let mut buffered_stream = StdIoWrapper::from(BufStream::new(storage_cur));
    fs::format_volume(&mut buffered_stream, opts).expect("format volume");

    let fs = TestFileSystem::new(buffered_stream, FsOptions::new()).expect("open fs");
    basic_fs_test(&fs);
    fs
}

#[test]
fn test_format_1mb() {
    let total_bytes = MB;
    let opts = FormatVolumeOptions::new();
    let fs = test_format_fs(opts, total_bytes);
    assert_eq!(fs.fat_type(), FatType::Fat12);
}

#[test]
fn test_format_8mb_1fat() {
    let total_bytes = 8 * MB;
    let opts = FormatVolumeOptions::new().fats(1);
    let fs = test_format_fs(opts, total_bytes);
    assert_eq!(fs.fat_type(), FatType::Fat16);
}

#[test]
fn test_format_50mb() {
    let total_bytes = 50 * MB;
    let opts = FormatVolumeOptions::new();
    let fs = test_format_fs(opts, total_bytes);
    assert_eq!(fs.fat_type(), FatType::Fat16);
}

#[test]
fn test_format_2gb_512sec() {
    let total_bytes = 2 * 1024 * MB;
    let opts = FormatVolumeOptions::new();
    let fs = test_format_fs(opts, total_bytes);
    assert_eq!(fs.fat_type(), FatType::Fat32);
}

#[test]
fn test_format_1gb_4096sec() {
    let total_bytes = 1024 * MB;
    let opts = FormatVolumeOptions::new().bytes_per_sector(4096);
    let fs = test_format_fs(opts, total_bytes);
    assert_eq!(fs.fat_type(), FatType::Fat32);
}

#[test]
fn test_format_empty_volume_label() {
    let total_bytes = 2 * 1024 * MB;
    let opts = FormatVolumeOptions::new();
    let fs = test_format_fs(opts, total_bytes);
    assert_eq!(fs.volume_label(), "NO NAME");
    assert_eq!(fs.read_volume_label_from_root_dir().unwrap(), None);
}

#[test]
fn test_format_volume_label_and_id() {
    let total_bytes = 2 * 1024 * MB;
    let opts = FormatVolumeOptions::new()
        .volume_id(1234)
        .volume_label(*b"VOLUMELABEL");
    let fs = test_format_fs(opts, total_bytes);
    assert_eq!(fs.volume_label(), "VOLUMELABEL");
    assert_eq!(
        fs.read_volume_label_from_root_dir().unwrap(),
        Some("VOLUMELABEL".to_string())
    );
    assert_eq!(fs.volume_id(), 1234);
}