kotlin_poet_rs/spec/
secondary_constructor.rs

1use crate::io::RenderKotlin;
2use crate::spec::{VisibilityModifier, Argument, CodeBlock, Parameter, PrimaryConstructor};
3use crate::spec::kdoc::{KdocSlot, mixin_kdoc_mutators};
4use crate::tokens;
5
6/// Defines [Class's secondary constructor](https://kotlinlang.org/docs/classes.html#constructors)
7#[derive(Debug, Clone)]
8pub struct SecondaryConstructor {
9    parameters: Vec<Parameter>,
10    delegate_parameters: Vec<Argument>,
11    visibility_modifier: VisibilityModifier,
12    body: Option<CodeBlock>,
13    kdoc: KdocSlot
14}
15
16impl SecondaryConstructor {
17    pub fn new() -> SecondaryConstructor {
18        SecondaryConstructor {
19            parameters: Vec::new(),
20            delegate_parameters: Vec::new(),
21            visibility_modifier: VisibilityModifier::default(),
22            body: None,
23            kdoc: KdocSlot::default()
24        }
25    }
26
27    pub fn parameter(mut self, parameter: Parameter) -> SecondaryConstructor {
28        self.parameters.push(parameter);
29        self
30    }
31
32    pub fn delegate_argument(mut self, argument: Argument) -> SecondaryConstructor {
33        self.delegate_parameters.push(argument);
34        self
35    }
36
37    pub fn body<CodeBlockLike: Into<CodeBlock>>(mut self, body: CodeBlockLike) -> SecondaryConstructor {
38        self.body = Some(body.into());
39        self
40    }
41
42    pub fn visibility_modifier(mut self, visibility_modifier: VisibilityModifier) -> SecondaryConstructor {
43        self.visibility_modifier = visibility_modifier;
44        self
45    }
46
47    mixin_kdoc_mutators!();
48}
49
50impl RenderKotlin for SecondaryConstructor {
51    fn render_into(&self, block: &mut CodeBlock) {
52        block.push_renderable(&self.kdoc);
53
54        let mut pc = PrimaryConstructor::new()
55            .visibility_modifier(self.visibility_modifier.clone());
56
57        for parameter in &self.parameters {
58            pc = pc.parameter(parameter.clone());
59        }
60
61        block.push_renderable(&pc);
62        block.push_space();
63        block.push_static_atom(tokens::COLON);
64        block.push_space();
65        block.push_static_atom(tokens::keyword::THIS);
66        block.push_round_brackets(|params_block| {
67            params_block.push_comma_separated(&self.delegate_parameters);
68        });
69        block.push_space();
70        block.push_curly_brackets(|body_block| {
71            if let Some(body) = &self.body {
72                body_block.push_renderable(body);
73            }
74        });
75    }
76}
77
78#[cfg(test)]
79mod tests {
80    use crate::io::RenderKotlin;
81    use crate::spec::{VisibilityModifier, Argument, CodeBlock, Parameter, SecondaryConstructor, Type, KDoc, Name};
82
83    #[test]
84    fn test_secondary_constructor() {
85        let secondary_constructor = SecondaryConstructor::new()
86            .visibility_modifier(VisibilityModifier::Public)
87            .parameter(Parameter::new(Name::from("name"), Type::string()))
88            .parameter(Parameter::new(Name::from("age"), Type::int()))
89            .delegate_argument(Argument::new_positional(CodeBlock::atom("name")))
90            .delegate_argument(Argument::new_positional(CodeBlock::atom("age")))
91            .body(CodeBlock::statement("println(42)"));
92
93        let rendered = secondary_constructor.render_string();
94        let expected = "public constructor(name: kotlin.String, age: kotlin.Int) : this(name, age) {\n    println(42)\n}";
95        assert_eq!(rendered, expected);
96    }
97
98    #[test]
99    fn test_secondary_constructor_with_kdoc() {
100        let secondary_constructor = SecondaryConstructor::new()
101            .kdoc(KDoc::from("Hello\nWorld"));
102
103        let rendered = secondary_constructor.render_string();
104        let expected = "/**\n * Hello\n * World\n */\npublic constructor() : this() {\n}";
105        assert_eq!(rendered, expected);
106    }
107}
kotlin_poet_rs/spec/secondary_constructor.rs

kotlin_poet_rs/spec/
secondary_constructor.rs
