parsers: Add with_payload/with_payloads helpers on Message

Maxime “pep” Buquet created 2 years ago

Signed-off-by: Maxime “pep” Buquet <[e-mail]>

Change summary

parsers/src/message.rs | 12 ++++++++++++
1 file changed, 12 insertions(+)

Detailed changes

parsers/src/message.rs

@@ -154,6 +154,18 @@ impl Message {
         self
     }
 
+    /// Set a payload inside this message.
+    pub fn with_payload<P: MessagePayload>(mut self, payload: P) -> Message {
+        self.payloads.push(payload.into());
+        self
+    }
+
+    /// Set the payloads of this message.
+    pub fn with_payloads(mut self, payloads: Vec<Element>) -> Message {
+        self.payloads = payloads;
+        self
+    }
+
     fn get_best<'a, T>(
         map: &'a BTreeMap<Lang, T>,
         preferred_langs: Vec<&str>,