xsd_parser/pipeline/optimizer/
dynamic_to_choice.rs

1use crate::models::{
2    meta::{ComplexMeta, ElementMeta, ElementMode, GroupMeta, MetaType, MetaTypeVariant},
3    Ident,
4};
5use crate::traits::VecHelper;
6
7use super::Optimizer;
8
9impl Optimizer {
10    /// This will use a enum that contains all known variants of the dynamic
11    /// type instead of a dynamic box.
12    ///
13    /// # Examples
14    ///
15    /// Consider the following XML schema.
16    /// ```xml
17    #[doc = include_str!("../../../tests/optimizer/abstract.xsd")]
18    /// ```
19    ///
20    /// Without this optimization this will result in the following code:
21    /// ```rust
22    #[doc = include_str!("../../../tests/optimizer/expected0/convert_dynamic_to_choice.rs")]
23    /// ```
24    ///
25    /// With this optimization the following code is generated:
26    /// ```rust
27    #[doc = include_str!("../../../tests/optimizer/expected1/convert_dynamic_to_choice.rs")]
28    /// ```
29    pub fn convert_dynamic_to_choice(mut self) -> Self {
30        use std::collections::btree_map::Entry;
31
32        tracing::debug!("convert_dynamic_to_choice");
33
34        let idents = self
35            .types
36            .items
37            .iter()
38            .filter_map(|(ident, ty)| {
39                if matches!(&ty.variant, MetaTypeVariant::Dynamic(_)) {
40                    Some(ident)
41                } else {
42                    None
43                }
44            })
45            .cloned()
46            .collect::<Vec<_>>();
47
48        for ident in idents {
49            let content_name = self.types.name_builder().shared_name("Content").finish();
50            let content_ident = Ident::new(content_name).with_ns(ident.ns);
51
52            let type_ = self.types.items.get_mut(&ident).unwrap();
53            let form = type_.form();
54            let MetaTypeVariant::Dynamic(x) = &mut type_.variant else {
55                crate::unreachable!();
56            };
57
58            let mut si = GroupMeta::default();
59            for derived in &x.derived_types {
60                si.elements.find_or_insert(derived.clone(), |ident| {
61                    ElementMeta::new(ident, derived.clone(), ElementMode::Element, form)
62                });
63            }
64
65            type_.variant = MetaTypeVariant::ComplexType(ComplexMeta {
66                content: Some(content_ident.clone()),
67                is_dynamic: true,
68                ..Default::default()
69            });
70
71            match self.types.items.entry(content_ident) {
72                Entry::Vacant(e) => {
73                    e.insert(MetaType::new(MetaTypeVariant::Choice(si)));
74                }
75                Entry::Occupied(_) => crate::unreachable!(),
76            }
77        }
78
79        self
80    }
81}