trl_codegen 1.2.1

This library provides auto generation of some common methods based on Rust macros
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

//! # modifier
//!
//! This module contains the `Modifier` enum which represents the `self` modifier

use proc_macro2::TokenStream;
use quote::quote;
use syn::{
    Token,
    parse::{Parse, ParseStream},
};

/// Represents the `self` modifier
#[derive(Debug, Clone)]
pub enum Modifier {
    /// fn get(self)
    Move,
    // fn get(&self)
    Ref,
    // fn get(&mut self)
    MutRef,
}

impl Parse for Modifier {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        if input.peek(Token![mut]) && input.peek2(Token![ref]) {
            input.parse::<Token![mut]>()?;
            input.parse::<Token![ref]>()?;
            Ok(Modifier::MutRef)
        } else if input.peek(Token![ref]) {
            input.parse::<Token![ref]>()?;
            Ok(Modifier::Ref)
        } else if input.peek(Token![move]) {
            input.parse::<Token![move]>()?;
            Ok(Modifier::Move)
        } else {
            Err(input.error("expected one of: `ref`, `mut ref`, `move`"))
        }
    }
}

impl Into<TokenStream> for Modifier {
    fn into(self) -> TokenStream {
        TokenStream::from(&self)
    }
}

impl From<&Modifier> for TokenStream {
    fn from(value: &Modifier) -> Self {
        match value {
            Modifier::Move => quote!(),
            Modifier::Ref => quote!(&),
            Modifier::MutRef => quote!(&mut),
        }
    }
}