1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use crate::{CodeBuffer, Statement};

/// An element with true statements. (if-else)
pub trait WithTrueStatements {
    /// Gets the true statements.
    fn true_statements(&self) -> &[Box<dyn Statement>];

    /// Adds the true statement.
    fn with_true_statement<S>(self, true_statement: S) -> Self
    where
        S: 'static + Statement;

    /// Adds the true statement.
    fn add_true_statement<S>(&mut self, true_statement: S)
    where
        S: 'static + Statement;

    /// Writes the true statements.
    fn write_true_statements(&self, b: &mut CodeBuffer, level: usize) {
        for statement in self.true_statements() {
            statement.write(b, level);
        }
    }

    /// Writes the curly-bracket true statement block.
    fn write_true_curly_statement_block(&self, b: &mut CodeBuffer, level: usize) {
        if self.true_statements().is_empty() {
            b.write("{}");
        } else {
            b.write("{");
            b.end_line();
            self.write_true_statements(b, level + 1);
            b.indent(level);
            b.write("}");
        }
    }
}