1#![recursion_limit = "1024"]
2#![deny(warnings, missing_copy_implementations)]
4#![allow(
6 clippy::needless_pass_by_value,
7 clippy::option_map_unwrap_or_else,
8 clippy::option_map_unwrap_or
9)]
10#![warn(
11 clippy::wrong_pub_self_convention,
12 clippy::mut_mut,
13 clippy::non_ascii_literal,
14 clippy::similar_names,
15 clippy::unicode_not_nfc,
16 clippy::if_not_else,
17 clippy::items_after_statements,
18 clippy::used_underscore_binding
19)]
20#![cfg_attr(feature = "nightly", feature(proc_macro_diagnostic, proc_macro_span))]
21
22extern crate proc_macro;
23extern crate proc_macro2;
24#[macro_use]
25extern crate quote;
26#[macro_use]
27extern crate syn;
28
29use proc_macro::TokenStream;
30
31mod diagnostic_shim;
32mod field;
33mod meta;
34mod model;
35mod resolved_at_shim;
36mod util;
37
38mod as_changeset;
39mod as_expression;
40mod associations;
41mod diesel_numeric_ops;
42mod from_sql_row;
43mod identifiable;
44mod insertable;
45mod query_id;
46mod queryable;
47mod queryable_by_name;
48mod sql_type;
49
50use diagnostic_shim::*;
51
52#[proc_macro_derive(
53 AsChangeset,
54 attributes(table_name, primary_key, column_name, changeset_options)
55)]
56pub fn derive_as_changeset(input: TokenStream) -> TokenStream {
57 expand_derive(input, as_changeset::derive)
58}
59
60#[proc_macro_derive(AsExpression, attributes(diesel, sql_type))]
61pub fn derive_as_expression(input: TokenStream) -> TokenStream {
62 expand_derive(input, as_expression::derive)
63}
64
65#[proc_macro_derive(Associations, attributes(belongs_to, column_name, table_name))]
66pub fn derive_associations(input: TokenStream) -> TokenStream {
67 expand_derive(input, associations::derive)
68}
69
70#[proc_macro_derive(DieselNumericOps)]
71pub fn derive_diesel_numeric_ops(input: TokenStream) -> TokenStream {
72 expand_derive(input, diesel_numeric_ops::derive)
73}
74
75#[proc_macro_derive(FromSqlRow, attributes(diesel))]
76pub fn derive_from_sql_row(input: TokenStream) -> TokenStream {
77 expand_derive(input, from_sql_row::derive)
78}
79
80#[proc_macro_derive(Identifiable, attributes(table_name, primary_key, column_name))]
81pub fn derive_identifiable(input: TokenStream) -> TokenStream {
82 expand_derive(input, identifiable::derive)
83}
84
85#[proc_macro_derive(Insertable, attributes(table_name, column_name, diesel))]
86pub fn derive_insertable(input: TokenStream) -> TokenStream {
87 expand_derive(input, insertable::derive)
88}
89
90#[proc_macro_derive(QueryId)]
91pub fn derive_query_id(input: TokenStream) -> TokenStream {
92 expand_derive(input, query_id::derive)
93}
94
95#[proc_macro_derive(Queryable, attributes(column_name, diesel))]
96pub fn derive_queryable(input: TokenStream) -> TokenStream {
97 expand_derive(input, queryable::derive)
98}
99
100#[proc_macro_derive(QueryableByName, attributes(table_name, column_name, sql_type, diesel))]
101pub fn derive_queryable_by_name(input: TokenStream) -> TokenStream {
102 expand_derive(input, queryable_by_name::derive)
103}
104
105#[proc_macro_derive(SqlType, attributes(postgres, sqlite_type, mysql_type))]
106pub fn derive_sql_type(input: TokenStream) -> TokenStream {
107 expand_derive(input, sql_type::derive)
108}
109
110fn expand_derive(
111 input: TokenStream,
112 f: fn(syn::DeriveInput) -> Result<proc_macro2::TokenStream, Diagnostic>,
113) -> TokenStream {
114 let item = syn::parse(input).unwrap();
115 match f(item) {
116 Ok(x) => x.into(),
117 Err(e) => {
118 e.emit();
119 "".parse().unwrap()
120 }
121 }
122}