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

use common_primitives::{
	messages::MessageResponse, msa::MessageSourceId, node::BlockNumber, schema::PayloadLocation,
};
use frame_support::{traits::Get, BoundedVec};
use multibase::Base;
use parity_scale_codec::{Decode, Encode, MaxEncodedLen};
use scale_info::TypeInfo;
use sp_std::{fmt::Debug, prelude::*};

/// Payloads stored offchain contain a tuple of (bytes(the payload reference), payload length).
pub type OffchainPayloadType = (Vec<u8>, u32);
/// Index of message in the block
pub type MessageIndex = u16;

/// A single message type definition.
#[derive(Default, Encode, Decode, PartialEq, Debug, TypeInfo, Eq, MaxEncodedLen)]
#[scale_info(skip_type_params(MaxDataSize))]
#[codec(mel_bound(MaxDataSize: MaxEncodedLen))]
pub struct Message<MaxDataSize>
where
	MaxDataSize: Get<u32> + Debug,
{
	///  Data structured by the associated schema's model.
	pub payload: BoundedVec<u8, MaxDataSize>,
	/// Message source account id of the Provider. This may be the same id as contained in `msa_id`,
	/// indicating that the original source MSA is acting as its own provider. An id differing from that
	/// of `msa_id` indicates that `provider_msa_id` was delegated by `msa_id` to send this message on
	/// its behalf.
	pub provider_msa_id: MessageSourceId,
	///  Message source account id (the original source).
	pub msa_id: Option<MessageSourceId>,
}

impl<MaxDataSize> Message<MaxDataSize>
where
	MaxDataSize: Get<u32> + Debug,
{
	/// Helper function to handle response type [`MessageResponse`] depending on the Payload Location (on chain or IPFS)
	pub fn map_to_response(
		&self,
		block_number: BlockNumber,
		payload_location: PayloadLocation,
		index: u16,
	) -> MessageResponse {
		match payload_location {
			PayloadLocation::OnChain => MessageResponse {
				provider_msa_id: self.provider_msa_id,
				index,
				block_number,
				msa_id: self.msa_id,
				payload: Some(self.payload.to_vec()),
				cid: None,
				payload_length: None,
			},
			PayloadLocation::IPFS => {
				let (binary_cid, payload_length) =
					OffchainPayloadType::decode(&mut &self.payload[..]).unwrap_or_default();
				MessageResponse {
					provider_msa_id: self.provider_msa_id,
					index,
					block_number,
					cid: Some(multibase::encode(Base::Base32Lower, binary_cid).as_bytes().to_vec()),
					payload_length: Some(payload_length),
					msa_id: None,
					payload: None,
				}
			}, // Message types of Itemized and Paginated are retrieved differently
			_ => MessageResponse {
				provider_msa_id: self.provider_msa_id,
				index,
				block_number,
				msa_id: None,
				payload: None,
				cid: None,
				payload_length: None,
			},
		}
	}
}