nias/
file_io.rs

1use std::fs::{remove_file, File};
2use std::io::prelude::*;
3#[cfg(target_family = "unix")]
4use std::os::unix::fs::PermissionsExt;
5use std::path::{Path, PathBuf};
6
7#[cfg(target_family = "unix")]
8fn create_file(path: PathBuf, make_executable: bool) -> Result<File, ()> {
9    let file = match File::create(&path) {
10        Ok(file) => file,
11        Err(_) => return Err(()),
12    };
13
14    if make_executable {
15        let metadata = match file.metadata() {
16            Ok(metadata) => metadata,
17            Err(_) => return Err(()),
18        };
19        let mut permissions = metadata.permissions();
20        permissions.set_mode(0o755);
21        if std::fs::set_permissions(&path, permissions).is_err() {
22            return Err(());
23        };
24    };
25
26    Ok(file)
27}
28
29#[cfg(target_family = "windows")]
30fn create_file(path: PathBuf, _make_executable: bool) -> Result<File, ()> {
31    match File::create(&path) {
32        Err(_) => Err(()),
33        Ok(file) => Ok(file),
34    }
35}
36
37pub fn get_file_writer(
38) -> fn(file_path: &str, contents: &str, make_executable: bool) -> Result<(), String> {
39    |file_path: &str, contents: &str, make_executable: bool| {
40        let path = PathBuf::from(file_path);
41        let mut file = match create_file(path, make_executable) {
42            Ok(f) => f,
43            Err(_) => return Err(format!("Failed to create file {}", file_path)),
44        };
45
46        match file.write_all(contents.as_bytes()) {
47            Ok(_) => Ok(()),
48            Err(_) => Err(format!("Failed to write contents to {}", file_path)),
49        }
50    }
51}
52
53pub fn get_file_existence_checker() -> fn(file_path: &str) -> Result<bool, ()> {
54    |file_path: &str| Ok(Path::new(file_path).exists())
55}
56
57pub fn get_file_reader() -> fn(file_path: &str) -> Result<String, ()> {
58    |file_path: &str| {
59        let mut file = match File::open(&Path::new(file_path)) {
60            Err(_) => return Err(()),
61            Ok(file) => file,
62        };
63        let mut contents = String::new();
64        match file.read_to_string(&mut contents) {
65            Err(_) => Err(()),
66            Ok(_) => Ok(contents),
67        }
68    }
69}
70
71pub fn get_file_remover() -> fn(file_path: &str) -> Result<(), String> {
72    |file_path: &str| match remove_file(file_path) {
73        Ok(_) => Ok(()),
74        Err(msg) => Err(format!(
75            "Failed to remove file: {}. Error details: {}",
76            file_path, msg
77        )),
78    }
79}
80
81#[cfg(test)]
82#[path = "file_io_test.rs"]
83mod file_io_tests;