diff options
Diffstat (limited to 'runtime/ops/os.rs')
-rw-r--r-- | runtime/ops/os.rs | 56 |
1 files changed, 38 insertions, 18 deletions
diff --git a/runtime/ops/os.rs b/runtime/ops/os.rs index f138bae31..654bbede1 100644 --- a/runtime/ops/os.rs +++ b/runtime/ops/os.rs @@ -4,32 +4,36 @@ use super::utils::into_string; use crate::permissions::Permissions; use crate::worker::ExitCode; use deno_core::error::{type_error, AnyError}; -use deno_core::op; use deno_core::url::Url; use deno_core::Extension; use deno_core::OpState; +use deno_core::{op, ExtensionBuilder}; use serde::Serialize; use std::collections::HashMap; use std::env; +fn init_ops(builder: &mut ExtensionBuilder) -> &mut ExtensionBuilder { + builder.ops(vec![ + op_env::decl(), + op_exec_path::decl(), + op_exit::decl(), + op_delete_env::decl(), + op_get_env::decl(), + op_getgid::decl(), + op_getuid::decl(), + op_hostname::decl(), + op_loadavg::decl(), + op_network_interfaces::decl(), + op_os_release::decl(), + op_set_env::decl(), + op_set_exit_code::decl(), + op_system_memory_info::decl(), + ]) +} + pub fn init(exit_code: ExitCode) -> Extension { - Extension::builder() - .ops(vec![ - op_env::decl(), - op_exec_path::decl(), - op_exit::decl(), - op_delete_env::decl(), - op_get_env::decl(), - op_getgid::decl(), - op_getuid::decl(), - op_hostname::decl(), - op_loadavg::decl(), - op_network_interfaces::decl(), - op_os_release::decl(), - op_set_env::decl(), - op_set_exit_code::decl(), - op_system_memory_info::decl(), - ]) + let mut builder = Extension::builder(); + init_ops(&mut builder) .state(move |state| { state.put::<ExitCode>(exit_code.clone()); Ok(()) @@ -37,6 +41,22 @@ pub fn init(exit_code: ExitCode) -> Extension { .build() } +pub fn init_for_worker() -> Extension { + let mut builder = Extension::builder(); + init_ops(&mut builder) + .middleware(|op| match op.name { + "op_exit" => noop_op::decl(), + "op_set_exit_code" => noop_op::decl(), + _ => op, + }) + .build() +} + +#[op] +fn noop_op(_code: i32) -> Result<(), AnyError> { + Ok(()) +} + #[op] fn op_exec_path(state: &mut OpState) -> Result<String, AnyError> { let current_exe = env::current_exe().unwrap(); |