loot / loot-condition-interpreter / 12774060573

fn evaluate_file_path(state: &State, file_path: &Path) -> Result<bool, Error> {

    Ok(resolve_path(state, file_path).exists())

}

fn is_match(game_type: GameType, regex: &Regex, file_name: &OsStr) -> bool {

    normalise_file_name(game_type, file_name)

        .to_str()

        .map(|s| regex.is_match(s))

        .unwrap_or(false)

}

fn evaluate_regex(

    game_type: GameType,

    data_path: &Path,

    parent_path: &Path,

    regex: &Regex,

    mut condition: impl FnMut() -> bool,

) -> Result<bool, Error> {

    let parent_path = data_path.join(parent_path);

    let dir_iterator = match read_dir(&parent_path) {

        Ok(i) => i,

        Err(_) => return Ok(false),

    for entry in dir_iterator {

        let entry = entry.map_err(|e| Error::IoError(parent_path.to_path_buf(), e))?;

        if is_match(game_type, regex, &entry.file_name()) && condition() {

            return Ok(true);

        }

    Ok(false)

}

fn evaluate_file_regex(state: &State, parent_path: &Path, regex: &Regex) -> Result<bool, Error> {

    for data_path in &state.additional_data_paths {

        let result = evaluate_regex(state.game_type, data_path, parent_path, regex, || true)?;

        if result {

            return Ok(true);

    evaluate_regex(

        state.game_type,

        &state.data_path,

        parent_path,

        regex,

        || true,

    )

}

fn evaluate_readable(state: &State, path: &Path) -> Result<bool, Error> {

    if path.is_dir() {

        Ok(read_dir(resolve_path(state, path)).is_ok())

        Ok(File::open(resolve_path(state, path)).is_ok())

}

fn evaluate_many(state: &State, parent_path: &Path, regex: &Regex) -> Result<bool, Error> {

    // Share the found_one state across all data paths because they're all

    // treated as if they were merged into one directory.

    let mut found_one = false;

    let mut condition = || {

        if found_one {

            true

            found_one = true;

            false

    };

    for data_path in &state.additional_data_paths {

        let result = evaluate_regex(

            state.game_type,

            data_path,

            parent_path,

            regex,

            &mut condition,

        )?;

        if result {

        }

    evaluate_regex(

        state.game_type,

        &state.data_path,

        parent_path,

        regex,

        &mut condition,

    )

}

fn evaluate_active_path(state: &State, path: &Path) -> Result<bool, Error> {

    Ok(path

        .to_str()

        .map(|s| state.active_plugins.contains(&s.to_lowercase()))

        .unwrap_or(false))

}

fn evaluate_active_regex(state: &State, regex: &Regex) -> Result<bool, Error> {

    Ok(state.active_plugins.iter().any(|p| regex.is_match(p)))

}

fn evaluate_is_master(state: &State, file_path: &Path) -> Result<bool, Error> {

    let game_id = match state.game_type {

        GameType::Oblivion => GameId::Oblivion,

    let path = resolve_path(state, file_path);

    let mut plugin = esplugin::Plugin::new(game_id, &path);

    plugin

        .parse_file(ParseOptions::header_only())

        .map(|_| plugin.is_master_file())

        .or(Ok(false))

}

fn evaluate_many_active(state: &State, regex: &Regex) -> Result<bool, Error> {

    let mut found_one = false;

    for active_plugin in &state.active_plugins {

        if regex.is_match(active_plugin) {

            if found_one {

                return Ok(true);

            } else {

                found_one = true;

            }

        }

    Ok(false)

}

fn lowercase(path: &Path) -> Option<String> {

    path.to_str().map(str::to_lowercase)

}

fn evaluate_checksum(state: &State, file_path: &Path, crc: u32) -> Result<bool, Error> {

    if let Ok(reader) = state.crc_cache.read() {

        if let Some(key) = lowercase(file_path) {

            if let Some(cached_crc) = reader.get(&key) {

                return Ok(*cached_crc == crc);

            }

    let path = resolve_path(state, file_path);

    if !path.is_file() {

        return Ok(false);

    }

    let io_error_mapper = |e| Error::IoError(file_path.to_path_buf(), e);

    let file = File::open(path).map_err(io_error_mapper)?;

    let mut reader = BufReader::new(file);

    let mut hasher = crc32fast::Hasher::new();

    let mut buffer = reader.fill_buf().map_err(io_error_mapper)?;

    while !buffer.is_empty() {

        hasher.write(buffer);

        let length = buffer.len();

        reader.consume(length);

        buffer = reader.fill_buf().map_err(io_error_mapper)?;

    let calculated_crc = hasher.finalize();

    if let Ok(mut writer) = state.crc_cache.write() {

        if let Some(key) = lowercase(file_path) {

            writer.insert(key, calculated_crc);

        }

    Ok(calculated_crc == crc)

}

fn lowercase_filename(path: &Path) -> Option<String> {

    path.file_name()

        .and_then(OsStr::to_str)

        .map(str::to_lowercase)

}

fn get_version(state: &State, file_path: &Path) -> Result<Option<Version>, Error> {

    if !file_path.is_file() {

        return Ok(None);

    }

    if let Some(key) = lowercase_filename(file_path) {

        if let Some(version) = state.plugin_versions.get(&key) {

            return Ok(Some(Version::from(version.as_str())));

        }

    if has_plugin_file_extension(state.game_type, file_path) {

        Ok(None)

        Version::read_file_version(file_path)

}

fn get_product_version(file_path: &Path) -> Result<Option<Version>, Error> {

    if file_path.is_file() {

        Version::read_product_version(file_path)

        Ok(None)

}

fn evaluate_version<F>(

    state: &State,

    file_path: &Path,

    given_version: &str,

    comparator: ComparisonOperator,

    read_version: F,

) -> Result<bool, Error>

where

    F: Fn(&State, &Path) -> Result<Option<Version>, Error>,

{

    let file_path = resolve_path(state, file_path);

    let actual_version = match read_version(state, &file_path)? {

        Some(v) => v,

            return Ok(comparator == ComparisonOperator::NotEqual

                || comparator == ComparisonOperator::LessThan

                || comparator == ComparisonOperator::LessThanOrEqual);

    let given_version = Version::from(given_version);

    match comparator {

        ComparisonOperator::Equal => Ok(actual_version == given_version),

        ComparisonOperator::NotEqual => Ok(actual_version != given_version),

        ComparisonOperator::LessThan => Ok(actual_version < given_version),

        ComparisonOperator::GreaterThan => Ok(actual_version > given_version),

        ComparisonOperator::LessThanOrEqual => Ok(actual_version <= given_version),

        ComparisonOperator::GreaterThanOrEqual => Ok(actual_version >= given_version),

}

    pub fn eval(&self, state: &State) -> Result<bool, Error> {

        if self.is_slow() {

            if let Ok(reader) = state.condition_cache.read() {

                if let Some(cached_result) = reader.get(self) {

                    return Ok(*cached_result);

                }

        }

        let result = match self {

            Function::FilePath(f) => evaluate_file_path(state, f),

            Function::FileRegex(p, r) => evaluate_file_regex(state, p, r),

            Function::Readable(p) => evaluate_readable(state, p),

            Function::ActivePath(p) => evaluate_active_path(state, p),

            Function::ActiveRegex(r) => evaluate_active_regex(state, r),

            Function::IsMaster(p) => evaluate_is_master(state, p),

            Function::Many(p, r) => evaluate_many(state, p, r),

            Function::ManyActive(r) => evaluate_many_active(state, r),

            Function::Checksum(path, crc) => evaluate_checksum(state, path, *crc),

            Function::Version(p, v, c) => evaluate_version(state, p, v, *c, get_version),

            Function::ProductVersion(p, v, c) => {

                evaluate_version(state, p, v, *c, |_, p| get_product_version(p))

        if self.is_slow() {

            if let Ok(function_result) = result {

                if let Ok(mut writer) = state.condition_cache.write() {

                    writer.insert(self.clone(), function_result);

                }

        }

        result

    }

    fn is_slow(&self) -> bool {

        !matches!(

            self,

    }

    fn state<T: Into<PathBuf>>(data_path: T) -> State {

        state_with_active_plugins(data_path, &[])

    }

    fn state_with_active_plugins<T: Into<PathBuf>>(data_path: T, active_plugins: &[&str]) -> State {

        state_with_data(data_path, Vec::default(), active_plugins, &[])

    }

    fn state_with_versions<T: Into<PathBuf>>(

        data_path: T,

        plugin_versions: &[(&str, &str)],

    ) -> State {

        state_with_data(data_path, Vec::default(), &[], plugin_versions)

    }

    fn state_with_data<T: Into<PathBuf>>(

        data_path: T,

        additional_data_paths: Vec<T>,

        active_plugins: &[&str],

        plugin_versions: &[(&str, &str)],

    ) -> State {

        let data_path = data_path.into();

        if !data_path.exists() {

            create_dir(&data_path).unwrap();

        }

        let additional_data_paths = additional_data_paths

            .into_iter()

            .map(|data_path| {

                let data_path: PathBuf = data_path.into();

                if !data_path.exists() {

                }

                data_path

            })

            .collect();

        State {

            game_type: GameType::Oblivion,

            data_path,

            additional_data_paths,

            active_plugins: active_plugins.iter().map(|s| s.to_lowercase()).collect(),

            crc_cache: RwLock::default(),

            plugin_versions: plugin_versions

                .iter()

                .map(|(p, v)| (p.to_lowercase(), v.to_string()))

                .collect(),

            condition_cache: RwLock::default(),

        }

    }

    fn regex(string: &str) -> Regex {

        RegexBuilder::new(string)

            .case_insensitive(true)

            .build()

            .unwrap()

    }

    fn make_path_unreadable(path: &Path) {

        let mut permissions = std::fs::metadata(&path).unwrap().permissions();

        permissions.set_mode(0o200);

        std::fs::set_permissions(&path, permissions).unwrap();

    }

    fn function_file_path_eval_should_return_true_if_the_file_exists_relative_to_the_data_path() {

        let function = Function::FilePath(PathBuf::from("Cargo.toml"));

        let state = state(".");

        assert!(function.eval(&state).unwrap());

    }

    fn function_file_path_eval_should_return_true_if_given_a_plugin_that_is_ghosted() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        copy(

            Path::new("tests/testing-plugins/Oblivion/Data/Blank.esp"),

            state.data_path.join("Blank.esp.ghost"),

        )

        .unwrap();

        let function = Function::FilePath(PathBuf::from("Blank.esp"));

        assert!(function.eval(&state).unwrap());

    }

    fn function_file_path_eval_should_not_check_for_ghosted_non_plugin_file() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        copy(

            Path::new("Cargo.toml"),

            state.data_path.join("Cargo.toml.ghost"),

        )

        .unwrap();

        let function = Function::FilePath(PathBuf::from("Cargo.toml"));

        assert!(!function.eval(&state).unwrap());

    }

    fn function_file_path_eval_should_return_false_if_the_file_does_not_exist() {

        let function = Function::FilePath(PathBuf::from("missing"));

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_file_regex_eval_should_be_false_if_no_directory_entries_match() {

        let function = Function::FileRegex(PathBuf::from("."), regex("missing"));

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_file_regex_eval_should_be_false_if_the_parent_path_part_is_not_a_directory() {

        let function = Function::FileRegex(PathBuf::from("missing"), regex("Cargo.*"));

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_file_regex_eval_should_be_true_if_a_directory_entry_matches() {

        let function = Function::FileRegex(

            PathBuf::from("tests/testing-plugins/Oblivion/Data"),

            regex("Blank\\.esp"),

        );

        let state = state(".");

        assert!(function.eval(&state).unwrap());

    }

    fn function_file_regex_eval_should_trim_ghost_plugin_extension_before_matching_against_regex() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        copy(

            Path::new("tests/testing-plugins/Oblivion/Data/Blank.esm"),

            state.data_path.join("Blank.esm.ghost"),

        )

        .unwrap();

        let function = Function::FileRegex(PathBuf::from("."), regex("^Blank\\.esm$"));

        assert!(function.eval(&state).unwrap());

    }

    fn function_file_regex_eval_should_check_all_configured_data_paths() {

        let function = Function::FileRegex(PathBuf::from("Data"), regex("Blank\\.esp"));

        let state = state_with_data("./src", vec!["./tests/testing-plugins/Oblivion"], &[], &[]);

        assert!(function.eval(&state).unwrap());

    }

    fn function_readable_eval_should_be_true_for_a_file_that_can_be_opened_as_read_only() {

        let function = Function::Readable(PathBuf::from("Cargo.toml"));

        let state = state(".");

        assert!(function.eval(&state).unwrap());

    }

    fn function_readable_eval_should_be_true_for_a_folder_that_can_be_read() {

        let function = Function::Readable(PathBuf::from("tests"));

        let state = state(".");

        assert!(function.eval(&state).unwrap());

    }

    fn function_readable_eval_should_be_false_for_a_file_that_does_not_exist() {

        let function = Function::Readable(PathBuf::from("missing"));

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_readable_eval_should_be_false_for_a_file_that_is_not_readable() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        let relative_path = "unreadable";

        let file_path = state.data_path.join(relative_path);

        std::fs::write(&file_path, "").unwrap();

        make_path_unreadable(&file_path);

        assert!(file_path.exists());

        let function = Function::Readable(PathBuf::from(relative_path));

        assert!(!function.eval(&state).unwrap());

    }

    fn function_readable_eval_should_be_false_for_a_folder_that_is_not_readable() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        let relative_path = "unreadable";

        let folder_path = state.data_path.join(relative_path);

        std::fs::create_dir(&folder_path).unwrap();

        make_path_unreadable(&folder_path);

        assert!(folder_path.exists());

        let function = Function::Readable(PathBuf::from(relative_path));

        assert!(!function.eval(&state).unwrap());

    }

    fn function_active_path_eval_should_be_true_if_the_path_is_an_active_plugin() {

        let function = Function::ActivePath(PathBuf::from("Blank.esp"));

        let state = state_with_active_plugins(".", &["Blank.esp"]);

        assert!(function.eval(&state).unwrap());

    }

    fn function_active_path_eval_should_be_case_insensitive() {

        let function = Function::ActivePath(PathBuf::from("Blank.esp"));

        let state = state_with_active_plugins(".", &["blank.esp"]);

        assert!(function.eval(&state).unwrap());

    }

    fn function_active_path_eval_should_be_false_if_the_path_is_not_an_active_plugin() {

        let function = Function::ActivePath(PathBuf::from("inactive.esp"));

        let state = state_with_active_plugins(".", &["Blank.esp"]);

        assert!(!function.eval(&state).unwrap());

    }

    fn function_active_regex_eval_should_be_true_if_the_regex_matches_an_active_plugin() {

        let function = Function::ActiveRegex(regex("Blank\\.esp"));

        let state = state_with_active_plugins(".", &["Blank.esp"]);

        assert!(function.eval(&state).unwrap());

    }

    fn function_active_regex_eval_should_be_false_if_the_regex_does_not_match_an_active_plugin() {

        let function = Function::ActiveRegex(regex("inactive\\.esp"));

        let state = state_with_active_plugins(".", &["Blank.esp"]);

        assert!(!function.eval(&state).unwrap());

    }

    fn function_is_master_eval_should_be_true_if_the_path_is_a_master_plugin() {

        let function = Function::IsMaster(PathBuf::from("Blank.esm"));

        let state = state("tests/testing-plugins/Oblivion/Data");

        assert!(function.eval(&state).unwrap());

    }

    fn function_is_master_eval_should_be_false_if_the_path_does_not_exist() {

        let function = Function::IsMaster(PathBuf::from("missing.esp"));

        let state = state("tests/testing-plugins/Oblivion/Data");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_is_master_eval_should_be_false_if_the_path_is_not_a_plugin() {

        let function = Function::IsMaster(PathBuf::from("Cargo.toml"));

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_is_master_eval_should_be_false_if_the_path_is_a_non_master_plugin() {

        let function = Function::IsMaster(PathBuf::from("Blank.esp"));

        let state = state("tests/testing-plugins/Oblivion/Data");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_many_eval_should_be_false_if_no_directory_entries_match() {

        let function = Function::Many(PathBuf::from("."), regex("missing"));

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_many_eval_should_be_false_if_the_parent_path_part_is_not_a_directory() {

        let function = Function::Many(PathBuf::from("missing"), regex("Cargo.*"));

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_many_eval_should_be_false_if_one_directory_entry_matches() {

        let function = Function::Many(

            PathBuf::from("tests/testing-plugins/Oblivion/Data"),

            regex("Blank\\.esp"),

        );

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_many_eval_should_be_true_if_more_than_one_directory_entry_matches() {

        let function = Function::Many(

            PathBuf::from("tests/testing-plugins/Oblivion/Data"),

            regex("Blank.*"),

        );

        let state = state(".");

        assert!(function.eval(&state).unwrap());

    }

    fn function_many_eval_should_trim_ghost_plugin_extension_before_matching_against_regex() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        copy(

            Path::new("tests/testing-plugins/Oblivion/Data/Blank.esm"),

            state.data_path.join("Blank.esm.ghost"),

        )

        .unwrap();

        copy(

            Path::new("tests/testing-plugins/Oblivion/Data/Blank.esp"),

            state.data_path.join("Blank.esp.ghost"),

        )

        .unwrap();

        let function = Function::Many(PathBuf::from("."), regex("^Blank\\.es(m|p)$"));

        assert!(function.eval(&state).unwrap());

    }

    fn function_many_eval_should_check_across_all_configured_data_paths() {

        let function = Function::Many(PathBuf::from("Data"), regex("Blank\\.esp"));

        let state = state_with_data(

            "./tests/testing-plugins/Skyrim",

            vec!["./tests/testing-plugins/Oblivion"],

            &[],

            &[],

        );

        assert!(function.eval(&state).unwrap());

    }

    fn function_many_active_eval_should_be_true_if_the_regex_matches_more_than_one_active_plugin() {

        let function = Function::ManyActive(regex("Blank.*"));

        let state = state_with_active_plugins(".", &["Blank.esp", "Blank.esm"]);

        assert!(function.eval(&state).unwrap());

    }

    fn function_many_active_eval_should_be_false_if_one_active_plugin_matches() {

        let function = Function::ManyActive(regex("Blank\\.esp"));

        let state = state_with_active_plugins(".", &["Blank.esp", "Blank.esm"]);

        assert!(!function.eval(&state).unwrap());

    }

    fn function_many_active_eval_should_be_false_if_the_regex_does_not_match_an_active_plugin() {

        let function = Function::ManyActive(regex("inactive\\.esp"));

        let state = state_with_active_plugins(".", &["Blank.esp", "Blank.esm"]);

        assert!(!function.eval(&state).unwrap());

    }

    fn function_checksum_eval_should_be_false_if_the_file_does_not_exist() {

        let function = Function::Checksum(PathBuf::from("missing"), 0x374E2A6F);

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_checksum_eval_should_be_false_if_the_file_checksum_does_not_equal_the_given_checksum(

    ) {

        let function = Function::Checksum(

            PathBuf::from("tests/testing-plugins/Oblivion/Data/Blank.esm"),

            0xDEADBEEF,

        );

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_checksum_eval_should_be_true_if_the_file_checksum_equals_the_given_checksum() {

        let function = Function::Checksum(

            PathBuf::from("tests/testing-plugins/Oblivion/Data/Blank.esm"),

            0x374E2A6F,

        );

        let state = state(".");

        assert!(function.eval(&state).unwrap());

    }

    fn function_checksum_eval_should_support_checking_the_crc_of_a_ghosted_plugin() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        copy(

            Path::new("tests/testing-plugins/Oblivion/Data/Blank.esm"),

            state.data_path.join("Blank.esm.ghost"),

        )

        .unwrap();

        let function = Function::Checksum(PathBuf::from("Blank.esm"), 0x374E2A6F);

        assert!(function.eval(&state).unwrap());

    }

    fn function_checksum_eval_should_not_check_for_ghosted_non_plugin_file() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        copy(

            Path::new("tests/testing-plugins/Oblivion/Data/Blank.bsa"),

            state.data_path.join("Blank.bsa.ghost"),

        )

        .unwrap();

        let function = Function::Checksum(PathBuf::from("Blank.bsa"), 0x22AB79D9);

        assert!(!function.eval(&state).unwrap());

    }

    fn function_checksum_eval_should_be_false_if_given_a_directory_path() {

        // The given CRC is the CRC-32 of the directory as calculated by 7-zip.

        let function = Function::Checksum(PathBuf::from("tests/testing-plugins"), 0xC9CD16C3);

        let state = state(".");

        assert!(!function.eval(&state).unwrap());

    }

    fn function_checksum_eval_should_cache_and_use_cached_crcs() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        copy(

            Path::new("tests/testing-plugins/Oblivion/Data/Blank.esm"),

            state.data_path.join("Blank.esm"),

        )

        .unwrap();

        let function = Function::Checksum(PathBuf::from("Blank.esm"), 0x374E2A6F);

        assert!(function.eval(&state).unwrap());

        copy(

            Path::new("tests/testing-plugins/Oblivion/Data/Blank.bsa"),

            state.data_path.join("Blank.esm"),

        )

        .unwrap();

        let function = Function::Checksum(PathBuf::from("Blank.esm"), 0x374E2A6F);

        assert!(function.eval(&state).unwrap());

    }

    fn function_eval_should_cache_results_and_use_cached_results() {

        let tmp_dir = tempdir().unwrap();

        let data_path = tmp_dir.path().join("Data");

        let state = state(data_path);

        copy(Path::new("Cargo.toml"), state.data_path.join("Cargo.toml")).unwrap();

        let function = Function::FilePath(PathBuf::from("Cargo.toml"));

        assert!(function.eval(&state).unwrap());

        remove_file(state.data_path.join("Cargo.toml")).unwrap();

        assert!(function.eval(&state).unwrap());

    }

    fn function_version_eval_should_be_true_if_the_path_does_not_exist_and_comparator_is_ne() {

        let function =

            Function::Version("missing".into(), "1.0".into(), ComparisonOperator::NotEqual);

        let state = state(".");

        assert!(function.eval(&state).unwrap());

    }

    fn function_version_eval_should_be_true_if_the_path_does_not_exist_and_comparator_is_lt() {

        let function =

            Function::Version("missing".into(), "1.0".into(), ComparisonOperator::LessThan);

        let state = state(".");

        assert!(function.eval(&state).unwrap());

    }

    fn function_version_eval_should_be_true_if_the_path_does_not_exist_and_comparator_is_lteq() {

        let function = Function::Version(

            "missing".into(),

            "1.0".into(),

            ComparisonOperator::LessThanOrEqual,

        );

        let state = state(".");

        assert!(function.eval(&state).unwrap());

    }