fix(bolt-cli): don't panic when stake is not found

chainbound · Dec 12, 2024 · 831bf7a · 831bf7a
1 parent 5e77002
commit 831bf7a
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 2 deletions.
diff --git a/bolt-cli/src/commands/operators.rs b/bolt-cli/src/commands/operators.rs
@@ -258,7 +258,19 @@ impl OperatorsCommand {
                     let mut total_collateral = Uint::from(0);
                     for (name, address) in deployments.collateral {
                         let stake =
-                            bolt_manager.getOperatorStake(address, address).call().await?._0;
+                            match bolt_manager.getOperatorStake(address, address).call().await {
+                                Ok(stake) => stake._0,
+                                Err(e) => match try_parse_contract_error::<
+                                    BoltEigenLayerMiddlewareErrors,
+                                >(e)?
+                                {
+                                    BoltEigenLayerMiddlewareErrors::KeyNotFound(_) => Uint::from(0),
+                                    other => unreachable!(
+                                        "Unexpected error with selector {:?}",
+                                        other.selector()
+                                    ),
+                                },
+                            };
                         if stake > Uint::from(0) {
                             total_collateral += stake;
                             info!(?address, token = %name, amount = ?stake, "Operator has collateral");
@@ -420,7 +432,19 @@ impl OperatorsCommand {
                     let mut total_collateral = Uint::from(0);
                     for (name, address) in deployments.collateral {
                         let stake =
-                            bolt_manager.getOperatorStake(address, address).call().await?._0;
+                            match bolt_manager.getOperatorStake(address, address).call().await {
+                                Ok(stake) => stake._0,
+                                Err(e) => match try_parse_contract_error::<
+                                    BoltSymbioticMiddlewareErrors,
+                                >(e)?
+                                {
+                                    BoltSymbioticMiddlewareErrors::KeyNotFound(_) => Uint::from(0),
+                                    other => unreachable!(
+                                        "Unexpected error with selector {:?}",
+                                        other.selector()
+                                    ),
+                                },
+                            };
                         if stake > Uint::from(0) {
                             total_collateral += stake;
                             info!(?address, token = %name, amount = ?stake, "Operator has collateral");

diff --git a/bolt-cli/src/contracts/bolt.rs b/bolt-cli/src/contracts/bolt.rs
@@ -64,6 +64,7 @@ sol! {
         error AlreadyRegistered();
         error NotOperator();
         error NotRegistered();
+        error KeyNotFound();
     }
 
     #[allow(missing_docs)]
@@ -87,5 +88,6 @@ sol! {
         error AlreadyRegistered();
         error NotOperator();
         error NotRegistered();
+        error KeyNotFound();
     }
 }