add exec root option

src/args.rs

@@ -25,6 +25,7 @@ pub enum HazeArgs {
         filter: Option<String>,
         service: Option<ExecService>,
         command: Vec<String>,
+        root: bool,
     },
     Occ {
         filter: Option<String>,
@@ -180,11 +181,20 @@ impl HazeArgs {
                     _ => None,
                 };
 
+                let root = match args.peek() {
+                    Some(arg) if arg.as_ref() == "root" => {
+                        args.next();
+                        true
+                    }
+                    _ => false,
+                };
+
                 let command = args.map(S::into).collect();
                 Ok(HazeArgs::Exec {
                     filter,
                     service,
                     command,
+                    root,
                 })
             }
             HazeCommand::Occ => Ok(HazeArgs::Occ {
@@ -508,6 +518,20 @@ fn test_arg_parse() {
             filter: None,
             service: None,
             command: vec!["foo".to_string(), "bar".to_string()],
+            root: false,
+        }
+    );
+    assert_eq!(
+        HazeArgs::parse(
+            &config,
+            vec!["haze", "exec", "root", "foo", "bar"].into_iter()
+        )
+        .unwrap(),
+        HazeArgs::Exec {
+            filter: None,
+            service: None,
+            command: vec!["foo".to_string(), "bar".to_string()],
+            root: true,
         }
     );
     assert_eq!(
@@ -520,6 +544,7 @@ fn test_arg_parse() {
             filter: Some("asdasd".to_string()),
             service: None,
             command: vec!["foo".to_string(), "bar".to_string()],
+            root: false,
         }
     );
     assert_eq!(
@@ -532,6 +557,7 @@ fn test_arg_parse() {
             filter: Some("asdasd".to_string()),
             service: Some(ExecService::Db),
             command: vec!["foo".to_string(), "bar".to_string()],
+            root: false,
         }
     );
     assert_eq!(

src/main.rs

@@ -4,7 +4,7 @@ use crate::args::{ExecService, HazeArgs};
 use crate::cloud::{Cloud, CloudOptions};
 use crate::config::HazeConfig;
 use crate::database::DatabaseFamily;
-use crate::exec::container_logs;
+use crate::exec::{container_logs, exec, exec_tty};
 use crate::git::checkout_all;
 use crate::help::help;
 use crate::image::update_image;
@@ -117,22 +117,24 @@ async fn main() -> Result<ExitCode> {
             filter,
             service,
             command,
+            root,
         } => {
             let cloud = Cloud::get_by_filter(&docker, filter, &config).await?;
             match service {
                 None => {
-                    cloud
+                    let command = if command.is_empty() {
-                        .exec(
+                        vec!["bash".to_string()]
-                            &docker,
+                    } else {
-                            if command.is_empty() {
+                        command
-                                vec!["bash".to_string()]
+                    };
-                            } else {
+                    let env = get_forward_env();
-                                command
+                    let tty = atty::is(atty::Stream::Stdout);
-                            },
+                    let user = if root { "root" } else { "haze" };
-                            atty::is(atty::Stream::Stdout),
+                    if tty {
-                            get_forward_env(),
+                        exec_tty(&docker, &cloud.id, user, command, env).await?;
-                        )
+                    } else {
-                        .await?;
+                        exec(&docker, &cloud.id, user, command, env, Some(stdout())).await?;
+                    }
                 }
                 Some(ExecService::Db) => {
                     cloud