Add methods to set invoice features in Features objects.

Author: valentinewallace

lightning/src/ln/features.rs

@@ -508,6 +508,16 @@ impl<T: sealed::VariableLengthOnion> Features<T> {
 	pub(crate) fn supports_variable_length_onion(&self) -> bool {
 		<T as sealed::VariableLengthOnion>::supports_feature(&self.flags)
 	}
+
+	/// Set these features as requiring the var_onion_optin feature.
+	pub fn set_supports_var_onion_optin(&mut self) {
+		<T as sealed::VariableLengthOnion>::set_optional_bit(&mut self.flags);
+	}
+
+	/// Set these features as supporting the var_onion_optin feature.
+	pub fn set_requires_var_onion_optin(&mut self) {
+		<T as sealed::VariableLengthOnion>::set_required_bit(&mut self.flags);
+	}
 }
 
 impl<T: sealed::StaticRemoteKey> Features<T> {
@@ -545,6 +555,16 @@ impl<T: sealed::PaymentSecret> Features<T> {
 	pub(crate) fn supports_payment_secret(&self) -> bool {
 		<T as sealed::PaymentSecret>::supports_feature(&self.flags)
 	}
+
+	/// Set these features as supporting the payment_secret feature.
+	pub fn set_supports_payment_secret(&mut self) {
+		<T as sealed::PaymentSecret>::set_optional_bit(&mut self.flags);
+	}
+
+	/// Set these features as requiring the payment_secret feature.
+	pub fn set_requires_payment_secret(&mut self) {
+		<T as sealed::PaymentSecret>::set_required_bit(&mut self.flags);
+	}
 }
 
 impl<T: sealed::BasicMPP> Features<T> {
@@ -557,6 +577,16 @@ impl<T: sealed::BasicMPP> Features<T> {
 	pub(crate) fn supports_basic_mpp(&self) -> bool {
 		<T as sealed::BasicMPP>::supports_feature(&self.flags)
 	}
+
+	/// Set these features as supporting the basic_mpp feature.
+	pub fn set_supports_basic_mpp(&mut self) {
+		<T as sealed::BasicMPP>::set_optional_bit(&mut self.flags);
+	}
+
+	/// Set these features as requiring the basic_mpp feature.
+	pub fn set_requires_basic_mpp(&mut self) {
+		<T as sealed::BasicMPP>::set_required_bit(&mut self.flags);
+	}
 }
 
 impl<T: sealed::ShutdownAnySegwit> Features<T> {