use std::cell::RefCell;
use std::net::IpAddr;
use std::net::SocketAddr;
use std::rc::Rc;
use deno_core::GarbageCollected;
use deno_core::InspectorMsg;
use deno_core::InspectorSessionKind;
use deno_core::JsRuntimeInspector;
use deno_core::ModuleSpecifier;
use deno_core::OpState;
use deno_core::op2;
use deno_core::v8;
use deno_inspector_server::InspectPublishUid;
use deno_inspector_server::InspectorServerUrl;
use deno_inspector_server::create_inspector_server;
use deno_inspector_server::stop_inspector_server;
use deno_permissions::PermissionsContainer;
#[op2(fast)]
pub fn op_inspector_enabled(state: &OpState) -> bool {
state.try_borrow::<InspectorServerUrl>().is_some()
}
#[op2(stack_trace)]
pub fn op_inspector_open(
state: &mut OpState,
port: Option<u16>,
#[string] host: Option<String>,
wait_for_session: bool,
) -> Result<(), InspectorOpenError> {
const DEFAULT_HOST: IpAddr =
IpAddr::V4(std::net::Ipv4Addr::new(127, 0, 0, 1));
const DEFAULT_PORT: u16 = 9229;
let host_ip: IpAddr = match &host {
Some(h) => h.parse().map_err(|e| {
InspectorOpenError::InvalidHost(format!(
"Invalid inspector host '{}': {}",
h, e
))
})?,
None => DEFAULT_HOST,
};
let port = port.unwrap_or(DEFAULT_PORT);
let addr = SocketAddr::new(host_ip, port);
state
.borrow_mut::<PermissionsContainer>()
.check_net(&(host_ip.to_string(), Some(port)), "inspector.open")?;
let server =
create_inspector_server(addr, "deno", InspectPublishUid::default())?;
let inspector = state.borrow::<Rc<JsRuntimeInspector>>().clone();
let main_module = state.borrow::<ModuleSpecifier>().to_string();
let inspector_url =
server.register_inspector(main_module, inspector, wait_for_session);
state.put(inspector_url);
Ok(())
}
#[op2(fast)]
pub fn op_inspector_close(state: &mut OpState) {
stop_inspector_server();
state.try_take::<InspectorServerUrl>();
}
#[op2]
#[string]
pub fn op_inspector_url(
state: &mut OpState,
) -> Result<Option<String>, InspectorConnectError> {
state
.borrow_mut::<PermissionsContainer>()
.check_sys("inspector", "inspector.url")?;
Ok(
state
.try_borrow::<InspectorServerUrl>()
.map(|url| url.0.to_string()),
)
}
#[op2(fast)]
pub fn op_inspector_wait(state: &OpState) -> bool {
match state.try_borrow::<Rc<JsRuntimeInspector>>() {
Some(inspector) => {
inspector.wait_for_session_and_break_on_next_statement();
true
}
None => false,
}
}
#[op2(fast)]
pub fn op_inspector_emit_protocol_event(
#[string] _event_name: String,
#[string] _params: String,
) {
}
struct JSInspectorSession {
session: RefCell<Option<deno_core::LocalInspectorSession>>,
}
unsafe impl GarbageCollected for JSInspectorSession {
fn trace(&self, _visitor: &mut deno_core::v8::cppgc::Visitor) {}
fn get_name(&self) -> &'static std::ffi::CStr {
c"JSInspectorSession"
}
}
#[derive(Debug, thiserror::Error, deno_error::JsError)]
pub enum InspectorOpenError {
#[class(inherit)]
#[error(transparent)]
Permission(
#[from]
#[inherit]
deno_permissions::PermissionCheckError,
),
#[class(inherit)]
#[error(transparent)]
Server(
#[from]
#[inherit]
deno_inspector_server::InspectorServerError,
),
#[class(generic)]
#[error("{0}")]
InvalidHost(String),
}
#[derive(Debug, thiserror::Error, deno_error::JsError)]
pub enum InspectorConnectError {
#[class(inherit)]
#[error(transparent)]
Permission(
#[from]
#[inherit]
deno_permissions::PermissionCheckError,
),
#[class(generic)]
#[error("connectToMainThread not supported")]
ConnectToMainThreadUnsupported,
}
#[op2(stack_trace)]
#[cppgc]
pub fn op_inspector_connect<'s>(
isolate: &v8::Isolate,
scope: &mut v8::PinScope<'s, '_>,
state: &mut OpState,
connect_to_main_thread: bool,
callback: v8::Local<'s, v8::Function>,
) -> Result<JSInspectorSession, InspectorConnectError> {
state
.borrow_mut::<PermissionsContainer>()
.check_sys("inspector", "inspector.Session.connect")?;
if connect_to_main_thread {
return Err(InspectorConnectError::ConnectToMainThreadUnsupported);
}
let context = scope.get_current_context();
let context = v8::Global::new(scope, context);
let callback = v8::Global::new(scope, callback);
let inspector = state.borrow::<Rc<JsRuntimeInspector>>().clone();
let isolate = unsafe { isolate.as_raw_isolate_ptr() };
let callback = Box::new(move |message: InspectorMsg| {
let mut isolate = unsafe { v8::Isolate::from_raw_isolate_ptr(isolate) };
v8::callback_scope!(unsafe let scope, &mut isolate);
let context = v8::Local::new(scope, context.clone());
let scope = &mut v8::ContextScope::new(scope, context);
v8::tc_scope!(let scope, scope);
let recv = v8::undefined(scope);
if let Some(message) = v8::String::new(scope, &message.content) {
let callback = v8::Local::new(scope, callback.clone());
callback.call(scope, recv.into(), &[message.into()]);
}
});
let session = JsRuntimeInspector::create_local_session(
inspector,
callback,
InspectorSessionKind::NonBlocking {
wait_for_disconnect: false,
},
);
Ok(JSInspectorSession {
session: RefCell::new(Some(session)),
})
}
#[op2(fast, reentrant)]
pub fn op_inspector_dispatch(
#[cppgc] inspector: &JSInspectorSession,
#[string] message: String,
) {
if let Some(session) = &mut *inspector.session.borrow_mut() {
session.dispatch(message);
}
}
#[op2(fast)]
pub fn op_inspector_disconnect(#[cppgc] inspector: &JSInspectorSession) {
inspector.session.borrow_mut().take();
}