transfer oracle data ix to constituent (#1643)

* transfer oracle data ix to constituent

* add lib entrypoint

* simplify more

* add spot market constraint

Author: moosecat2

programs/drift/src/instructions/lp_pool.rs

@@ -20,8 +20,10 @@ use crate::{
             ConstituentTargetBaseFixed, LPPool, TargetsDatum, WeightValidationFlags,
         },
         oracle::OraclePriceData,
+        oracle_map::OracleMap,
         perp_market::{AmmCacheFixed, CacheInfo, AMM_POSITIONS_CACHE},
         perp_market_map::MarketSet,
+        spot_market::SpotMarket,
         spot_market_map::get_writable_spot_market_set_from_many,
         state::State,
         user::MarketType,
@@ -912,6 +914,46 @@ pub fn handle_lp_pool_remove_liquidity<'c: 'info, 'info>(
     Ok(())
 }
 
+pub fn handle_update_constituent_oracle_info<'c: 'info, 'info>(
+    ctx: Context<'_, '_, 'c, 'info, UpdateConstituentOracleInfo<'info>>,
+) -> Result<()> {
+    let clock = Clock::get()?;
+    let mut constituent = ctx.accounts.constituent.load_mut()?;
+    let spot_market = ctx.accounts.spot_market.load()?;
+
+    let oracle_id = spot_market.oracle_id();
+    let mut oracle_map = OracleMap::load_one(
+        &ctx.accounts.oracle,
+        clock.slot,
+        Some(ctx.accounts.state.oracle_guard_rails),
+    )?;
+
+    let oracle_data = oracle_map.get_price_data(&oracle_id)?;
+    let oracle_data_slot = clock.slot - oracle_data.delay.max(0i64).cast::<u64>()?;
+    if constituent.last_oracle_slot < oracle_data_slot {
+        constituent.last_oracle_price = oracle_data.price;
+        constituent.last_oracle_slot = oracle_data_slot;
+    }
+
+    Ok(())
+}
+
+#[derive(Accounts)]
+pub struct UpdateConstituentOracleInfo<'info> {
+    pub state: Box<Account<'info, State>>,
+    #[account(mut)]
+    pub keeper: Signer<'info>,
+    #[account(mut)]
+    pub constituent: AccountLoader<'info, Constituent>,
+    #[account(
+        owner = crate::ID,
+        constraint = spot_market.load()?.market_index == constituent.load()?.spot_market_index
+    )]
+    pub spot_market: AccountLoader<'info, SpotMarket>,
+    /// CHECK: checked when loading oracle in oracle map
+    pub oracle: AccountInfo<'info>,
+}
+
 #[derive(Accounts)]
 pub struct UpdateConstituentTargetBase<'info> {
     pub state: Box<Account<'info, State>>,

programs/drift/src/lib.rs

@@ -1859,6 +1859,12 @@ pub mod drift {
     ) -> Result<()> {
         handle_end_lp_swap(ctx)
     }
+
+    pub fn update_constituent_oracle_info<'c: 'info, 'info>(
+        ctx: Context<'_, '_, 'c, 'info, UpdateConstituentOracleInfo<'info>>,
+    ) -> Result<()> {
+        handle_update_constituent_oracle_info(ctx)
+    }
 }
 
 #[cfg(not(feature = "no-entrypoint"))]