program: subaccount-stake-for-users-stats

Author: 0xbigz

programs/drift/src/instructions/user.rs

@@ -32,6 +32,7 @@ use crate::instructions::optional_accounts::{
 };
 use crate::instructions::SpotFulfillmentType;
 use crate::math::casting::Cast;
+use crate::math::constants::GOV_SPOT_MARKET_INDEX;
 use crate::math::liquidation::is_user_being_liquidated;
 use crate::math::margin::calculate_margin_requirement_and_total_collateral_and_liability_info;
 use crate::math::margin::meets_initial_margin_requirement;
@@ -852,6 +853,10 @@ pub fn handle_withdraw<'c: 'info, 'info>(
 
     spot_market.validate_max_token_deposits_and_borrows(is_borrow)?;
 
+    if spot_market.market_index == GOV_SPOT_MARKET_INDEX && user_stats.gov_stake_sub_account_active()? {
+        user_stats.update_gov_stake()?;
+    }
+
     Ok(())
 }
 

programs/drift/src/state/user.rs

@@ -1654,7 +1654,8 @@ pub struct UserStats {
     /// Second bit: 1 if user was referred, 0 otherwise
     pub referrer_status: u8,
     pub disable_update_perp_bid_ask_twap: bool,
-    pub padding1: [u8; 1],
+    // use user subaccount for if_staked_gov_token_amount value (-1 is sub_account 0)
+    pub gov_stake_sub_account: i8,
     /// whether the user has a FuelOverflow account
     pub fuel_overflow_status: u8,
     /// accumulated fuel for token amounts of insurance