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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use std::convert::TryFrom;

use proc_macro2::{Span, TokenStream};
use quote::ToTokens;
use syn::{self, spanned::Spanned, Block, LitStr};

/// A wrapper for expressions/blocks which automatically adds the start and end
/// braces.
///
/// - **full access** to variables environment.
/// - **full access** to control-flow of the environment via `return`, `?` etc.
#[derive(Debug, Clone)]
pub struct BlockContents(Block);

impl BlockContents {
    pub fn is_empty(&self) -> bool {
        self.0.stmts.is_empty()
    }

    pub fn span(&self) -> Span {
        self.0.span()
    }
}

impl ToTokens for BlockContents {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        self.0.to_tokens(tokens)
    }
}

impl TryFrom<&'_ LitStr> for BlockContents {
    type Error = syn::Error;

    fn try_from(s: &LitStr) -> Result<Self, Self::Error> {
        let mut block_str = s.value();
        block_str.insert(0, '{');
        block_str.push('}');
        LitStr::new(&block_str, s.span()).parse().map(Self)
    }
}

impl From<syn::Expr> for BlockContents {
    fn from(v: syn::Expr) -> Self {
        Self(Block {
            brace_token: syn::token::Brace(v.span()),
            stmts: vec![syn::Stmt::Expr(v, None)],
        })
    }
}

impl darling::FromMeta for BlockContents {
    fn from_value(value: &syn::Lit) -> darling::Result<Self> {
        if let syn::Lit::Str(s) = value {
            let contents = BlockContents::try_from(s)?;
            if contents.is_empty() {
                Err(darling::Error::unknown_value("").with_span(s))
            } else {
                Ok(contents)
            }
        } else {
            Err(darling::Error::unexpected_lit_type(value))
        }
    }

    fn from_expr(expr: &syn::Expr) -> darling::Result<Self> {
        if let syn::Expr::Lit(lit) = expr {
            Self::from_value(&lit.lit)
        } else {
            Ok(Self::from(expr.clone()))
        }
    }
}

#[cfg(test)]
mod test {
    use std::convert::TryInto;

    use super::*;
    use proc_macro2::Span;

    fn parse(s: &str) -> Result<BlockContents, syn::Error> {
        (&LitStr::new(s, Span::call_site())).try_into()
    }

    #[test]
    #[should_panic(expected = r#"cannot parse"#)]
    fn block_invalid_token_trees() {
        parse("let x = 2; { x+1").unwrap();
    }

    #[test]
    fn block_delimited_token_tree() {
        let expr = parse("let x = 2; { x+1 }").unwrap();
        assert_eq!(
            quote!(#expr).to_string(),
            quote!({
                let x = 2;
                {
                    x + 1
                }
            })
            .to_string()
        );
    }

    #[test]
    fn block_single_token_tree() {
        let expr = parse("42").unwrap();
        assert_eq!(quote!(#expr).to_string(), quote!({ 42 }).to_string());
    }
}