pkenum_core 0.3.1

Core logic for pkenum.
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

use syn::{ Ident, ItemEnum, };
use crate::token::EnumVariant;

/// Represents a Rust enum.
pub struct EnumToken {
    /// The name of the enum.
    ///
    /// # Example
    /// `ident` would equate to `IpAddrKind`
    ///
    /// ```
    /// enum IpAddrKind {
    ///     V4,
    ///     V6,
    /// }
    /// ```
    pub ident: Ident,

    /// Variants in the enum.
    ///
    /// # Example
    /// `variants` would equate to `V4` and `V6`.
    ///
    /// ```
    /// enum IpAddrKind {
    ///     V4,
    ///     V6(String),
    /// }
    /// ```
    pub variants: Vec<EnumVariant>,
}

impl EnumToken {
    /// Initiaizes `EnumToken` from `ItemEnum`.
    pub fn from_ast(item: ItemEnum) -> Self {
        EnumToken {
            ident: item.ident,
            variants: item.variants
                .into_iter()
                .map(EnumVariant::from_variant)
                .collect()
        }
    }
}