kosame_dsl 0.3.0

Macro-based Rust ORM focused on developer ergonomics
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
86
87
88
89
90
91
92
93
94
95

mod field;
mod from;
mod group_by;
mod having;
mod limit;
mod offset;
mod order_by;
mod returning;
mod select;
mod set;
mod values;
mod r#where;
mod with;

pub use field::*;
pub use from::*;
pub use group_by::*;
pub use having::*;
pub use limit::*;
pub use offset::*;
pub use order_by::*;
pub use returning::*;
pub use select::*;
pub use set::*;
pub use values::*;
pub use r#where::*;
pub use with::*;

use crate::{
    command::SelectCombinator,
    parse_option::ParseOption,
    pretty::{PrettyPrint, Printer},
};

pub fn peek_clause(input: syn::parse::ParseStream) -> bool {
    With::peek(input)
        || From::peek(input)
        || Where::peek(input)
        || GroupBy::peek(input)
        || Having::peek(input)
        || OrderBy::peek(input)
        || Limit::peek(input)
        || Offset::peek(input)
        || Returning::peek(input)
        || Set::peek(input)
        || Values::peek(input)
        || SelectCombinator::peek(input)
}

pub struct Clause<'a> {
    keywords: &'a [&'a dyn PrettyPrint],
    body: &'a dyn PrettyPrint,
    first: bool,
}

impl<'a> Clause<'a> {
    pub fn new(keywords: &'a [&'a dyn PrettyPrint], body: &'a dyn PrettyPrint) -> Self {
        Self {
            keywords,
            body,
            first: false,
        }
    }

    pub fn new_first(keywords: &'a [&'a dyn PrettyPrint], body: &'a dyn PrettyPrint) -> Self {
        Self {
            keywords,
            body,
            first: true,
        }
    }
}

impl PrettyPrint for Clause<'_> {
    fn pretty_print(&self, printer: &mut Printer<'_>) {
        if !self.first {
            printer.scan_break();
            " ".pretty_print(printer);
        }
        printer.scan_trivia(!self.first, true);

        for (index, keyword) in self.keywords.iter().enumerate() {
            keyword.pretty_print(printer);
            if index < self.keywords.len() - 1 {
                " ".pretty_print(printer);
            }
        }
        printer.scan_indent(1);
        printer.scan_break();
        " ".pretty_print(printer);
        printer.scan_trivia(false, true);
        self.body.pretty_print(printer);
        printer.scan_indent(-1);
    }
}