Add MessageContext for async payments.

Author: valentinewallace

lightning/src/blinded_path/message.rs

@@ -98,6 +98,10 @@ pub enum MessageContext {
 	///
 	/// [`OffersMessage`]: crate::onion_message::offers::OffersMessage
 	Offers(OffersContext),
+	/// Represents the data specific to [`AsyncPaymentsMessage`]s.
+	///
+	/// [`AsyncPaymentsMessage`]: crate::onion_message::async_payments::AsyncPaymentsMessage
+	AsyncPayments(AsyncPaymentsContext),
 	/// Represents custom data received in a Custom Onion Message.
 	Custom(Vec<u8>),
 }
@@ -119,9 +123,22 @@ pub enum OffersContext {
 	},
 }
 
+/// Contains data specific to [`AsyncPaymentsMessage`]s.
+///
+/// [`AsyncPaymentsMessage`]: crate::onion_message::async_payments::AsyncPaymentsMessage
+#[derive(Clone, Debug)]
+pub enum AsyncPaymentsContext {
+	/// Represents an outbound async payment context.
+	OutboundPayment {
+		/// Payment id of the outbound payment.
+		payment_id: PaymentId
+	},
+}
+
 impl_writeable_tlv_based_enum!(MessageContext, ;
 	(0, Offers),
 	(1, Custom),
+	(2, AsyncPayments),
 );
 
 impl_writeable_tlv_based_enum!(OffersContext,
@@ -131,6 +148,12 @@ impl_writeable_tlv_based_enum!(OffersContext,
 	},
 ;);
 
+impl_writeable_tlv_based_enum!(AsyncPaymentsContext,
+	(0, OutboundPayment) => {
+		(0, payment_id, required),
+	},
+;);
+
 /// Construct blinded onion message hops for the given `intermediate_nodes` and `recipient_node_id`.
 pub(super) fn blinded_hops<T: secp256k1::Signing + secp256k1::Verification>(
 	secp_ctx: &Secp256k1<T>, intermediate_nodes: &[ForwardNode], recipient_node_id: PublicKey,

lightning/src/onion_message/messenger.rs

@@ -1463,7 +1463,7 @@ where
 						let context = match context {
 							None => OffersContext::Unknown {},
 							Some(MessageContext::Offers(context)) => context,
-							Some(MessageContext::Custom(_)) => {
+							Some(MessageContext::Custom(_)) | Some(MessageContext::AsyncPayments(_)) => {
 								debug_assert!(false, "Shouldn't have triggered this case.");
 								return
 							}
@@ -1486,7 +1486,7 @@ where
 						let context = match context {
 							None => None,
 							Some(MessageContext::Custom(data)) => Some(data),
-							Some(MessageContext::Offers(_)) => {
+							Some(MessageContext::Offers(_)) | Some(MessageContext::AsyncPayments(_)) => {
 								debug_assert!(false, "Shouldn't have triggered this case.");
 								return
 							}