azalea_inventory_macros/
lib.rs

1mod location_enum;
2mod menu_enum;
3mod menu_impl;
4mod parse_macro;
5mod utils;
6
7use parse_macro::{DeclareMenus, Field};
8use proc_macro::TokenStream;
9use proc_macro2::Span;
10use quote::quote;
11use syn::{Ident, parse_macro_input};
12
13#[proc_macro]
14pub fn declare_menus(input: TokenStream) -> TokenStream {
15    let mut input = parse_macro_input!(input as DeclareMenus);
16
17    // implicitly add a `player` field at the end unless an `inventory` field
18    // is present
19    for menu in &mut input.menus {
20        let mut inventory_field_missing = true;
21        for field in &menu.fields {
22            if matches!(field.name.to_string().as_str(), "inventory" | "player") {
23                inventory_field_missing = false;
24            }
25        }
26        if inventory_field_missing {
27            menu.fields.push(Field {
28                name: Ident::new("player", Span::call_site()),
29                length: 36,
30            })
31        }
32    }
33
34    let menu_enum = menu_enum::generate(&input);
35    let menu_impl = menu_impl::generate(&input);
36    let location_enum = location_enum::generate(&input);
37
38    quote! {
39        #menu_enum
40        #menu_impl
41
42        #location_enum
43    }
44    .into()
45}