sat-rs/satrs-mib/codegen/src/lib.rs

use quote::{format_ident, quote, ToTokens};
use syn::{parse_macro_input, ItemConst, LitStr};

#[proc_macro_attribute]
pub fn resultcode(
    args: proc_macro::TokenStream,
    item: proc_macro::TokenStream,
) -> proc_macro::TokenStream {
    // Handle attributes first.
    let mut info_str: Option<LitStr> = None;
    let res_code_parser = syn::meta::parser(|meta| {
        if meta.path.is_ident("info") {
            info_str = Some(meta.value()?.parse()?);
            Ok(())
        } else {
            Err(meta.error("unsupported property for resultcode attribute"))
        }
    });
    parse_macro_input!(args with res_code_parser);
    let item = parse_macro_input!(item as ItemConst);

    // Generate additional generated info struct used for introspection.
    let result_code_name = &item.ident;
    let name_as_str = result_code_name.to_string();
    let gen_struct_name = format_ident!("{}_EXT", result_code_name);
    let info_str = info_str.map_or(String::from(""), |v| v.value());
    // TODO: Group string
    let generated_struct = quote! {
        const #gen_struct_name: satrs_mib::res_code::ResultU16Info =
            satrs_mib::res_code::ResultU16Info::const_new(
                #name_as_str,
                &#result_code_name,
                "",
                #info_str
            );
    };

    // The input constant returncode is written to the output in any case.
    let mut output = item.to_token_stream();
    output.extend(generated_struct);
    output.into()
}