use check_diff::{
    CheckDiffError, CheckDiffRunners, CodeFormatter, check_diff, search_for_rs_files,
};
use std::fs::File;
use tempfile::Builder;

struct DoNothingFormatter;

impl CodeFormatter for DoNothingFormatter {
    fn format_code<'a>(
        &self,
        _code: &'a str,
        _config: &Option<Vec<String>>,
    ) -> Result<String, CheckDiffError> {
        Ok(String::new())
    }
}

/// Formatter that adds a white space to the end of the codd
struct AddWhiteSpaceFormatter;

impl CodeFormatter for AddWhiteSpaceFormatter {
    fn format_code<'a>(
        &self,
        code: &'a str,
        _config: &Option<Vec<String>>,
    ) -> Result<String, CheckDiffError> {
        let result = code.to_string() + " ";
        Ok(result)
    }
}

#[test]
fn search_for_files_correctly_non_nested() -> Result<(), Box<dyn std::error::Error>> {
    let dir = Builder::new().tempdir_in("").unwrap();
    let file_path = dir.path().join("test.rs");
    let _tmp_file = File::create(file_path)?;

    let iter = search_for_rs_files(dir.path());

    let mut count = 0;
    for _ in iter {
        count += 1;
    }

    assert_eq!(count, 1);

    Ok(())
}

#[test]
fn search_for_files_correctly_nested() -> Result<(), Box<dyn std::error::Error>> {
    let dir = Builder::new().tempdir_in("").unwrap();
    let file_path = dir.path().join("test.rs");
    let _tmp_file = File::create(file_path)?;

    let nested_dir = Builder::new().tempdir_in(dir.path()).unwrap();
    let nested_file_path = nested_dir.path().join("nested.rs");
    let _ = File::create(nested_file_path)?;

    let iter = search_for_rs_files(dir.path());

    let mut count = 0;
    for _ in iter {
        count += 1;
    }

    assert_eq!(count, 2);

    Ok(())
}

#[test]
fn check_diff_test_no_formatting_difference() -> Result<(), CheckDiffError> {
    let runners = CheckDiffRunners::new(DoNothingFormatter, DoNothingFormatter);

    let dir = Builder::new().tempdir_in("").unwrap();
    let file_path = dir.path().join("test.rs");
    let _tmp_file = File::create(file_path)?;

    let errors = check_diff(None, runners, dir.path());
    assert_eq!(errors, 0);
    Ok(())
}

#[test]
fn check_diff_test_formatting_difference() -> Result<(), CheckDiffError> {
    let runners = CheckDiffRunners::new(DoNothingFormatter, AddWhiteSpaceFormatter);
    let dir = Builder::new().tempdir_in("").unwrap();
    let file_path = dir.path().join("test.rs");
    let _tmp_file = File::create(file_path)?;

    let errors = check_diff(None, runners, dir.path());
    assert_ne!(errors, 0);
    Ok(())
}