teo_parser/ast/
middleware.rs1use crate::ast::argument_list_declaration::ArgumentListDeclaration;
2use crate::ast::callable_variant::CallableVariant;
3use crate::ast::identifier::Identifier;
4use crate::{declare_container_node, impl_container_node_defaults, node_child_fn, node_optional_child_fn};
5use crate::ast::doc_comment::DocComment;
6use crate::format::Writer;
7use crate::traits::has_availability::HasAvailability;
8use crate::traits::info_provider::InfoProvider;
9use crate::traits::write::Write;
10
11#[derive(Debug, Copy, Clone)]
12pub enum MiddlewareType {
13    RequestMiddleware,
14    HandlerMiddleware,
15}
16
17declare_container_node!(MiddlewareDeclaration, named, availability,
18    pub(crate) identifier: usize,
19    pub(crate) argument_list_declaration: Option<usize>,
20    pub(crate) comment: Option<usize>,
21    pub(crate) middleware_type: MiddlewareType,
22);
23
24impl_container_node_defaults!(MiddlewareDeclaration, named, availability);
25
26impl MiddlewareDeclaration {
27
28    node_optional_child_fn!(comment, DocComment);
29
30    node_child_fn!(identifier, Identifier);
31
32    node_optional_child_fn!(argument_list_declaration, ArgumentListDeclaration);
33
34    pub fn callable_variants(&self) -> Vec<CallableVariant> {
35        vec![CallableVariant {
36            generics_declarations: vec![],
37            argument_list_declaration: self.argument_list_declaration(),
38            generics_constraints: vec![],
39            pipeline_input: None,
40            pipeline_output: None,
41        }]
42    }
43
44    pub fn middleware_type(&self) -> MiddlewareType {
45        self.middleware_type
46    }
47}
48
49impl InfoProvider for MiddlewareDeclaration {
50    fn namespace_skip(&self) -> usize {
51        1
52    }
53}
54
55impl Write for MiddlewareDeclaration {
56    fn write<'a>(&'a self, writer: &mut Writer<'a>) {
57        writer.write_children(self, self.children.values());
58    }
59    fn is_block_level_element(&self) -> bool {
60        true
61    }
62}