program: add_update_perp_pnl_pool (#1810)

* program: add_update_perp_pnl_pool

* test

* CHANGELOG

Author: crispheaney

CHANGELOG.md

@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Features
 
 - program: add new settle pnl invariants ([#1812](https://github.com/drift-labs/protocol-v2/pull/1812))
+- program: add update_perp_market_pnl_pool ([#1810](https://github.com/drift-labs/protocol-v2/pull/1810))
 
 ### Fixes
 

programs/drift/src/instructions/admin.rs

@@ -1967,6 +1967,36 @@ pub fn handle_deposit_into_perp_market_fee_pool<'c: 'info, 'info>(
     Ok(())
 }
 
+#[access_control(
+    perp_market_valid(&ctx.accounts.perp_market)
+)]
+pub fn handle_update_perp_market_pnl_pool<'c: 'info, 'info>(
+    ctx: Context<'_, '_, 'c, 'info, UpdatePerpMarketPnlPool<'info>>,
+    amount: u64,
+) -> Result<()> {
+    let perp_market = &mut load_mut!(ctx.accounts.perp_market)?;
+
+    let spot_market = &mut load_mut!(ctx.accounts.spot_market)?;
+
+    controller::spot_balance::update_spot_balances(
+        amount.cast::<u128>()?,
+        &SpotBalanceType::Deposit,
+        spot_market,
+        &mut perp_market.pnl_pool,
+        false,
+    )?;
+
+    validate_spot_market_vault_amount(spot_market, ctx.accounts.spot_market_vault.amount)?;
+
+    msg!(
+        "updating perp market {} pnl pool with amount {}",
+        perp_market.market_index,
+        amount
+    );
+
+    Ok(())
+}
+
 #[access_control(
     deposit_not_paused(&ctx.accounts.state)
     spot_market_valid(&ctx.accounts.spot_market)
@@ -5261,6 +5291,29 @@ pub struct SettleExpiredMarketPoolsToRevenuePool<'info> {
     pub perp_market: AccountLoader<'info, PerpMarket>,
 }
 
+#[derive(Accounts)]
+pub struct UpdatePerpMarketPnlPool<'info> {
+    #[account(
+        has_one = admin
+    )]
+    pub state: Box<Account<'info, State>>,
+    pub admin: Signer<'info>,
+    #[account(
+        seeds = [b"spot_market", 0_u16.to_le_bytes().as_ref()],
+        bump,
+        mut
+    )]
+    pub spot_market: AccountLoader<'info, SpotMarket>,
+    #[account(
+        mut,
+        seeds = [b"spot_market_vault".as_ref(), 0_u16.to_le_bytes().as_ref()],
+        bump,
+    )]
+    pub spot_market_vault: Box<InterfaceAccount<'info, TokenAccount>>,
+    #[account(mut)]
+    pub perp_market: AccountLoader<'info, PerpMarket>,
+}
+
 #[derive(Accounts)]
 pub struct DepositIntoMarketFeePool<'info> {
     #[account(

programs/drift/src/lib.rs

@@ -1087,6 +1087,13 @@ pub mod drift {
         handle_deposit_into_perp_market_fee_pool(ctx, amount)
     }
 
+    pub fn update_perp_market_pnl_pool<'c: 'info, 'info>(
+        ctx: Context<'_, '_, 'c, 'info, UpdatePerpMarketPnlPool<'info>>,
+        amount: u64,
+    ) -> Result<()> {
+        handle_update_perp_market_pnl_pool(ctx, amount)
+    }
+
     pub fn deposit_into_spot_market_vault<'c: 'info, 'info>(
         ctx: Context<'_, '_, 'c, 'info, DepositIntoSpotMarketVault<'info>>,
         amount: u64,

sdk/src/adminClient.ts

@@ -1079,6 +1079,42 @@ export class AdminClient extends DriftClient {
 		});
 	}
 
+	public async updatePerpMarketPnlPool(
+		perpMarketIndex: number,
+		amount: BN
+	): Promise<TransactionSignature> {
+		const updatePerpMarketPnlPoolIx = await this.getUpdatePerpMarketPnlPoolIx(
+			perpMarketIndex,
+			amount
+		);
+
+		const tx = await this.buildTransaction(updatePerpMarketPnlPoolIx);
+
+		const { txSig } = await this.sendTransaction(tx, [], this.opts);
+
+		return txSig;
+	}
+
+	public async getUpdatePerpMarketPnlPoolIx(
+		perpMarketIndex: number,
+		amount: BN
+	): Promise<TransactionInstruction> {
+		return await this.program.instruction.updatePerpMarketPnlPool(amount, {
+			accounts: {
+				admin: this.isSubscribed
+					? this.getStateAccount().admin
+					: this.wallet.publicKey,
+				state: await this.getStatePublicKey(),
+				perpMarket: await getPerpMarketPublicKey(
+					this.program.programId,
+					perpMarketIndex
+				),
+				spotMarket: this.getQuoteSpotMarketAccount().pubkey,
+				spotMarketVault: this.getQuoteSpotMarketAccount().vault,
+			},
+		});
+	}
+
 	public async depositIntoSpotMarketVault(
 		spotMarketIndex: number,
 		amount: BN,

sdk/src/idl/drift.json

@@ -4832,6 +4832,42 @@
         }
       ]
     },
+    {
+      "name": "updatePerpMarketPnlPool",
+      "accounts": [
+        {
+          "name": "state",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "admin",
+          "isMut": false,
+          "isSigner": true
+        },
+        {
+          "name": "spotMarket",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "spotMarketVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "perpMarket",
+          "isMut": true,
+          "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "amount",
+          "type": "u64"
+        }
+      ]
+    },
     {
       "name": "depositIntoSpotMarketVault",
       "accounts": [

tests/admin.ts

@@ -6,9 +6,11 @@ import {
 	BN,
 	ExchangeStatus,
 	getPythLazerOraclePublicKey,
+	getTokenAmount,
 	loadKeypair,
 	OracleGuardRails,
 	OracleSource,
+	SpotBalanceType,
 	TestClient,
 	Wallet,
 } from '../sdk/src';
@@ -19,13 +21,15 @@ import {
 	initializeQuoteSpotMarket,
 	mockOracleNoProgram,
 	mockUSDCMint,
+	mockUserUSDCAccount,
 } from './testHelpers';
 import { PublicKey } from '@solana/web3.js';
 import {
 	BankrunContextWrapper,
 	Connection,
 } from '../sdk/src/bankrun/bankrunConnection';
 import { TestBulkAccountLoader } from '../sdk/src/accounts/testBulkAccountLoader';
+import { createTransferCheckedInstruction } from '@solana/spl-token';
 
 describe('admin', () => {
 	const chProgram = anchor.workspace.Drift as Program;
@@ -36,6 +40,10 @@ describe('admin', () => {
 
 	let usdcMint;
 
+	let userUSDCAccount;
+
+	const usdcAmount = new BN(10 * 10 ** 6);
+
 	let bankrunContextWrapper: BankrunContextWrapper;
 
 	before(async () => {
@@ -72,6 +80,13 @@ describe('admin', () => {
 			},
 		});
 
+		userUSDCAccount = await mockUserUSDCAccount(
+			usdcMint,
+			usdcAmount,
+			bankrunContextWrapper,
+			driftClient.wallet.publicKey
+		);
+
 		await driftClient.initialize(usdcMint.publicKey, true);
 		await driftClient.subscribe();
 		await driftClient.initializeUserAccount(0);
@@ -457,6 +472,38 @@ describe('admin', () => {
 		assert(perpMarket.amm.ammSpreadAdjustment == ammSpreadAdjustment);
 	});
 
+	it('update pnl pool', async () => {
+		const quoteVault = driftClient.getSpotMarketAccount(0).vault;
+
+		const splTransferIx = createTransferCheckedInstruction(
+			userUSDCAccount.publicKey,
+			usdcMint.publicKey,
+			quoteVault,
+			driftClient.wallet.publicKey,
+			usdcAmount.toNumber(),
+			6
+		);
+
+		const tx = await driftClient.buildTransaction(splTransferIx);
+		// @ts-ignore
+		await driftClient.sendTransaction(tx);
+
+		await driftClient.updatePerpMarketPnlPool(0, usdcAmount);
+
+		await driftClient.fetchAccounts();
+
+		const perpMarket = driftClient.getPerpMarketAccount(0);
+		const spotMarket = driftClient.getSpotMarketAccount(0);
+
+		const tokenAmount = getTokenAmount(
+			perpMarket.pnlPool.scaledBalance,
+			spotMarket,
+			SpotBalanceType.DEPOSIT
+		);
+
+		assert(tokenAmount.eq(usdcAmount));
+	});
+
 	it('Update admin', async () => {
 		const newAdminKey = PublicKey.default;