1mod attrs;
2mod codegen;
3mod sanitization;
4#[cfg(test)]
5mod testing;
6mod tree;
7
8use std::env;
9use std::path::PathBuf;
10
11use proc_macro2::TokenStream;
12use syn::spanned::Spanned;
13use syn::{parse2, Item};
14use testdata_rt::GlobSpec;
15
16use crate::attrs::{ArgAttrs, MacroArgs};
17use crate::codegen::generate;
18
19#[proc_macro_attribute]
20pub fn files(
21 raw_args: proc_macro::TokenStream,
22 raw_item: proc_macro::TokenStream,
23) -> proc_macro::TokenStream {
24 match files2(raw_args.into(), raw_item.into()) {
25 Ok(tokens) => tokens.into(),
26 Err(err) => err.to_compile_error().into(),
27 }
28}
29
30fn files2(raw_args: TokenStream, raw_item: TokenStream) -> Result<TokenStream, syn::Error> {
31 let span = raw_args.span();
32 let args = MacroArgs::parse(raw_args)?;
33
34 let item: Item = parse2(raw_item)?;
35 let item = if let Item::Fn(item) = item {
36 item
37 } else {
38 return Err(syn::Error::new(
39 item.span(),
40 "expected function after #[testdata]",
41 ));
42 };
43
44 let mut args_attrs = Vec::new();
45 for arg in &item.sig.inputs {
46 let attrs = ArgAttrs::parse(arg)?;
47 args_attrs.push(attrs);
48 }
49
50 let mut spec = GlobSpec::new();
51 if let Some(root) = &args.root {
52 spec.root = root.into();
53 }
54 for attrs in &args_attrs {
55 spec.args
56 .push(testdata_rt::ArgSpec::parse(&attrs.glob).map_err(|e| syn::Error::new(span, e))?);
57 }
58
59 let cwd = env::var_os("CARGO_MANIFEST_DIR")
60 .ok_or_else(|| syn::Error::new(span, "Missing CARGO_MANIFEST_DIR"))?;
61 let cwd = PathBuf::from(cwd);
62 let stems = spec.glob_from(&cwd).map_err(|e| syn::Error::new(span, e))?;
63
64 Ok(generate(&spec, &args, &item, &stems))
65}