adding file.temp_file to eldritch

docs/_docs/user-guide/eldritch.md

@@ -381,6 +381,12 @@ The <b>file.replace</b> method finds the first string matching a regex pattern i
 
 The <b>file.replace_all</b> method finds all strings matching a regex pattern in the specified file and replaces them with the value. Please consult the [Rust Regex Docs](https://rust-lang-nursery.github.io/rust-cookbook/text/regex.html) for more information on pattern matching.
 
+### file.tmp_file
+
+`file.tmp_file(name: str) -> str`
+
+The <b> file.temp</b> method returns the path of a new temporary file with the name provided as the argument.
+
 ### file.template
 
 `file.template(template_path: str, dst: str, args: Dict<String, Value>, autoescape: bool) -> None`

implants/lib/eldritch/src/file/mod.rs

@@ -14,6 +14,7 @@ mod read_impl;
 mod remove_impl;
 mod replace_all_impl;
 mod replace_impl;
+mod temp_file_impl;
 mod template_impl;
 mod timestomp_impl;
 mod write_impl;
@@ -176,4 +177,10 @@ fn methods(builder: &mut MethodsBuilder) {
         follow_impl::follow(path, f, eval)?;
         Ok(NoneType{})
     }
+
+    #[allow(unused_variables)]
+    fn temp_file(this: &FileLibrary, name: String) -> anyhow::Result<String> {
+        temp_file_impl::temp_file(name)
+    }
+
 }

implants/lib/eldritch/src/file/temp_file_impl.rs

@@ -0,0 +1,46 @@
+use anyhow::Result;
+use std::fs;
+use std::path::PathBuf;
+use tempfile::NamedTempFile;
+
+pub fn temp_file(name: String) -> Result<String> {
+    //create a file in temp folder
+    let tmp_file = NamedTempFile::new()?;
+    let tdir: PathBuf = tmp_file.path().parent().unwrap().into();
+    let new_path = tdir.join(name);
+    let (_tf, tpath) = tmp_file.keep()?;
+
+    fs::rename(&tpath, &new_path)?;
+
+    Ok(String::from(new_path.to_str().unwrap()).clone())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use std::path::Path;
+
+    #[test]
+    fn test_temp_file() -> anyhow::Result<()> {
+        // Create file
+        let p = temp_file("foo".to_string())?;
+        // check if file exists
+        assert!(Path::new(&p).exists());
+
+        Ok(())
+    }
+    #[test]
+    fn test_temp_no_file() -> anyhow::Result<()> {
+        // Create file and then delete it (so we know it doesnt exist)
+        let p = temp_file("foo".to_string())?;
+        if Path::new(&p).exists() {
+            // delete the file
+            fs::remove_file(&p)?;
+        }
+
+        // check if file exists
+        assert!(!Path::new(&p).exists());
+
+        Ok(())
+    }
+}

implants/lib/eldritch/src/runtime/mod.rs

@@ -89,7 +89,7 @@ mod tests {
                 parameters: HashMap::new(),
                 file_names: Vec::new(),
             },
-            want_text: format!("{}\n", r#"["append", "compress", "copy", "exists", "find", "follow", "is_dir", "is_file", "list", "mkdir", "moveto", "parent_dir", "read", "remove", "replace", "replace_all", "template", "timestomp", "write"]"#),
+            want_text: format!("{}\n", r#"["append", "compress", "copy", "exists", "find", "follow", "is_dir", "is_file", "list", "mkdir", "moveto", "parent_dir", "read", "remove", "replace", "replace_all", "temp_file", "template", "timestomp", "write"]"#),
             want_error: None,
         },
         process_bindings: TestCase {