From b5dc4f89dfdd68db92d5ed9902a0ae143f2c4900 Mon Sep 17 00:00:00 2001
From: Zhouhui Tian <125243011+zhouhuitian@users.noreply.github.com>
Date: Wed, 10 Jan 2024 00:38:26 +0800
Subject: [PATCH] Add missing litentry-cli commands for request-vc (#2394)

* add LITStaking

* add  BRC20AmountHolder
---
 .../cli/src/trusted_base_cli/commands/litentry/request_vc.rs  | 4 ++++
 .../trusted_base_cli/commands/litentry/request_vc_direct.rs   | 2 ++
 2 files changed, 6 insertions(+)

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
index b740d93d2a..64723c63dd 100644
--- 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
@@ -104,6 +104,8 @@ pub enum Command {
 	#[clap(subcommand)]
 	EVMAmountHolding(EVMAmountHoldingCommand),
 	CryptoSummary,
+	LITStaking,
+	BRC20AmountHolder,
 }
 
 #[derive(Args, Debug)]
@@ -476,6 +478,8 @@ impl RequestVcCommand {
 				EVMAmountHoldingCommand::Trx => Assertion::EVMAmountHolding(EVMTokenType::Trx),
 			},
 			Command::CryptoSummary => Assertion::CryptoSummary,
+			Command::LITStaking => Assertion::LITStaking,
+			Command::BRC20AmountHolder => Assertion::BRC20AmountHolder,
 		};
 
 		let key = Self::random_aes_key();
diff --git a/tee-worker/cli/src/trusted_base_cli/commands/litentry/request_vc_direct.rs b/tee-worker/cli/src/trusted_base_cli/commands/litentry/request_vc_direct.rs
index d909a05212..54da244d3d 100644
--- a/tee-worker/cli/src/trusted_base_cli/commands/litentry/request_vc_direct.rs
+++ b/tee-worker/cli/src/trusted_base_cli/commands/litentry/request_vc_direct.rs
@@ -254,6 +254,8 @@ impl RequestVcDirectCommand {
 				EVMAmountHoldingCommand::Trx => Assertion::EVMAmountHolding(EVMTokenType::Trx),
 			},
 			Command::CryptoSummary => Assertion::CryptoSummary,
+			Command::LITStaking => Assertion::LITStaking,
+			Command::BRC20AmountHolder => Assertion::BRC20AmountHolder,
 		};
 
 		let mut key: RequestAesKey = RequestAesKey::default();