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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135

/*
 * @Author: plucky
 * @Date: 2022-10-21 16:53:21
 * @LastEditTime: 2024-05-17 16:24:23
 */

//! Derive macro for sqlx to implement Create, Read, Update, and Delete (CRUD) methods.
//! # Use
//! adding the following to your project's Cargo.toml:
//! ```toml
//! [dependencies]
//! co-orm = { virsion = "0.3", features = ["mysql"] }
//! sqlx = { version = "0.7", features = ["mysql"] }
//! ```
//! features: mysql, postgres, sqlite, mssql
//! 
//! # Examples 
//! ``` no_run,ignore
//! use co_orm::{Crud, args};
//! 
//! #[derive(Debug, Crud, sqlx::FromRow)]
//! #[co_orm(rename = "users")] // rename table name
//! pub struct User {
//!     // #[co_orm(id)] // default first field is primary key
//!     #[co_orm(seq)] // sequence field, insert will ignore this field
//!     pub id: u64,
//!     #[co_orm(rename = "name")] // rename field name
//!     #[co_orm(by)] // generate query_by_field,update_by_field,delete_by_field
//!     pub name: String,
//!     #[co_orm(update)] // generate method update_xxx. 
//!     pub password: String,
//!     #[co_orm(skip)] // ignore field
//!     #[sqlx(skip)]
//!     pub addr: Option<String>,
//!     // #[co_orm(skip_insert)] // insert will skip this field.
//!     // pub update_at: Option<NaiveDateTime>,
//! }
//! 
//! // use crud
//! let u = User::get(&pool, 1).await;
//! println!("get {:?}", u);
//! let u = User::get_by(&pool, "where id=?", args!(1)).await;
//! println!("get_by {:?}", u);
//! ```

pub use co_orm_macros::Crud;
pub use co_orm_macros::FromRow;


/// sqlx::query_as
/// ``` no_run,ignore
/// query_as!(User, "select * from users where name = ?", name).fetch_one(&pool).await
/// ```
#[macro_export]
macro_rules! query_as (
    ($out_struct:path, $query:expr) => ( {
        sqlx::query_as::<_, $out_struct>($query)
    });
    ($out_struct:path, $query:expr, $($args:expr),*) => ( {
        sqlx::query_as::<_, $out_struct>($query)
        $(.bind($args))*
    })
);

/// sqlx::query
/// ``` no_run,ignore
/// query!("insert into users (name, password) values (?,?)", name, password).execute(&pool).await
/// ```
#[macro_export]
macro_rules! query (
    ($query:expr) => ( {
        sqlx::query($query)
    });
    ($query:expr, $($args:expr),*) => ( {
        sqlx::query($query)
        $(.bind($args))*
    })
);




#[cfg(feature = "mysql")]
#[macro_export]
macro_rules! args {
    // ($sql:expr) => {
    //     args!($sql,);
    // };
   
    ($($args:expr),*) => {{
        use sqlx::Arguments;
        let mut sqlargs = sqlx::mysql::MySqlArguments::default();
        $(sqlargs.add($args);)*
        sqlargs
        
        
    }};
}

#[cfg(feature = "postgres")]
#[macro_export]
macro_rules! args {
    ($($args:expr),*) => {{
        use sqlx::Arguments;
        let mut sqlargs = sqlx::postgres::PgArguments::default();
        $(sqlargs.add($args);)*
        sqlargs
    }};

}

#[cfg(feature = "sqlite")]
#[macro_export]
macro_rules! args {
    ($($args:expr),*) => {{
        use sqlx::Arguments;
        let mut sqlargs = sqlx::sqlite::SqliteArguments::default();
        $(sqlargs.add($args);)*
        sqlargs
    }};
}

#[cfg(feature = "mssql")]
#[macro_export]
macro_rules! args {
    ($($args:expr),*) => {{
        use sqlx::Arguments;
        let mut sqlargs = sqlx::mssql::MssqlArguments::default();
        $(sqlargs.add($args);)*
        sqlargs
    }};

}