Add endpoint for marking btc tx as tainted.

chainflip-io · Oct 28, 2024 · a3835f1 · a3835f1
1 parent 6cbcde1
commit a3835f1
Show file tree

Hide file tree

Showing 4 changed files with 35 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/api/bin/chainflip-broker-api/Cargo.toml b/api/bin/chainflip-broker-api/Cargo.toml
@@ -23,6 +23,7 @@ workspace = true
 
 [dependencies]
 chainflip-api = { workspace = true }
+cf-chains = { workspace = true, default-features = true }
 cf-utilities = { workspace = true, default-features = true }
 custom-rpc = { workspace = true }
 

diff --git a/api/bin/chainflip-broker-api/src/main.rs b/api/bin/chainflip-broker-api/src/main.rs
@@ -8,8 +8,8 @@ use chainflip_api::{
 	self,
 	primitives::{AccountRole, Affiliates, Asset, BasisPoints, CcmChannelMetadata, DcaParameters},
 	settings::StateChain,
-	AccountId32, AddressString, BrokerApi, OperatorApi, RefundParameters, StateChainApi,
-	SwapDepositAddress, SwapPayload, WithdrawFeesDetail,
+	AccountId32, AddressString, BrokerApi, DepositMonitorApi, OperatorApi, RefundParameters,
+	StateChainApi, SwapDepositAddress, SwapPayload, WithdrawFeesDetail,
 };
 use clap::Parser;
 use futures::FutureExt;
@@ -100,6 +100,9 @@ pub trait Rpc {
 		affiliate_fees: Option<Affiliates<AccountId32>>,
 		dca_parameters: Option<DcaParameters>,
 	) -> RpcResult<SwapPayload>;
+
+	#[method(name = "mark_btc_transaction_as_tainted", aliases = ["broker_markBtcTransactionAsTainted"])]
+	async fn mark_btc_transaction_as_tainted(&self, tx_id: cf_chains::btc::Hash) -> RpcResult<()>;
 }
 
 pub struct RpcServerImpl {
@@ -194,6 +197,14 @@ impl RpcServer for RpcServerImpl {
 			)
 			.await?)
 	}
+
+	async fn mark_btc_transaction_as_tainted(&self, tx_id: cf_chains::btc::Hash) -> RpcResult<()> {
+		self.api
+			.deposit_monitor_api()
+			.mark_btc_transaction_as_tainted(tx_id)
+			.await
+			.map_err(BrokerApiError::Other)
+	}
 }
 
 #[derive(Parser, Debug, Clone, Default)]

diff --git a/api/lib/src/lib.rs b/api/lib/src/lib.rs
@@ -158,6 +158,10 @@ impl StateChainApi {
 		self.state_chain_client.clone()
 	}
 
+	pub fn deposit_monitor_api(&self) -> Arc<impl DepositMonitorApi> {
+		self.state_chain_client.clone()
+	}
+
 	pub fn query_api(&self) -> queries::QueryApi {
 		queries::QueryApi { state_chain_client: self.state_chain_client.clone() }
 	}
@@ -175,6 +179,8 @@ impl BrokerApi for StateChainClient {
 impl OperatorApi for StateChainClient {}
 #[async_trait]
 impl ValidatorApi for StateChainClient {}
+#[async_trait]
+impl DepositMonitorApi for StateChainClient {}
 
 #[async_trait]
 pub trait ValidatorApi: SimpleSubmissionApi {
@@ -628,6 +634,20 @@ pub fn clean_foreign_chain_address(chain: ForeignChain, address: &str) -> Result
 	})
 }
 
+#[async_trait]
+pub trait DepositMonitorApi:
+	SignedExtrinsicApi + StorageApi + Sized + Send + Sync + 'static
+{
+	async fn mark_btc_transaction_as_tainted(&self, tx_id: cf_chains::btc::Hash) -> Result<()> {
+		let _ = self
+			.submit_signed_extrinsic(state_chain_runtime::RuntimeCall::BitcoinIngressEgress(
+				pallet_cf_ingress_egress::Call::mark_transaction_as_tainted { tx_id },
+			))
+			.await;
+		Ok(())
+	}
+}
+
 #[derive(Debug, Zeroize, PartialEq, Eq)]
 /// Public and Secret keys as bytes
 pub struct KeyPair {