You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.0 KiB
39 lines
1.0 KiB
3 years ago
|
use std::{env::temp_dir, fs, ops::Drop, os::unix::fs::PermissionsExt, path::PathBuf};
|
||
|
|
use uuid::Uuid;
|
||
|
|
|
||
|
|
pub struct TempFile {
|
||
|
|
path: PathBuf,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl TempFile {
|
||
|
|
pub fn get_path(&self) -> String {
|
||
|
|
self.path.to_string_lossy().to_string()
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn new() -> Self {
|
||
|
|
let name = Uuid::simple(&Uuid::new_v4()).to_string();
|
||
|
|
let mut path = temp_dir();
|
||
|
|
path.push(name);
|
||
|
|
Self { path }
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn write_all(&self, data: &[u8]) -> Result<(), String> {
|
||
|
|
fs::write(&self.path, data).map_err(|e| e.to_string())
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn write_exec(data: &[u8]) -> Result<Self, (String, String)> {
|
||
|
|
let this = Self::new();
|
||
|
|
let path = this.get_path();
|
||
|
|
this.write_all(data).map_err(|e| (path.clone(), e))?;
|
||
|
|
let perms = fs::Permissions::from_mode(0o555);
|
||
|
|
fs::set_permissions(&path, perms).map_err(|e| (path, e.to_string()))?;
|
||
|
|
Ok(this)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Drop for TempFile {
|
||
|
|
fn drop(&mut self) {
|
||
|
|
fs::remove_file(&self.path).ok();
|
||
|
|
}
|
||
|
|
}
|