f - implement CustomMessageHandler in macro lol

Author: jkczyz

lightning/src/ln/peer_handler.rs

@@ -85,6 +85,35 @@ macro_rules! composite_custom_message_handler {
 			)*
 		}
 
+		impl CustomMessageHandler for $handler {
+			fn handle_custom_message(
+				&self, msg: Self::CustomMessage, sender_node_id: &PublicKey
+			) -> Result<(), LightningError> {
+				match msg {
+					$(
+						$message::$variant(message) => {
+							$crate::ln::peer_handler::CustomMessageHandler::handle_custom_message(
+								&self.$field, message, sender_node_id
+							)
+						},
+					)*
+				}
+			}
+
+			fn get_and_clear_pending_msg(&self) -> Vec<(PublicKey, Self::CustomMessage)> {
+				vec![].into_iter()
+					$(
+						.chain(
+							self.$field
+								.get_and_clear_pending_msg()
+								.into_iter()
+								.map(|(pubkey, message)| (pubkey, $message::$variant(message)))
+						)
+					)*
+					.collect()
+			}
+		}
+
 		impl wire::CustomMessageReader for $handler {
 			type CustomMessage = $message;
 			fn read<R: io::Read>(