genco_macros/
lib.rs

1//! [<img alt="github" src="https://img.shields.io/badge/github-udoprog/genco-8da0cb?style=for-the-badge&logo=github" height="20">](https://github.com/udoprog/genco)
2//! [<img alt="crates.io" src="https://img.shields.io/crates/v/genco-macros.svg?style=for-the-badge&color=fc8d62&logo=rust" height="20">](https://crates.io/crates/genco-macros)
3//! [<img alt="docs.rs" src="https://img.shields.io/badge/docs.rs-genco--macros-66c2a5?style=for-the-badge&logoColor=white&logo=data:image/svg+xml;base64,PHN2ZyByb2xlPSJpbWciIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyIgdmlld0JveD0iMCAwIDUxMiA1MTIiPjxwYXRoIGZpbGw9IiNmNWY1ZjUiIGQ9Ik00ODguNiAyNTAuMkwzOTIgMjE0VjEwNS41YzAtMTUtOS4zLTI4LjQtMjMuNC0zMy43bC0xMDAtMzcuNWMtOC4xLTMuMS0xNy4xLTMuMS0yNS4zIDBsLTEwMCAzNy41Yy0xNC4xIDUuMy0yMy40IDE4LjctMjMuNCAzMy43VjIxNGwtOTYuNiAzNi4yQzkuMyAyNTUuNSAwIDI2OC45IDAgMjgzLjlWMzk0YzAgMTMuNiA3LjcgMjYuMSAxOS45IDMyLjJsMTAwIDUwYzEwLjEgNS4xIDIyLjEgNS4xIDMyLjIgMGwxMDMuOS01MiAxMDMuOSA1MmMxMC4xIDUuMSAyMi4xIDUuMSAzMi4yIDBsMTAwLTUwYzEyLjItNi4xIDE5LjktMTguNiAxOS45LTMyLjJWMjgzLjljMC0xNS05LjMtMjguNC0yMy40LTMzLjd6TTM1OCAyMTQuOGwtODUgMzEuOXYtNjguMmw4NS0zN3Y3My4zek0xNTQgMTA0LjFsMTAyLTM4LjIgMTAyIDM4LjJ2LjZsLTEwMiA0MS40LTEwMi00MS40di0uNnptODQgMjkxLjFsLTg1IDQyLjV2LTc5LjFsODUtMzguOHY3NS40em0wLTExMmwtMTAyIDQxLjQtMTAyLTQxLjR2LS42bDEwMi0zOC4yIDEwMiAzOC4ydi42em0yNDAgMTEybC04NSA0Mi41di03OS4xbDg1LTM4Ljh2NzUuNHptMC0xMTJsLTEwMiA0MS40LTEwMi00MS40di0uNmwxMDItMzguMiAxMDIgMzguMnYuNnoiPjwvcGF0aD48L3N2Zz4K" height="20">](https://docs.rs/genco-macros)
4
5#![recursion_limit = "256"]
6#![allow(clippy::type_complexity)]
7#![cfg_attr(proc_macro_span, feature(proc_macro_span))]
8
9extern crate proc_macro;
10
11use proc_macro2::Span;
12use syn::parse::{ParseStream, Parser as _};
13
14struct Ctxt {
15    receiver: syn::Ident,
16    module: syn::Path,
17}
18
19impl Default for Ctxt {
20    fn default() -> Self {
21        let mut module = syn::Path {
22            leading_colon: None,
23            segments: syn::punctuated::Punctuated::default(),
24        };
25
26        module
27            .segments
28            .push(syn::Ident::new("genco", Span::call_site()).into());
29
30        Self {
31            receiver: syn::Ident::new("__genco_macros_toks", Span::call_site()),
32            module,
33        }
34    }
35}
36
37mod ast;
38mod cursor;
39mod encoder;
40mod fake;
41mod quote;
42mod quote_fn;
43mod quote_in;
44mod requirements;
45mod static_buffer;
46mod string_parser;
47
48#[proc_macro]
49pub fn quote(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
50    let cx = Ctxt::default();
51    let parser = crate::quote::Quote::new(&cx);
52
53    let parser = move |stream: ParseStream| parser.parse(stream);
54
55    let (req, output) = match parser.parse(input) {
56        Ok(data) => data,
57        Err(e) => return proc_macro::TokenStream::from(e.to_compile_error()),
58    };
59
60    let check = req.into_check(&cx.receiver);
61
62    let Ctxt { receiver, module } = &cx;
63
64    let gen = q::quote! {{
65        let mut #receiver = #module::tokens::Tokens::new();
66
67        {
68            let mut #receiver = &mut #receiver;
69            #output
70        }
71
72        #check
73        #receiver
74    }};
75
76    gen.into()
77}
78
79#[proc_macro]
80pub fn quote_in(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
81    let quote_in = syn::parse_macro_input!(input as quote_in::QuoteIn);
82    quote_in.stream.into()
83}
84
85#[proc_macro]
86pub fn quote_fn(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
87    let quote_fn = syn::parse_macro_input!(input as quote_fn::QuoteFn);
88    quote_fn.stream.into()
89}
genco_macros/lib.rs

genco_macros/
lib.rs
