++

dfinity · Sep 9, 2024 · a4e0a53 · a4e0a53
1 parent 9519f63
commit a4e0a53
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/api/src/lib.rs b/src/api/src/lib.rs
@@ -5,6 +5,7 @@ use cycles_ledger_client::WithdrawFromError;
 #[derive(Debug, CandidType, Deserialize, Clone, Eq, PartialEq)]
 #[non_exhaustive]
 pub enum PaymentError {
+    UnsupportedPaymentType,
     LedgerUnreachable {
         ledger: Principal,
     },
@@ -18,6 +19,7 @@ pub enum PaymentError {
     },
 }
 
+#[derive(Debug, CandidType, Deserialize, Clone, Eq, PartialEq)]
 #[non_exhaustive]
 pub enum PaymentType {
     AttachedCycles,

diff --git a/src/example/paid_service/src/lib.rs b/src/example/paid_service/src/lib.rs
@@ -46,11 +46,12 @@ async fn cost_1b(payment: PaymentType) -> Result<String, PaymentError> {
         PaymentType::AttachedCycles => {
             AttachedCyclesPayment::default().deduct(fee).await?;
         }
-        PaymentType::Icrc2 => {
+        PaymentType::Icrc2Cycles(_payer) => {
             let mut guard = Icrc2CyclesPaymentGuard::new(own_canister_id());
             guard.ledger_canister_id = payment_ledger();
             guard.deduct(fee).await?;
         }
+        _ => return Err(PaymentError::UnsupportedPaymentType),
     };
     Ok("Yes, you paid 1 billion cycles!".to_string())
 }