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

#[cfg(feature = "json")]
#[cfg_attr(docsrs, doc(cfg(feature = "json")))]
pub mod json {
	use crate::error::MessageError;
	use crate::message::{Message, Action, PacketBytes};
	use bytes::BytesRead;

	use serde::{Serialize, de::DeserializeOwned};


	pub fn encode<T, A, B>(value: T) -> Result<Message<A, B>, MessageError>
	where
		T: Serialize,
		A: Action,
		B: PacketBytes
	{
		let mut msg = Message::new();
		serde_json::to_writer(msg.body_mut(), &value)
			.map_err(MessageError::Json)?;

		Ok(msg)
	}

	pub fn decode<A, B, T>(msg: Message<A, B>) -> Result<T, MessageError>
	where
		A: Action,
		B: PacketBytes,
		T: DeserializeOwned
	{
		serde_json::from_slice(msg.body().as_slice())
			.map_err(MessageError::Json)
	}
}

#[cfg(feature = "protobuf")]
#[cfg_attr(docsrs, doc(cfg(feature = "protobuf")))]
pub mod protobuf {
	use crate::error::MessageError;
	use crate::message::{Message, Action, PacketBytes};
	use bytes::BytesRead;

	use fire_protobuf::encode::{EncodeMessage, MessageEncoder};
	use fire_protobuf::decode::DecodeMessage;


	pub fn encode<T, A, B>(mut value: T) -> Result<Message<A, B>, MessageError>
	where
		T: EncodeMessage,
		A: Action,
		B: PacketBytes
	{
		let mut msg = Message::new();
		let mut encoder = MessageEncoder::new(msg.body_mut());
		value.encode(None, &mut encoder)
			.map_err(MessageError::EncodeError)?;

		Ok(msg)
	}

	pub fn decode<A, B, T>(msg: Message<A, B>) -> Result<T, MessageError>
	where
		A: Action,
		B: PacketBytes,
		T: for<'a> DecodeMessage<'a>
	{
		T::parse_from_bytes(msg.body().as_slice())
			.map_err(MessageError::DecodeError)
	}
}