Skip to content

Commit cccdef7

Browse files
committed
f - add OnionMessageContentsType trait
1 parent 8d74148 commit cccdef7

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

lightning/src/onion_message/messenger.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ use crate::ln::features::{InitFeatures, NodeFeatures};
2323
use crate::ln::msgs::{self, OnionMessageHandler};
2424
use crate::ln::onion_utils;
2525
use crate::ln::peer_handler::IgnoringMessageHandler;
26-
pub use super::packet::{CustomOnionMessageContents, OnionMessageContents};
26+
pub use super::packet::{CustomOnionMessageContents, OnionMessageContents, OnionMessageContentsType};
2727
use super::offers::OffersMessageHandler;
2828
use super::packet::{BIG_PACKET_HOP_DATA_LEN, ForwardControlTlvs, Packet, Payload, ReceiveControlTlvs, SMALL_PACKET_HOP_DATA_LEN};
2929
use crate::util::logger::Logger;
@@ -161,7 +161,7 @@ where
161161
/// enqueued for sending.
162162
///
163163
/// [`OnionMessage`]: crate::ln::msgs::OnionMessage
164-
pub struct PendingOnionMessage<T> {
164+
pub struct PendingOnionMessage<T: OnionMessageContentsType> {
165165
/// The message contents to send in an [`OnionMessage`].
166166
///
167167
/// [`OnionMessage`]: crate::ln::msgs::OnionMessage

lightning/src/onion_message/packet.rs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,13 @@ pub enum OnionMessageContents<T: CustomOnionMessageContents> {
118118
Custom(T),
119119
}
120120

121+
/// A type used within a variant of [`OnionMessageContents`].
122+
pub trait OnionMessageContentsType {}
123+
124+
impl OnionMessageContentsType for OffersMessage {}
125+
126+
impl<T: CustomOnionMessageContents> OnionMessageContentsType for T {}
127+
121128
impl<T: CustomOnionMessageContents> OnionMessageContents<T> {
122129
/// Returns the type that was used to decode the message payload.
123130
///

0 commit comments

Comments
 (0)