toql_sql_expr_macro 0.4.0

Library with sql expr macro for Toql
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109

//! The `sql_expr!` macro compiles an SQL expression into program code.
//!
//! ### Example
//! ```rust, ignore
//! use toql_sql_expr_macro::sql_expr;
//!
//! let e = sql_expr!("SELECT ..id FROM User .. WHERE ..age <= <age>");
//! ```

#![recursion_limit = "512"]

extern crate proc_macro;

extern crate syn;

#[macro_use]
extern crate quote;

use syn::parse_macro_input;

use proc_macro::TokenStream;

mod sql_expr_macro;

#[proc_macro]
pub fn sql_expr(input: TokenStream) -> TokenStream {
    let _ = env_logger::try_init(); // Avoid multiple init
                                    // eprintln!("{:?}", input);

    let ast = parse_macro_input!(input as sql_expr_macro::SqlExprMacro);

    let gen = sql_expr_macro::parse(&ast.query, &mut ast.arguments.iter());

    match gen {
        Ok(o) => {
            tracing::debug!(
                "Source code for `{}`:\n{}",
                ast.query.value(),
                o.to_string()
            );
            TokenStream::from(o)
        }
        Err(e) => {
            tracing::debug!(
                "Source code for `{}`:\n{}",
                ast.query.value(),
                e.to_string()
            );
            TokenStream::from(e)
        }
    }
}

#[test]
fn literal_and_alias() {
    use sql_expr_macro::SqlExprMacro;
    let input = "\"SELECT ..id FROM Table .. JOIN OtherTable ...\"";

    let m = syn::parse_str(input);
    assert_eq!(m.is_ok(), true);

    let SqlExprMacro { query, arguments } = m.unwrap();
    let f = sql_expr_macro::parse(&query, &mut arguments.iter());
    assert_eq!(f.is_ok(), true);

    assert_eq!(f.unwrap().to_string(), "{ let mut t = toql :: sql_expr :: SqlExpr :: new ( ) ; \
        t . extend ( toql :: sql_expr :: SqlExpr :: from ( vec ! [ \
            toql :: sql_expr :: SqlExprToken :: Literal ( String :: from ( \"SELECT \" ) ) , \
            toql :: sql_expr :: SqlExprToken :: SelfAlias , \
            toql :: sql_expr :: SqlExprToken :: Literal ( String :: from ( \".id FROM Table \" ) ) , \
            toql :: sql_expr :: SqlExprToken :: SelfAlias , \
            toql :: sql_expr :: SqlExprToken :: Literal ( String :: from ( \" JOIN OtherTable \" ) ) , \
            toql :: sql_expr :: SqlExprToken :: OtherAlias ] ) ) ; t }");
}
#[test]
fn placeholders_and_aux_params() {
    use sql_expr_macro::SqlExprMacro;
    let input = "\"SELECT ?, <aux_parm>\"";

    let m = syn::parse_str(input);
    assert_eq!(m.is_ok(), true);

    let SqlExprMacro { query, arguments } = m.unwrap();
    let f = sql_expr_macro::parse(&query, &mut arguments.iter());
    assert_eq!(f.is_ok(), true);

    assert_eq!(f.unwrap().to_string(), "{ let mut t = toql :: sql_expr :: SqlExpr :: new ( ) ; \
    t . extend ( toql :: sql_expr :: SqlExpr :: from ( vec ! [ \
        toql :: sql_expr :: SqlExprToken :: Literal ( String :: from ( \"SELECT \" ) ) , \
        toql :: sql_expr :: SqlExprToken :: UnresolvedArg , \
        toql :: sql_expr :: SqlExprToken :: Literal ( String :: from ( \", \" ) ) , \
        toql :: sql_expr :: SqlExprToken :: AuxParam ( String :: from ( \"aux_parm\" ) ) ] ) ) ; t }");
}
#[test]
fn quotes() {
    use sql_expr_macro::SqlExprMacro;
    let input = "\"SELECT '''?', '<aux_parm>'\"";

    let m = syn::parse_str(input);
    assert_eq!(m.is_ok(), true);

    let SqlExprMacro { query, arguments } = m.unwrap();
    let f = sql_expr_macro::parse(&query, &mut arguments.iter());
    assert_eq!(f.is_ok(), true);

    assert_eq!(f.unwrap().to_string(), "{ let mut t = toql :: sql_expr :: SqlExpr :: new ( ) ; \
    t . extend ( toql :: sql_expr :: SqlExpr :: from ( vec ! [ \
        toql :: sql_expr :: SqlExprToken :: Literal ( String :: from ( \"SELECT \'\'\'?\', \'<aux_parm>\'\" ) ) ] ) ) ; t }");
}