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
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101

use std::convert::TryFrom;

use serde::{Deserialize, Serialize};

use crate::Error;

/// Available serialization formats.
#[derive(Copy, Clone, Debug, Deserialize, PartialEq, Serialize)]
pub enum Format {
    /// JSON serialization (via the `serde_json` crate).
    ///
    /// Enabled with the `json_fmt` feature.
    #[cfg(feature = "json_fmt")]
    JSON,

    /// MsgPack serialization (via the `rmp-serde` crate).
    ///
    /// Enabled with the `msgpack_fmt` feature.
    #[cfg(feature = "msgpack_fmt")]
    MsgPack,

    /// MsgPack serialization (via the `toml` crate).
    ///
    /// Enabled with the `toml_fmt` feature.
    #[cfg(feature = "toml_fmt")]
    TOML,

    /// YAML serialization (via the `serde_yaml` crate)
    ///
    /// Enabled with the `yaml_fmt` feature.
    #[cfg(feature = "yaml_fmt")]
    YAML,
}

impl TryFrom<&str> for Format {
    type Error = Error;
    fn try_from(v: &str) -> crate::Result<Format> {
        #[cfg(feature = "json_fmt")]
        {
            if v == "json" {
                return Ok(Format::JSON);
            }
        }
        #[cfg(feature = "msgpack_fmt")]
        {
            if v == "msgpack" {
                return Ok(Format::MsgPack);
            }
        }

        #[cfg(feature = "toml_fmt")]
        {
            if v == "toml" {
                return Ok(Format::TOML);
            }
        }

        #[cfg(feature = "yaml_fmt")]
        {
            if v == "yaml" {
                return Ok(Format::YAML);
            }
        }

        Err(Error::new(format!("Unknown format {}", v)))
    }
}

impl AsRef<str> for Format {
    fn as_ref(&self) -> &str {
        #[cfg(feature = "json_fmt")]
        {
            if self == &Format::JSON {
                return "json";
            }
        }

        #[cfg(feature = "msgpack_fmt")]
        {
            if self == &Format::MsgPack {
                return "msgpack";
            }
        }

        #[cfg(feature = "toml_fmt")]
        {
            if self == &Format::TOML {
                return "toml";
            }
        }

        #[cfg(feature = "yaml_fmt")]
        {
            if self == &Format::YAML {
                return "yaml";
            }
        }

        "unknown"
    }
}