1use alef_codegen::generators::trait_bridge::{
7 BridgeOutput, TraitBridgeGenerator, TraitBridgeSpec, bridge_param_type as param_type, gen_bridge_all,
8 host_function_path, to_camel_case, visitor_param_type,
9};
10use alef_core::config::TraitBridgeConfig;
11use alef_core::ir::{ApiSurface, MethodDef, TypeDef, TypeRef};
12use std::collections::HashMap;
13
14pub use alef_codegen::generators::trait_bridge::find_bridge_param;
19
20pub fn find_options_field_binding<'a>(
23 func: &alef_core::ir::FunctionDef,
24 bridges: &'a [TraitBridgeConfig],
25) -> Option<(usize, &'a TraitBridgeConfig)> {
26 for bridge in bridges {
27 if bridge.bind_via != alef_core::config::BridgeBinding::OptionsField {
28 continue;
29 }
30 if let Some(options_type) = &bridge.options_type {
31 for (idx, param) in func.params.iter().enumerate() {
32 let matches = match ¶m.ty {
34 alef_core::ir::TypeRef::Named(n) => n == options_type,
35 alef_core::ir::TypeRef::Optional(inner) => {
36 if let alef_core::ir::TypeRef::Named(n) = inner.as_ref() {
37 n == options_type
38 } else {
39 false
40 }
41 }
42 _ => false,
43 };
44 if matches {
45 return Some((idx, bridge));
46 }
47 }
48 }
49 }
50 None
51}
52
53pub struct NapiBridgeGenerator {
56 pub core_import: String,
58 pub type_paths: HashMap<String, String>,
60 pub error_type: String,
62}
63
64impl TraitBridgeGenerator for NapiBridgeGenerator {
65 fn foreign_object_type(&self) -> &str {
66 "napi::bindgen_prelude::Object<'static>"
67 }
68
69 fn bridge_imports(&self) -> Vec<String> {
70 vec![
71 "napi::bindgen_prelude::{JsObjectValue, ToNapiValue, Unknown, Object}".to_string(),
72 "napi::JsValue".to_string(),
73 "std::sync::Arc".to_string(),
74 ]
75 }
76
77 fn gen_sync_method_body(&self, method: &MethodDef, spec: &TraitBridgeSpec) -> String {
78 let name = &method.name;
79 let has_error = method.error_type.is_some();
80
81 let js_args_exprs = build_napi_args(method, spec.bridge_config);
83 let inner_tuple_ty = unknown_tuple_type(js_args_exprs.len());
84 let args_tuple_ty = if js_args_exprs.is_empty() {
85 inner_tuple_ty.clone()
86 } else {
87 format!("napi::bindgen_prelude::FnArgs<{inner_tuple_ty}>")
88 };
89
90 let empty_args = js_args_exprs.is_empty();
91 let tuple_args = if empty_args {
92 String::new()
93 } else if js_args_exprs.len() == 1 {
94 format!("({},)", js_args_exprs[0])
95 } else {
96 format!("({})", js_args_exprs.join(", "))
97 };
98
99 let error_lookup =
100 spec.make_error("format!(\"Method '{}' not found on bridge object: {}\", self.cached_name, e)");
101 let error_call = spec.make_error(&format!(
102 "format!(\"Plugin '{{}}' method '{}' failed: {{}}\", self.cached_name, e)",
103 name
104 ));
105 let error_coercion = spec.make_error(&format!(
106 "format!(\"Failed to extract return value from method '{}': {{}}\", e)",
107 name
108 ));
109 let error_parse = spec.make_error(&format!(
110 "format!(\"Plugin '{{}}' failed to parse return value for method '{}'\", self.cached_name)",
111 name
112 ));
113
114 let has_default_impl = method.has_default_impl;
115 if matches!(method.return_type, TypeRef::Unit) {
116 crate::template_env::render(
117 "sync_method_unit_return.jinja",
118 minijinja::context! {
119 method_name => name,
120 args_tuple_ty => args_tuple_ty,
121 has_error => has_error,
122 has_default_impl => has_default_impl,
123 empty_args => empty_args,
124 tuple_args => tuple_args,
125 error_lookup => error_lookup,
126 error_call => error_call,
127 },
128 )
129 } else {
130 crate::template_env::render(
131 "sync_method_non_unit_return.jinja",
132 minijinja::context! {
133 method_name => name,
134 args_tuple_ty => args_tuple_ty,
135 has_error => has_error,
136 has_default_impl => has_default_impl,
137 empty_args => empty_args,
138 tuple_args => tuple_args,
139 error_lookup => error_lookup,
140 error_call => error_call,
141 error_coercion => error_coercion,
142 error_parse => error_parse,
143 },
144 )
145 }
146 }
147
148 fn gen_async_method_body(&self, method: &MethodDef, spec: &TraitBridgeSpec) -> String {
149 let name = &method.name;
150
151 let js_args_exprs = build_napi_args(method, spec.bridge_config);
153 let inner_tuple_ty = unknown_tuple_type(js_args_exprs.len());
154 let args_tuple_ty = if js_args_exprs.is_empty() {
155 inner_tuple_ty.clone()
156 } else {
157 format!("napi::bindgen_prelude::FnArgs<{inner_tuple_ty}>")
158 };
159
160 let empty_args = js_args_exprs.is_empty();
161 let tuple_args = if empty_args {
162 String::new()
163 } else if js_args_exprs.len() == 1 {
164 format!("({},)", js_args_exprs[0])
165 } else {
166 format!("({})", js_args_exprs.join(", "))
167 };
168
169 let error_lookup = spec.make_error("format!(\"Method '{}' not found on bridge object: {}\", cached_name, e)");
170 let error_call = spec.make_error(&format!(
171 "format!(\"Plugin '{{}}' method '{}' failed: {{}}\", cached_name, e)",
172 name
173 ));
174 let error_coercion = spec.make_error(&format!(
175 "format!(\"Failed to extract return value from method '{}': {{}}\", e)",
176 name
177 ));
178 let error_parse = spec.make_error(&format!(
179 "\"Failed to parse return value for method '{}'\".to_string()",
180 name
181 ));
182
183 if matches!(method.return_type, TypeRef::Unit) {
184 crate::template_env::render(
185 "async_method_unit_return.jinja",
186 minijinja::context! {
187 method_name => name,
188 args_tuple_ty => args_tuple_ty,
189 empty_args => empty_args,
190 tuple_args => tuple_args,
191 error_lookup => error_lookup,
192 error_call => error_call,
193 },
194 )
195 } else {
196 crate::template_env::render(
197 "async_method_non_unit_return.jinja",
198 minijinja::context! {
199 method_name => name,
200 args_tuple_ty => args_tuple_ty,
201 empty_args => empty_args,
202 tuple_args => tuple_args,
203 error_lookup => error_lookup,
204 error_call => error_call,
205 error_coercion => error_coercion,
206 error_parse => error_parse,
207 },
208 )
209 }
210 }
211
212 fn gen_constructor(&self, spec: &TraitBridgeSpec) -> String {
213 let wrapper = spec.wrapper_name();
214 let required_methods = spec
215 .required_methods()
216 .iter()
217 .map(|m| {
218 minijinja::context! {
219 name => &m.name,
220 }
221 })
222 .collect::<Vec<_>>();
223
224 crate::template_env::render(
225 "trait_bridge_constructor.jinja",
226 minijinja::context! {
227 wrapper_name => wrapper,
228 required_methods => required_methods,
229 },
230 )
231 }
232
233 fn gen_unregistration_fn(&self, spec: &TraitBridgeSpec) -> String {
234 let Some(unregister_fn) = spec.bridge_config.unregister_fn.as_deref() else {
235 return String::new();
236 };
237 let host_path = host_function_path(spec, unregister_fn);
238 let camel = to_camel_case(unregister_fn);
239 crate::template_env::render(
240 "unregistration_fn.jinja",
241 minijinja::context! {
242 unregister_fn => unregister_fn,
243 camel_fn_name => camel,
244 host_path => host_path,
245 },
246 )
247 }
248
249 fn gen_clear_fn(&self, spec: &TraitBridgeSpec) -> String {
250 let Some(clear_fn) = spec.bridge_config.clear_fn.as_deref() else {
251 return String::new();
252 };
253 let host_path = host_function_path(spec, clear_fn);
254 let camel = to_camel_case(clear_fn);
255 crate::template_env::render(
256 "clear_fn.jinja",
257 minijinja::context! {
258 clear_fn => clear_fn,
259 camel_fn_name => camel,
260 host_path => host_path,
261 },
262 )
263 }
264
265 fn gen_registration_fn(&self, spec: &TraitBridgeSpec) -> String {
266 let Some(register_fn) = spec.bridge_config.register_fn.as_deref() else {
267 return String::new();
268 };
269 let Some(registry_getter) = spec.bridge_config.registry_getter.as_deref() else {
270 return String::new();
271 };
272 let wrapper = spec.wrapper_name();
273 let trait_path = spec.trait_path();
274
275 let extra = spec
276 .bridge_config
277 .register_extra_args
278 .as_deref()
279 .map(|a| format!(", {a}"))
280 .unwrap_or_default();
281
282 crate::template_env::render(
283 "registration_fn.jinja",
284 minijinja::context! {
285 register_fn => register_fn,
286 wrapper => wrapper,
287 trait_path => trait_path,
288 registry_getter => registry_getter,
289 extra_args => extra,
290 },
291 )
292 }
293}
294
295pub fn gen_trait_bridge(
297 trait_type: &TypeDef,
298 bridge_cfg: &TraitBridgeConfig,
299 core_import: &str,
300 error_type: &str,
301 error_constructor: &str,
302 api: &ApiSurface,
303) -> BridgeOutput {
304 let type_paths: HashMap<String, String> = api
306 .types
307 .iter()
308 .map(|t| (t.name.clone(), t.rust_path.replace('-', "_")))
309 .chain(
310 api.enums
311 .iter()
312 .map(|e| (e.name.clone(), e.rust_path.replace('-', "_"))),
313 )
314 .chain(
317 api.excluded_type_paths
318 .iter()
319 .map(|(name, path)| (name.clone(), path.replace('-', "_"))),
320 )
321 .collect();
322
323 let is_visitor_bridge = bridge_cfg.type_alias.is_some()
325 && bridge_cfg.register_fn.is_none()
326 && bridge_cfg.super_trait.is_none()
327 && trait_type.methods.iter().all(|m| m.has_default_impl);
328
329 if is_visitor_bridge {
330 let struct_name = format!("Js{}Bridge", bridge_cfg.trait_name);
331 let trait_path = trait_type.rust_path.replace('-', "_");
332 let code = gen_visitor_bridge(
333 trait_type,
334 bridge_cfg,
335 &struct_name,
336 &trait_path,
337 core_import,
338 &type_paths,
339 );
340 BridgeOutput { imports: vec![], code }
341 } else {
342 let generator = NapiBridgeGenerator {
344 core_import: core_import.to_string(),
345 type_paths: type_paths.clone(),
346 error_type: error_type.to_string(),
347 };
348 let spec = TraitBridgeSpec {
349 trait_def: trait_type,
350 bridge_config: bridge_cfg,
351 core_import,
352 wrapper_prefix: "Js",
353 type_paths,
354 error_type: error_type.to_string(),
355 error_constructor: error_constructor.to_string(),
356 };
357 gen_bridge_all(&spec, &generator)
358 }
359}
360
361fn gen_visitor_bridge(
366 trait_type: &TypeDef,
367 bridge_cfg: &TraitBridgeConfig,
368 struct_name: &str,
369 trait_path: &str,
370 core_crate: &str,
371 type_paths: &HashMap<String, String>,
372) -> String {
373 let mut method_impls = String::with_capacity(4096);
374 for method in &trait_type.methods {
375 if method.trait_source.is_some() {
376 continue;
377 }
378 gen_visitor_method_napi(
379 &mut method_impls,
380 method,
381 trait_path,
382 core_crate,
383 bridge_cfg,
384 type_paths,
385 );
386 }
387
388 crate::template_env::render(
389 "visitor_bridge.jinja",
390 minijinja::context! {
391 core_crate => core_crate,
392 struct_name => struct_name,
393 trait_path => trait_path,
394 method_impls => method_impls,
395 },
396 )
397}
398
399fn unknown_tuple_type(count: usize) -> String {
401 if count == 0 {
402 return "()".to_string();
403 }
404 let parts = vec!["napi::bindgen_prelude::Unknown"; count];
405 format!("({}{})", parts.join(", "), if count == 1 { "," } else { "" })
406}
407
408fn gen_visitor_method_napi(
410 out: &mut String,
411 method: &MethodDef,
412 _trait_path: &str,
413 _core_crate: &str,
414 bridge_cfg: &TraitBridgeConfig,
415 type_paths: &HashMap<String, String>,
416) {
417 let name = &method.name;
418 let js_method_name = to_camel_case(name);
419
420 let mut sig_parts = vec!["&mut self".to_string()];
421 for p in &method.params {
422 let ty_str = visitor_param_type(&p.ty, p.is_ref, p.optional, type_paths);
423 sig_parts.push(format!("{}: {}", p.name, ty_str));
424 }
425 let signature = sig_parts.join(", ");
426
427 let return_type = match &method.return_type {
428 TypeRef::Named(n) => type_paths
429 .get(n.as_str())
430 .map(|p| p.replace('-', "_"))
431 .unwrap_or_else(|| n.clone()),
432 other => param_type(other, "", false, type_paths),
433 };
434
435 let arg_count = method.params.len();
436 let empty_args = arg_count == 0;
437 let inner_tuple_ty = unknown_tuple_type(arg_count);
438 let args_tuple_ty = if empty_args {
439 inner_tuple_ty
440 } else {
441 format!("napi::bindgen_prelude::FnArgs<{inner_tuple_ty}>")
442 };
443
444 let js_args_exprs = build_napi_args(method, bridge_cfg);
445 let arg_exprs: Vec<String> = js_args_exprs
446 .iter()
447 .map(|expr| expr.replace("self.env()", "__env"))
448 .collect();
449
450 let tuple_args = if arg_count == 1 {
451 "(arg_0,)".to_string()
452 } else if arg_count > 0 {
453 let arg_names: Vec<String> = (0..arg_count).map(|i| format!("arg_{i}")).collect();
454 format!("({})", arg_names.join(", "))
455 } else {
456 String::new()
457 };
458
459 out.push_str(&crate::template_env::render(
460 "visitor_method.jinja",
461 minijinja::context! {
462 method_name => name,
463 js_method_name => js_method_name,
464 signature => signature,
465 return_type => return_type,
466 empty_args => empty_args,
467 arg_exprs => arg_exprs,
468 tuple_args => tuple_args,
469 args_tuple_ty => args_tuple_ty,
470 },
471 ));
472}
473
474fn build_napi_args(method: &MethodDef, bridge_cfg: &TraitBridgeConfig) -> Vec<String> {
478 method
479 .params
480 .iter()
481
482 .map(|p| {
483 if let TypeRef::Named(n) = &p.ty {
484 if Some(n.as_str()) == bridge_cfg.context_type.as_deref() {
485 return format!(
486 "match nodecontext_to_js_object(&self.env(), {}{}) {{ Ok(o) => o.to_unknown(), Err(_) => unsafe {{ \
487 let r = napi::bindgen_prelude::ToNapiValue::to_napi_value(self.env().raw(), napi::bindgen_prelude::Null).unwrap_or(std::ptr::null_mut()); \
488 napi::bindgen_prelude::Unknown::from_raw_unchecked(self.env().raw(), r) }} \
489 }}",
490 if p.is_ref { "" } else { "&" },
491 p.name
492 );
493 }
494 }
495 if p.optional && matches!(&p.ty, TypeRef::String) && p.is_ref {
497 return format!(
498 "match {name} {{ \
499 Some(s) => match self.env().create_string(s) {{ \
500 Ok(v) => v.to_unknown(), \
501 Err(_) => unsafe {{ \
502 let r = napi::bindgen_prelude::ToNapiValue::to_napi_value(self.env().raw(), napi::bindgen_prelude::Null).unwrap_or(std::ptr::null_mut()); \
503 napi::bindgen_prelude::Unknown::from_raw_unchecked(self.env().raw(), r) }} \
504 }}, \
505 None => unsafe {{ \
506 let r = napi::bindgen_prelude::ToNapiValue::to_napi_value(self.env().raw(), napi::bindgen_prelude::Null).unwrap_or(std::ptr::null_mut()); \
507 napi::bindgen_prelude::Unknown::from_raw_unchecked(self.env().raw(), r) }} \
508 }}",
509 name = p.name
510 );
511 }
512 if matches!(&p.ty, TypeRef::String) && p.is_ref {
514 return format!(
515 "match self.env().create_string({name}) {{ \
516 Ok(s) => s.to_unknown(), \
517 Err(_) => unsafe {{ \
518 let r = napi::bindgen_prelude::ToNapiValue::to_napi_value(self.env().raw(), napi::bindgen_prelude::Null).unwrap_or(std::ptr::null_mut()); \
519 napi::bindgen_prelude::Unknown::from_raw_unchecked(self.env().raw(), r) }} \
520 }}",
521 name = p.name
522 );
523 }
524 if matches!(&p.ty, TypeRef::String) {
526 return format!(
527 "match self.env().create_string({name}.as_str()) {{ \
528 Ok(s) => s.to_unknown(), \
529 Err(_) => unsafe {{ \
530 let r = napi::bindgen_prelude::ToNapiValue::to_napi_value(self.env().raw(), napi::bindgen_prelude::Null).unwrap_or(std::ptr::null_mut()); \
531 napi::bindgen_prelude::Unknown::from_raw_unchecked(self.env().raw(), r) }} \
532 }}",
533 name = p.name
534 );
535 }
536 if matches!(&p.ty, TypeRef::Primitive(alef_core::ir::PrimitiveType::Bool)) {
538 return format!(
539 "unsafe {{ \
540 let r = napi::bindgen_prelude::ToNapiValue::to_napi_value(self.env().raw(), {name}).unwrap_or(std::ptr::null_mut()); \
541 napi::bindgen_prelude::Unknown::from_raw_unchecked(self.env().raw(), r) }}",
542 name = p.name
543 );
544 }
545 if matches!(&p.ty, TypeRef::Primitive(alef_core::ir::PrimitiveType::U32)) {
547 return format!(
548 "match self.env().create_uint32({name}) {{ Ok(n) => n.to_unknown(), Err(_) => unsafe {{ \
549 let r = napi::bindgen_prelude::ToNapiValue::to_napi_value(self.env().raw(), napi::bindgen_prelude::Null).unwrap_or(std::ptr::null_mut()); \
550 napi::bindgen_prelude::Unknown::from_raw_unchecked(self.env().raw(), r) }} \
551 }}",
552 name = p.name
553 );
554 }
555 if matches!(&p.ty, TypeRef::Primitive(alef_core::ir::PrimitiveType::Usize)) {
556 return format!(
557 "match self.env().create_uint32({name} as u32) {{ Ok(n) => n.to_unknown(), Err(_) => unsafe {{ \
558 let r = napi::bindgen_prelude::ToNapiValue::to_napi_value(self.env().raw(), napi::bindgen_prelude::Null).unwrap_or(std::ptr::null_mut()); \
559 napi::bindgen_prelude::Unknown::from_raw_unchecked(self.env().raw(), r) }} \
560 }}",
561 name = p.name
562 );
563 }
564 format!(
567 "match self.env().create_string(&format!(\"{{:?}}\", {name})) {{ Ok(s) => s.to_unknown(), Err(_) => unsafe {{ \
568 let r = napi::bindgen_prelude::ToNapiValue::to_napi_value(self.env().raw(), napi::bindgen_prelude::Null).unwrap_or(std::ptr::null_mut()); \
569 napi::bindgen_prelude::Unknown::from_raw_unchecked(self.env().raw(), r) }} \
570 }}",
571 name = p.name
572 )
573 })
574 .collect()
575}
576
577#[allow(clippy::too_many_arguments)]
581pub fn gen_bridge_function(
582 func: &alef_core::ir::FunctionDef,
583 bridge_param_idx: usize,
584 bridge_cfg: &TraitBridgeConfig,
585 mapper: &dyn alef_codegen::type_mapper::TypeMapper,
586 _cfg: &alef_codegen::generators::RustBindingConfig<'_>,
587 _adapter_bodies: &alef_codegen::generators::AdapterBodies,
588 opaque_types: &ahash::AHashSet<String>,
589 core_import: &str,
590) -> String {
591 use alef_core::ir::TypeRef;
592
593 let struct_name = format!("Js{}Bridge", bridge_cfg.trait_name);
594 let handle_path = format!("{core_import}::visitor::VisitorHandle");
595 let param_name = &func.params[bridge_param_idx].name;
596 let bridge_param = &func.params[bridge_param_idx];
597 let is_optional = bridge_param.optional || matches!(&bridge_param.ty, TypeRef::Optional(_));
598
599 let is_options_field_binding = matches!(bridge_cfg.bind_via, alef_core::config::BridgeBinding::OptionsField);
601
602 let options_param_idx = if is_options_field_binding {
604 func.params.iter().enumerate().find(|(_, p)| {
605 matches!(&p.ty, TypeRef::Named(n) if bridge_cfg.options_type.as_ref().is_some_and(|opt_type| n == opt_type))
606 }).map(|(i, _)| i)
607 } else {
608 None
609 };
610
611 let mut sig_parts = vec![];
614 for (idx, p) in func.params.iter().enumerate() {
615 if is_options_field_binding && Some(idx) == options_param_idx {
616 let ty = if p.optional || (idx > 0 && func.params[..idx].iter().any(|pp| pp.optional)) {
618 format!("Option<{}>", mapper.map_type(&p.ty))
619 } else {
620 mapper.map_type(&p.ty)
621 };
622 sig_parts.push(format!("{}: {}", p.name, ty));
623 } else if idx == bridge_param_idx {
624 if is_optional {
625 sig_parts.push(format!("{}: Option<napi::bindgen_prelude::Object>", p.name));
626 } else {
627 sig_parts.push(format!("{}: napi::bindgen_prelude::Object", p.name));
628 }
629 } else {
630 let promoted = idx > bridge_param_idx || func.params[..idx].iter().any(|pp| pp.optional);
631 let ty = if p.optional || promoted {
632 format!("Option<{}>", mapper.map_type(&p.ty))
633 } else {
634 mapper.map_type(&p.ty)
635 };
636 sig_parts.push(format!("{}: {}", p.name, ty));
637 }
638 }
639
640 let params_str = sig_parts.join(", ");
641 let return_type = mapper.map_type(&func.return_type);
642 let ret = mapper.wrap_return(&return_type, func.error_type.is_some());
643
644 let err_conv = ".map_err(|e| napi::Error::new(napi::Status::GenericFailure, e.to_string()))";
645
646 let bridge_wrap = if is_optional {
648 crate::template_env::render(
649 "bridge_optional_wrap.jinja",
650 minijinja::context! {
651 param_name => param_name,
652 struct_name => struct_name,
653 handle_path => handle_path,
654 },
655 )
656 } else {
657 crate::template_env::render(
658 "bridge_required_wrap.jinja",
659 minijinja::context! {
660 param_name => param_name,
661 struct_name => struct_name,
662 handle_path => handle_path,
663 },
664 )
665 };
666
667 let serde_bindings: String = func
669 .params
670 .iter()
671 .enumerate()
672 .filter(|(idx, p)| {
673 if *idx == bridge_param_idx {
674 return false;
675 }
676 let named = match &p.ty {
677 TypeRef::Named(n) => Some(n.as_str()),
678 TypeRef::Optional(inner) => {
679 if let TypeRef::Named(n) = inner.as_ref() {
680 Some(n.as_str())
681 } else {
682 None
683 }
684 }
685 _ => None,
686 };
687 named.is_some_and(|n| !opaque_types.contains(n))
688 })
689 .map(|(_, p)| {
690 let name = &p.name;
691 let core_path = format!(
692 "{core_import}::{}",
693 match &p.ty {
694 TypeRef::Named(n) => n.clone(),
695 TypeRef::Optional(inner) =>
696 if let TypeRef::Named(n) = inner.as_ref() {
697 n.clone()
698 } else {
699 String::new()
700 },
701 _ => String::new(),
702 }
703 );
704 let template_name = if p.optional || matches!(&p.ty, TypeRef::Optional(_)) {
705 "named_core_binding_optional.jinja"
706 } else {
707 "named_core_binding_required.jinja"
708 };
709 crate::template_env::render(
710 template_name,
711 minijinja::context! {
712 name => name,
713 core_path => core_path,
714 },
715 )
716 })
717 .collect();
718
719 let call_args: Vec<String> = func
721 .params
722 .iter()
723 .enumerate()
724 .map(|(idx, p)| {
725 if idx == bridge_param_idx {
726 return p.name.clone();
727 }
728 match &p.ty {
729 TypeRef::Named(n) if opaque_types.contains(n.as_str()) => {
730 if p.optional {
731 format!("{}.as_ref().map(|v| &v.inner)", p.name)
732 } else {
733 format!("&{}.inner", p.name)
734 }
735 }
736 TypeRef::Named(_) => format!("{}_core", p.name),
737 TypeRef::Optional(inner) => {
738 if let TypeRef::Named(n) = inner.as_ref() {
739 if opaque_types.contains(n.as_str()) {
740 format!("{}.as_ref().map(|v| &v.inner)", p.name)
741 } else {
742 format!("{}_core", p.name)
743 }
744 } else {
745 p.name.clone()
746 }
747 }
748 TypeRef::String | TypeRef::Char => {
749 if p.is_ref {
750 format!("&{}", p.name)
751 } else {
752 p.name.clone()
753 }
754 }
755 _ => p.name.clone(),
756 }
757 })
758 .collect();
759 let call_args_str = call_args.join(", ");
760
761 let core_fn_path = {
762 let path = func.rust_path.replace('-', "_");
763 if path.starts_with(core_import) {
764 path
765 } else {
766 format!("{core_import}::{}", func.name)
767 }
768 };
769 let core_call = format!("{core_fn_path}({call_args_str})");
770
771 let return_wrap = match &func.return_type {
772 TypeRef::Named(name) if opaque_types.contains(name.as_str()) => {
773 format!("{name} {{ inner: std::sync::Arc::new(val) }}")
774 }
775 TypeRef::Named(_) => "val.into()".to_string(),
776 TypeRef::String | TypeRef::Bytes => "val.into()".to_string(),
777 _ => "val".to_string(),
778 };
779
780 let body = render_bridge_function_body(
781 func.error_type.is_some(),
782 &return_wrap,
783 &bridge_wrap,
784 &serde_bindings,
785 &core_call,
786 err_conv,
787 );
788
789 let js_name = {
790 let mut result = String::with_capacity(func.name.len());
791 let mut capitalize_next = false;
792 for (i, c) in func.name.chars().enumerate() {
793 if c == '_' {
794 capitalize_next = true;
795 } else if capitalize_next {
796 result.extend(c.to_uppercase());
797 capitalize_next = false;
798 } else if i == 0 {
799 result.extend(c.to_lowercase());
800 } else {
801 result.push(c);
802 }
803 }
804 result
805 };
806 let js_name_attr = if js_name != func.name {
807 format!("(js_name = \"{}\")", js_name)
808 } else {
809 String::new()
810 };
811
812 let func_name = &func.name;
813 crate::template_env::render(
814 "bridge_function.jinja",
815 minijinja::context! {
816 has_error => func.error_type.is_some(),
817 js_name_attr => js_name_attr,
818 func_name => func_name,
819 params_str => params_str,
820 ret => ret,
821 body => body,
822 },
823 )
824}
825
826#[allow(clippy::too_many_arguments)]
830pub fn gen_options_field_bridge_function(
831 func: &alef_core::ir::FunctionDef,
832 options_param_idx: usize,
833 bridge_cfg: &TraitBridgeConfig,
834 mapper: &dyn alef_codegen::type_mapper::TypeMapper,
835 _cfg: &alef_codegen::generators::RustBindingConfig<'_>,
836 opaque_types: &ahash::AHashSet<String>,
837 core_import: &str,
838) -> String {
839 use alef_core::ir::TypeRef;
840
841 let struct_name = format!("Js{}Bridge", bridge_cfg.trait_name);
842 let handle_path = format!("{core_import}::visitor::VisitorHandle");
843 let options_param = &func.params[options_param_idx];
844 let options_name = &options_param.name;
845
846 let ir_param_optional = matches!(&options_param.ty, TypeRef::Optional(_));
853
854 let visitor_kwarg = bridge_cfg.param_name.as_deref().unwrap_or("visitor");
856 let field_name = bridge_cfg.resolved_options_field().unwrap_or(visitor_kwarg);
857
858 let params_str = {
861 let mut sig_parts = vec![];
862 for (i, p) in func.params.iter().enumerate() {
863 let ty = mapper.map_type(&p.ty);
864 if i == options_param_idx && !ir_param_optional {
865 sig_parts.push(format!("{}: Option<{ty}>", p.name));
866 } else {
867 sig_parts.push(format!("{}: {ty}", p.name));
868 }
869 }
870 sig_parts.push(format!("{visitor_kwarg}: Option<napi::bindgen_prelude::Object>"));
872 sig_parts.join(", ")
873 };
874
875 let return_type = mapper.map_type(&func.return_type);
876 let ret = mapper.wrap_return(&return_type, func.error_type.is_some());
877
878 let err_conv = ".map_err(|e| napi::Error::new(napi::Status::GenericFailure, e.to_string()))";
879
880 let visitor_wrap = format!(
883 "let {visitor_kwarg}_handle: Option<{handle_path}> = {visitor_kwarg}.map(|v| {{\n \
884 let bridge = {struct_name}::new(v);\n \
885 std::sync::Arc::new(std::sync::Mutex::new(bridge)) as {handle_path}\n\
886 }});"
887 );
888
889 let options_convert = format!(
894 "let {options_name}_core: Option<{core_import}::ConversionOptions> = {options_name}.map(|o| {{\n \
895 let mut result: {core_import}::ConversionOptions = o.into();\n \
896 if {visitor_kwarg}_handle.is_some() {{\n \
897 result.{field_name} = {visitor_kwarg}_handle.clone();\n \
898 }}\n \
899 result\n \
900 }}).or_else(|| {{\n \
901 if {visitor_kwarg}_handle.is_some() {{\n \
902 Some({core_import}::ConversionOptions {{\n \
903 {field_name}: {visitor_kwarg}_handle.clone(),\n \
904 ..Default::default()\n \
905 }})\n \
906 }} else {{\n \
907 None\n \
908 }}\n \
909 }});"
910 );
911
912 let call_args: String = func
914 .params
915 .iter()
916 .enumerate()
917 .map(|(idx, p)| {
918 if idx == options_param_idx {
919 format!("{options_name}_core")
920 } else {
921 match &p.ty {
922 TypeRef::Named(n) if opaque_types.contains(n.as_str()) => {
923 if p.optional {
924 format!("{}.as_ref().map(|v| &v.inner)", p.name)
925 } else {
926 format!("&{}.inner", p.name)
927 }
928 }
929 TypeRef::Named(_) => format!("{}.into()", p.name),
930 TypeRef::Optional(inner) => {
931 if let TypeRef::Named(n) = inner.as_ref() {
932 if opaque_types.contains(n.as_str()) {
933 format!("{}.as_ref().map(|v| &v.inner)", p.name)
934 } else {
935 format!("{}.map(Into::into)", p.name)
936 }
937 } else {
938 p.name.clone()
939 }
940 }
941 TypeRef::String | TypeRef::Char => {
942 if p.is_ref {
943 format!("&{}", p.name)
944 } else {
945 p.name.clone()
946 }
947 }
948 _ => p.name.clone(),
949 }
950 }
951 })
952 .collect::<Vec<_>>()
953 .join(", ");
954
955 let core_fn_path = {
956 let path = func.rust_path.replace('-', "_");
957 if path.starts_with(core_import) {
958 path
959 } else {
960 format!("{core_import}::{}", func.name)
961 }
962 };
963 let core_call = format!("{core_fn_path}({call_args})");
964
965 let return_wrap = match &func.return_type {
966 TypeRef::Named(name) if opaque_types.contains(name.as_str()) => {
967 format!("{name} {{ inner: std::sync::Arc::new(val) }}")
968 }
969 TypeRef::Named(_) => "val.into()".to_string(),
970 TypeRef::String | TypeRef::Bytes => "val.into()".to_string(),
971 _ => "val".to_string(),
972 };
973
974 let body = if func.error_type.is_some() {
976 if return_wrap == "val" {
977 format!("{visitor_wrap}\n {options_convert}\n {core_call}{err_conv}")
978 } else {
979 format!("{visitor_wrap}\n {options_convert}\n {core_call}.map(|val| {return_wrap}){err_conv}")
980 }
981 } else {
982 format!("{visitor_wrap}\n {options_convert}\n {core_call}")
983 };
984
985 let mut out = String::with_capacity(1024);
986 if func.error_type.is_some() {
987 out.push_str("#[allow(clippy::missing_errors_doc)]\n");
988 }
989 out.push_str("#[napi]\n");
990 let func_name = &func.name;
991 out.push_str(&crate::template_env::render(
992 "trait_bridge_fn_wrapper.jinja",
993 minijinja::context! {
994 func_name => func_name,
995 params_str => params_str,
996 return_type => ret,
997 body => body,
998 },
999 ));
1000
1001 out
1002}
1003
1004fn render_bridge_function_body(
1005 has_error: bool,
1006 return_wrap: &str,
1007 bridge_wrap: &str,
1008 serde_bindings: &str,
1009 core_call: &str,
1010 err_conv: &str,
1011) -> String {
1012 let template_name = match (has_error, return_wrap == "val") {
1013 (true, true) => "bridge_function_body_error.jinja",
1014 (true, false) => "bridge_function_body_error_mapped.jinja",
1015 (false, _) => "bridge_function_body_plain.jinja",
1016 };
1017 crate::template_env::render(
1018 template_name,
1019 minijinja::context! {
1020 bridge_wrap => bridge_wrap,
1021 serde_bindings => serde_bindings,
1022 core_call => core_call,
1023 err_conv => err_conv,
1024 return_wrap => return_wrap,
1025 },
1026 )
1027}