feat(napi-derive): catch_unwind attribute

crates/backend/src/ast.rs

@@ -26,6 +26,7 @@ pub struct NapiFn {
   pub writable: bool,
   pub enumerable: bool,
   pub configurable: bool,
+  pub catch_unwind: bool,
 }
 
 #[derive(Debug, Clone)]

crates/backend/src/codegen/fn.rs

@@ -69,6 +69,20 @@ impl TryToTokens for NapiFn {
       }
     };
 
+    let function_call = if self.catch_unwind {
+      quote! {
+        {
+          std::panic::catch_unwind(|| { #function_call })
+            .map_err(|e| napi::Error::new(napi::Status::GenericFailure, format!("{:?}", e)))
+            .and_then(|r| r)
+        }
+      }
+    } else {
+      quote! {
+        #function_call
+      }
+    };
+
     (quote! {
       #(#attrs)*
       #[doc(hidden)]

crates/macro/src/parser/attrs.rs

@@ -42,6 +42,7 @@ pub struct BindgenAttrs {
 macro_rules! attrgen {
   ($mac:ident) => {
     $mac! {
+      (catch_unwind, CatchUnwind(Span)),
       (js_name, JsName(Span, String, Span)),
       (constructor, Constructor(Span)),
       (factory, Factory(Span)),

crates/macro/src/parser/mod.rs

@@ -695,6 +695,7 @@ fn napi_fn_from_decl(
       writable: opts.writable(),
       enumerable: opts.enumerable(),
       configurable: opts.configurable(),
+      catch_unwind: opts.catch_unwind().is_some(),
     }
   })
 }
@@ -747,6 +748,12 @@ impl ParseNapi for syn::ItemStruct {
         "#[napi(return_if_invalid)] can only be applied to a function or method."
       );
     }
+    if opts.catch_unwind().is_some() {
+      bail_span!(
+        self,
+        "#[napi(catch_unwind)] can only be applied to a function or method."
+      );
+    }
     if opts.object().is_some() && opts.custom_finalize().is_some() {
       bail_span!(self, "Custom finalize is not supported for #[napi(object)]");
     }
@@ -776,6 +783,12 @@ impl ParseNapi for syn::ItemImpl {
         "#[napi(return_if_invalid)] can only be applied to a function or method."
       );
     }
+    if opts.catch_unwind().is_some() {
+      bail_span!(
+        self,
+        "#[napi(catch_unwind)] can only be applied to a function or method."
+      );
+    }
     // #[napi] macro will be remove from impl items after converted to ast
     let napi = self.convert_to_ast(opts);
     self.to_tokens(tokens);
@@ -802,6 +815,12 @@ impl ParseNapi for syn::ItemEnum {
         "#[napi(return_if_invalid)] can only be applied to a function or method."
       );
     }
+    if opts.catch_unwind().is_some() {
+      bail_span!(
+        self,
+        "#[napi(catch_unwind)] can only be applied to a function or method."
+      );
+    }
     let napi = self.convert_to_ast(opts);
     self.to_tokens(tokens);
 
@@ -826,6 +845,12 @@ impl ParseNapi for syn::ItemConst {
         "#[napi(return_if_invalid)] can only be applied to a function or method."
       );
     }
+    if opts.catch_unwind().is_some() {
+      bail_span!(
+        self,
+        "#[napi(catch_unwind)] can only be applied to a function or method."
+      );
+    }
     let napi = self.convert_to_ast(opts);
     self.to_tokens(tokens);
     napi

crates/napi/src/bindgen_runtime/module_register.rs

@@ -38,7 +38,7 @@ impl<T> PersistedSingleThreadVec<T> {
       .inner
       .lock()
       .expect("Acquire persisted thread vec lock failed");
-    f(&mut *locked);
+    f(&mut locked);
   }
 
   fn push(&self, item: T) {

crates/napi/src/env.rs

@@ -854,7 +854,7 @@ impl Env {
       ))?;
       let type_id = unknown_tagged_object as *const TypeId;
       if *type_id == TypeId::of::<T>() {
-        Box::from_raw(unknown_tagged_object as *mut TaggedObject<T>);
+        drop(Box::from_raw(unknown_tagged_object as *mut TaggedObject<T>));
         Ok(())
       } else {
         Err(Error::new(

examples/napi/__test__/typegen.spec.ts.md

@@ -103,6 +103,7 @@ Generated by [AVA](https://avajs.dev).
     }␊
     export function enumToI32(e: CustomNumEnum): number␊
     export function throwError(): void␊
+    export function panic(): void␊
     export function createExternal(size: number): ExternalObject<number>␊
     export function createExternalString(content: string): ExternalObject<string>␊
     export function getExternal(external: ExternalObject<number>): number␊

examples/napi/__test__/values.spec.ts

@@ -31,6 +31,7 @@ import {
   mapOption,
   readFile,
   throwError,
+  panic,
   readPackageJson,
   getPackageJsonName,
   getBuffer,
@@ -163,7 +164,7 @@ test('enum', (t) => {
   t.is(enumToI32(CustomNumEnum.Eight), 8)
 })
 
-test.only('class', (t) => {
+test('class', (t) => {
   const dog = new Animal(Kind.Dog, '旺财')
 
   t.is(dog.name, '旺财')
@@ -371,6 +372,7 @@ test('Option', (t) => {
 
 test('Result', (t) => {
   t.throws(() => throwError(), void 0, 'Manual Error')
+  t.throws(() => panic(), void 0, `Don't panic`)
 })
 
 test('function ts type override', (t) => {

examples/napi/index.d.ts

@@ -93,6 +93,7 @@ export const enum CustomNumEnum {
 }
 export function enumToI32(e: CustomNumEnum): number
 export function throwError(): void
+export function panic(): void
 export function createExternal(size: number): ExternalObject<number>
 export function createExternalString(content: string): ExternalObject<string>
 export function getExternal(external: ExternalObject<number>): number

examples/napi/src/error.rs

@@ -1,6 +1,11 @@
 use napi::bindgen_prelude::*;
 
 #[napi]
-fn throw_error() -> Result<()> {
+pub fn throw_error() -> Result<()> {
   Err(Error::new(Status::InvalidArg, "Manual Error".to_owned()))
 }
+
+#[napi(catch_unwind)]
+pub fn panic() {
+  panic!("Don't panic");
+}

examples/napi/src/lib.rs

@@ -1,6 +1,6 @@
 #![allow(dead_code)]
 #![allow(unreachable_code)]
-#![allow(clippy::blacklisted_name)]
+#![allow(clippy::disallowed_names)]
 
 #[macro_use]
 extern crate napi_derive;