novel-cli 0.17.0

use std::path::{Path, PathBuf};

use color_eyre::eyre::{self, Result};

pub fn ensure_executable_exists<T>(name: T) -> Result<()>
where
    T: AsRef<str>,
{
    let name = name.as_ref();

    if let Err(error) = which::which(name) {
        eyre::bail!("{}: `{}`", error, name);
    }

    Ok(())
}

pub fn ensure_epub_file<T>(path: T) -> Result<()>
where
    T: AsRef<Path>,
{
    if !is_epub_file(&path)? {
        eyre::bail!("File `{}` is not epub file", path.as_ref().display())
    }

    Ok(())
}

pub fn ensure_epub_dir<T>(path: T) -> Result<()>
where
    T: AsRef<Path>,
{
    if !is_epub_dir(&path)? {
        eyre::bail!(
            "Directory `{}` is not epub directory",
            path.as_ref().display()
        )
    }

    Ok(())
}

pub fn is_markdown_or_txt_file<T>(path: T) -> Result<bool>
where
    T: AsRef<Path>,
{
    Ok(is_some_file(&path, "md")? || is_some_file(&path, "txt")?)
}

pub fn is_epub_file<T>(path: T) -> Result<bool>
where
    T: AsRef<Path>,
{
    is_some_file(path, "epub")
}

pub fn try_get_markdown_or_txt_file_name_in_dir<T>(path: T) -> Result<Option<PathBuf>>
where
    T: AsRef<Path>,
{
    let path = path.as_ref();

    ensure_exists(path)?;

    let markdown_file_name = PathBuf::from(path.file_stem().unwrap()).with_extension("md");
    let markdown_file_path = path.join(markdown_file_name);

    let txt_file_name = PathBuf::from(path.file_stem().unwrap()).with_extension("txt");
    let txt_file_path = path.join(txt_file_name);

    if markdown_file_path.is_file() && txt_file_path.is_file() {
        eyre::bail!("Both markdown and txt files exist in the directory");
    }

    if markdown_file_path.is_file() {
        Ok(Some(dunce::canonicalize(markdown_file_path)?))
    } else if txt_file_path.is_file() {
        Ok(Some(dunce::canonicalize(txt_file_path)?))
    } else {
        Ok(None)
    }
}

pub fn is_mdbook_dir<T>(path: T) -> Result<bool>
where
    T: AsRef<Path>,
{
    let path = path.as_ref();

    ensure_exists(path)?;

    if !path.is_dir() {
        return Ok(false);
    }

    let src_path = path.join("src");
    let toml_path = path.join("book.toml");

    Ok(src_path.is_dir() && toml_path.is_file())
}

pub fn is_epub_dir<T>(path: T) -> Result<bool>
where
    T: AsRef<Path>,
{
    let path = path.as_ref();

    ensure_exists(path)?;

    if !path.is_dir() {
        return Ok(false);
    }

    let epub_path = path.join("EPUB");
    let meta_path = path.join("META-INF");
    let mimetype_path = path.join("mimetype");

    Ok(epub_path.is_dir() && meta_path.is_dir() && mimetype_path.is_file())
}

fn is_some_file<T, E>(path: T, extension: E) -> Result<bool>
where
    T: AsRef<Path>,
    E: AsRef<str>,
{
    let path = path.as_ref();

    ensure_exists(path)?;

    if !path.is_file() {
        return Ok(false);
    }

    Ok(path
        .extension()
        .is_some_and(|ext| ext == extension.as_ref()))
}

fn ensure_exists<T>(path: T) -> Result<()>
where
    T: AsRef<Path>,
{
    let path = path.as_ref();

    eyre::ensure!(
        path.try_exists()?,
        "File or directory `{}` does not exist",
        path.display()
    );

    Ok(())
}

pub fn ensure_pandoc_dir<T>(path: T) -> Result<()>
where
    T: AsRef<Path>,
{
    if !is_pandoc_dir(&path)? {
        eyre::bail!(
            "Directory `{}` is not pandoc directory",
            path.as_ref().display()
        )
    }

    Ok(())
}

pub fn ensure_mdbook_dir<T>(path: T) -> Result<()>
where
    T: AsRef<Path>,
{
    if !is_mdbook_dir(&path)? {
        eyre::bail!(
            "Directory `{}` is not mdbook directory",
            path.as_ref().display()
        )
    }

    Ok(())
}

pub fn is_pandoc_dir<T>(path: T) -> Result<bool>
where
    T: AsRef<Path>,
{
    let path = path.as_ref();

    ensure_exists(path)?;

    if !path.is_dir() {
        return Ok(false);
    }

    let markdown = path.join(path.file_stem().unwrap()).with_extension("md");

    Ok(markdown.is_file())
}