syn-ext 0.5.0

Human friendly or editable extension for syn
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

use syn::punctuated::{Pair, Punctuated};

/// Extension for [syn::punctuated::Punctuated]
pub trait PunctuatedExt<T, P> {
    /// Removes and returns the element at position index, popping all elements after it and push them back.
    fn remove(&mut self, index: usize) -> Pair<T, P>
    where
        P: Default;
}

impl<T, P> PunctuatedExt<T, P> for Punctuated<T, P> {
    fn remove(&mut self, index: usize) -> Pair<T, P>
    where
        P: Default,
    {
        let mut stack = Vec::new();
        for _ in index + 1..self.len() {
            stack.push(self.pop().unwrap().into_value());
        }
        let removed = self.pop().unwrap();
        while let Some(item) = stack.pop() {
            self.push(item)
        }
        removed
    }
}

#[cfg(test)]
#[cfg(all(feature = "parsing", feature = "derive"))]
mod test {
    use super::*;
    use crate::assert_quote_eq;
    use crate::meta::MetaList1;
    use syn::parse_quote;

    #[test]
    fn test_remove() {
        let mut list: MetaList1 = parse_quote!(meta(a, b, c, d));
        list.nested.remove(2);
        let expected: MetaList1 = parse_quote!(meta(a, b, d));
        assert_quote_eq!(list, expected);
        list.nested.remove(0);
        let expected: MetaList1 = parse_quote!(meta(b, d));
        assert_quote_eq!(list, expected);
    }
}