Move compute_output_value as part of package member functions

Author: Antoine Riard

lightning/src/chain/onchain_utils.rs

@@ -521,9 +521,6 @@ impl PackageTemplate {
 	pub(crate) fn aggregable(&self) -> bool {
 		self.aggregable
 	}
-	pub(crate) fn feerate(&self) -> u64 {
-		self.feerate_previous
-	}
 	pub(crate) fn set_feerate(&mut self, new_feerate: u64) {
 		self.feerate_previous = new_feerate;
 	}
@@ -662,6 +659,27 @@ impl PackageTemplate {
 			},
 		}
 	}
+	pub(crate) fn compute_package_output<F: Deref, L: Deref>(&self, predicted_weight: usize, input_amounts: u64, fee_estimator: &F, logger: &L) -> Option<(u64, u64)>
+		where F::Target: FeeEstimator,
+		      L::Target: Logger,
+	{
+		// If old feerate is 0, first iteration of this claim, use normal fee calculation
+		if self.feerate_previous != 0 {
+			if let Some((new_fee, feerate)) = feerate_bump(predicted_weight, input_amounts, self.feerate_previous, fee_estimator, logger) {
+				// If new computed fee is superior at the whole claimable amount burn all in fees
+				if new_fee > input_amounts {
+					return Some((0, feerate));
+				} else {
+					return Some((input_amounts - new_fee, feerate));
+				}
+			}
+		} else {
+			if let Some((new_fee, feerate)) = subtract_high_prio_fee(input_amounts, predicted_weight, fee_estimator, logger) {
+					return Some((input_amounts - new_fee, feerate));
+			}
+		}
+		None
+	}
 	pub (crate) fn build_package(txid: Txid, vout: u32, input_solving_data: PackageSolvingData, absolute_timelock: u32, aggregable: bool, height_original: u32) -> Self {
 		let malleability = match input_solving_data {
 			PackageSolvingData::RevokedOutput(..) => { PackageMalleability::Malleable },
@@ -813,31 +831,6 @@ fn feerate_bump<F: Deref, L: Deref>(predicted_weight: usize, input_amounts: u64,
 	Some((new_fee, new_fee * 1000 / (predicted_weight as u64)))
 }
 
-/// Deduce a new proposed fee from the claiming transaction output value.
-/// If the new proposed fee is superior to the consumed outpoint's value, burn everything in miner's
-/// fee to deter counterparties attacker.
-pub(crate) fn compute_output_value<F: Deref, L: Deref>(predicted_weight: usize, input_amounts: u64, previous_feerate: u64, fee_estimator: &F, logger: &L) -> Option<(u64, u64)>
-	where F::Target: FeeEstimator,
-	      L::Target: Logger,
-{
-	// If old feerate is 0, first iteration of this claim, use normal fee calculation
-	if previous_feerate != 0 {
-		if let Some((new_fee, feerate)) = feerate_bump(predicted_weight, input_amounts, previous_feerate, fee_estimator, logger) {
-			// If new computed fee is superior at the whole claimable amount burn all in fees
-			if new_fee > input_amounts {
-				return Some((0, feerate));
-			} else {
-				return Some((input_amounts - new_fee, feerate));
-			}
-		}
-	} else {
-		if let Some((new_fee, feerate)) = subtract_high_prio_fee(input_amounts, predicted_weight, fee_estimator, logger) {
-				return Some((input_amounts - new_fee, feerate));
-		}
-	}
-	None
-}
-
 /// In LN, output claimed are time-sensitive, which means we have to spend them before reaching some timelock expiration. At in-channel
 /// output detection, we generate a first version of a claim tx and associate to it a height timer. A height timer is an absolute block
 /// height than once reached we should generate a new bumped "version" of the claim tx to be sure than we safely claim outputs before

lightning/src/chain/onchaintx.rs

@@ -337,7 +337,7 @@ impl<ChannelSigner: Sign> OnchainTxHandler<ChannelSigner> {
 		let amt = cached_request.package_amount();
 		if cached_request.is_malleable() {
 			let predicted_weight = cached_request.package_weight(&self.destination_script);
-			if let Some((output_value, new_feerate)) = onchain_utils::compute_output_value(predicted_weight, amt, cached_request.feerate(), fee_estimator, logger) {
+			if let Some((output_value, new_feerate)) = cached_request.compute_package_output(predicted_weight, amt, fee_estimator, logger) {
 				assert!(new_feerate != 0);
 
 				let transaction = cached_request.finalize_package(self, output_value, self.destination_script.clone(), logger).unwrap();