use std::env::var;
use std::path::{Path, PathBuf};
use std::sync::OnceLock;
use crate::normalize_path;
use crate::python_util::get_sys_path;
use crate::style::colors::*;
use crate::style::RESET;
fn fallback_erg_path() -> PathBuf {
#[allow(deprecated)]
std::env::home_dir().map_or(PathBuf::from(".erg"), |path| path.join(".erg"))
}
fn _erg_path() -> PathBuf {
let path = var("ERG_PATH").unwrap_or_else(|_| env!("CARGO_ERG_PATH").to_string());
PathBuf::from(path).canonicalize().unwrap_or_else(|_| {
let fallback = fallback_erg_path();
if !fallback.exists() {
eprintln!("{RED}[ERR] ERG_PATH not found{RESET}");
}
fallback
})
}
fn _erg_std_path() -> PathBuf {
_erg_path()
.join("lib")
.join("std")
.canonicalize()
.unwrap_or_else(|_| {
eprintln!("{RED}[ERR] ERG_PATH/lib/std not found{RESET}");
fallback_erg_path().join("lib/std")
})
}
fn _erg_std_decl_path() -> PathBuf {
_erg_path()
.join("lib")
.join("std.d")
.canonicalize()
.unwrap_or_else(|_| {
eprintln!("{RED}[ERR] ERG_PATH/lib/std.d not found {RESET}");
fallback_erg_path().join("lib/std.d")
})
}
fn _erg_pystd_path() -> PathBuf {
_erg_path()
.join("lib")
.join("pystd")
.canonicalize()
.unwrap_or_else(|_| {
eprintln!("{RED}[ERR] ERG_PATH/lib/pystd not found {RESET}");
fallback_erg_path().join("lib/pystd")
})
}
fn _erg_external_lib_path() -> PathBuf {
_erg_path()
.join("lib")
.join("external")
.canonicalize()
.unwrap_or_else(|_| {
eprintln!("{RED}[ERR] ERG_PATH/lib/external not found {RESET}");
fallback_erg_path().join("lib/external")
})
}
fn _python_site_packages() -> impl Iterator<Item = PathBuf> {
get_sys_path(None)
.unwrap_or_default()
.into_iter()
.filter(|p| p.ends_with("site-packages"))
.map(|p| {
p.canonicalize().unwrap_or_else(|_| {
eprintln!("{RED}[ERR] ERG_PATH/lib/external not found {RESET}");
fallback_erg_path().join("lib/external")
})
})
}
pub static ERG_PATH: OnceLock<PathBuf> = OnceLock::new();
pub static ERG_STD_PATH: OnceLock<PathBuf> = OnceLock::new();
pub static ERG_STD_DECL_PATH: OnceLock<PathBuf> = OnceLock::new();
pub static ERG_PYSTD_PATH: OnceLock<PathBuf> = OnceLock::new();
pub static ERG_EXTERNAL_LIB_PATH: OnceLock<PathBuf> = OnceLock::new();
pub static PYTHON_SITE_PACKAGES: OnceLock<Vec<PathBuf>> = OnceLock::new();
pub fn erg_path() -> &'static PathBuf {
ERG_PATH.get_or_init(|| normalize_path(_erg_path())) }
pub fn erg_std_path() -> &'static PathBuf {
ERG_STD_PATH.get_or_init(|| normalize_path(_erg_std_path()))
}
pub fn erg_std_decl_path() -> &'static PathBuf {
ERG_STD_DECL_PATH.get_or_init(|| normalize_path(_erg_std_decl_path()))
}
pub fn erg_pystd_path() -> &'static PathBuf {
ERG_PYSTD_PATH.get_or_init(|| normalize_path(_erg_pystd_path()))
}
pub fn erg_py_external_lib_path() -> &'static PathBuf {
ERG_EXTERNAL_LIB_PATH.get_or_init(|| normalize_path(_erg_external_lib_path()))
}
pub fn python_site_packages() -> &'static Vec<PathBuf> {
PYTHON_SITE_PACKAGES.get_or_init(|| _python_site_packages().collect())
}
pub fn is_std_decl_path(path: &Path) -> bool {
path.starts_with(erg_pystd_path().as_path())
|| path.starts_with(erg_std_decl_path().as_path())
|| path.starts_with(erg_py_external_lib_path().as_path())
}
pub fn is_pystd_main_module(path: &Path) -> bool {
let mut path = PathBuf::from(path);
if path.ends_with("__init__.d.er") {
path.pop();
path.pop();
} else {
path.pop();
}
path == erg_pystd_path().as_path()
}