denoland-deno/cli/napi/sym/lib.rs

// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.

use proc_macro::TokenStream;
use quote::quote;
use serde::Deserialize;

static NAPI_EXPORTS: &str = include_str!("./symbol_exports.json");

#[derive(Deserialize)]
struct SymbolExports {
  pub symbols: Vec<String>,
}

#[proc_macro_attribute]
pub fn napi_sym(_attr: TokenStream, item: TokenStream) -> TokenStream {
  let func = syn::parse::<syn::ItemFn>(item).expect("expected a function");

  let exports: SymbolExports =
    serde_json::from_str(NAPI_EXPORTS).expect("failed to parse exports");
  let name = &func.sig.ident;
  assert!(
    exports.symbols.contains(&name.to_string()),
    "tools/napi/sym/symbol_exports.json is out of sync!"
  );

  let block = &func.block;
  let inputs = &func.sig.inputs;
  let generics = &func.sig.generics;
  TokenStream::from(quote! {
      // SAFETY: it's an NAPI function.
      #[no_mangle]
      pub unsafe extern "C" fn #name #generics (#inputs) -> napi_status {
        #block
      }
  })
}