1
use proc_macro::TokenStream;
2

            
3
use proc_macro2::Span;
4
use quote::quote;
5
use syn::punctuated::Punctuated;
6
use syn::token::Comma;
7
use syn::{
8
    parenthesized, parse::Parse, parse::ParseStream, parse_macro_input, Ident, ItemFn, LitInt,
9
    LitStr, Path, Result,
10
};
11

            
12
#[derive(Debug)]
13
struct RuleSetAndPriority {
14
    rule_set: LitStr,
15
    priority: LitInt,
16
}
17

            
18
impl Parse for RuleSetAndPriority {
19
99
    fn parse(input: ParseStream) -> Result<Self> {
20
99
        let content;
21
99
        parenthesized!(content in input);
22
99
        let rule_set: LitStr = content.parse()?;
23
99
        let _: Comma = content.parse()?;
24
99
        let priority: LitInt = content.parse()?;
25
99
        Ok(RuleSetAndPriority { rule_set, priority })
26
99
    }
27
}
28

            
29
#[derive(Debug)]
30
struct RegisterRuleArgs {
31
    pub rule_sets: Vec<RuleSetAndPriority>,
32
}
33

            
34
impl Parse for RegisterRuleArgs {
35
99
    fn parse(input: ParseStream) -> Result<Self> {
36
99
        let rule_sets = Punctuated::<RuleSetAndPriority, Comma>::parse_terminated(input)?;
37
99
        Ok(RegisterRuleArgs {
38
99
            rule_sets: rule_sets.into_iter().collect(),
39
99
        })
40
99
    }
41
}
42

            
43
/**
44
 * Register a rule with the given rule sets and priorities.
45
 */
46
#[proc_macro_attribute]
47
99
pub fn register_rule(arg_tokens: TokenStream, item: TokenStream) -> TokenStream {
48
99
    let func = parse_macro_input!(item as ItemFn);
49
99
    let rule_ident = &func.sig.ident;
50
99
    let static_name = format!("CONJURE_GEN_RULE_{}", rule_ident).to_uppercase();
51
99
    let static_ident = Ident::new(&static_name, rule_ident.span());
52

            
53
99
    let args = parse_macro_input!(arg_tokens as RegisterRuleArgs);
54
99
    let rule_sets = args
55
99
        .rule_sets
56
99
        .iter()
57
99
        .map(|rule_set| {
58
99
            let rule_set_name = &rule_set.rule_set;
59
99
            let priority = &rule_set.priority;
60
99
            quote! {
61
99
                (#rule_set_name, #priority as u8)
62
99
            }
63
99
        })
64
99
        .collect::<Vec<_>>();
65

            
66
99
    let expanded = quote! {
67
99
        #func
68
99

            
69
99
        use ::conjure_core::rule_engine::_dependencies::*; // ToDo idk if we need to explicitly do that?
70
99

            
71
99
        #[::conjure_core::rule_engine::_dependencies::distributed_slice(::conjure_core::rule_engine::RULES_DISTRIBUTED_SLICE)]
72
99
        pub static #static_ident: ::conjure_core::rule_engine::Rule<'static> = ::conjure_core::rule_engine::Rule {
73
99
            name: stringify!(#rule_ident),
74
99
            application: #rule_ident,
75
99
            rule_sets: &[#(#rule_sets),*],
76
99
        };
77
99
    };
78

            
79
99
    TokenStream::from(expanded)
80
99
}
81

            
82
21
fn parse_parenthesized<T: Parse>(input: ParseStream) -> Result<Vec<T>> {
83
21
    let content;
84
21
    parenthesized!(content in input);
85

            
86
21
    let mut paths = Vec::new();
87
21
    while !content.is_empty() {
88
15
        let path = content.parse()?;
89
15
        paths.push(path);
90
15
        if content.is_empty() {
91
15
            break;
92
        }
93
        content.parse::<Comma>()?;
94
    }
95

            
96
21
    Ok(paths)
97
21
}
98

            
99
struct RuleSetArgs {
100
    name: LitStr,
101
    priority: LitInt,
102
    dependencies: Vec<LitStr>,
103
    solver_families: Vec<Path>,
104
}
105

            
106
impl Parse for RuleSetArgs {
107
15
    fn parse(input: ParseStream) -> Result<Self> {
108
15
        let name = input.parse()?;
109
15
        input.parse::<Comma>()?;
110
15
        let priority = input.parse()?;
111

            
112
15
        if input.is_empty() {
113
            return Ok(Self {
114
                name,
115
                priority,
116
                dependencies: Vec::new(),
117
                solver_families: Vec::new(),
118
            });
119
15
        }
120
15

            
121
15
        input.parse::<Comma>()?;
122
15
        let dependencies = parse_parenthesized::<LitStr>(input)?;
123

            
124
15
        if input.is_empty() {
125
9
            return Ok(Self {
126
9
                name,
127
9
                priority,
128
9
                dependencies,
129
9
                solver_families: Vec::new(),
130
9
            });
131
6
        }
132
6

            
133
6
        input.parse::<Comma>()?;
134
6
        let solver_families = parse_parenthesized::<Path>(input)?;
135

            
136
6
        Ok(Self {
137
6
            name,
138
6
            priority,
139
6
            dependencies,
140
6
            solver_families,
141
6
        })
142
15
    }
143
}
144

            
145
/**
146
* Register a rule set with the given name, priority, and dependencies.
147
*
148
* # Example
149
* ```rust
150
 * use conjure_macros::register_rule_set;
151
 * register_rule_set!("MyRuleSet", 10, ("DependencyRuleSet", "AnotherRuleSet"));
152
* ```
153
 */
154
#[proc_macro]
155
15
pub fn register_rule_set(args: TokenStream) -> TokenStream {
156
    let RuleSetArgs {
157
15
        name,
158
15
        priority,
159
15
        dependencies,
160
15
        solver_families,
161
15
    } = parse_macro_input!(args as RuleSetArgs);
162

            
163
15
    let static_name = format!("CONJURE_GEN_RULE_SET_{}", name.value()).to_uppercase();
164
15
    let static_ident = Ident::new(&static_name, Span::call_site());
165

            
166
15
    let dependencies = quote! {
167
15
        #(#dependencies),*
168
15
    };
169

            
170
15
    let solver_families = quote! {
171
15
        #(#solver_families),*
172
15
    };
173

            
174
15
    let expanded = quote! {
175
15
        use ::conjure_core::rule_engine::_dependencies::*; // ToDo idk if we need to explicitly do that?
176
15
        #[::conjure_core::rule_engine::_dependencies::distributed_slice(::conjure_core::rule_engine::RULE_SETS_DISTRIBUTED_SLICE)]
177
15
        pub static #static_ident: ::conjure_core::rule_engine::RuleSet<'static> = ::conjure_core::rule_engine::RuleSet::new(#name, #priority, &[#dependencies], &[#solver_families]);
178
15
    };
179
15

            
180
15
    TokenStream::from(expanded)
181
15
}