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
102
103
104
105
106
107
108
109
110
111

use serde::Serialize;

use crate::{Error, Format};

/// Serialize a struct in the given format, outputting a vector of bytes.
///
/// Obviously, only formats enabled with feature flags will be supported.
/// ## Example
/// ```
/// use polyglot::{ser, Format};
/// use serde::Serialize;
///
/// #[derive(Serialize)]
/// pub struct Person {
///     pub name: String
/// }
///
/// let p = Person{name: String::from("John")};
/// let out_bytes = ser::to_vec(&p, Format::TOML).unwrap();
/// ```
pub fn to_vec<T: Serialize>(val: &T, format: Format) -> crate::Result<Vec<u8>> {
    #[cfg(feature = "json_fmt")]
    {
        if format == Format::JSON {
            return Ok(serde_json::to_vec(val)?);
        }
    }

    #[cfg(feature = "msgpack_fmt")]
    {
        if format == Format::MsgPack {
            return Ok(rmp_serde::to_vec(val)?);
        }
    }

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

    Err(Error::new("Library Error - Unimplemented Serialize"))
}

/// Serialize a struct in the given format, outputting a string.
///
/// Obviously, only formats enabled with feature flags will be supported.
///
/// # Errors
/// Serializing to a string using msgpack will return an error due to MessagePack being a binary
/// format.
///
/// # Examples
/// ```
/// use polyglot::{ser, Format};
/// use serde::Serialize;
///
/// #[derive(Serialize)]
/// pub struct Person {
///     pub name: String
/// }
///
/// let p = Person{name: String::from("John")};
/// let out_str = ser::to_string(&p, Format::JSON).unwrap();
/// assert_eq!(&out_str, "{\"name\":\"John\"}");
/// ```
pub fn to_string<T: Serialize>(val: &T, format: Format) -> crate::Result<String> {
    #[cfg(feature = "json_fmt")]
    {
        if format == Format::JSON {
            return Ok(serde_json::to_string(val)?);
        }
    }

    #[cfg(feature = "msgpack_fmt")]
    {
        if format == Format::MsgPack {
            return Err(Error::new(
                "MessagePack is a binary format. Serializing to string is unsupported.",
            ));
        }
    }

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

    Err(Error::new("Library Error - Unimplemented Serialize"))
}

pub fn to_writer<T: Serialize, V: std::io::Write>(
    mut w: V,
    val: &T,
    format: Format,
) -> crate::Result<()> {
    #[cfg(feature = "json_fmt")]
    {
        if format == Format::JSON {
            serde_json::to_writer(w, val)?;
            return Ok(());
        }
    }

    let vec_data = crate::to_vec(val, format)?;
    w.write_all(&vec_data)?;
    Ok(())
}