1use alef_core::ir::{CoreWrapper, PrimitiveType, TypeDef, TypeRef};
2use std::fmt::Write;
3
4use super::ConversionConfig;
5use super::helpers::{core_prim_str, core_type_path, is_newtype, is_tuple_type_name, needs_i64_cast};
6
7pub fn gen_from_binding_to_core(typ: &TypeDef, core_import: &str) -> String {
10 gen_from_binding_to_core_cfg(typ, core_import, &ConversionConfig::default())
11}
12
13pub fn gen_from_binding_to_core_cfg(typ: &TypeDef, core_import: &str, config: &ConversionConfig) -> String {
15 let core_path = core_type_path(typ, core_import);
16 let binding_name = format!("{}{}", config.type_name_prefix, typ.name);
17 let mut out = String::with_capacity(256);
18 if typ.has_stripped_cfg_fields {
21 writeln!(out, "#[allow(clippy::needless_update)]").ok();
22 }
23 let uses_builder_pattern = config.option_duration_on_defaults
25 && typ.has_default
26 && typ
27 .fields
28 .iter()
29 .any(|f| !f.optional && matches!(f.ty, TypeRef::Duration));
30 if uses_builder_pattern {
31 writeln!(out, "#[allow(clippy::field_reassign_with_default)]").ok();
32 }
33 writeln!(out, "impl From<{binding_name}> for {core_path} {{").ok();
34 writeln!(out, " fn from(val: {binding_name}) -> Self {{").ok();
35
36 if is_newtype(typ) {
38 let field = &typ.fields[0];
39 let inner_expr = match &field.ty {
40 TypeRef::Named(_) => "val._0.into()".to_string(),
41 TypeRef::Path => "val._0.into()".to_string(),
42 TypeRef::Duration => "std::time::Duration::from_millis(val._0)".to_string(),
43 _ => "val._0".to_string(),
44 };
45 writeln!(out, " Self({inner_expr})").ok();
46 writeln!(out, " }}").ok();
47 write!(out, "}}").ok();
48 return out;
49 }
50
51 let has_optionalized_duration = config.option_duration_on_defaults
56 && typ.has_default
57 && typ
58 .fields
59 .iter()
60 .any(|f| !f.optional && matches!(f.ty, TypeRef::Duration));
61
62 if has_optionalized_duration {
63 writeln!(out, " let mut __result = {core_path}::default();").ok();
65 let optionalized = config.optionalize_defaults && typ.has_default;
66 for field in &typ.fields {
67 if field.cfg.is_some() {
69 continue;
70 }
71 if field.sanitized {
72 continue;
74 }
75 if !config.exclude_types.is_empty()
77 && super::helpers::field_references_excluded_type(&field.ty, config.exclude_types)
78 {
79 continue;
80 }
81 if !field.optional && matches!(field.ty, TypeRef::Duration) {
83 let cast = if config.cast_large_ints_to_i64 { " as u64" } else { "" };
84 writeln!(
85 out,
86 " if let Some(__v) = val.{} {{ __result.{} = std::time::Duration::from_millis(__v{cast}); }}",
87 field.name, field.name
88 )
89 .ok();
90 continue;
91 }
92 let conversion = if optionalized && !field.optional {
93 gen_optionalized_field_to_core(&field.name, &field.ty, config)
94 } else {
95 field_conversion_to_core_cfg(&field.name, &field.ty, field.optional, config)
96 };
97 if let Some(expr) = conversion.strip_prefix(&format!("{}: ", field.name)) {
99 writeln!(out, " __result.{} = {};", field.name, expr).ok();
100 }
101 }
102 writeln!(out, " __result").ok();
103 writeln!(out, " }}").ok();
104 write!(out, "}}").ok();
105 return out;
106 }
107
108 writeln!(out, " Self {{").ok();
109 let optionalized = config.optionalize_defaults && typ.has_default;
110 for field in &typ.fields {
111 if field.cfg.is_some() {
116 continue;
117 }
118 let references_excluded = !config.exclude_types.is_empty()
125 && super::helpers::field_references_excluded_type(&field.ty, config.exclude_types);
126 if references_excluded && typ.has_stripped_cfg_fields {
127 continue;
128 }
129 let conversion = if field.sanitized || references_excluded {
130 format!("{}: Default::default()", field.name)
131 } else if optionalized && !field.optional {
132 gen_optionalized_field_to_core(&field.name, &field.ty, config)
135 } else {
136 field_conversion_to_core_cfg(&field.name, &field.ty, field.optional, config)
137 };
138 let conversion = if let Some(newtype_path) = &field.newtype_wrapper {
144 if let Some(expr) = conversion.strip_prefix(&format!("{}: ", field.name)) {
145 match &field.ty {
148 TypeRef::Optional(_) => format!("{}: ({expr}).map({newtype_path})", field.name),
149 TypeRef::Vec(_) => {
150 format!("{}: ({expr}).into_iter().map({newtype_path}).collect()", field.name)
151 }
152 _ if field.optional => format!("{}: ({expr}).map({newtype_path})", field.name),
153 _ => format!("{}: {newtype_path}({expr})", field.name),
154 }
155 } else {
156 conversion
157 }
158 } else {
159 conversion
160 };
161 let conversion = if field.is_boxed && matches!(&field.ty, TypeRef::Named(_)) {
163 if let Some(expr) = conversion.strip_prefix(&format!("{}: ", field.name)) {
164 if field.optional {
165 format!("{}: {}.map(Box::new)", field.name, expr)
167 } else {
168 format!("{}: Box::new({})", field.name, expr)
169 }
170 } else {
171 conversion
172 }
173 } else {
174 conversion
175 };
176 let conversion = apply_core_wrapper_to_core(
178 &conversion,
179 &field.name,
180 &field.core_wrapper,
181 &field.vec_inner_core_wrapper,
182 field.optional,
183 );
184 writeln!(out, " {conversion},").ok();
185 }
186 if typ.has_stripped_cfg_fields {
188 writeln!(out, " ..Default::default()").ok();
189 }
190 writeln!(out, " }}").ok();
191 writeln!(out, " }}").ok();
192 write!(out, "}}").ok();
193 out
194}
195
196pub(super) fn gen_optionalized_field_to_core(name: &str, ty: &TypeRef, config: &ConversionConfig) -> String {
199 match ty {
200 TypeRef::Json => {
201 format!("{name}: val.{name}.as_ref().and_then(|s| serde_json::from_str(s).ok()).unwrap_or_default()")
202 }
203 TypeRef::Named(_) => {
204 format!("{name}: val.{name}.map(Into::into).unwrap_or_default()")
206 }
207 TypeRef::Primitive(PrimitiveType::F32) if config.cast_f32_to_f64 => {
208 format!("{name}: val.{name}.map(|v| v as f32).unwrap_or(0.0)")
209 }
210 TypeRef::Primitive(PrimitiveType::F32 | PrimitiveType::F64) => {
211 format!("{name}: val.{name}.unwrap_or(0.0)")
212 }
213 TypeRef::Primitive(p) if config.cast_large_ints_to_i64 && needs_i64_cast(p) => {
214 let core_ty = core_prim_str(p);
215 format!("{name}: val.{name}.map(|v| v as {core_ty}).unwrap_or_default()")
216 }
217 TypeRef::Optional(inner)
218 if config.cast_large_ints_to_i64
219 && matches!(inner.as_ref(), TypeRef::Primitive(p) if needs_i64_cast(p)) =>
220 {
221 if let TypeRef::Primitive(p) = inner.as_ref() {
222 let core_ty = core_prim_str(p);
223 format!("{name}: val.{name}.map(|v| v as {core_ty})")
224 } else {
225 field_conversion_to_core(name, ty, false)
226 }
227 }
228 TypeRef::Duration if config.cast_large_ints_to_i64 => {
229 format!("{name}: val.{name}.map(|v| std::time::Duration::from_millis(v as u64)).unwrap_or_default()")
230 }
231 TypeRef::Duration => {
232 format!("{name}: val.{name}.map(std::time::Duration::from_millis).unwrap_or_default()")
233 }
234 TypeRef::Path => {
235 format!("{name}: val.{name}.map(Into::into).unwrap_or_default()")
236 }
237 TypeRef::Optional(inner) if matches!(inner.as_ref(), TypeRef::Path) => {
238 format!("{name}: val.{name}.map(|s| std::path::PathBuf::from(s))")
240 }
241 TypeRef::Optional(_) => {
242 format!("{name}: val.{name}.map(Some)")
245 }
246 TypeRef::Char => {
248 format!("{name}: val.{name}.and_then(|s| s.chars().next()).unwrap_or('*')")
249 }
250 TypeRef::Vec(inner) => match inner.as_ref() {
251 TypeRef::Json => {
252 format!(
253 "{name}: val.{name}.map(|v| v.into_iter().filter_map(|s| serde_json::from_str(&s).ok()).collect()).unwrap_or_default()"
254 )
255 }
256 TypeRef::Named(_) => {
257 format!("{name}: val.{name}.map(|v| v.into_iter().map(Into::into).collect()).unwrap_or_default()")
258 }
259 TypeRef::Primitive(p) if config.cast_large_ints_to_i64 && needs_i64_cast(p) => {
260 let core_ty = core_prim_str(p);
261 format!(
262 "{name}: val.{name}.map(|v| v.into_iter().map(|x| x as {core_ty}).collect()).unwrap_or_default()"
263 )
264 }
265 _ => format!("{name}: val.{name}.unwrap_or_default()"),
266 },
267 TypeRef::Map(k, v) if matches!(v.as_ref(), TypeRef::Json) => {
268 let k_is_json = matches!(k.as_ref(), TypeRef::Json);
270 let k_expr = if k_is_json {
271 "serde_json::from_str(&k).unwrap_or_default()"
272 } else {
273 "k"
274 };
275 format!(
276 "{name}: val.{name}.unwrap_or_default().into_iter().map(|(k, v)| ({k_expr}, serde_json::from_str(&v).unwrap_or(serde_json::json!(v)))).collect()"
277 )
278 }
279 TypeRef::Map(k, _v) if matches!(k.as_ref(), TypeRef::Json) => {
280 format!(
282 "{name}: val.{name}.unwrap_or_default().into_iter().map(|(k, v)| (serde_json::from_str(&k).unwrap_or_default(), v)).collect()"
283 )
284 }
285 TypeRef::Map(_, _) => {
286 format!("{name}: val.{name}.unwrap_or_default().into_iter().collect()")
288 }
289 _ => {
290 format!("{name}: val.{name}.unwrap_or_default()")
292 }
293 }
294}
295
296pub fn field_conversion_to_core(name: &str, ty: &TypeRef, optional: bool) -> String {
298 match ty {
299 TypeRef::Primitive(_) | TypeRef::String | TypeRef::Bytes | TypeRef::Unit => {
301 format!("{name}: val.{name}")
302 }
303 TypeRef::Json => {
305 if optional {
306 format!("{name}: val.{name}.as_ref().and_then(|s| serde_json::from_str(s).ok())")
307 } else {
308 format!("{name}: serde_json::from_str(&val.{name}).unwrap_or_default()")
309 }
310 }
311 TypeRef::Char => {
313 if optional {
314 format!("{name}: val.{name}.and_then(|s| s.chars().next())")
315 } else {
316 format!("{name}: val.{name}.chars().next().unwrap_or('*')")
317 }
318 }
319 TypeRef::Duration => {
321 if optional {
322 format!("{name}: val.{name}.map(std::time::Duration::from_millis)")
323 } else {
324 format!("{name}: std::time::Duration::from_millis(val.{name})")
325 }
326 }
327 TypeRef::Path => {
329 if optional {
330 format!("{name}: val.{name}.map(Into::into)")
331 } else {
332 format!("{name}: val.{name}.into()")
333 }
334 }
335 TypeRef::Named(type_name) if is_tuple_type_name(type_name) => {
338 format!("{name}: val.{name}")
339 }
340 TypeRef::Named(_) => {
341 if optional {
342 format!("{name}: val.{name}.map(Into::into)")
343 } else {
344 format!("{name}: val.{name}.into()")
345 }
346 }
347 TypeRef::Map(k, v) if matches!(v.as_ref(), TypeRef::Json) => {
349 let k_expr = if matches!(k.as_ref(), TypeRef::Json) {
350 "serde_json::from_str(&k).unwrap_or_default()"
351 } else {
352 "k"
353 };
354 if optional {
355 format!(
356 "{name}: val.{name}.map(|m| m.into_iter().map(|(k, v)| ({k_expr}, serde_json::from_str(&v).unwrap_or_default())).collect())"
357 )
358 } else {
359 format!(
360 "{name}: val.{name}.into_iter().map(|(k, v)| ({k_expr}, serde_json::from_str(&v).unwrap_or_default())).collect()"
361 )
362 }
363 }
364 TypeRef::Optional(inner) => match inner.as_ref() {
366 TypeRef::Json => format!("{name}: val.{name}.as_ref().and_then(|s| serde_json::from_str(s).ok())"),
367 TypeRef::Named(_) | TypeRef::Path => format!("{name}: val.{name}.map(Into::into)"),
368 TypeRef::Vec(vi) if matches!(vi.as_ref(), TypeRef::Named(_)) => {
369 format!("{name}: val.{name}.map(|v| v.into_iter().map(Into::into).collect())")
370 }
371 _ => format!("{name}: val.{name}"),
372 },
373 TypeRef::Vec(inner) => match inner.as_ref() {
375 TypeRef::Json => {
376 if optional {
377 format!(
378 "{name}: val.{name}.map(|v| v.into_iter().filter_map(|s| serde_json::from_str(&s).ok()).collect())"
379 )
380 } else {
381 format!("{name}: val.{name}.into_iter().filter_map(|s| serde_json::from_str(&s).ok()).collect()")
382 }
383 }
384 TypeRef::Named(type_name) if is_tuple_type_name(type_name) => {
386 format!("{name}: val.{name}")
387 }
388 TypeRef::Named(_) => {
389 if optional {
390 format!("{name}: val.{name}.map(|v| v.into_iter().map(Into::into).collect())")
391 } else {
392 format!("{name}: val.{name}.into_iter().map(Into::into).collect()")
393 }
394 }
395 _ => format!("{name}: val.{name}"),
396 },
397 TypeRef::Map(k, v) => {
400 let has_named_key = matches!(k.as_ref(), TypeRef::Named(n) if !is_tuple_type_name(n));
401 let has_named_val = matches!(v.as_ref(), TypeRef::Named(n) if !is_tuple_type_name(n));
402 let has_json_val = matches!(v.as_ref(), TypeRef::Json);
403 let has_json_key = matches!(k.as_ref(), TypeRef::Json);
404 if has_json_val || has_json_key || has_named_key || has_named_val {
405 let k_expr = if has_json_key {
406 "serde_json::from_str(&k).unwrap_or(serde_json::Value::String(k))"
407 } else if has_named_key {
408 "k.into()"
409 } else {
410 "k"
411 };
412 let v_expr = if has_json_val {
413 "serde_json::from_str(&v).unwrap_or(serde_json::Value::String(v))"
414 } else if has_named_val {
415 "v.into()"
416 } else {
417 "v"
418 };
419 if optional {
420 format!("{name}: val.{name}.map(|m| m.into_iter().map(|(k, v)| ({k_expr}, {v_expr})).collect())")
421 } else {
422 format!("{name}: val.{name}.into_iter().map(|(k, v)| ({k_expr}, {v_expr})).collect()")
423 }
424 } else {
425 if optional {
427 format!("{name}: val.{name}.map(|m| m.into_iter().collect())")
428 } else {
429 format!("{name}: val.{name}.into_iter().collect()")
430 }
431 }
432 }
433 }
434}
435
436pub fn field_conversion_to_core_cfg(name: &str, ty: &TypeRef, optional: bool, config: &ConversionConfig) -> String {
438 if optional && matches!(ty, TypeRef::Optional(_)) {
442 let inner_expr = field_conversion_to_core_cfg(name, ty, false, config);
445 if let Some(expr) = inner_expr.strip_prefix(&format!("{name}: ")) {
447 return format!("{name}: ({expr}).map(Some)");
448 }
449 return inner_expr;
450 }
451
452 if config.map_uses_jsvalue {
454 let is_nested_vec = matches!(ty, TypeRef::Vec(inner) if matches!(inner.as_ref(), TypeRef::Vec(_)));
455 let is_vec_json = matches!(ty, TypeRef::Vec(inner) if matches!(inner.as_ref(), TypeRef::Json));
456 let is_map = matches!(ty, TypeRef::Map(_, _));
457 if is_nested_vec || is_map || is_vec_json {
458 if optional {
459 return format!(
460 "{name}: val.{name}.as_ref().and_then(|v| serde_wasm_bindgen::from_value(v.clone()).ok())"
461 );
462 }
463 return format!("{name}: serde_wasm_bindgen::from_value(val.{name}.clone()).unwrap_or_default()");
464 }
465 if let TypeRef::Optional(inner) = ty {
466 let is_inner_nested = matches!(inner.as_ref(), TypeRef::Vec(vi) if matches!(vi.as_ref(), TypeRef::Vec(_)));
467 let is_inner_vec_json = matches!(inner.as_ref(), TypeRef::Vec(vi) if matches!(vi.as_ref(), TypeRef::Json));
468 let is_inner_map = matches!(inner.as_ref(), TypeRef::Map(_, _));
469 if is_inner_nested || is_inner_map || is_inner_vec_json {
470 return format!(
471 "{name}: val.{name}.as_ref().and_then(|v| serde_wasm_bindgen::from_value(v.clone()).ok())"
472 );
473 }
474 }
475 }
476
477 if config.vec_named_to_string {
481 if let TypeRef::Vec(_) = ty {
482 if optional {
483 return format!("{name}: val.{name}.as_ref().and_then(|s| serde_json::from_str(s).ok())");
484 }
485 return format!("{name}: serde_json::from_str(&val.{name}).unwrap_or_default()");
486 }
487 }
488 if config.json_to_string && matches!(ty, TypeRef::Json) {
490 return format!("{name}: Default::default()");
491 }
492 if config.map_uses_jsvalue && matches!(ty, TypeRef::Json) {
494 if optional {
495 return format!("{name}: val.{name}.as_ref().and_then(|v| serde_wasm_bindgen::from_value(v.clone()).ok())");
496 }
497 return format!("{name}: serde_wasm_bindgen::from_value(val.{name}.clone()).unwrap_or_default()");
498 }
499 if !config.cast_large_ints_to_i64
500 && !config.cast_f32_to_f64
501 && !config.json_to_string
502 && !config.vec_named_to_string
503 {
504 return field_conversion_to_core(name, ty, optional);
505 }
506 match ty {
508 TypeRef::Primitive(p) if config.cast_large_ints_to_i64 && needs_i64_cast(p) => {
509 let core_ty = core_prim_str(p);
510 if optional {
511 format!("{name}: val.{name}.map(|v| v as {core_ty})")
512 } else {
513 format!("{name}: val.{name} as {core_ty}")
514 }
515 }
516 TypeRef::Primitive(PrimitiveType::F32) if config.cast_f32_to_f64 => {
518 if optional {
519 format!("{name}: val.{name}.map(|v| v as f32)")
520 } else {
521 format!("{name}: val.{name} as f32")
522 }
523 }
524 TypeRef::Duration if config.cast_large_ints_to_i64 => {
525 if optional {
526 format!("{name}: val.{name}.map(|v| std::time::Duration::from_millis(v as u64))")
527 } else {
528 format!("{name}: std::time::Duration::from_millis(val.{name} as u64)")
529 }
530 }
531 TypeRef::Optional(inner) if matches!(inner.as_ref(), TypeRef::Primitive(p) if needs_i64_cast(p)) => {
532 if let TypeRef::Primitive(p) = inner.as_ref() {
533 let core_ty = core_prim_str(p);
534 format!("{name}: val.{name}.map(|v| v as {core_ty})")
535 } else {
536 field_conversion_to_core(name, ty, optional)
537 }
538 }
539 TypeRef::Vec(inner)
541 if config.cast_large_ints_to_i64
542 && matches!(inner.as_ref(), TypeRef::Primitive(p) if needs_i64_cast(p)) =>
543 {
544 if let TypeRef::Primitive(p) = inner.as_ref() {
545 let core_ty = core_prim_str(p);
546 if optional {
547 format!("{name}: val.{name}.map(|v| v.into_iter().map(|x| x as {core_ty}).collect())")
548 } else {
549 format!("{name}: val.{name}.into_iter().map(|v| v as {core_ty}).collect()")
550 }
551 } else {
552 field_conversion_to_core(name, ty, optional)
553 }
554 }
555 TypeRef::Map(k, v)
557 if config.cast_large_ints_to_i64 && matches!(v.as_ref(), TypeRef::Primitive(p) if needs_i64_cast(p)) =>
558 {
559 if let TypeRef::Primitive(p) = v.as_ref() {
560 let core_ty = core_prim_str(p);
561 if optional {
562 format!("{name}: val.{name}.map(|m| m.into_iter().map(|(k, v)| (k, v as {core_ty})).collect())")
563 } else {
564 format!("{name}: val.{name}.into_iter().map(|(k, v)| (k, v as {core_ty})).collect()")
565 }
566 } else {
567 field_conversion_to_core(name, ty, optional)
568 }
569 }
570 TypeRef::Vec(inner)
572 if config.cast_f32_to_f64 && matches!(inner.as_ref(), TypeRef::Primitive(PrimitiveType::F32)) =>
573 {
574 if optional {
575 format!("{name}: val.{name}.map(|v| v.into_iter().map(|x| x as f32).collect())")
576 } else {
577 format!("{name}: val.{name}.into_iter().map(|v| v as f32).collect()")
578 }
579 }
580 TypeRef::Optional(inner)
582 if config.cast_f32_to_f64
583 && matches!(inner.as_ref(), TypeRef::Vec(vi) if matches!(vi.as_ref(), TypeRef::Primitive(PrimitiveType::F32))) =>
584 {
585 format!("{name}: val.{name}.map(|v| v.into_iter().map(|x| x as f32).collect())")
586 }
587 _ => field_conversion_to_core(name, ty, optional),
589 }
590}
591
592fn apply_core_wrapper_to_core(
595 conversion: &str,
596 name: &str,
597 core_wrapper: &CoreWrapper,
598 vec_inner_core_wrapper: &CoreWrapper,
599 optional: bool,
600) -> String {
601 if *vec_inner_core_wrapper == CoreWrapper::Arc {
603 return conversion
604 .replace(
605 ".map(Into::into).collect()",
606 ".map(|v| std::sync::Arc::new(v.into())).collect()",
607 )
608 .replace(
609 "map(|v| v.into_iter().map(Into::into)",
610 "map(|v| v.into_iter().map(|v| std::sync::Arc::new(v.into()))",
611 );
612 }
613
614 match core_wrapper {
615 CoreWrapper::None => conversion.to_string(),
616 CoreWrapper::Cow => {
617 if let Some(expr) = conversion.strip_prefix(&format!("{name}: ")) {
621 if optional {
622 format!("{name}: {expr}.map(Into::into)")
623 } else if expr == format!("val.{name}") {
624 format!("{name}: val.{name}.into()")
625 } else if expr == "Default::default()" {
626 conversion.to_string()
629 } else {
630 format!("{name}: ({expr}).into()")
631 }
632 } else {
633 conversion.to_string()
634 }
635 }
636 CoreWrapper::Arc => {
637 if let Some(expr) = conversion.strip_prefix(&format!("{name}: ")) {
639 if expr == "Default::default()" {
640 conversion.to_string()
643 } else if optional {
644 format!("{name}: {expr}.map(|v| std::sync::Arc::new(v))")
645 } else {
646 format!("{name}: std::sync::Arc::new({expr})")
647 }
648 } else {
649 conversion.to_string()
650 }
651 }
652 CoreWrapper::Bytes => {
653 if let Some(expr) = conversion.strip_prefix(&format!("{name}: ")) {
655 if optional {
656 format!("{name}: {expr}.map(Into::into)")
657 } else if expr == format!("val.{name}") {
658 format!("{name}: val.{name}.into()")
659 } else if expr == "Default::default()" {
660 conversion.to_string()
663 } else {
664 format!("{name}: ({expr}).into()")
665 }
666 } else {
667 conversion.to_string()
668 }
669 }
670 CoreWrapper::ArcMutex => {
671 if let Some(expr) = conversion.strip_prefix(&format!("{name}: ")) {
673 if optional {
674 format!("{name}: {expr}.map(|v| std::sync::Arc::new(std::sync::Mutex::new(v.into())))")
675 } else if expr == format!("val.{name}") {
676 format!("{name}: std::sync::Arc::new(std::sync::Mutex::new(val.{name}.into()))")
677 } else {
678 format!("{name}: std::sync::Arc::new(std::sync::Mutex::new(({expr}).into()))")
679 }
680 } else {
681 conversion.to_string()
682 }
683 }
684 }
685}