use std::path::Path;

/// Return `true` if the [`Path`] is named `pyproject.toml`.
pub fn is_pyproject_toml(path: &Path) -> bool {
    path.file_name()
        .is_some_and(|name| name == "pyproject.toml")
}

/// Return `true` if the [`Path`] appears to be that of a Python interface definition file (`.pyi`).
pub fn is_python_stub_file(path: &Path) -> bool {
    path.extension().is_some_and(|ext| ext == "pyi")
}

/// Return `true` if the [`Path`] appears to be that of a Jupyter notebook (`.ipynb`).
pub fn is_jupyter_notebook(path: &Path) -> bool {
    path.extension().is_some_and(|ext| ext == "ipynb")
}

/// Return `true` if a [`Path`] should use the name of its parent directory as its module name.
pub fn is_module_file(path: &Path) -> bool {
    path.file_name().is_some_and(|file_name| {
        file_name == "__init__.py"
            || file_name == "__init__.pyi"
            || file_name == "__main__.py"
            || file_name == "__main__.pyi"
    })
}

#[cfg(test)]
mod tests {
    use std::path::Path;

    use crate::path::is_jupyter_notebook;

    #[test]
    fn test_is_jupyter_notebook() {
        let path = Path::new("foo/bar/baz.ipynb");
        assert!(is_jupyter_notebook(path));

        let path = Path::new("foo/bar/baz.py");
        assert!(!is_jupyter_notebook(path));
    }
}