feat: P-584 added bitcoin holding amount VC (#2689)

* feat: P-584 added bitcoin holding amount vc using new dataprovider https://blockchain.info * fix clippy * fix: call multiaddr BlockChainInfo API to query balance for BTC, add new ts-tests for it with bumped vc-jsonschema version --------- Co-authored-by: higherordertech <higherordertech>
litentry · May 2, 2024 · 7208568 · 7208568
1 parent 5863796
commit 7208568
Show file tree

Hide file tree

Showing 18 changed files with 453 additions and 6 deletions.
diff --git a/bitacross-worker/docker/docker-compose.yml b/bitacross-worker/docker/docker-compose.yml
@@ -140,6 +140,7 @@ services:
       - CONTEST_LEGEND_DISCORD_ROLE_ID=CONTEST_LEGEND_DISCORD_ROLE_ID
       - CONTEST_POPULARITY_DISCORD_ROLE_ID=CONTEST_POPULARITY_DISCORD_ROLE_ID
       - CONTEST_PARTICIPANT_DISCORD_ROLE_ID=CONTEST_PARTICIPANT_DISCORD_ROLE_ID
+      - BLOCKCHAIN_INFO_API_URL=http://localhost:19527/blockchain_info/
     networks:
       - litentry-test-network
     healthcheck:

diff --git a/bitacross-worker/docker/multiworker-docker-compose.yml b/bitacross-worker/docker/multiworker-docker-compose.yml
@@ -141,6 +141,7 @@ services:
       - CONTEST_LEGEND_DISCORD_ROLE_ID=CONTEST_LEGEND_DISCORD_ROLE_ID
       - CONTEST_POPULARITY_DISCORD_ROLE_ID=CONTEST_POPULARITY_DISCORD_ROLE_ID
       - CONTEST_PARTICIPANT_DISCORD_ROLE_ID=CONTEST_PARTICIPANT_DISCORD_ROLE_ID
+      - BLOCKCHAIN_INFO_API_URL=http://localhost:19527/blockchain_info/
     networks:
       - litentry-test-network
     healthcheck:
@@ -196,6 +197,7 @@ services:
       - CONTEST_LEGEND_DISCORD_ROLE_ID=CONTEST_LEGEND_DISCORD_ROLE_ID
       - CONTEST_POPULARITY_DISCORD_ROLE_ID=CONTEST_POPULARITY_DISCORD_ROLE_ID
       - CONTEST_PARTICIPANT_DISCORD_ROLE_ID=CONTEST_PARTICIPANT_DISCORD_ROLE_ID
+      - BLOCKCHAIN_INFO_API_URL=http://localhost:19527/blockchain_info/
     networks:
       - litentry-test-network
     healthcheck:
@@ -251,6 +253,7 @@ services:
       - CONTEST_LEGEND_DISCORD_ROLE_ID=CONTEST_LEGEND_DISCORD_ROLE_ID
       - CONTEST_POPULARITY_DISCORD_ROLE_ID=CONTEST_POPULARITY_DISCORD_ROLE_ID
       - CONTEST_PARTICIPANT_DISCORD_ROLE_ID=CONTEST_PARTICIPANT_DISCORD_ROLE_ID
+      - BLOCKCHAIN_INFO_API_URL=http://localhost:19527/blockchain_info/
     networks:
       - litentry-test-network
     healthcheck:

diff --git a/local-setup/.env.dev b/local-setup/.env.dev
@@ -55,3 +55,4 @@ LITENTRY_ARCHIVE_URL=http://localhost:19527
 MORALIS_API_URL=http://localhost:19527/moralis/
 MORALIS_SOLANA_API_URL=http://localhost:19527/moralis_solana/
 KARAT_DAO_API_URL=http://localhost:19527/karat_dao/
+BLOCKCHAIN_INFO_API_URL=http://localhost:19527/blockchain_info/
diff --git a/primitives/core/src/assertion/web3_token.rs b/primitives/core/src/assertion/web3_token.rs
@@ -72,6 +72,8 @@ pub enum Web3TokenType {
 	Sol,
 	#[codec(index = 24)]
 	Mcrt,
+	#[codec(index = 25)]
+	Btc,
 }
 
 impl Web3TokenType {
@@ -88,6 +90,13 @@ impl Web3TokenType {
 			Self::Nfp => vec![Web3Network::Bsc],
 			Self::Sol | Self::Mcrt =>
 				vec![Web3Network::Bsc, Web3Network::Ethereum, Web3Network::Solana],
+			Self::Btc => vec![
+				Web3Network::BitcoinP2tr,
+				Web3Network::BitcoinP2pkh,
+				Web3Network::BitcoinP2sh,
+				Web3Network::BitcoinP2wpkh,
+				Web3Network::BitcoinP2wsh,
+			],
 			_ => vec![Web3Network::Ethereum],
 		}
 	}

diff --git a/tee-worker/cli/src/trusted_base_cli/commands/litentry/request_vc.rs b/tee-worker/cli/src/trusted_base_cli/commands/litentry/request_vc.rs
@@ -265,6 +265,7 @@ pub enum TokenHoldingAmountCommand {
 	Nfp,
 	Sol,
 	Mcrt,
+	Btc,
 }
 
 #[derive(Subcommand, Debug)]
@@ -606,6 +607,7 @@ impl Command {
 				TokenHoldingAmountCommand::Nfp => TokenHoldingAmount(Web3TokenType::Nfp),
 				TokenHoldingAmountCommand::Sol => TokenHoldingAmount(Web3TokenType::Sol),
 				TokenHoldingAmountCommand::Mcrt => TokenHoldingAmount(Web3TokenType::Mcrt),
+				TokenHoldingAmountCommand::Btc => TokenHoldingAmount(Web3TokenType::Btc),
 			},
 			Command::PlatformUser(arg) => match arg {
 				PlatformUserCommand::KaratDaoUser => PlatformUser(PlatformUserType::KaratDaoUser),

diff --git a/tee-worker/client-api/parachain-api/prepare-build/interfaces/vc/definitions.ts b/tee-worker/client-api/parachain-api/prepare-build/interfaces/vc/definitions.ts
@@ -184,6 +184,7 @@ export default {
                 "Nfp",
                 "Sol",
                 "Mcrt",
+                "Btc",
             ],
         },
         // PlatformUserType

diff --git a/tee-worker/docker/docker-compose.yml b/tee-worker/docker/docker-compose.yml
@@ -142,6 +142,7 @@ services:
       - CONTEST_PARTICIPANT_DISCORD_ROLE_ID=CONTEST_PARTICIPANT_DISCORD_ROLE_ID
       - LITENTRY_ARCHIVE_URL=http://localhost:19527
       - VIP3_URL=http://localhost:19527
+      - BLOCKCHAIN_INFO_API_URL=http://localhost:19527/blockchain_info/
     networks:
       - litentry-test-network
     healthcheck:

diff --git a/tee-worker/docker/multiworker-docker-compose.yml b/tee-worker/docker/multiworker-docker-compose.yml
@@ -143,6 +143,7 @@ services:
       - CONTEST_PARTICIPANT_DISCORD_ROLE_ID=CONTEST_PARTICIPANT_DISCORD_ROLE_ID
       - LITENTRY_ARCHIVE_URL=http://localhost:19527
       - VIP3_URL=http://localhost:19527
+      - BLOCKCHAIN_INFO_API_URL=http://localhost:19527/blockchain_info/
     networks:
       - litentry-test-network
     healthcheck:
@@ -200,6 +201,7 @@ services:
       - CONTEST_PARTICIPANT_DISCORD_ROLE_ID=CONTEST_PARTICIPANT_DISCORD_ROLE_ID
       - LITENTRY_ARCHIVE_URL=http://localhost:19527
       - VIP3_URL=http://localhost:19527
+      - BLOCKCHAIN_INFO_API_URL=http://localhost:19527/blockchain_info/
     networks:
       - litentry-test-network
     healthcheck:
@@ -257,6 +259,7 @@ services:
       - CONTEST_PARTICIPANT_DISCORD_ROLE_ID=CONTEST_PARTICIPANT_DISCORD_ROLE_ID
       - LITENTRY_ARCHIVE_URL=http://localhost:19527
       - VIP3_URL=http://localhost:19527
+      - BLOCKCHAIN_INFO_API_URL=http://localhost:19527/blockchain_info/
     networks:
       - litentry-test-network
     healthcheck:

diff --git a/tee-worker/litentry/core/assertion-build-v2/src/token_holding_amount/mod.rs b/tee-worker/litentry/core/assertion-build-v2/src/token_holding_amount/mod.rs
@@ -151,6 +151,9 @@ mod tests {
 		data_provider_config
 			.set_nodereal_api_chain_network_url(url.clone() + "/nodereal_jsonrpc/")
 			.unwrap();
+		data_provider_config
+			.set_blockchain_info_api_url(url.clone() + "/blockchain_info/")
+			.unwrap();
 		data_provider_config.set_achainable_url(url.clone()).unwrap();
 		data_provider_config.set_moralis_api_url(url.clone() + "/moralis/").unwrap();
 		data_provider_config
@@ -608,4 +611,52 @@ mod tests {
 			},
 		}
 	}
+
+	#[test]
+	fn build_btc_holding_amount_works() {
+		let data_provider_config: DataProviderConfig = init();
+		// bc1pgr5fw4p9gl9me0vzjklnlnap669caxc0gsk4j62gff2qktlw6naqm4m3d0
+		let address = decode_hex(
+			"0x02e8c39e82aaaa143c3def8d3c7084a539b227244ac9067c3f7fc86cb73a0b7aed"
+				.as_bytes()
+				.to_vec(),
+		)
+		.unwrap()
+		.as_slice()
+		.try_into()
+		.unwrap();
+		let identities: Vec<IdentityNetworkTuple> =
+			vec![(Identity::Bitcoin(address), vec![Web3Network::BitcoinP2tr])];
+
+		let req = crate_assertion_build_request(Web3TokenType::Btc, identities);
+
+		match build(&req, Web3TokenType::Btc, &data_provider_config) {
+			Ok(credential) => {
+				log::info!("build btc TokenHoldingAmount done");
+				assert_eq!(
+					*(credential.credential_subject.assertions.first().unwrap()),
+					AssertionLogic::And {
+						items: vec![
+							create_token_assertion_logic(Web3TokenType::Btc),
+							create_network_address_assertion_logics(Web3TokenType::Btc),
+							Box::new(AssertionLogic::Item {
+								src: "$holding_amount".into(),
+								op: Op::GreaterEq,
+								dst: "1600".into()
+							}),
+							Box::new(AssertionLogic::Item {
+								src: "$holding_amount".into(),
+								op: Op::LessThan,
+								dst: "3000".into()
+							})
+						]
+					}
+				);
+				assert_eq!(*(credential.credential_subject.values.first().unwrap()), true);
+			},
+			Err(e) => {
+				panic!("build btc TokenHoldingAmount failed with error {:?}", e);
+			},
+		}
+	}
 }
diff --git a/tee-worker/litentry/core/common/src/web3_token/mod.rs b/tee-worker/litentry/core/common/src/web3_token/mod.rs
@@ -58,6 +58,7 @@ impl TokenName for Web3TokenType {
 			Self::Nfp => "NFP",
 			Self::Sol => "SOL",
 			Self::Mcrt => "MCRT",
+			Self::Btc => "BTC",
 		}
 	}
 }
@@ -214,7 +215,11 @@ impl TokenDecimals for Web3TokenType {
 			// Wbtc
 			(Self::Wbtc, Web3Network::Bsc) | (Self::Wbtc, Web3Network::Ethereum) |
 			// Mcrt
-			(Self::Mcrt, Web3Network::Solana) => 8,
+			(Self::Mcrt, Web3Network::Solana) |
+			// Btc
+			(Self::Btc, Web3Network::BitcoinP2tr) | (Self::Btc, Web3Network::BitcoinP2pkh) |
+			 (Self::Btc, Web3Network::BitcoinP2sh) | (Self::Btc, Web3Network::BitcoinP2wpkh) |
+			  (Self::Btc, Web3Network::BitcoinP2wsh) => 8,
 			// Usdc
 			(Self::Usdc, Web3Network::Ethereum) |
 			// Usdt

diff --git a/tee-worker/litentry/core/credentials/src/credential_schema.rs b/tee-worker/litentry/core/credentials/src/credential_schema.rs
@@ -94,8 +94,7 @@ pub fn get_schema_url(assertion: &Assertion) -> String {
 
 		Assertion::LITStaking => format!("{BASE_URL}/17-token-holding-amount/1-1-0.json"),
 
-		Assertion::TokenHoldingAmount(_) | Assertion::EVMAmountHolding(_) =>
-			format!("{BASE_URL}/21-evm-holding-amount/1-1-0.json"),
+		Assertion::EVMAmountHolding(_) => format!("{BASE_URL}/21-evm-holding-amount/1-1-0.json"),
 
 		Assertion::BRC20AmountHolder =>
 			format!("{BASE_URL}/20-token-holding-amount-list/1-1-0.json"),
@@ -104,6 +103,9 @@ pub fn get_schema_url(assertion: &Assertion) -> String {
 
 		Assertion::PlatformUser(_) => format!("{BASE_URL}/24-platform-user/1-1-0.json"),
 
+		Assertion::TokenHoldingAmount(_) =>
+			format!("{BASE_URL}/25-token-holding-amount/1-1-0.json"),
+
 		Assertion::NftHolder(_) => format!("{BASE_URL}/26-nft-holder/1-1-0.json"),
 	}
 }