Jito Bell: Subscribe rebalance directed event

jito-bell/src/event_parser/jito_steward.rs

@@ -7,7 +7,8 @@ use solana_pubkey::Pubkey;
 
 use crate::events::jito_steward::{
     AutoAddValidatorEvent, AutoRemoveValidatorEvent, DecreaseComponents, EpochMaintenanceEvent,
-    InstantUnstakeComponents, RebalanceEvent, ScoreComponents, StateTransition,
+    InstantUnstakeComponents, RebalanceDirectedEvent, RebalanceEvent, ScoreComponents,
+    StateTransition,
 };
 
 const PROGRAM_LOG: &str = "Program log: ";
@@ -20,6 +21,7 @@ pub enum JitoStewardEvent {
     EpochMaintenance(EpochMaintenanceEvent),
     StateTransition(StateTransition),
     Rebalance(RebalanceEvent),
+    RebalanceDirected(RebalanceDirectedEvent),
     DecreaseComponents(DecreaseComponents),
     ScoreComponents(ScoreComponents),
     InstantUnstake(InstantUnstakeComponents),
@@ -33,6 +35,7 @@ impl std::fmt::Display for JitoStewardEvent {
             JitoStewardEvent::EpochMaintenance(_) => write!(f, "epoch_maintenance"),
             JitoStewardEvent::StateTransition(_) => write!(f, "state_transition"),
             JitoStewardEvent::Rebalance(_) => write!(f, "rebalance"),
+            JitoStewardEvent::RebalanceDirected(_) => write!(f, "rebalance_directed"),
             JitoStewardEvent::DecreaseComponents(_) => write!(f, "decrease_components"),
             JitoStewardEvent::ScoreComponents(_) => write!(f, "score_components"),
             JitoStewardEvent::InstantUnstake(_) => write!(f, "instant_unstake"),
@@ -112,6 +115,14 @@ impl JitoStewardEvent {
             }
         }
 
+        // RebalanceDirectedEvent
+        if discriminator == RebalanceDirectedEvent::DISCRIMINATOR {
+            match RebalanceDirectedEvent::try_from_slice(event_data) {
+                Ok(event) => return Some(JitoStewardEvent::RebalanceDirected(event)),
+                Err(e) => debug!("Failed to deserialize RebalanceDirectedEvent: {e}"),
+            }
+        }
+
         // DecreaseComponents
         if discriminator == DecreaseComponents::DISCRIMINATOR {
             match DecreaseComponents::try_from_slice(event_data) {

jito-bell/src/events/jito_steward.rs

@@ -82,6 +82,19 @@ pub enum RebalanceTypeTag {
     Decrease,
 }
 
+#[derive(Debug, Clone, BorshDeserialize)]
+pub struct RebalanceDirectedEvent {
+    pub vote_account: Pubkey,
+    pub epoch: u16,
+    pub rebalance_type_tag: RebalanceTypeTag,
+    pub increase_lamports: u64,
+    pub decrease_components: DecreaseComponents,
+}
+
+impl RebalanceDirectedEvent {
+    pub const DISCRIMINATOR: [u8; 8] = [120, 27, 117, 235, 104, 42, 132, 75];
+}
+
 /// Deprecated: This struct is no longer emitted but is kept to allow parsing of old events.
 /// Because the event discriminator is based on struct name, it's important to rename the struct if
 /// fields are changed.

jito-bell/src/lib.rs

@@ -349,6 +349,42 @@ impl JitoBellHandler {
 
                                 (desc, Some(amount_sol), Some("SOL"))
                             }
+                            JitoStewardEvent::RebalanceDirected(rebalance) => {
+                                let (change_type, amount_lamports) =
+                                    if rebalance.increase_lamports > 0 {
+                                        ("Stake Increase", rebalance.increase_lamports)
+                                    } else {
+                                        (
+                                            "Stake Decrease",
+                                            rebalance.decrease_components.total_unstake_lamports,
+                                        )
+                                    };
+
+                                let amount_sol = amount_lamports as f64 / 1_000_000_000.0;
+                                let type_emoji = if rebalance.increase_lamports > 0 {
+                                    "📈"
+                                } else {
+                                    "📉"
+                                };
+
+                                let validator_short =
+                                    self.shorten_pubkey(&rebalance.vote_account.to_string(), 4, 4);
+
+                                let desc = format!(
+                                    "{} *{}* | {:.2} SOL\n\
+                                    \n\
+                                    Validator: `{}`\n\
+                                    Epoch: {} | Type: {:?}",
+                                    type_emoji,
+                                    change_type,
+                                    amount_sol,
+                                    validator_short,
+                                    rebalance.epoch,
+                                    rebalance.rebalance_type_tag
+                                );
+
+                                (desc, Some(amount_sol), Some("SOL"))
+                            }
                             _ => {
                                 debug!("Unhandled event type: {:?}", jito_steward_event);
                                 ("Unknown event".to_string(), None, None)

jito_bell_config_sample.yaml

@@ -33,6 +33,25 @@ programs:
               description: "Whale rebalance detected"
               destinations: ["stake_pool_alerts_slack"]
 
+      rebalance_directed:
+        thresholds:
+          - value: 1.0 # SOL (minimum to notify)
+            notification:
+              description: "Small rebalance detected"
+              destinations: ["stake_pool_alerts_slack"]
+          - value: 100.0 # SOL
+            notification:
+              description: "Medium rebalance detected"
+              destinations: ["stake_pool_alerts_slack"]
+          - value: 1000.0 # SOL
+            notification:
+              description: "Large rebalance detected"
+              destinations: ["stake_pool_alerts_slack"]
+          - value: 10000.0 # SOL
+            notification:
+              description: "Whale rebalance detected"
+              destinations: ["stake_pool_alerts_slack"]
+
       epoch_maintenance:
         destinations: ["stake_pool_alerts_slack"]
         description: "Epoch maintenance completed"