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
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
#![forbid(unsafe_code)]

mod attribute;
mod derive;
mod utils;

use syn::{parse_macro_input, AttributeArgs, DeriveInput};

/// Derives the Ent trait and additional typed functionality
///
/// ```
/// use entity::{Ent, Id, WeakDatabaseRc};
///
/// /// Define an entity and derive all associated ent functionality
/// ///
/// /// The entity must also implement clone as this is a requirement of
/// /// the Ent trait
/// ///
/// /// If using serde, this struct will need to implement serialize and
/// /// deserialize itself AND include the attribute ent(typetag)
/// #[derive(Clone, Ent)]
/// pub struct PageEnt {
///     /// Required and can only be specified once to indicate the struct
///     /// field that contains the ent's id
///     #[ent(id)]
///     id: Id,
///
///     /// Required and can only be specified once to indicate the struct
///     /// field that contains the database. Must be an option!
///     ///
///     /// If using serde, this field will need to be skipped via serde(skip)
///     /// as it will not be serialized and, when deserializing, will be
///     /// filled in with the database automatically
///     #[ent(database)]
///     database: WeakDatabaseRc,
///
///     /// Required and can only be specified once to indicate the struct
///     /// field that contains the timestamp of when the ent was created
///     #[ent(created)]
///     created: u64,
///
///     /// Required and can only be specified once to indicate the struct
///     /// field that contains the timestamp of when the ent was last updated
///     #[ent(last_updated)]
///     last_updated: u64,
///
///     /// A public ent field that is indexed, meaning that searches for this
///     /// ent by its title should be faster, but this will also take up
///     /// more space in the database
///     #[ent(field(indexed))]
///     title: String,
///
///     /// A public ent field that is not indexed
///     #[ent(field)]
///     url: String,
///
///     /// An edge out to a ContentEnt that is shallowly connected, meaning
///     /// that when this ent is deleted, the ent connected by this edge
///     /// will remove this ent if it is reversely-connected
///     #[ent(edge(policy = "shallow", type = "ContentEnt"))]
///     header: Id,
///
///     /// An optional edge out to a ContentEnt that is deeply connected,
///     /// meaning that when this ent is deleted, the ent connected by this
///     /// edge will also be deleted
///     #[ent(edge(policy = "deep", type = "ContentEnt"))]
///     subheader: Option<Id>,
///
///     /// An edge out to zero or more ContentEnt, defaulting to doing
///     /// nothing special when this ent is deleted
///     #[ent(edge(type = "ContentEnt"))]
///     paragraphs: Vec<Id>,
/// }
///
/// #[derive(Clone, Ent)]
/// pub struct ContentEnt {
///     #[ent(id)]
///     id: Id,
///
///     #[ent(database)]
///     database: WeakDatabaseRc,
///
///     #[ent(created)]
///     created: u64,
///
///     #[ent(last_updated)]
///     last_updated: u64,
///
///     #[ent(field)]
///     text: String,
/// }
/// ```
#[proc_macro_derive(Ent, attributes(ent))]
pub fn derive_ent(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_ent)(input)
}

#[proc_macro_derive(EntDebug, attributes(ent))]
pub fn derive_ent_debug(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_ent_debug)(input)
}

#[proc_macro_derive(EntType, attributes(ent))]
pub fn derive_ent_type(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_ent_type)(input)
}

#[proc_macro_derive(EntBuilder, attributes(ent))]
pub fn derive_ent_builder(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_ent_builder)(input)
}

#[proc_macro_derive(EntLoader, attributes(ent))]
pub fn derive_ent_loader(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_ent_loader)(input)
}

#[proc_macro_derive(EntQuery, attributes(ent))]
pub fn derive_ent_query(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_ent_query)(input)
}

#[proc_macro_derive(EntTypedFields, attributes(ent))]
pub fn derive_ent_typed_fields(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_ent_typed_fields)(input)
}

#[proc_macro_derive(EntTypedEdges, attributes(ent))]
pub fn derive_ent_typed_edges(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_ent_typed_edges)(input)
}

#[proc_macro_derive(EntWrapper, attributes(ent))]
pub fn derive_ent_wrapper(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_ent_wrapper)(input)
}

#[proc_macro_derive(ValueLike)]
pub fn derive_value_like(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_value_like)(input)
}

#[proc_macro_derive(IntoValue)]
pub fn derive_into_value(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_into_value)(input)
}

#[proc_macro_derive(TryFromValue)]
pub fn derive_try_from_value(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    utils::do_derive(derive::do_derive_try_from_value)(input)
}

/// Injects elements needed for an ent to be derived.
///
/// ```
/// use entity::{simple_ent};
///
/// #[simple_ent]
/// pub struct MyEnt {
///     name: String,
///     value: u32,
/// }
/// ```
#[proc_macro_attribute]
pub fn simple_ent(
    args: proc_macro::TokenStream,
    input: proc_macro::TokenStream,
) -> proc_macro::TokenStream {
    let args = parse_macro_input!(args as AttributeArgs);
    let input = parse_macro_input!(input as DeriveInput);

    let expanded = utils::entity_crate()
        .and_then(|root| attribute::do_simple_ent(root, args, input))
        .unwrap_or_else(|x| x.write_errors());

    proc_macro::TokenStream::from(expanded)
}