Skip to main content

alef_backend_napi/
trait_bridge.rs

1//! NAPI-RS-specific trait bridge code generation.
2//!
3//! Generates Rust wrapper structs that implement Rust traits by delegating
4//! to JavaScript objects via NAPI-RS.
5
6use 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
14/// Find the first parameter index and bridge config where the parameter's named type
15/// matches a trait bridge's `type_alias`.
16///
17/// Returns `None` when no bridge applies.
18pub use alef_codegen::generators::trait_bridge::find_bridge_param;
19
20/// Find a bridge config that uses options_field binding and a parameter of the options_type.
21/// This complements find_bridge_param which only handles FunctionParam bindings.
22pub 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                // Check if param type is Named(options_type) or Optional(Named(options_type))
33                let matches = match &param.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
53/// NAPI-specific trait bridge generator.
54/// Implements code generation for bridging JavaScript objects to Rust traits.
55pub struct NapiBridgeGenerator {
56    /// Core crate import path (e.g., `"kreuzberg"`).
57    pub core_import: String,
58    /// Map of type name → fully-qualified Rust path for type references.
59    pub type_paths: HashMap<String, String>,
60    /// Error type name (e.g., `"KreuzbergError"`).
61    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        // Get the JS function from the object
82        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        // Build the JS function call
152        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
295/// Generate all trait bridge code for a given trait type and bridge config.
296pub 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    // Build type name → rust_path lookup (converted to String-owned HashMap)
305    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        // Include excluded types so trait methods referencing them (e.g. `&InternalDocument`)
315        // are qualified with the full Rust path rather than emitting the bare type name.
316        .chain(
317            api.excluded_type_paths
318                .iter()
319                .map(|(name, path)| (name.clone(), path.replace('-', "_"))),
320        )
321        .collect();
322
323    // Visitor-style bridge: all methods have defaults, no registry, no super-trait.
324    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        // Use the IR-driven TraitBridgeGenerator infrastructure
343        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
361/// Generate a visitor-style bridge wrapping a `napi::bindgen_prelude::Object`.
362///
363/// Every trait method checks if the JS object has a matching camelCase property,
364/// then calls it with converted arguments and maps the JS return value to `VisitResult`.
365fn 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
399/// Build the Function args tuple type string for a given number of Unknown args.
400fn 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
408/// Generate a single visitor method that checks for a camelCase JS property and calls it.
409fn 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
474/// Build NAPI argument expressions for a visitor method.
475///
476/// Returns one expression per parameter, each producing a `napi::bindgen_prelude::Unknown`.
477fn 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            // Option<&str>
496            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            // &str
513            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            // String (owned)
525            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            // Bool
537            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            // u32 / usize: create_uint32 needs a u32; usize requires the cast but u32 does not.
546            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            // Vec<String> or &[String] - serialize to JSON string as fallback
565            // Default: serialize as debug string
566            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/// Generate a NAPI free function that has one parameter replaced by
578/// `Option<napi::bindgen_prelude::Object>` (a trait bridge). The bridge is constructed
579/// before calling the core function.
580#[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    // Check if this is an options_field binding pattern (visitor embedded in options struct)
600    let is_options_field_binding = matches!(bridge_cfg.bind_via, alef_core::config::BridgeBinding::OptionsField);
601
602    // Find the options parameter when using options_field binding
603    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    // Build parameter list: bridge param becomes Option<Object>, no explicit env param
612    // (napi v3 does not implement FromNapiValue for Env; env is obtained from the Object)
613    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            // For options_field binding, visitor is extracted from options, not a separate param
617            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    // Bridge wrapping code: constructor is infallible (transmute-based).
647    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    // Use From/Into for non-bridge Named params — the generated bindings have From impls.
668    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    // Build call args
720    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/// Generate a NAPI free function where a trait bridge is embedded in an options struct field.
827/// The visitor is extracted from options before the Into conversion, wrapped in a bridge,
828/// and manually injected back into the converted core options.
829#[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    // Bridge functions always treat the options param as optional: callers may pass
847    // undefined/null (no options) or an options object (with or without visitor).
848    // Even if the IR marks the param as non-optional (e.g. because has_default types
849    // get their Option<> stripped during IR parsing), we force Option<T> behavior here.
850
851    // Whether the IR already marks the options param as Optional<T>.
852    let ir_param_optional = matches!(&options_param.ty, TypeRef::Optional(_));
853
854    // Name of the visitor parameter that will be appended to the function signature.
855    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    // Build parameter list; force the options param to Option<T> if the IR didn't already,
859    // and append the visitor parameter.
860    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        // Append visitor parameter (always optional for JS compatibility)
871        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    // Generate visitor wrapping (wrap the visitor parameter into a VisitorHandle).
881    // This mirrors PyO3's approach: take visitor as a separate parameter and wrap it.
882    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    // Generate options conversion with visitor injection.
890    // The From<JsConversionOptions> impl post-processes the visitor field to forward
891    // val.visitor through JsHtmlVisitorBridge, so we let the From impl handle it.
892    // The separate visitor kwarg (if provided) overrides options.visitor.
893    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    // Build call args, replacing options param with the _core version
913    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    // Build function body with visitor wrapping and options conversion.
975    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}