bee_block/output/token_scheme/
mod.rs

1// Copyright 2020-2022 IOTA Stiftung
2// SPDX-License-Identifier: Apache-2.0
3
4mod simple;
5
6pub use self::simple::SimpleTokenScheme;
7use crate::Error;
8
9///
10#[derive(Clone, Debug, Eq, PartialEq, Ord, PartialOrd, derive_more::From, packable::Packable)]
11#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
12#[packable(unpack_error = Error)]
13#[packable(tag_type = u8, with_error = Error::InvalidTokenSchemeKind)]
14pub enum TokenScheme {
15    ///
16    #[packable(tag = SimpleTokenScheme::KIND)]
17    Simple(SimpleTokenScheme),
18}
19
20impl TokenScheme {
21    /// Returns the token scheme kind of a [`TokenScheme`].
22    pub fn kind(&self) -> u8 {
23        match self {
24            Self::Simple(_) => SimpleTokenScheme::KIND,
25        }
26    }
27}
28
29#[cfg(feature = "dto")]
30#[allow(missing_docs)]
31pub mod dto {
32    use derive_more::From;
33    use serde::{Deserialize, Serialize};
34
35    pub use super::simple::dto::SimpleTokenSchemeDto;
36    use super::*;
37    use crate::error::dto::DtoError;
38
39    #[derive(Clone, Debug, Eq, PartialEq, Serialize, Deserialize, From)]
40    #[serde(untagged)]
41    pub enum TokenSchemeDto {
42        /// A simple token scheme.
43        Simple(SimpleTokenSchemeDto),
44    }
45
46    impl From<&TokenScheme> for TokenSchemeDto {
47        fn from(value: &TokenScheme) -> Self {
48            match value {
49                TokenScheme::Simple(v) => Self::Simple(v.into()),
50            }
51        }
52    }
53
54    impl TryFrom<&TokenSchemeDto> for TokenScheme {
55        type Error = DtoError;
56
57        fn try_from(value: &TokenSchemeDto) -> Result<Self, Self::Error> {
58            Ok(match value {
59                TokenSchemeDto::Simple(v) => Self::Simple(v.try_into()?),
60            })
61        }
62    }
63}
bee_block/output/token_scheme/mod.rs

bee_block/output/token_scheme/
mod.rs
