From 118467e07989d8c2abc42a0f87745811649635a8 Mon Sep 17 00:00:00 2001
From: veeso <christian.visintin1997@gmail.com>
Date: Thu, 4 Mar 2021 16:24:34 +0100
Subject: [PATCH] Exec on Localhos

---
 src/host/mod.rs | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/src/host/mod.rs b/src/host/mod.rs
index 3f78b36..841d1a1 100644
--- a/src/host/mod.rs
+++ b/src/host/mod.rs
@@ -46,6 +46,7 @@ pub enum HostErrorType {
     FileNotAccessible,
     FileAlreadyExists,
     CouldNotCreateFile,
+    ExecutionFailed,
     DeleteFailed,
 }
 
@@ -79,6 +80,7 @@ impl std::fmt::Display for HostError {
             HostErrorType::FileNotAccessible => "Could not access file",
             HostErrorType::FileAlreadyExists => "File already exists",
             HostErrorType::CouldNotCreateFile => "Could not create file",
+            HostErrorType::ExecutionFailed => "Could not run command",
             HostErrorType::DeleteFailed => "Could not delete file",
         };
         match &self.ioerr {
@@ -142,8 +144,13 @@ impl Localhost {
         if !self.file_exists(new_dir.as_path()) {
             return Err(HostError::new(HostErrorType::NoSuchFileOrDirectory, None));
         }
+        // Change directory
+        if std::env::set_current_dir(new_dir.as_path()).is_err() {
+            return Err(HostError::new(HostErrorType::NoSuchFileOrDirectory, None));
+        }
         let prev_dir: PathBuf = self.wrkdir.clone(); // Backup location
                                                      // Update working directory
+                                                     // Change dir
         self.wrkdir = new_dir;
         // Scan new directory
         self.files = match self.scan_dir(self.wrkdir.as_path()) {
@@ -430,6 +437,23 @@ impl Localhost {
         })
     }
 
+    /// ### exec
+    ///
+    /// Execute a command on localhost
+    pub fn exec(&self, cmd: &str) -> Result<String, HostError> {
+        // Make command
+        let args: Vec<&str> = cmd.split(' ').collect();
+        let cmd: &str = args.first().unwrap();
+        let argv: &[&str] = &args[1..];
+        match std::process::Command::new(cmd).args(argv).output() {
+            Ok(output) => match std::str::from_utf8(&output.stdout) {
+                Ok(s) => Ok(s.to_string()),
+                Err(_) => Ok(String::new()),
+            },
+            Err(err) => Err(HostError::new(HostErrorType::ExecutionFailed, Some(err))),
+        }
+    }
+
     /// ### chmod
     ///
     /// Change file mode to file, according to UNIX permissions
@@ -942,6 +966,14 @@ mod tests {
         assert!(host.stat(test_file_path.as_path()).is_ok());
     }
 
+    #[test]
+    fn test_host_exec() {
+        let tmpdir: tempfile::TempDir = tempfile::TempDir::new().unwrap();
+        let host: Localhost = Localhost::new(PathBuf::from(tmpdir.path())).ok().unwrap();
+        // Execute
+        assert_eq!(host.exec("echo 5").ok().unwrap().as_str(), "5\n");
+    }
+
     #[test]
     fn test_host_fmt_error() {
         let err: HostError = HostError::new(
@@ -956,6 +988,10 @@ mod tests {
             format!("{}", HostError::new(HostErrorType::DeleteFailed, None)),
             String::from("Could not delete file")
         );
+        assert_eq!(
+            format!("{}", HostError::new(HostErrorType::ExecutionFailed, None)),
+            String::from("Could not run command")
+        );
         assert_eq!(
             format!("{}", HostError::new(HostErrorType::DirNotAccessible, None)),
             String::from("Could not access directory")