openharmony_ability_derive/
lib.rs

1use darling::FromMeta;
2use proc_macro::TokenStream;
3use syn::ItemFn;
4
5#[derive(FromMeta, Default, Debug)]
6struct AbilityArgs {
7    #[darling(default)]
8    webview: bool,
9    #[darling(default)]
10    protocol: Option<String>,
11}
12
13#[proc_macro_attribute]
14pub fn ability(attr: TokenStream, item: TokenStream) -> TokenStream {
15    let ast = syn::parse_macro_input!(item as ItemFn);
16    let fn_name = &ast.sig.ident;
17    let block = &ast.block;
18    let arg = &ast.sig.inputs;
19
20    let args = if attr.is_empty() {
21        AbilityArgs::default()
22    } else {
23        match darling::ast::NestedMeta::parse_meta_list(proc_macro2::TokenStream::from(attr)) {
24            Ok(list) => match AbilityArgs::from_list(&list) {
25                Ok(args) => args,
26                Err(e) => {
27                    return TokenStream::from(e.write_errors());
28                }
29            },
30            Err(e) => {
31                return TokenStream::from(e.to_compile_error());
32            }
33        }
34    };
35
36    let protocol_registrations = args
37        .protocol
38        .as_ref()
39        .map(|protocols| {
40            protocols
41                .split(",")
42                .map(|protocol| {
43                    let protocol_lit = syn::LitStr::new(protocol, proc_macro2::Span::call_site());
44                    quote::quote! {
45                        openharmony_ability::native_web::CustomProtocol::add_protocol_with_option(#protocol_lit,
46                            openharmony_ability::native_web::CustomProtocolOption::Standard |
47                            openharmony_ability::native_web::CustomProtocolOption::CorsEnabled |
48                            openharmony_ability::native_web::CustomProtocolOption::CspBypassing |
49                            openharmony_ability::native_web::CustomProtocolOption::FetchEnabled |
50                            openharmony_ability::native_web::CustomProtocolOption::CodeCacheEnabled
51                        );
52                    }
53                })
54                .collect::<Vec<_>>()
55        })
56        .unwrap_or_default();
57
58    let render = quote::quote! {
59        #[napi_derive_ohos::napi]
60        pub fn render<'a>(
61            env: &'a napi_ohos::Env,
62            helper: napi_ohos::bindgen_prelude::ObjectRef,
63            slot: openharmony_ability::arkui::ArkUIHandle,
64        ) -> napi_ohos::Result<()> {
65            let root = openharmony_ability::render(env, helper, slot, (*APP).clone())?;
66            ROOT_NODE.replace(Some(root));
67            Ok(())
68        }
69    };
70
71    // Register custom protocol if protocol is specified and webview is enabled
72    let protocol_registrations_apply = if args.protocol.is_some() && args.webview {
73        quote::quote! {
74            #[napi_derive_ohos::napi]
75            pub fn register_custom_protocol<'a>(
76                env: &'a napi_ohos::Env,
77            ) -> napi_ohos::Result<()> {
78                #(#protocol_registrations)*
79
80                openharmony_ability::native_web::CustomProtocol::register();
81
82                Ok(())
83            }
84        }
85    } else {
86        quote::quote! {}
87    };
88
89    let f = quote::quote! {
90        pub(crate) fn #fn_name(#arg) #block
91
92        mod openharmony_ability_mod {
93            use super::*;
94
95            static APP: std::sync::LazyLock<openharmony_ability::OpenHarmonyApp> =
96                std::sync::LazyLock::new(|| openharmony_ability::OpenHarmonyApp::new());
97
98            thread_local! {
99                pub static ROOT_NODE: std::cell::RefCell<Option<openharmony_ability::arkui::RootNode>> = std::cell::RefCell::new(None);
100            }
101
102            #protocol_registrations_apply
103
104            #[napi_derive_ohos::napi]
105            pub fn init<'a>(
106                env: &'a napi_ohos::Env,
107            ) -> napi_ohos::Result<openharmony_ability::ApplicationLifecycle<'a>> {
108                let lifecycle_handle = openharmony_ability::create_lifecycle_handle(env, (*APP).clone())?;
109                #fn_name((*APP).clone());
110                Ok(lifecycle_handle)
111            }
112
113            #render
114        }
115    };
116
117    f.into()
118}
openharmony_ability_derive/lib.rs

openharmony_ability_derive/
lib.rs
