eliott/ort
1
0
Fork 0
mirror of https://github.com/pykeio/ort synced 2026-04-25 16:34:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat(sys): allow overriding cache dir with environment variable

Browse Source
This commit is contained in:
Carson M.
2025-08-07 12:27:54 -05:00
parent 5f96a2d585
commit f3cd995d06
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ort-sys/src/internal/dirs.rs
View File

@@ -2,6 +2,8 @@
#![allow(unused)]
use std::path::PathBuf;
pub const PYKE_ROOT: &str = "ort.pyke.io";
#[cfg(all(target_os = "windows", target_arch = "x86"))]
@@ -95,7 +97,7 @@ mod windows {
}
#[cfg(target_os = "windows")]
#[must_use]
pub fn cache_dir() -> Option<std::path::PathBuf> {
fn cache_dir_default() -> Option<PathBuf> {
self::windows::known_folder_local_app_data().map(|h| h.join(PYKE_ROOT))
}
@@ -174,7 +176,7 @@ mod unix {
#[cfg(target_os = "linux")]
#[must_use]
pub fn cache_dir() -> Option<std::path::PathBuf> {
fn cache_dir_default() -> Option<PathBuf> {
std::env::var_os("XDG_CACHE_HOME")
.and_then(self::unix::is_absolute_path)
.or_else(|| self::unix::home_dir().map(|h| h.join(".cache").join(PYKE_ROOT)))
@@ -182,6 +184,10 @@ pub fn cache_dir() -> Option<std::path::PathBuf> {
#[cfg(target_os = "macos")]
#[must_use]
pub fn cache_dir() -> Option<std::path::PathBuf> {
fn cache_dir_default() -> Option<PathBuf> {
self::unix::home_dir().map(|h| h.join("Library/Caches").join(PYKE_ROOT))
}
pub fn cache_dir() -> Option<PathBuf> {
std::env::var_os("ORT_CACHE_DIR").map(PathBuf::from).or_else(cache_dir_default)
}