teo_parser/ast/
struct_declaration.rs

1use std::collections::BTreeMap;
2use maplit::btreemap;
3use crate::ast::doc_comment::DocComment;
4use crate::ast::function_declaration::FunctionDeclaration;
5use crate::ast::generics::{GenericsConstraint, GenericsDeclaration};
6use crate::ast::identifier::Identifier;
7use crate::{declare_container_node, impl_container_node_defaults, node_child_fn, node_children_iter, node_children_iter_fn, node_optional_child_fn};
8use crate::format::Writer;
9use crate::r#type::keyword::Keyword;
10use crate::r#type::reference::Reference;
11use crate::r#type::Type;
12use crate::r#type::Type::StructObject;
13use crate::traits::has_availability::HasAvailability;
14use crate::traits::info_provider::InfoProvider;
15use crate::traits::named_identifiable::NamedIdentifiable;
16use crate::traits::write::Write;
17
18declare_container_node!(StructDeclaration, named, availability,
19    pub(crate) comment: Option<usize>,
20    pub(crate) identifier: usize,
21    pub(crate) generics_declaration: Option<usize>,
22    pub(crate) generics_constraint: Option<usize>,
23    pub(crate) function_declarations: Vec<usize>,
24);
25
26impl_container_node_defaults!(StructDeclaration, named, availability);
27
28node_children_iter!(StructDeclaration, FunctionDeclaration, FunctionsIter, function_declarations);
29
30impl StructDeclaration {
31
32    node_optional_child_fn!(comment, DocComment);
33    node_child_fn!(identifier, Identifier);
34    node_optional_child_fn!(generics_declaration, GenericsDeclaration);
35    node_optional_child_fn!(generics_constraint, GenericsConstraint);
36    node_children_iter_fn!(function_declarations, FunctionsIter);
37
38    pub fn instance_function(&self, name: &str) -> Option<&FunctionDeclaration> {
39        self.function_declarations().find(|f| !f.r#static && f.identifier().name() == name)
40    }
41
42    pub fn static_function(&self, name: &str) -> Option<&FunctionDeclaration> {
43        self.function_declarations().find(|f| f.r#static && f.identifier().name() == name)
44    }
45
46    pub fn keywords_map(&self) -> BTreeMap<Keyword, Type> {
47        btreemap! {
48            Keyword::SelfIdentifier => StructObject(Reference::new(self.path.clone(), self.string_path.clone()), if let Some(generics_declaration) = self.generics_declaration() {
49                generics_declaration.identifiers().map(|i| Type::GenericItem(i.name.clone())).collect()
50            } else {
51                vec![]
52            })
53        }
54    }
55}
56
57impl InfoProvider for StructDeclaration {
58    fn namespace_skip(&self) -> usize {
59        1
60    }
61}
62
63impl Write for StructDeclaration {
64    fn write<'a>(&'a self, writer: &mut Writer<'a>) {
65        writer.write_children(self, self.children.values());
66    }
67}
teo_parser/ast/struct_declaration.rs

teo_parser/ast/
struct_declaration.rs
