rkt_codegen 0.6.0

use std::error::Error;
use std::path::Path;

use devise::ext::SpanDiagnosticExt;
use proc_macro2::TokenStream;
use syn::{Ident, LitStr};

pub fn _macro(input: proc_macro::TokenStream) -> devise::Result<TokenStream> {
    let root_glob = syn::parse::<LitStr>(input)?;
    let tests = entry_to_tests(&root_glob)
        .map_err(|e| root_glob.span().error(format!("failed to read: {}", e)))?;

    Ok(quote!(#(#tests)*))
}

fn entry_to_tests(root_glob: &LitStr) -> Result<Vec<TokenStream>, Box<dyn Error>> {
    let manifest_dir = std::env::var("CARGO_MANIFEST_DIR").expect("MANIFEST_DIR");
    let full_glob = Path::new(&manifest_dir).join(root_glob.value());

    let mut tests = vec![];
    for path in glob::glob(&full_glob.to_string_lossy()).map_err(Box::new)? {
        let path = path.map_err(Box::new)?;
        let name = path
            .file_name()
            .and_then(|f| f.to_str())
            .map(|name| {
                name.trim_matches(|c| char::is_numeric(c) || c == '-')
                    .replace(['-', '.'], "_")
            })
            .ok_or("invalid file name")?;

        let ident = Ident::new(&name.to_lowercase(), root_glob.span());
        let full_path = Path::new(&manifest_dir).join(&path).display().to_string();
        tests.push(quote_spanned!(root_glob.span() =>
            #[allow(unused_doc_comments)]
            mod #ident {
                macro_rules! doc_comment { ($x:expr) => (#[doc = $x] extern {}); }
                doc_comment!(include_str!(#full_path));
            }
        ));
    }

    if tests.is_empty() {
        return Err(format!("glob '{}' evaluates to 0 files", full_glob.display()).into());
    }

    Ok(tests)
}