NyxErinys/crunchy-cli
1
0
Fork 0
mirror of https://github.com/crunchy-labs/crunchy-cli.git synced 2026-01-21 12:12:00 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add function to get temp directory

Browse source
This commit is contained in:
ByteDream 2023-04-09 15:52:21 +02:00
parent a2820c0399
commit ad06096f34
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
crunchy-cli-core/src/utils/os.rs
View file

@ -19,17 +19,21 @@ pub fn has_ffmpeg() -> bool {
} }
} }
/// Get the temp directory either by the specified `CRUNCHY_CLI_TEMP_DIR` env variable or the dir
/// provided by the os.
pub fn temp_directory() -> PathBuf {
env::var("CRUNCHY_CLI_TEMP_DIR").map_or(env::temp_dir(), |d| PathBuf::from(d))
}
/// Any tempfile should be created with this function. The prefix and directory of every file /// Any tempfile should be created with this function. The prefix and directory of every file
/// created with this method stays the same which is helpful to query all existing tempfiles and /// created with this method stays the same which is helpful to query all existing tempfiles and
/// e.g. remove them in a case of ctrl-c. Having one function also good to prevent mistakes like /// e.g. remove them in a case of ctrl-c. Having one function also good to prevent mistakes like
/// setting the wrong prefix if done manually. /// setting the wrong prefix if done manually.
pub fn tempfile<S: AsRef<str>>(suffix: S) -> io::Result<NamedTempFile> { pub fn tempfile<S: AsRef<str>>(suffix: S) -> io::Result<NamedTempFile> {
let tmp_dir = env::var("CRUNCHY_CLI_TEMP_DIR").map_or(env::temp_dir(), |d| PathBuf::from(d));
let tempfile = Builder::default() let tempfile = Builder::default()
.prefix(".crunchy-cli_") .prefix(".crunchy-cli_")
.suffix(suffix.as_ref()) .suffix(suffix.as_ref())
.tempfile_in(tmp_dir)?; .tempfile_in(temp_directory())?;
debug!( debug!(
"Created temporary file: {}", "Created temporary file: {}",
tempfile.path().to_string_lossy() tempfile.path().to_string_lossy()