Loans: multiple cashflows

Cargo.toml

@@ -82,6 +82,7 @@ rev_slice = { version = "0.1.5", default-features = false }
 impl-trait-for-tuples = "0.2.1"
 num-traits = { version = "0.2", default-features = false }
 num_enum = { version = "0.5.3", default-features = false }
+chrono = { version = "0.4", default-features = false }
 
 # Cumulus
 cumulus-pallet-aura-ext = { git = "https://github.com/paritytech/polkadot-sdk", default-features = false, branch = "release-polkadot-v1.1.0" }

pallets/loans/Cargo.toml

@@ -13,6 +13,7 @@ documentation.workspace = true
 targets = ["x86_64-unknown-linux-gnu"]
 
 [dependencies]
+chrono = { workspace = true }
 parity-scale-codec = { workspace = true }
 scale-info = { workspace = true }
 

pallets/loans/docs/types.md

@@ -4,28 +4,31 @@
 set namespaceSeparator ::
 hide methods
 
-enum Maturity {
-    Fixed::date: Seconds
-    Fixed::extension: Seconds
-}
+package cashflow {
+    enum Maturity {
+        Fixed::date: Seconds
+        Fixed::extension: Seconds
+    }
 
-enum InterestPayments {
-    None
-}
+    enum InterestPayments {
+        None
+        Monthly::reference_day: u8
+    }
 
-enum PayDownSchedule {
-    None
-}
+    enum PayDownSchedule {
+        None
+    }
 
-class RepaymentSchedule {
-    maturity: Maturity
-    interest_payments: InterestPayments
-    pay_down_schedule: PayDownSchedule
-}
+    class RepaymentSchedule {
+        maturity: Maturity
+        interest_payments: InterestPayments
+        pay_down_schedule: PayDownSchedule
+    }
 
-RepaymentSchedule *--> Maturity
-RepaymentSchedule *---> PayDownSchedule
-RepaymentSchedule *----> InterestPayments
+    RepaymentSchedule *--> Maturity
+    RepaymentSchedule *--> PayDownSchedule
+    RepaymentSchedule *--> InterestPayments
+}
 
 enum BorrowRestrictions {
     NoWrittenOff

pallets/loans/src/benchmarking.rs

@@ -16,7 +16,7 @@ use cfg_traits::{
 	benchmarking::FundedPoolBenchmarkHelper,
 	changes::ChangeGuard,
 	interest::{CompoundingSchedule, InterestAccrual, InterestRate},
-	Permissions, PoolWriteOffPolicyMutate, Seconds, TimeAsSecs, ValueProvider,
+	Permissions, PoolWriteOffPolicyMutate, TimeAsSecs, ValueProvider,
 };
 use cfg_types::{
 	adjustments::Adjustment,
@@ -40,13 +40,12 @@ use crate::{
 	},
 	pallet::*,
 	types::{
+		cashflow::{InterestPayments, Maturity, PayDownSchedule, RepaymentSchedule},
 		valuation::{DiscountedCashFlow, ValuationMethod},
-		BorrowRestrictions, InterestPayments, LoanRestrictions, Maturity, PayDownSchedule,
-		RepayRestrictions, RepaymentSchedule,
+		BorrowRestrictions, LoanRestrictions, RepayRestrictions,
 	},
 };
 
-const OFFSET: Seconds = 120;
 const COLLECION_ID: u16 = 42;
 const COLLATERAL_VALUE: u128 = 1_000_000;
 const FUNDS: u128 = 1_000_000_000;
@@ -124,10 +123,12 @@ where
 	}
 
 	fn base_loan(item_id: T::ItemId) -> LoanInfo<T> {
+		let maturity_offset = 40 * 365 * 24 * 3600; // 40 years
+
 		LoanInfo {
 			schedule: RepaymentSchedule {
-				maturity: Maturity::fixed(T::Time::now() + OFFSET),
-				interest_payments: InterestPayments::None,
+				maturity: Maturity::fixed(T::Time::now() + maturity_offset),
+				interest_payments: InterestPayments::Monthly(1),
 				pay_down_schedule: PayDownSchedule::None,
 			},
 			collateral: (COLLECION_ID.into(), item_id),
@@ -144,7 +145,7 @@ where
 					probability_of_default: T::Rate::zero(),
 					loss_given_default: T::Rate::zero(),
 					discount_rate: InterestRate::Fixed {
-						rate_per_year: T::Rate::one(),
+						rate_per_year: T::Rate::saturating_from_rational(1, 5000),
 						compounding: CompoundingSchedule::Secondly,
 					},
 				}),

pallets/loans/src/entities/changes.rs

@@ -7,8 +7,9 @@ use crate::{
 	entities::input::{PrincipalInput, RepaidInput},
 	pallet::Config,
 	types::{
-		policy::WriteOffRule, valuation::ValuationMethod, InterestPayments, Maturity,
-		PayDownSchedule,
+		cashflow::{InterestPayments, Maturity, PayDownSchedule},
+		policy::WriteOffRule,
+		valuation::ValuationMethod,
 	},
 };
 

pallets/loans/src/entities/loans.rs

@@ -14,7 +14,7 @@
 	},
 	DispatchError,
 };
-use sp_std::collections::btree_map::BTreeMap;
+use sp_std::{collections::btree_map::BTreeMap, vec::Vec};
 
 use crate::{
 	entities::{
@@ -27,9 +27,10 @@
 	},
 	pallet::{AssetOf, Config, Error},
 	types::{
+		cashflow::{CashflowPayment, RepaymentSchedule},
 		policy::{WriteOffStatus, WriteOffTrigger},
 		BorrowLoanError, BorrowRestrictions, CloseLoanError, CreateLoanError, LoanRestrictions,
-		MutationError, RepaidAmount, RepayLoanError, RepayRestrictions, RepaymentSchedule,
+		MutationError, RepaidAmount, RepayLoanError, RepayRestrictions,
 	},
 };
 
@@ -70,7 +71,7 @@
 		T::InterestAccrual::validate_rate(&self.interest_rate)?;
 
 		ensure!(
-			self.schedule.is_valid(now),
+			self.schedule.is_valid(now)?,
 			Error::<T>::from(CreateLoanError::InvalidRepaymentSchedule)
 		);
 
@@ -222,6 +223,10 @@
 		&self.borrower
 	}
 
+	pub fn origination_date(&self) -> Seconds {
+		self.origination_date
+	}
+
 	pub fn maturity_date(&self) -> Seconds {
 		self.schedule.maturity.date()
 	}
@@ -237,13 +242,24 @@
 		}
 	}
 
+	pub fn principal(&self) -> Result<T::Balance, DispatchError> {
+		Ok(self
+			.total_borrowed
+			.ensure_sub(self.total_repaid.principal)?)
+	}
+
+	pub fn expected_cashflows(&self) -> Result<Vec<CashflowPayment<T::Balance>>, DispatchError> {
+		self.schedule.generate_cashflows(
+			self.repayments_on_schedule_until,
+			self.principal()?,
+			self.pricing.interest().rate(),
+		)
+	}
+
 	pub fn write_off_status(&self) -> WriteOffStatus<T::Rate> {
 		WriteOffStatus {
 			percentage: self.write_off_percentage,
-			penalty: match &self.pricing {
-				ActivePricing::Internal(inner) => inner.interest.penalty(),
-				ActivePricing::External(inner) => inner.interest.penalty(),
-			},
+			penalty: self.pricing.interest().penalty(),
 		}
 	}
 
@@ -345,6 +361,18 @@
 			Error::<T>::from(BorrowLoanError::MaturityDatePassed)
 		);
 
+		let expected_payment = self.schedule.expected_payment(
+			self.repayments_on_schedule_until,
+			self.principal()?,
+			self.pricing.interest().rate(),
+			now,
+		)?;
+
+		ensure!(
+			self.total_repaid.effective()? >= expected_payment,
+			Error::<T>::from(BorrowLoanError::PaymentOverdue)
+		);
+
 		Ok(())
 	}
 
@@ -362,6 +390,8 @@
 			}
 		}
 
+		self.repayments_on_schedule_until = T::Time::now();
+
 		Ok(())
 	}
 
@@ -378,11 +408,8 @@
 	) -> Result<RepaidInput<T>, DispatchError> {
 		let (max_repay_principal, outstanding_interest) = match &self.pricing {
 			ActivePricing::Internal(inner) => {
-				amount.principal.internal()?;
-
-				let principal = self
-					.total_borrowed
-					.ensure_sub(self.total_repaid.principal)?;
+				let _ = amount.principal.internal()?;
+				let principal = self.principal()?;
 
 				(principal, inner.outstanding_interest(principal)?)
 			}
@@ -440,24 +467,20 @@
 	}
 
 	pub fn write_off(&mut self, new_status: &WriteOffStatus<T::Rate>) -> DispatchResult {
-		let penalty = new_status.penalty;
-		match &mut self.pricing {
-			ActivePricing::Internal(inner) => inner.interest.set_penalty(penalty)?,
-			ActivePricing::External(inner) => inner.interest.set_penalty(penalty)?,
-		}
+		self.pricing
+			.interest_mut()
+			.set_penalty(new_status.penalty)?;
 
 		self.write_off_percentage = new_status.percentage;
 
 		Ok(())
 	}
 
 	fn ensure_can_close(&self) -> DispatchResult {
-		let can_close = match &self.pricing {
-			ActivePricing::Internal(inner) => !inner.interest.has_debt(),
-			ActivePricing::External(inner) => !inner.interest.has_debt(),
-		};
-
-		ensure!(can_close, Error::<T>::from(CloseLoanError::NotFullyRepaid));
+		ensure!(
+			!self.pricing.interest().has_debt(),
+			Error::<T>::from(CloseLoanError::NotFullyRepaid)
+		);
 
 		Ok(())
 	}
@@ -500,10 +523,7 @@
 				.maturity
 				.extends(extension)
 				.map_err(|_| Error::<T>::from(MutationError::MaturityExtendedTooMuch))?,
-			LoanMutation::InterestRate(rate) => match &mut self.pricing {
-				ActivePricing::Internal(inner) => inner.interest.set_base_rate(rate)?,
-				ActivePricing::External(inner) => inner.interest.set_base_rate(rate)?,
-			},
+			LoanMutation::InterestRate(rate) => self.pricing.interest_mut().set_base_rate(rate)?,
 			LoanMutation::InterestPayments(payments) => self.schedule.interest_payments = payments,
 			LoanMutation::PayDownSchedule(schedule) => self.schedule.pay_down_schedule = schedule,
 			LoanMutation::Internal(mutation) => match &mut self.pricing {
@@ -519,7 +539,7 @@
 
 	#[cfg(feature = "runtime-benchmarks")]
 	pub fn set_maturity(&mut self, duration: Seconds) {
-		self.schedule.maturity = crate::types::Maturity::fixed(duration);
+		self.schedule.maturity = crate::types::cashflow::Maturity::fixed(duration);
 	}
 }
 
@@ -555,9 +575,7 @@
 
 		Ok(match &active_loan.pricing {
 			ActivePricing::Internal(inner) => {
-				let principal = active_loan
-					.total_borrowed
-					.ensure_sub(active_loan.total_repaid.principal)?;
+				let principal = active_loan.principal()?;
 
 				Self {
 					present_value,

pallets/loans/src/entities/pricing.rs

@@ -2,7 +2,7 @@ use frame_support::RuntimeDebugNoBound;
 use parity_scale_codec::{Decode, Encode, MaxEncodedLen};
 use scale_info::TypeInfo;
 
-use crate::pallet::Config;
+use crate::{entities::interest::ActiveInterestRate, pallet::Config};
 
 pub mod external;
 pub mod internal;
@@ -28,3 +28,19 @@ pub enum ActivePricing<T: Config> {
 	/// Internal attributes
 	External(external::ExternalActivePricing<T>),
 }
+
+impl<T: Config> ActivePricing<T> {
+	pub fn interest(&self) -> &ActiveInterestRate<T> {
+		match self {
+			Self::Internal(inner) => &inner.interest,
+			Self::External(inner) => &inner.interest,
+		}
+	}
+
+	pub fn interest_mut(&mut self) -> &mut ActiveInterestRate<T> {
+		match self {
+			Self::Internal(inner) => &mut inner.interest,
+			Self::External(inner) => &mut inner.interest,
+		}
+	}
+}

pallets/loans/src/lib.rs

@@ -107,6 +107,7 @@
 	use sp_std::{collections::btree_map::BTreeMap, vec, vec::Vec};
 	use types::{
 		self,
+		cashflow::CashflowPayment,
 		policy::{self, WriteOffRule, WriteOffStatus},
 		BorrowLoanError, CloseLoanError, CreateLoanError, MutationError, RepayLoanError,
 		WrittenOffError,
@@ -1199,7 +1200,7 @@
 		) -> Result<PortfolioInfoOf<T>, DispatchError> {
 			ActiveLoans::<T>::get(pool_id)
 				.into_iter()
-				.map(|(loan_id, loan)| Ok((loan_id, (pool_id, loan).try_into()?)))
+				.map(|(loan_id, loan)| Ok((loan_id, ActiveLoanInfo::try_from((pool_id, loan))?)))
 				.collect()
 		}
 
@@ -1210,9 +1211,20 @@
 			ActiveLoans::<T>::get(pool_id)
 				.into_iter()
 				.find(|(id, _)| *id == loan_id)
-				.map(|(_, loan)| (pool_id, loan).try_into())
+				.map(|(_, loan)| ActiveLoanInfo::try_from((pool_id, loan)))
 				.transpose()
 		}
+
+		pub fn expected_cashflows(
+			pool_id: T::PoolId,
+			loan_id: T::LoanId,
+		) -> Result<Vec<CashflowPayment<T::Balance>>, DispatchError> {
+			ActiveLoans::<T>::get(pool_id)
+				.into_iter()
+				.find(|(id, _)| *id == loan_id)
+				.map(|(_, loan)| loan.expected_cashflows())
+				.ok_or(Error::<T>::LoanNotActiveOrNotFound)?
+		}
 	}
 
 	// TODO: This implementation can be cleaned once #908 be solved