better_comprehension 3.0.3

collection and iterator comprehensions for Rust
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
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

use crate::iter_clause::IterClause;
use crate::mapping::{Mapping, MappingElse};
use quote::quote;
use syn::parse::ParseStream;

/*-----------------BTreeSetComprehension------------------- */
#[derive(Debug)]
pub struct BTreeSetComprehension {
    pub mapping: Mapping,
    pub iter_clauses: Vec<IterClause>,
}

impl quote::ToTokens for BTreeSetComprehension {
    fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {
        let BTreeSetComprehension {
            mapping:
                Mapping {
                    left_key,
                    left_value,
                    right_expr,
                },
            iter_clauses,
        } = self;

        if left_value.is_some() {
            panic!("BTreeSet isn't key-value collection");
        }

        let mut nested_code = match right_expr {
            None => quote! {
                __b_tree_set_comprehension.insert(#left_key);
            },
            Some(MappingElse {
                conditions,
                else_key,
                else_value,
            }) => {
                if else_value.is_some() {
                    panic!("BTreeSet isn't key-value collection");
                }

                quote! {
                    if #conditions {
                        __b_tree_set_comprehension.insert(#left_key);
                    } else {
                        __b_tree_set_comprehension.insert(#else_key);
                    }
                }
            }
        };

        nested_code = crate::eager_evaluation::handle_nested_loops(iter_clauses, nested_code);
        nested_code = quote! {
            {
                use ::std::collections::BTreeSet;
                let mut __b_tree_set_comprehension = BTreeSet::new();
                #nested_code
                __b_tree_set_comprehension
            }
        };

        tokens.extend(nested_code);
    }
}

impl syn::parse::Parse for BTreeSetComprehension {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        let (mapping, iter_clauses) = crate::common_parse(input);

        Ok(Self {
            mapping,
            iter_clauses,
        })
    }
}