sqler_macros 0.0.1-beta

procedural macros for sqler crate
Documentation 
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

#[derive(Debug)]
struct Sg {
    need_leading_space: bool,
    sg: String,
    need_trailing_space: bool,
    len: usize,
}

impl Sg {
    fn new(need_leading_space: bool, sg: &str, need_trailing_space: bool) -> Sg {
        let mut len = sg.len();
        if need_leading_space {
            len += 1;
        }

        if need_trailing_space {
            len += 1;
        }

        Sg {
            need_leading_space,
            sg: String::from(sg),
            need_trailing_space,
            len,
        }
    }
}

pub struct StmtSg {
    sgs: Vec<Sg>,
    len: usize,
}

impl StmtSg {
    pub fn new() -> StmtSg {
        Self {
            sgs: Vec::new(),
            len: 0,
        }
    }

    /* pub fn construct_with_one(need_leading_space: bool, sg:String, need_trailing_space: bool) -> StmtSg {

    } */

    #[inline]
    pub fn len(&self) -> usize {
        self.len
    }

    pub fn add_token(
        &mut self,
        need_leading_space: bool,
        sg: &str,
        need_trailing_space: bool,
    ) -> &mut Self {
        let sg = Sg::new(need_leading_space, sg, need_trailing_space);
        self.len += sg.len;
        self.sgs.push(sg);

        self
    }

    pub fn construct(&self) -> String {
        let mut stmt_sg = String::from('"');

        let mut trailing_space_added = false;
        for sg in &self.sgs {
            if sg.need_leading_space && !trailing_space_added {
                stmt_sg.push(' ');
            }

            stmt_sg.push_str(&sg.sg);

            if sg.need_trailing_space {
                stmt_sg.push(' ');
                trailing_space_added = true;
            } else {
                trailing_space_added = false;
            }
        }
        stmt_sg.push('"');
        stmt_sg
    }
}