--no-prompt flag for non-interactive environments (#1913)

1 files changed, 22 insertions, 6 deletions

diff --git a/src/permissions.rs b/src/permissions.rs
index 03ffd20cb..5f9588a1e 100644
--- a/src/permissions.rs
+++ b/src/permissions.rs
@@ -18,6 +18,7 @@ pub struct DenoPermissions {
   pub allow_net: AtomicBool,
   pub allow_env: AtomicBool,
   pub allow_run: AtomicBool,
+  pub no_prompts: AtomicBool,
 }
 
 impl DenoPermissions {
@@ -28,6 +29,7 @@ impl DenoPermissions {
       allow_env: AtomicBool::new(flags.allow_env),
       allow_net: AtomicBool::new(flags.allow_net),
       allow_run: AtomicBool::new(flags.allow_run),
+      no_prompts: AtomicBool::new(flags.no_prompts),
     }
   }
 
@@ -36,7 +38,7 @@ impl DenoPermissions {
       return Ok(());
     };
     // TODO get location (where access occurred)
-    let r = permission_prompt("access to run a subprocess");
+    let r = self.try_permissions_prompt("access to run a subprocess");
     if r.is_ok() {
       self.allow_run.store(true, Ordering::SeqCst);
     }
@@ -48,7 +50,8 @@ impl DenoPermissions {
       return Ok(());
     };
     // TODO get location (where access occurred)
-    let r = permission_prompt(&format!("read access to \"{}\"", filename));;
+    let r =
+      self.try_permissions_prompt(&format!("read access to \"{}\"", filename));;
     if r.is_ok() {
       self.allow_read.store(true, Ordering::SeqCst);
     }
@@ -60,7 +63,8 @@ impl DenoPermissions {
       return Ok(());
     };
     // TODO get location (where access occurred)
-    let r = permission_prompt(&format!("write access to \"{}\"", filename));;
+    let r =
+      self.try_permissions_prompt(&format!("write access to \"{}\"", filename));;
     if r.is_ok() {
       self.allow_write.store(true, Ordering::SeqCst);
     }
@@ -72,8 +76,10 @@ impl DenoPermissions {
       return Ok(());
     };
     // TODO get location (where access occurred)
-    let r =
-      permission_prompt(&format!("network access to \"{}\"", domain_name));
+    let r = self.try_permissions_prompt(&format!(
+      "network access to \"{}\"",
+      domain_name
+    ));
     if r.is_ok() {
       self.allow_net.store(true, Ordering::SeqCst);
     }
@@ -85,13 +91,22 @@ impl DenoPermissions {
       return Ok(());
     };
     // TODO get location (where access occurred)
-    let r = permission_prompt(&"access to environment variables");
+    let r = self.try_permissions_prompt(&"access to environment variables");
     if r.is_ok() {
       self.allow_env.store(true, Ordering::SeqCst);
     }
     r
   }
 
+  /// Try to present the user with a permission prompt
+  /// will error with permission_denied if no_prompts is enabled
+  fn try_permissions_prompt(&self, message: &str) -> DenoResult<()> {
+    if self.no_prompts.load(Ordering::SeqCst) {
+      return Err(permission_denied());
+    }
+    permission_prompt(message)
+  }
+
   pub fn allows_run(&self) -> bool {
     return self.allow_run.load(Ordering::SeqCst);
   }
@@ -144,6 +159,7 @@ impl DenoPermissions {
       allow_env: AtomicBool::new(false),
       allow_net: AtomicBool::new(false),
       allow_run: AtomicBool::new(false),
+      ..Default::default()
     }
   }
 }