1 files changed, 40 insertions, 0 deletions
diff --git a/cli/ops/mod.rs b/cli/ops/mod.rs
index 06a55bade..3858400a3 100644
--- a/cli/ops/mod.rs
+++ b/cli/ops/mod.rs
@@ -29,3 +29,43 @@ pub mod tty;
 pub mod web_worker;
 pub mod websocket;
 pub mod worker_host;
+
+use crate::metrics::metrics_op;
+use deno_core::json_op_async;
+use deno_core::json_op_sync;
+use deno_core::BufVec;
+use deno_core::ErrBox;
+use deno_core::JsRuntime;
+use deno_core::OpState;
+use deno_core::ZeroCopyBuf;
+use serde_json::Value;
+use std::cell::RefCell;
+use std::future::Future;
+use std::rc::Rc;
+
+pub fn reg_json_async<F, R>(rt: &mut JsRuntime, name: &'static str, op_fn: F)
+where
+  F: Fn(Rc<RefCell<OpState>>, Value, BufVec) -> R + 'static,
+  R: Future<Output = Result<Value, ErrBox>> + 'static,
+{
+  rt.register_op(name, metrics_op(json_op_async(op_fn)));
+}
+
+pub fn reg_json_sync<F>(rt: &mut JsRuntime, name: &'static str, op_fn: F)
+where
+  F: Fn(&mut OpState, Value, &mut [ZeroCopyBuf]) -> Result<Value, ErrBox>
+    + 'static,
+{
+  rt.register_op(name, metrics_op(json_op_sync(op_fn)));
+}
+
+/// Helper for extracting the commonly used state. Used for sync ops.
+pub fn cli_state(state: &OpState) -> Rc<crate::state::State> {
+  state.borrow::<Rc<crate::state::State>>().clone()
+}
+
+/// Helper for extracting the commonly used state. Used for async ops.
+pub fn cli_state2(state: &Rc<RefCell<OpState>>) -> Rc<crate::state::State> {
+  let state = state.borrow();
+  state.borrow::<Rc<crate::state::State>>().clone()
+}