diff --git a/codegen/src/types/mod.rs b/codegen/src/types/mod.rs index f2950ab5d9..bd3896bb92 100644 --- a/codegen/src/types/mod.rs +++ b/codegen/src/types/mod.rs @@ -365,6 +365,23 @@ impl Module { pub fn ident(&self) -> &Ident { &self.name } + + /// Returns this `Module`s child `mod`s. + pub fn children(&self) -> impl Iterator { + self.children.iter() + } + + /// Returns the generated types. + pub fn types( + &self, + ) -> impl Iterator, &TypeDefGen)> { + self.types.iter() + } + + /// Returns the root `mod` used for resolving type paths. + pub fn root_mod(&self) -> &Ident { + &self.root_mod + } } #[derive(Debug, Clone)]