feat: add update pyth token mint instruction

pyth-network · Sep 3, 2024 · 72f3d27 · 72f3d27
1 parent fac01eb
commit 72f3d27
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/staking/programs/integrity-pool/src/context.rs b/staking/programs/integrity-pool/src/context.rs
@@ -53,6 +53,21 @@ pub struct UpdateY<'info> {
     pub system_program: Program<'info, System>,
 }
 
+#[derive(Accounts)]
+pub struct UpdatePythTokenMint<'info> {
+    pub reward_program_authority: Signer<'info>,
+
+    #[account(
+        mut,
+        seeds = [POOL_CONFIG.as_bytes()],
+        bump,
+        has_one = reward_program_authority @ IntegrityPoolError::InvalidRewardProgramAuthority,
+    )]
+    pub pool_config: Account<'info, PoolConfig>,
+
+    pub system_program: Program<'info, System>,
+}
+
 #[derive(Accounts)]
 pub struct UpdateDelegationFee<'info> {
     pub reward_program_authority: Signer<'info>,

diff --git a/staking/programs/integrity-pool/src/lib.rs b/staking/programs/integrity-pool/src/lib.rs
@@ -57,6 +57,14 @@ pub mod integrity_pool {
         Ok(())
     }
 
+    pub fn update_pyth_token_mint(
+        ctx: Context<UpdatePythTokenMint>,
+        pyth_token_mint: Pubkey,
+    ) -> Result<()> {
+        ctx.accounts.pool_config.pyth_token_mint = pyth_token_mint;
+        Ok(())
+    }
+
     pub fn update_delegation_fee(
         ctx: Context<UpdateDelegationFee>,
         delegation_fee: frac64,