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

mod delim;
mod r#macro;
mod printer;
mod ring_buffer;
mod rust;
mod span;
mod text;
mod token;
mod trivia;

pub use delim::*;
pub use r#macro::*;
pub use printer::*;
pub use ring_buffer::*;
pub use span::*;
pub use token::*;
pub use trivia::*;

use syn::parse::Parse;

pub fn pretty_print_str<T>(
    source_text: &str,
    initial_space: isize,
    initial_indent: isize,
) -> syn::Result<String>
where
    T: Parse + PrettyPrint,
{
    let ast: T = syn::parse_str(source_text)?;
    let trivia = Lexer::new(source_text).collect::<Vec<_>>();

    let mut printer = Printer::new(&trivia, initial_space, initial_indent);
    ast.pretty_print(&mut printer);
    Ok(printer.eof())
}

pub fn pretty_print_ast<T>(ast: &T) -> String
where
    T: PrettyPrint,
{
    let mut printer = Printer::new(&[], 0, 0);
    ast.pretty_print(&mut printer);
    printer.eof()
}

pub trait PrettyPrint {
    fn pretty_print(&self, printer: &mut Printer<'_>);
}

impl<T> PrettyPrint for Option<T>
where
    T: PrettyPrint,
{
    fn pretty_print(&self, printer: &mut Printer<'_>) {
        if let Some(inner) = self {
            inner.pretty_print(printer);
        }
    }
}

impl<T> PrettyPrint for [T]
where
    T: PrettyPrint,
{
    fn pretty_print(&self, printer: &mut Printer<'_>) {
        for item in self {
            item.pretty_print(printer);
        }
    }
}

impl<T> PrettyPrint for syn::punctuated::Punctuated<T, syn::Token![,]>
where
    T: PrettyPrint,
{
    fn pretty_print(&self, printer: &mut Printer<'_>) {
        for (index, item) in self.pairs().enumerate() {
            item.value().pretty_print(printer);
            if item.punct().is_some() {
                printer.scan_no_break_trivia();
            }
            if index == self.len() - 1 {
                printer.scan_text(",".into(), TextMode::Break);
                printer.advance_cursor(",");
            } else {
                item.punct().unwrap().pretty_print(printer);
                printer.scan_same_line_trivia();
                printer.scan_break();
                " ".pretty_print(printer);
                printer.scan_trivia(true, true);
            }
        }
    }
}