neon/context/
internal.rs

1use std::{cell::RefCell, ffi::c_void, mem::MaybeUninit};
2
3use crate::{
4    context::{Cx, ModuleContext},
5    handle::Handle,
6    result::NeonResult,
7    sys::{self, raw},
8    types::{private::ValueInternal, JsObject},
9};
10
11#[repr(C)]
12#[derive(Clone, Copy)]
13pub struct Env(raw::Env);
14
15impl From<raw::Env> for Env {
16    fn from(env: raw::Env) -> Self {
17        Self(env)
18    }
19}
20
21thread_local! {
22    #[allow(unused)]
23    pub(crate) static IS_RUNNING: RefCell<bool> = const { RefCell::new(false) };
24}
25
26impl Env {
27    pub(crate) fn to_raw(self) -> raw::Env {
28        let Self(ptr) = self;
29        ptr
30    }
31
32    pub(super) unsafe fn try_catch<T, F>(self, f: F) -> Result<T, raw::Local>
33    where
34        F: FnOnce() -> Result<T, crate::result::Throw>,
35    {
36        let result = f();
37        let mut local: MaybeUninit<raw::Local> = MaybeUninit::zeroed();
38
39        if sys::error::catch_error(self.to_raw(), local.as_mut_ptr()) {
40            Err(local.assume_init())
41        } else if let Ok(result) = result {
42            Ok(result)
43        } else {
44            panic!("try_catch: unexpected Err(Throw) when VM is not in a throwing state");
45        }
46    }
47}
48
49pub trait ContextInternal<'cx>: Sized {
50    fn cx(&self) -> &Cx<'cx>;
51    fn cx_mut(&mut self) -> &mut Cx<'cx>;
52    fn env(&self) -> Env {
53        self.cx().env
54    }
55}
56
57fn default_main(mut cx: ModuleContext) -> NeonResult<()> {
58    #[cfg(feature = "tokio-rt-multi-thread")]
59    crate::executor::tokio::init(&mut cx)?;
60    crate::registered().export(&mut cx)
61}
62
63fn init(cx: ModuleContext) -> NeonResult<()> {
64    if crate::macro_internal::MAIN.len() > 1 {
65        panic!("The `neon::main` macro must only be used once");
66    }
67
68    if let Some(main) = crate::macro_internal::MAIN.first() {
69        main(cx)
70    } else {
71        default_main(cx)
72    }
73}
74
75#[no_mangle]
76unsafe extern "C" fn napi_register_module_v1(env: *mut c_void, m: *mut c_void) -> *mut c_void {
77    let env = env.cast();
78
79    sys::setup(env);
80
81    IS_RUNNING.with(|v| {
82        *v.borrow_mut() = true;
83    });
84
85    let env = Env(env);
86    let exports = Handle::new_internal(JsObject::from_local(env, m.cast()));
87    let _ = ModuleContext::with(env, exports, init);
88
89    m
90}
neon/context/internal.rs

neon/context/
internal.rs
