rust_writer_proc 1.0.4

This crate is useful for editing a Rust file from Rust code
Documentation 
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

// SPDX-License-Identifier: GPL-3.0

mod expand;

use crate::parse::{MacroAttrs, MacroFinderMutatorParsed, MacroLocalParsed};
use proc_macro::TokenStream;
use syn::{parse_macro_input, ItemStruct, LifetimeParam};

pub(crate) fn finder(attrs: TokenStream, item: TokenStream) -> TokenStream {
	let attrs_list = parse_macro_input!(attrs as MacroAttrs);
	let struct_def = parse_macro_input!(item as ItemStruct);

	match MacroFinderMutatorParsed::try_from(attrs_list, struct_def) {
		Ok(parsed) => expand::expand_finder(parsed).into(),
		Err(err) => err.to_compile_error().into(),
	}
}

pub(crate) fn local_finder(attrs: TokenStream, item: TokenStream) -> TokenStream {
	let mut finished = item.clone();

	let visit_lifetime = parse_macro_input!(attrs as LifetimeParam);
	let struct_def = parse_macro_input!(item as ItemStruct);

	let generated: TokenStream = match MacroLocalParsed::try_from(struct_def, "found") {
		Ok(parsed) => expand::expand_local_finder(visit_lifetime, parsed).into(),
		Err(err) => err.to_compile_error().into(),
	};

	finished.extend(generated);
	finished
}