hugo_to_json 0.3.9

use crate::constants;
use crate::operation_result::*;

use std::fmt;
use std::path::{Component, PathBuf};

use walkdir::DirEntry;

#[derive(Debug)]
pub struct FileLocation {
    /// The file's extension
    pub extension: String,
    /// The file's absolute path
    pub absolute_path: String,
    /// The name of the file including the extension
    pub file_name: String,
    /// The name of the file, without the extension. `foo.rs` becomes `foo`
    pub file_stem: String,
    // The directory relative to the content directory. Given `./blog/content/sub/post/example.md` and a content_dir of `./blog/content` produce `sub/post`
    pub relative_directory_to_content: String,
}

impl FileLocation {
    pub fn new(file: &DirEntry, content_dir: &PathBuf) -> Result<Self, OperationResult> {
        if file.file_type().is_dir() {
            return Err(OperationResult::Skip(Skip::new(
                file.path().to_str().unwrap_or_default(),
                "Not a file",
            )));
        }

        let path = file.path();
        let absolute_path = path.to_string_lossy().into_owned();
        let extension = file
            .path()
            .extension()
            .ok_or_else(|| PathError::new(&absolute_path, "Failed to determine extension."))?;
        let file_name = path
            .file_name()
            .ok_or_else(|| PathError::new(&absolute_path, "Failed to retrieve file name."))?;
        let file_stem = path
            .file_stem()
            .ok_or_else(|| PathError::new(&absolute_path, "Failed to retrieve file stem."))?;

        // Get the subdirectory path. Given ./blog/content/sub/post/example.md and a root_dir of ./blog/content produce sub/post
        let relative_directory_to_content: String = path
            .strip_prefix(content_dir)
            .map_err(|_| PathError::new(&absolute_path, "Failed to retrieve sub directory."))?
            .components()
            .take_while(|comp: &Component| comp.as_os_str() != file_name)
            .filter_map(|comp: Component| comp.as_os_str().to_str())
            .collect::<Vec<&str>>()
            .join(constants::FORWARD_SLASH);

        let extension = extension.to_string_lossy().into_owned();
        let file_name = file_name.to_string_lossy().into_owned();
        let file_stem = file_stem.to_string_lossy().into_owned();

        Ok(Self {
            extension,
            absolute_path,
            file_name,
            file_stem,
            relative_directory_to_content,
        })
    }
}

impl fmt::Display for FileLocation {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{}", self.absolute_path)
    }
}

#[cfg(test)]
mod tests {}