rusty_bind_parser/parsing/
module.rs

1use std::collections::HashSet;
2
3use anyhow::{anyhow, Context as _};
4use const_format::formatcp;
5
6use super::{
7    Context,
8    Parser,
9    EXTERN_EXCEPTION_TRAIT_MODULE_NAME,
10    RUST_MODULE_NAME,
11    TRAITS_MODULE_NAME,
12};
13use crate::model::r#enum::Enum;
14use crate::model::Module;
15
16impl Parser<syn::ItemMod> for Module {
17    fn parse(data: &syn::ItemMod, context: &mut Context) -> anyhow::Result<Self> {
18        let syn_enums = get_enums_from_module(data, context)?;
19
20        let enums: HashSet<Enum> = syn_enums
21            .iter()
22            .map(|item| Parser::parse(*item, context))
23            .collect::<Result<_, _>>()?;
24
25        if enums.len() != syn_enums.len() {
26            return Err(anyhow!("Duplicated enums found"));
27        }
28
29        let syn_exception_trait =
30            get_extern_mod_from_module(data, context, EXTERN_EXCEPTION_TRAIT_MODULE_NAME).context(
31                formatcp!("Cant find extern mod `{EXTERN_EXCEPTION_TRAIT_MODULE_NAME}`"),
32            )?;
33        let exception_trait = Parser::parse(&syn_exception_trait.items, context)
34            .context(formatcp!("Error in `{EXTERN_EXCEPTION_TRAIT_MODULE_NAME}`"))?;
35
36        let syn_traits_module = get_extern_mod_from_module(data, context, TRAITS_MODULE_NAME)
37            .context(formatcp!("Cant find extern mod `{TRAITS_MODULE_NAME}`"))?;
38        let traits_module = Parser::parse(&syn_traits_module.items, context)
39            .context(formatcp!("Error in `{TRAITS_MODULE_NAME}`"))?;
40
41        let syn_rust_module = get_extern_mod_from_module(data, context, RUST_MODULE_NAME)
42            .context(formatcp!("Cant find extern mod `{RUST_MODULE_NAME}`"))?;
43        let rust_module = Parser::parse(&syn_rust_module.items, context)
44            .context(formatcp!("Error in `{RUST_MODULE_NAME}`"))?;
45
46        let unknown = context.name_tracker.get_unknown();
47        if !unknown.is_empty() {
48            return Err(anyhow!("Some types are unknown: {unknown:?}"));
49        }
50
51        Ok(Self {
52            enums,
53            exception_trait,
54            traits_module,
55            rust_module,
56        })
57    }
58}
59
60fn get_enums_from_module<'a>(
61    module: &'a syn::ItemMod,
62    context: &Context,
63) -> anyhow::Result<Vec<&'a syn::ItemEnum>> {
64    Ok(module
65        .content
66        .as_ref()
67        .context("The module is empty")?
68        .1
69        .iter()
70        .filter_map(|item| match item {
71            syn::Item::Enum(enum_item) => Some(enum_item),
72            _ => None,
73        })
74        .filter(|item| context.build_context.check_cfg_attrs(&item.attrs))
75        .collect())
76}
77
78fn get_extern_mod_from_module<'a>(
79    module: &'a syn::ItemMod,
80    context: &Context,
81    extern_lang: &str,
82) -> Option<&'a syn::ItemForeignMod> {
83    module.content.as_ref().and_then(|(_, items)| {
84        items.iter().find_map(|module_item| match module_item {
85            syn::Item::ForeignMod(
86                rust_module @ syn::ItemForeignMod {
87                    abi:
88                        syn::Abi {
89                            name: Some(lang), ..
90                        },
91                    ..
92                },
93            ) if lang.value() == extern_lang
94                && context.build_context.check_cfg_attrs(&rust_module.attrs) =>
95            {
96                Some(rust_module)
97            }
98            _ => None,
99        })
100    })
101}
rusty_bind_parser/parsing/module.rs

rusty_bind_parser/parsing/
module.rs
