databoxer 0.1.0-rc.1

//! Contains functions for path manipulation

use std::{ffi::OsString, fs};
use std::path::{Path, PathBuf};
use crate::core::encryption::boxfile;
use crate::core::error::IOErrorKind;
use crate::{log_error, log_info, log_warn, Error, Result};

/// Opens and parses provided path, returning a flattened list of all found paths. Verifies if the
/// given paths exists. In case of a directory being provided returns all paths inside of it. Can
/// be optionally be marked to search recursively all files within all inner directories.
pub fn parse_paths(input_paths: Vec<PathBuf>, recursive: bool) -> Vec<PathBuf> {
    let mut file_paths: Vec<PathBuf> = Vec::new();

    for path in input_paths {
        if path.is_dir() {
            if let Err(err) = read_dir(&path, &mut file_paths, recursive) {
                log_warn!("Unable to read directory \"{}\": {}", path.display(), err);
                continue;
            }
        } else if path.is_file() {
            file_paths.push(path);
        } else if !path.exists() {
            let target_name = path.file_stem().unwrap().to_os_string();

            match search_for_original(path.parent().unwrap(), target_name) {
                Ok(box_path) => file_paths.push(box_path),
                Err(err) => {
                    log_error!("Unable to find \"{}\" ({})", path.display(), err);
                    continue;
                }
            }
        }
    }
    file_paths
}

fn read_dir(dir_path: &Path, file_paths: &mut Vec<PathBuf>, recursive: bool) -> Result<()> {
    for entry in fs::read_dir(dir_path)? {
        let path = entry?.path();

        if path.is_dir() && recursive {
            read_dir(&path, file_paths, true)?;
        } else if path.is_file() {
            file_paths.push(path);
        }
    }

    Ok(())
}

/// Searches `.box` files within a directory for one which matches its original name with provided
fn search_for_original(dir_path: &Path, target_name: OsString) -> Result<PathBuf> {
    for entry in fs::read_dir(dir_path)? {
        let path = entry?.path();

        if !path.is_file() || path.extension().unwrap() != "box" { continue; }

        let boxfile = boxfile::Boxfile::parse(&path)?;
        let (original_name, _) = boxfile.file_info();

        if target_name.eq(original_name) {
            log_info!("Found an encrypted (.box) file with the same original name: {}", path.display());
            return Ok(path)
        }
    }

    Err(Error::IOError(IOErrorKind::NotFound(target_name.into())))
}