dev_tool/

file_util.rs

//! 文件工具
use std::fs::{self, File};
use std::io::{self, Read, Write};
use std::path::Path;

/// 文件工具结构体
pub struct FileUtil;

impl FileUtil {
    /// 罗列指定路径下的子目录及文件。
    ///
    /// 参数 `path` 是要罗列内容的目标路径。
    /// 参数 `recurse` 决定是否以递归方式罗列子文件夹内的内容。
    ///
    /// 返回值是一个包含所有子目录及文件路径的 `Vec<String>`。
    /// 如果在读取目录时发生错误，将返回一个空的 `Vec`。
    pub fn list(path: &Path, recurse: bool) -> Vec<String> {
        let mut result = Vec::new();
        let entries = match fs::read_dir(path) {
            Ok(entries) => entries,
            Err(_) => return result,
        };

        for entry in entries {
            let entry = match entry {
                Ok(entry) => entry,
                Err(_) => continue,
            };
            let path = entry.path();
            if path.is_dir() {
                if recurse {
                    result.extend(Self::list(&path, true));
                }
                result.push(path.to_string_lossy().to_string());
            } else {
                result.push(path.to_string_lossy().to_string());
            }
        }
        result
    }

    /// 读取本地文件内容并以UTF - 8字符串形式返回。
    ///
    /// 参数 `path` 是要读取的文件路径。
    ///
    /// 如果文件成功打开并读取，将返回包含文件内容的 `String`。
    /// 如果在打开或读取文件时发生I/O错误，将返回对应的 `io::Error`。
    pub fn read_string(path: &Path) -> Result<String, io::Error> {
        let mut file = File::open(path)?;
        let mut contents = String::new();
        file.read_to_string(&mut contents)?;
        Ok(contents)
    }

    /// 读取本地文件内容并以字节数组形式返回。
    ///
    /// 参数 `path` 是要读取的文件路径。
    ///
    /// 如果文件成功打开并读取，将返回包含文件内容的 `Vec<u8>`。
    /// 如果在打开或读取文件时发生I/O错误，将返回对应的 `io::Error`。
    pub fn read_bytes(path: &Path) -> Result<Vec<u8>, io::Error> {
        let mut file = File::open(path)?;
        let mut contents = Vec::new();
        file.read_to_end(&mut contents)?;
        Ok(contents)
    }

    /// 将给定的字符串内容以覆盖方式写入到指定路径的文本文件。
    ///
    /// 参数 `path` 是要写入的文件路径。
    /// 参数 `content` 是要写入文件的字符串内容。
    ///
    /// 如果文件成功创建并写入内容，将返回 `Ok(())`。
    /// 如果在创建或写入文件时发生I/O错误，将返回对应的 `io::Error`。
    pub fn write_string(path: &Path, content: String) -> Result<(), io::Error> {
        let mut file = File::create(path)?;
        file.write_all(content.as_bytes())?;
        Ok(())
    }

    /// 将给定的字符串内容以追加方式写入到指定路径的文件。
    ///
    /// 参数 `path` 是要写入的文件路径。
    /// 参数 `content` 是要追加到文件的字符串内容。
    ///
    /// 如果文件成功打开并追加内容，将返回 `Ok(())`。
    /// 如果在打开或写入文件时发生I/O错误，将返回对应的 `io::Error`。
    pub fn append_string(path: &Path, content: String) -> Result<(), io::Error> {
        let mut file = File::options().append(true).open(path)?;
        file.write_all(content.as_bytes())?;
        Ok(())
    }

    /// 将给定的字节数组内容写入到指定路径的文件。
    ///
    /// 参数 `path` 是要写入的文件路径。
    /// 参数 `bytes` 是要写入文件的字节数组内容。
    ///
    /// 如果文件成功创建并写入内容，将返回 `Ok(())`。
    /// 如果在创建或写入文件时发生I/O错误，将返回对应的 `io::Error`。
    pub fn write_bytes(path: &Path, bytes: &[u8]) -> Result<(), io::Error> {
        let mut file = File::create(path)?;
        file.write_all(bytes)?;
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::fs::remove_file;
    use std::path::PathBuf;
    static root_dir: &str = "d:/tmp/";

    #[test]
    fn test_list() {
        let temp_dir = Path::new(root_dir);
        let sub_dir = temp_dir.join("sub_dir");
        std::fs::create_dir(&sub_dir).expect("Failed to create sub dir");
        let file_path = sub_dir.join("test_file.txt");
        std::fs::File::create(&file_path).expect("Failed to create file");

        let result = FileUtil::list(temp_dir, true);
        assert!(result.contains(&file_path.to_string_lossy().to_string()));
    }

    #[test]
    fn test_read_string() -> io::Result<()>{
        let temp_path = Path::new(root_dir).join("temp_file.txt");
        println!("{:?}", temp_path);
        let mut temp_file = File::options().write(true).create(true).open(temp_path.as_path())?;
        let content = "test content".to_string();
        temp_file.write_all(content.as_bytes()).expect("Failed to write to temp file");

        let result = FileUtil::read_string(temp_path.as_path());
        assert!(result.is_ok());
        assert_eq!(result.unwrap(), content);
        Ok(())
    }

    #[test]
    fn test_read_bytes() {
        let temp_path = Path::new(root_dir).join("temp_file.txt");
        let mut temp_file = File::options().write(true).open(temp_path.as_path()).unwrap();
        let content = [1, 2, 3];
        temp_file.write_all(&content).expect("Failed to write to temp file");

        let result = FileUtil::read_bytes(temp_path.as_path());
        assert!(result.is_ok());
    }

    #[test]
    fn test_write_string() {
        let temp_path = Path::new(root_dir).join("temp_file.txt");
        let mut temp_file = File::options().write(true).open(&temp_path).unwrap();
        let content = "test write content".to_string();

        let result = FileUtil::write_string(&temp_path, content.clone());
        assert!(result.is_ok());

        let read_result = FileUtil::read_string(&temp_path);
        assert!(read_result.is_ok());
        assert_eq!(read_result.unwrap(), content);
    }

    #[test]
    fn test_append_string() {
        let temp_path = Path::new(root_dir).join("temp_file.txt");
        let mut temp_file = File::open(&temp_path).unwrap();
        let initial_content = "initial content".to_string();
        let append_content = " appended content".to_string();
        FileUtil::write_string(&temp_path, initial_content.clone()).expect("Failed to write initial content");

        let append_result = FileUtil::append_string(&temp_path, append_content.clone());
        assert!(append_result.is_ok());

        let read_result = FileUtil::read_string(&temp_path);
        assert!(read_result.is_ok());
        assert_eq!(read_result.unwrap(), initial_content + &append_content);
    }

    #[test]
    fn test_write_bytes() {
        let temp_path = Path::new(root_dir).join("temp_file.txt");
        let mut temp_file = File::open(&temp_path).unwrap();
        let content = [4, 5, 6];

        let result = FileUtil::write_bytes(&temp_path, &content);
        assert!(result.is_ok());

        let read_result = FileUtil::read_bytes(&temp_path);
        assert!(read_result.is_ok());
        assert_eq!(read_result.unwrap(), content);
    }
}