rusty_bind_parser/swift/
enums.rs

1use std::fmt::Display;
2
3use syn::ItemEnum;
4
5use crate::enum_helpers::{enum_tag_name, is_primitive_enum};
6use crate::extern_module_translator::ExternModuleTranslator;
7
8/// Translates rust enum into its C equivalents.
9/// For any Rust enum only a C enum class representation is generated
10/// (for both unit and field enums).
11///
12pub fn translate_c_enums(emt: &ExternModuleTranslator) -> String {
13    emt.shared_enums
14        .iter()
15        .map(|e| {
16            if is_primitive_enum(e) {
17                translate_c_enum(e)
18            } else {
19                create_complex_enum_tag(e)
20            }
21        })
22        .collect()
23}
24
25fn translate_c_enum(enum_item: &ItemEnum) -> String {
26    let enum_name = enum_item.ident.to_string();
27    let variants = enum_item.variants.iter().map(|variant| &variant.ident);
28    create_c_enum_class(&enum_name, variants)
29}
30
31fn create_complex_enum_tag(enum_item: &ItemEnum) -> String {
32    let enum_tag_name = enum_tag_name(enum_item.ident.to_string().as_str());
33    let variants = enum_item.variants.iter().map(|variant| &variant.ident);
34    create_c_enum_class(&enum_tag_name, variants)
35}
36
37pub fn create_c_enum_class<T: Display>(name: &str, variants: impl Iterator<Item = T>) -> String {
38    let variants = variants
39        .map(|id| format!("    {name}_{id}"))
40        .collect::<Vec<_>>()
41        .join(",\n");
42    format!("\nenum {name} {{\n{variants}\n}};\n")
43}
44
45#[cfg(test)]
46mod tests {
47    use pretty_assertions::assert_eq;
48
49    use super::*;
50    use crate::utils::helpers;
51
52    #[test]
53    fn test_creating_enum_class_definition() {
54        let enum_item = helpers::get_enum_item();
55        let definition = translate_c_enum(&enum_item);
56
57        let expected_definition = "
58enum En1 {
59    En1_V1,
60    En1_V2
61};
62";
63        assert_eq!(definition, expected_definition);
64    }
65}
rusty_bind_parser/swift/enums.rs

rusty_bind_parser/swift/
enums.rs
