-
Notifications
You must be signed in to change notification settings - Fork 16
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
9f6233f
commit e533759
Showing
5 changed files
with
308 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
141 changes: 141 additions & 0 deletions
141
programs/drift_vaults/src/instructions/force_withdraw.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,141 @@ | ||
| use crate::constraints::*; | ||
| use crate::cpi::{TokenTransferCPI, WithdrawCPI}; | ||
| use crate::{declare_vault_seeds, AccountMapProvider}; | ||
| use crate::{Vault, VaultDepositor}; | ||
| use anchor_lang::prelude::*; | ||
| use anchor_spl::token::{self, Transfer}; | ||
| use anchor_spl::token::{Token, TokenAccount}; | ||
| use drift::cpi::accounts::Withdraw as DriftWithdraw; | ||
| use drift::instructions::optional_accounts::AccountMaps; | ||
| use drift::program::Drift; | ||
| use drift::state::user::User; | ||
|
|
||
| pub fn force_withdraw<'info>(ctx: Context<'_, '_, '_, 'info, ForceWithdraw<'info>>) -> Result<()> { | ||
| let clock = &Clock::get()?; | ||
| let mut vault = ctx.accounts.vault.load_mut()?; | ||
| let mut vault_depositor = ctx.accounts.vault_depositor.load_mut()?; | ||
|
|
||
| let user = ctx.accounts.drift_user.load()?; | ||
| let spot_market_index = vault.spot_market_index; | ||
|
|
||
| let AccountMaps { | ||
| perp_market_map, | ||
| spot_market_map, | ||
| mut oracle_map, | ||
| } = ctx.load_maps(clock.slot, Some(spot_market_index))?; | ||
|
|
||
| let vault_equity = | ||
| vault.calculate_equity(&user, &perp_market_map, &spot_market_map, &mut oracle_map)?; | ||
|
|
||
| let (withdraw_amount, _) = | ||
| vault_depositor.withdraw(vault_equity, &mut vault, clock.unix_timestamp)?; | ||
|
|
||
| msg!("force_withdraw_amount: {}", withdraw_amount); | ||
|
|
||
| drop(vault); | ||
| drop(user); | ||
|
|
||
| ctx.drift_withdraw(withdraw_amount)?; | ||
|
|
||
| ctx.token_transfer(withdraw_amount)?; | ||
|
|
||
| Ok(()) | ||
| } | ||
|
|
||
| #[derive(Accounts)] | ||
| pub struct ForceWithdraw<'info> { | ||
| #[account( | ||
| mut, | ||
| constraint = is_manager_for_vault(&vault, &manager)? | ||
| )] | ||
| pub vault: AccountLoader<'info, Vault>, | ||
| pub manager: Signer<'info>, | ||
| #[account( | ||
| mut, | ||
| constraint = is_vault_for_vault_depositor(&vault_depositor, &vault)?, | ||
| )] | ||
| pub vault_depositor: AccountLoader<'info, VaultDepositor>, | ||
| pub authority: Signer<'info>, | ||
| #[account( | ||
| mut, | ||
| seeds = [b"vault_token_account".as_ref(), vault.key().as_ref()], | ||
| bump, | ||
| )] | ||
| pub vault_token_account: Box<Account<'info, TokenAccount>>, | ||
| #[account( | ||
| mut, | ||
| constraint = is_user_stats_for_vault(&vault, &drift_user_stats)? | ||
| )] | ||
| /// CHECK: checked in drift cpi | ||
| pub drift_user_stats: AccountInfo<'info>, | ||
| #[account( | ||
| mut, | ||
| constraint = is_user_for_vault(&vault, &drift_user.key())? | ||
| )] | ||
| /// CHECK: checked in drift cpi | ||
| pub drift_user: AccountLoader<'info, User>, | ||
| /// CHECK: checked in drift cpi | ||
| pub drift_state: AccountInfo<'info>, | ||
| #[account( | ||
| mut, | ||
| token::mint = vault_token_account.mint | ||
| )] | ||
| pub drift_spot_market_vault: Box<Account<'info, TokenAccount>>, | ||
| /// CHECK: checked in drift cpi | ||
| pub drift_signer: AccountInfo<'info>, | ||
| #[account( | ||
| mut, | ||
| token::authority = vault_depositor.load()?.authority, | ||
| token::mint = vault_token_account.mint | ||
| )] | ||
| pub user_token_account: Box<Account<'info, TokenAccount>>, | ||
| pub drift_program: Program<'info, Drift>, | ||
| pub token_program: Program<'info, Token>, | ||
| } | ||
|
|
||
| impl<'info> WithdrawCPI for Context<'_, '_, '_, 'info, ForceWithdraw<'info>> { | ||
| fn drift_withdraw(&self, amount: u64) -> Result<()> { | ||
| declare_vault_seeds!(self.accounts.vault, seeds); | ||
| let spot_market_index = self.accounts.vault.load()?.spot_market_index; | ||
|
|
||
| let cpi_accounts = DriftWithdraw { | ||
| state: self.accounts.drift_state.to_account_info().clone(), | ||
| user: self.accounts.drift_user.to_account_info().clone(), | ||
| user_stats: self.accounts.drift_user_stats.to_account_info().clone(), | ||
| authority: self.accounts.vault.to_account_info().clone(), | ||
| spot_market_vault: self | ||
| .accounts | ||
| .drift_spot_market_vault | ||
| .to_account_info() | ||
| .clone(), | ||
| drift_signer: self.accounts.drift_signer.to_account_info().clone(), | ||
| user_token_account: self.accounts.vault_token_account.to_account_info().clone(), | ||
| token_program: self.accounts.token_program.to_account_info().clone(), | ||
| }; | ||
|
|
||
| let drift_program = self.accounts.drift_program.to_account_info().clone(); | ||
| let cpi_context = CpiContext::new_with_signer(drift_program, cpi_accounts, seeds) | ||
| .with_remaining_accounts(self.remaining_accounts.into()); | ||
| drift::cpi::withdraw(cpi_context, spot_market_index, amount, false)?; | ||
|
|
||
| Ok(()) | ||
| } | ||
| } | ||
|
|
||
| impl<'info> TokenTransferCPI for Context<'_, '_, '_, 'info, ForceWithdraw<'info>> { | ||
| fn token_transfer(&self, amount: u64) -> Result<()> { | ||
| declare_vault_seeds!(self.accounts.vault, seeds); | ||
|
|
||
| let cpi_accounts = Transfer { | ||
| from: self.accounts.vault_token_account.to_account_info().clone(), | ||
| to: self.accounts.user_token_account.to_account_info().clone(), | ||
| authority: self.accounts.vault.to_account_info().clone(), | ||
| }; | ||
| let token_program = self.accounts.token_program.to_account_info().clone(); | ||
| let cpi_context = CpiContext::new_with_signer(token_program, cpi_accounts, seeds); | ||
|
|
||
| token::transfer(cpi_context, amount)?; | ||
|
|
||
| Ok(()) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters