use super::visitor_bridge::gen_visitor_bridge;
use crate::codegen::generators::trait_bridge::{TraitBridgeGenerator, TraitBridgeSpec, gen_bridge_all};
use crate::core::config::TraitBridgeConfig;
use crate::core::ir::{ApiSurface, MethodDef, TypeDef, TypeRef};
use std::collections::HashMap;
pub fn gen_trait_bridge(
trait_type: &TypeDef,
bridge_cfg: &TraitBridgeConfig,
core_import: &str,
error_type: &str,
error_constructor: &str,
api: &ApiSurface,
) -> anyhow::Result<String> {
if bridge_cfg.exclude_languages.contains(&"ruby".to_string()) {
return Ok(String::new());
}
let trait_path = trait_type.rust_path.replace('-', "_");
let type_paths: HashMap<String, String> = api
.types
.iter()
.map(|t| (t.name.clone(), t.rust_path.replace('-', "_")))
.chain(
api.enums
.iter()
.map(|e| (e.name.clone(), e.rust_path.replace('-', "_"))),
)
.chain(
api.excluded_type_paths
.iter()
.map(|(name, path)| (name.clone(), path.replace('-', "_"))),
)
.collect();
let is_visitor_bridge = bridge_cfg.type_alias.is_some()
&& bridge_cfg.register_fn.is_none()
&& bridge_cfg.super_trait.is_none()
&& trait_type.methods.iter().all(|m| m.has_default_impl);
if is_visitor_bridge {
let struct_name = crate::codegen::generators::trait_bridge::bridge_wrapper_name("Rb", bridge_cfg);
let mut out = String::with_capacity(8192);
gen_visitor_bridge(
&mut out,
trait_type,
bridge_cfg,
&struct_name,
&trait_path,
core_import,
&type_paths,
api,
)?;
Ok(out)
} else {
let struct_param_types =
crate::codegen::generators::trait_bridge::native_marshalled_struct_params(trait_type, api);
let binding_to_core = crate::codegen::conversions::convertible_types(api);
let struct_return_types: std::collections::HashSet<String> =
crate::codegen::generators::trait_bridge::native_marshalled_struct_returns(trait_type, api)
.into_iter()
.filter(|name| binding_to_core.contains(name.as_str()))
.collect();
let forwardable_defaulted =
crate::codegen::generators::trait_bridge::forwardable_defaulted_method_names(trait_type, api);
let generator = MagnusBridgeGenerator {
core_import: core_import.to_string(),
type_paths: type_paths.clone(),
error_type: error_type.to_string(),
error_constructor: error_constructor.to_string(),
struct_param_types,
struct_return_types,
forwardable_defaulted,
};
let lifetime_type_names: std::collections::HashSet<String> = api
.types
.iter()
.filter(|typ| typ.has_lifetime_params)
.map(|typ| typ.name.clone())
.collect();
let spec = TraitBridgeSpec {
trait_def: trait_type,
bridge_config: bridge_cfg,
core_import,
wrapper_prefix: "Rb",
type_paths,
lifetime_type_names,
error_type: error_type.to_string(),
error_constructor: error_constructor.to_string(),
};
let output = gen_bridge_all(&spec, &generator);
let mut prefixed = String::with_capacity(output.imports.len() * 64 + output.code.len());
let imports_to_emit: Vec<_> = output
.imports
.iter()
.filter(|imp| *imp != "magnus::prelude::*")
.collect();
for imp in &imports_to_emit {
prefixed.push_str("#[allow(unused_imports)]\n");
prefixed.push_str("use ");
prefixed.push_str(imp);
prefixed.push_str(" as _;\n");
}
prefixed.push_str(&output.code);
Ok(prefixed)
}
}
struct MagnusBridgeGenerator {
core_import: String,
type_paths: HashMap<String, String>,
error_type: String,
error_constructor: String,
struct_param_types: std::collections::HashSet<String>,
struct_return_types: std::collections::HashSet<String>,
forwardable_defaulted: std::collections::HashSet<String>,
}
impl MagnusBridgeGenerator {
fn error_path(&self) -> String {
if self.error_type.contains("::") || self.error_type.contains('<') {
self.error_type.clone()
} else {
format!("{}::{}", self.core_import, self.error_type)
}
}
fn make_error(&self, msg_expr: &str) -> String {
self.error_constructor.replace("{msg}", msg_expr)
}
}
impl TraitBridgeGenerator for MagnusBridgeGenerator {
fn foreign_object_type(&self) -> &str {
"magnus::value::Opaque<magnus::Value>"
}
fn gen_method_presence_check(&self, method: &MethodDef, _spec: &TraitBridgeSpec) -> Option<String> {
self.forwardable_defaulted
.contains(&method.name)
.then(|| format!("self.has_{}", method.name))
}
fn gen_lifecycle_presence_check(&self, method: &MethodDef, _spec: &TraitBridgeSpec) -> Option<String> {
Some(format!("self.has_{}", method.name))
}
fn extra_bridge_fields(&self, spec: &TraitBridgeSpec) -> Vec<(String, String)> {
let mut fields: Vec<(String, String)> = spec
.trait_def
.methods
.iter()
.filter(|m| self.forwardable_defaulted.contains(&m.name))
.map(|m| (format!("has_{}", m.name), "bool".to_string()))
.collect();
if spec.bridge_config.super_trait.is_some() {
fields.push(("has_initialize".to_string(), "bool".to_string()));
fields.push(("has_shutdown".to_string(), "bool".to_string()));
}
fields
}
fn bridge_imports(&self) -> Vec<String> {
vec![
"magnus::value::InnerValue".to_string(),
"magnus::TryConvert".to_string(),
]
}
fn gen_sync_method_body(&self, method: &MethodDef, _spec: &TraitBridgeSpec) -> String {
let name = &method.name;
let has_error = method.error_type.is_some();
let is_unit = matches!(method.return_type, TypeRef::Unit);
let args: Vec<String> = method.params.iter().map(|p| self.ruby_arg_expr(p)).collect();
let call = if args.is_empty() {
format!("value.funcall::<_, _, magnus::Value>(\"{name}\", ())")
} else {
let args_tuple = if args.len() == 1 {
format!("({},)", args[0])
} else {
format!("({})", args.join(", "))
};
format!("value.funcall::<_, _, magnus::Value>(\"{name}\", {args_tuple})")
};
let err_expr = if has_error {
self.make_error(&format!("format!(\"Ruby method '{name}' failed: {{}}\", e)"))
} else {
String::new()
};
let mut body = crate::backends::magnus::template_env::render(
"sync_method_body.rs.jinja",
minijinja::context! {
call => call,
has_error => has_error,
is_unit => is_unit,
err_expr => err_expr,
},
);
if !is_unit {
body.push_str(&self.return_conversion(method, has_error, ""));
}
body
}
fn gen_async_method_body(&self, method: &MethodDef, _spec: &TraitBridgeSpec) -> String {
let name = &method.name;
let has_error = method.error_type.is_some();
let is_unit = matches!(method.return_type, TypeRef::Unit);
let conversions: Vec<String> = method
.params
.iter()
.map(|p| match (&p.ty, p.is_ref) {
(TypeRef::String, true) => format!("let {}_owned = {}.to_string();\n", p.name, p.name),
(TypeRef::Bytes, true) => format!("let {}_owned = {}.to_vec();\n", p.name, p.name),
(TypeRef::Path, true) => format!("let {}_owned = {}.to_path_buf();\n", p.name, p.name),
_ => format!("let {}_owned = {}.clone();\n", p.name, p.name),
})
.collect();
let conversion_bindings = conversions.join("");
let return_type_rust = if is_unit {
"()".to_string()
} else {
self.return_rust_type(&method.return_type)
};
let err_path = self.error_path();
let result_ty = if has_error {
format!("std::result::Result<{return_type_rust}, {err_path}>")
} else {
return_type_rust.clone()
};
let conversions = format!(
"let inner = self.inner;\n\
let cached_name = self.cached_name.clone();\n\
// cached_name is referenced both inside the spawn_blocking closure and after\n\
// the await for the JoinError fallback, so clone once for each consumer.\n\
let cached_name_for_blocking = cached_name.clone();\n\
{conversion_bindings}\n",
conversion_bindings = conversion_bindings,
);
let args: Vec<String> = method
.params
.iter()
.map(|p| {
let param_name = if matches!(&p.ty, TypeRef::String) && p.is_ref {
format!("{}_owned.as_str()", p.name)
} else {
format!("{}_owned", p.name)
};
self.ruby_arg_expr_custom(&p.ty, ¶m_name)
})
.collect();
let call = if args.is_empty() {
format!("value.funcall::<_, _, magnus::Value>(\"{name}\", ())")
} else {
let args_tuple = if args.len() == 1 {
format!("({},)", args[0])
} else {
format!("({})", args.join(", "))
};
format!("value.funcall::<_, _, magnus::Value>(\"{name}\", {args_tuple})")
};
let err_expr_call = self.make_error(&format!(
"format!(\"Plugin '{{}}' method '{name}' failed: {{}}\", cached_name_for_blocking, e)"
));
let err_expr_join = if has_error {
self.make_error("format!(\"spawn_blocking failed for '{}': {}\", cached_name, e)")
} else {
String::new()
};
crate::backends::magnus::template_env::render(
"trait_bridge_async_method_body.rs.jinja",
minijinja::context! {
conversions => conversions,
call => call,
has_error => has_error,
is_unit => is_unit,
result_ty => result_ty,
err_expr_call => err_expr_call,
err_expr_join => err_expr_join,
return_conversion => self.return_conversion(method, has_error, " "),
},
)
}
fn gen_constructor(&self, spec: &TraitBridgeSpec) -> String {
let wrapper = spec.wrapper_name();
let required_methods: Vec<_> = spec.required_methods().iter().map(|m| m.name.as_str()).collect();
let mut optional_methods: Vec<_> = spec
.trait_def
.methods
.iter()
.filter(|m| self.forwardable_defaulted.contains(&m.name))
.map(|m| m.name.as_str())
.collect();
if spec.bridge_config.super_trait.is_some() {
optional_methods.push("initialize");
optional_methods.push("shutdown");
}
crate::backends::magnus::template_env::render(
"trait_bridge_constructor.rs.jinja",
minijinja::context! {
wrapper => wrapper,
required_methods => required_methods,
optional_methods => optional_methods,
},
)
}
fn gen_unregistration_fn(&self, spec: &TraitBridgeSpec) -> String {
let Some(unregister_fn) = spec.bridge_config.unregister_fn.as_deref() else {
return String::new();
};
let host_path = crate::codegen::generators::trait_bridge::host_function_path(spec, unregister_fn);
let func = format!(
"pub fn {unregister_fn}(name: String) -> Result<(), magnus::Error> {{\n\
{host_path}(&name).map_err(|e| {{\n\
let ruby = unsafe {{ magnus::Ruby::get_unchecked() }};\n\
magnus::Error::new(ruby.exception_runtime_error(), format!(\"{{}}\", e))\n\
}})\n\
}}\n"
);
func
}
fn gen_clear_fn(&self, spec: &TraitBridgeSpec) -> String {
let Some(clear_fn) = spec.bridge_config.clear_fn.as_deref() else {
return String::new();
};
let host_path = crate::codegen::generators::trait_bridge::host_function_path(spec, clear_fn);
let mut out = String::with_capacity(512);
let func = format!(
"pub fn {clear_fn}() -> Result<(), magnus::Error> {{\n\
{host_path}().map_err(|e| {{\n\
let ruby = unsafe {{ magnus::Ruby::get_unchecked() }};\n\
magnus::Error::new(ruby.exception_runtime_error(), format!(\"{{}}\", e))\n\
}})\n\
}}\n"
);
out.push_str(&func);
out
}
fn gen_registration_fn(&self, spec: &TraitBridgeSpec) -> String {
let Some(register_fn) = spec.bridge_config.register_fn.as_deref() else {
return String::new();
};
let Some(registry_getter) = spec.bridge_config.registry_getter.as_deref() else {
return String::new();
};
let wrapper = spec.wrapper_name();
let trait_path = spec.trait_path();
let required_methods: Vec<_> = spec
.required_methods()
.iter()
.map(|m| format!("\"{}\"", m.name))
.collect();
let required_methods = required_methods.join(", ");
let register_extra_args = spec
.bridge_config
.register_extra_args
.as_deref()
.map(|a| format!(", {a}"))
.unwrap_or_default();
crate::backends::magnus::template_env::render(
"trait_bridge_registration_fn.rs.jinja",
minijinja::context! {
register_fn => register_fn,
registry_getter => registry_getter,
wrapper => wrapper,
trait_path => trait_path,
required_methods => required_methods,
register_extra_args => register_extra_args,
},
)
}
}
impl MagnusBridgeGenerator {
fn return_rust_type(&self, ty: &TypeRef) -> String {
match ty {
TypeRef::Primitive(p) => {
use crate::core::ir::PrimitiveType::*;
match p {
Bool => "bool",
U8 => "u8",
U16 => "u16",
U32 => "u32",
U64 => "u64",
I8 => "i8",
I16 => "i16",
I32 => "i32",
I64 => "i64",
F32 => "f32",
F64 => "f64",
Usize => "usize",
Isize => "isize",
}
.to_string()
}
TypeRef::String => "String".to_string(),
TypeRef::Bytes => "Vec<u8>".to_string(),
TypeRef::Vec(inner) => format!("Vec<{}>", self.return_rust_type(inner)),
TypeRef::Optional(inner) => format!("Option<{}>", self.return_rust_type(inner)),
TypeRef::Named(name) => self
.type_paths
.get(name.as_str())
.cloned()
.unwrap_or_else(|| format!("{}::{}", self.core_import, name)),
TypeRef::Unit => "()".to_string(),
TypeRef::Map(k, v) => format!(
"std::collections::HashMap<{}, {}>",
self.return_rust_type(k),
self.return_rust_type(v)
),
TypeRef::Json => "serde_json::Value".to_string(),
TypeRef::Duration => "std::time::Duration".to_string(),
TypeRef::Char => "char".to_string(),
TypeRef::Path => "std::path::PathBuf".to_string(),
}
}
fn needs_json_marshalling(&self, ty: &TypeRef) -> bool {
match ty {
TypeRef::Named(_) | TypeRef::Json => true,
TypeRef::Vec(inner) | TypeRef::Optional(inner) => self.needs_json_marshalling(inner),
TypeRef::Map(k, v) => self.needs_json_marshalling(k) || self.needs_json_marshalling(v),
_ => false,
}
}
fn return_conversion(&self, method: &MethodDef, has_error: bool, indent: &str) -> String {
let rust_ty = self.return_rust_type(&method.return_type);
let err_non_json = if has_error {
self.make_error(&format!(
"format!(\"Ruby method '{}' returned non-JSON value: {{}}\", e)",
method.name
))
} else {
String::new()
};
let err_deserialize = if has_error {
self.make_error(&format!(
"format!(\"Failed to deserialize Ruby '{}' return value: {{}}\", e)",
method.name
))
} else {
String::new()
};
let err_convert = if has_error {
self.make_error(&format!(
"format!(\"Failed to convert Ruby '{}' return value: {{}}\", e)",
method.name
))
} else {
String::new()
};
crate::backends::magnus::template_env::render(
"trait_bridge_return_conversion.rs.jinja",
minijinja::context! {
has_error => has_error,
needs_json => self.needs_json_marshalling(&method.return_type),
native_return_binding => self.native_struct_return(&method.return_type),
indent => indent,
rust_ty => rust_ty,
err_non_json => err_non_json,
err_deserialize => err_deserialize,
err_convert => err_convert,
},
)
}
fn native_struct_return<'a>(&self, ty: &'a TypeRef) -> Option<&'a str> {
match ty {
TypeRef::Named(n) if self.struct_return_types.contains(n) => Some(n.as_str()),
_ => None,
}
}
fn is_native_struct_param(&self, name: &str) -> bool {
self.struct_param_types.contains(name)
}
fn ruby_arg_expr(&self, p: &crate::core::ir::ParamDef) -> String {
self.ruby_arg_expr_custom(&p.ty, &p.name)
}
fn ruby_arg_expr_custom(&self, ty: &TypeRef, var: &str) -> String {
match ty {
TypeRef::String => format!(
"{{ let ruby = unsafe {{ magnus::Ruby::get_unchecked() }}; ruby.str_new(AsRef::<str>::as_ref(&{var})).as_value() }}"
),
TypeRef::Bytes => format!(
"{{ let ruby = unsafe {{ magnus::Ruby::get_unchecked() }}; ruby.str_new(String::from_utf8_lossy(AsRef::<[u8]>::as_ref(&{var})).as_ref()).as_value() }}"
),
TypeRef::Named(n) if self.is_native_struct_param(n) => format!(
"{{ let ruby = unsafe {{ magnus::Ruby::get_unchecked() }}; use magnus::IntoValue; {n}::from({var}.clone()).into_value_with(&ruby) }}"
),
TypeRef::Named(_) | TypeRef::Json => format!(
"{{ let ruby = unsafe {{ magnus::Ruby::get_unchecked() }}; serde_json::to_string(&{var}).ok().map(|s| ruby.str_new(s.as_str()).as_value()).unwrap_or_else(|| ruby.qnil().as_value()) }}"
),
TypeRef::Vec(_) | TypeRef::Map(_, _) | TypeRef::Optional(_) => format!(
"{{ let ruby = unsafe {{ magnus::Ruby::get_unchecked() }}; serde_json::to_string(&{var}).ok().map(|s| ruby.str_new(s.as_str()).as_value()).unwrap_or_else(|| ruby.qnil().as_value()) }}"
),
TypeRef::Path => format!(
"{{ let ruby = unsafe {{ magnus::Ruby::get_unchecked() }}; ruby.str_new(<_ as AsRef<std::path::Path>>::as_ref(&{var}).to_string_lossy().as_ref()).as_value() }}"
),
_ => var.to_string(),
}
}
}
#[cfg(test)]
mod forwarding_tests {
use super::*;
fn make_generator() -> MagnusBridgeGenerator {
MagnusBridgeGenerator {
core_import: "sample_core".to_string(),
type_paths: HashMap::new(),
error_type: "SampleError".to_string(),
error_constructor: "SampleError::Message { message: {msg} }".to_string(),
struct_param_types: std::collections::HashSet::new(),
struct_return_types: std::collections::HashSet::new(),
forwardable_defaulted: std::collections::HashSet::new(),
}
}
fn make_trait() -> crate::core::ir::TypeDef {
crate::core::ir::TypeDef {
name: "OcrBackend".to_string(),
rust_path: "sample_core::OcrBackend".to_string(),
is_trait: true,
is_opaque: true,
methods: vec![
crate::core::ir::MethodDef {
name: "supports_language".to_string(),
receiver: Some(crate::core::ir::ReceiverKind::Ref),
return_type: crate::core::ir::TypeRef::Primitive(crate::core::ir::PrimitiveType::Bool),
..Default::default()
},
crate::core::ir::MethodDef {
name: "supports_table_detection".to_string(),
has_default_impl: true,
receiver: Some(crate::core::ir::ReceiverKind::Ref),
return_type: crate::core::ir::TypeRef::Primitive(crate::core::ir::PrimitiveType::Bool),
..Default::default()
},
],
..Default::default()
}
}
#[test]
fn presence_uses_construction_time_flag_and_wrapper_gains_field() {
let mut generator = make_generator();
generator
.forwardable_defaulted
.insert("supports_table_detection".to_string());
let trait_def = make_trait();
let bridge = crate::core::config::TraitBridgeConfig {
trait_name: "OcrBackend".to_string(),
..crate::core::config::TraitBridgeConfig::default()
};
let spec = TraitBridgeSpec {
trait_def: &trait_def,
bridge_config: &bridge,
core_import: "sample_core",
wrapper_prefix: "Rb",
type_paths: HashMap::new(),
lifetime_type_names: std::collections::HashSet::new(),
error_type: "SampleError".to_string(),
error_constructor: "SampleError::Message { message: {msg} }".to_string(),
};
let check = generator
.gen_method_presence_check(&trait_def.methods[1], &spec)
.unwrap();
assert_eq!(check, "self.has_supports_table_detection");
let fields = generator.extra_bridge_fields(&spec);
assert_eq!(
fields,
vec![("has_supports_table_detection".to_string(), "bool".to_string())]
);
let ctor = generator.gen_constructor(&spec);
assert!(
ctor.contains("has_supports_table_detection: rb_obj.respond_to(\"supports_table_detection\", false)"),
"constructor must capture the flag under the GVL:\n{ctor}"
);
let output = crate::codegen::generators::trait_bridge::gen_bridge_all(&spec, &generator);
assert!(
output.code.contains("has_supports_table_detection: bool,"),
"wrapper struct must declare the flag field:\n{}",
output.code
);
assert!(
output
.code
.contains("RbOcrBackendBridgeDefaultSupportsTableDetection(self).supports_table_detection()"),
"fallback must run the Rust default via the delegate:\n{}",
output.code
);
}
}