diff options
Diffstat (limited to 'src/permissions.rs')
| -rw-r--r-- | src/permissions.rs | 28 |
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() } } } |