hedge-web3 0.2.24 → 0.2.26

package/src/instructions/redeemVault.ts

@@ -3,24 +3,20 @@ import { getOrCreateAssociatedTokenAccount, TOKEN_PROGRAM_ID } from '@solana/spl
 // import { TokenInstructions } from '@project-serum/serum'
 import {
   Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
+  PublicKey, Signer,
   SystemProgram,
   Transaction,
-  TransactionInstruction,
+  TransactionInstruction
 } from '@solana/web3.js'
-import { getLinkedListAccounts } from '../utils/getLinkedListAccounts'
 import {
   findAssociatedTokenAddress,
   getHedgeMintPublicKey,
-  getPoolPublicKeyForMint,
-  getVaultTypeAccountPublicKey,
-  getUshMintPublicKey,
-  getVaultSystemStatePublicKey,
+  getPoolPublicKeyForMint, getUshMintPublicKey,
+  getVaultSystemStatePublicKey
 } from '../Constants'
-import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 import { Vault } from '../idl/vault'
+import { getLinkedListAccounts } from '../utils/getLinkedListAccounts'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 
 export async function redeemVault(
   program: Program<Vault>,
@@ -30,7 +26,7 @@ export async function redeemVault(
   redeemAmount: number,
   transactionOverrideTime?: number
 ): Promise<PublicKey> {
-  const ushMintPublickey = await getUshMintPublicKey()
+  const ushMintPublickey = await getUshMintPublicKey(program.programId)
 
   // Prep the user to get USH back out at some point
   const payerUshAccount = await getOrCreateAssociatedTokenAccount(
@@ -116,11 +112,12 @@ export async function redeemVaultInstruction(
   redeemAmount: BN,
   transactionOverrideTime?: number
 ): Promise<TransactionInstruction> {
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
-  const ushMintPublickey = await getUshMintPublicKey()
-  const hedgeMintPublickey = await getHedgeMintPublicKey()
-  const [hedgeStakingPoolPublicKey] = await getPoolPublicKeyForMint(hedgeMintPublickey)
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
+  const ushMintPublickey = await getUshMintPublicKey(program.programId)
+  const hedgeMintPublickey = await getHedgeMintPublicKey(program.programId)
+  const [hedgeStakingPoolPublicKey] = await getPoolPublicKeyForMint(program.programId, hedgeMintPublickey)
   const hedgeStakingPoolAssociatedUshTokenAccount = await findAssociatedTokenAddress(
+    program.programId,
     hedgeStakingPoolPublicKey,
     ushMintPublickey
   )

package/src/instructions/referralClaimFees.ts

@@ -1,30 +1,20 @@
-import { BN, Idl, Program, Provider } from '@project-serum/anchor'
-import { TokenInstructions } from '@project-serum/serum'
-import { ASSOCIATED_TOKEN_PROGRAM_ID, TOKEN_PROGRAM_ID, getOrCreateAssociatedTokenAccount } from '@solana/spl-token'
+import { Program, Provider } from '@project-serum/anchor'
+import { getOrCreateAssociatedTokenAccount, TOKEN_PROGRAM_ID } from '@solana/spl-token'
 import {
-  Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
+  PublicKey, Signer,
   SystemProgram,
   SYSVAR_RENT_PUBKEY,
   Transaction,
-  TransactionInstruction,
+  TransactionInstruction
 } from '@solana/web3.js'
 import {
-  getVaultSystemStatePublicKey,
-  getReferralAccountPublicKey,
-  getHedgeMintPublicKey,
-  getReferralStatePublicKey,
-  getUshMintPublicKey,
-  getPoolPublicKeyForMint
+  getHedgeMintPublicKey, getPoolPublicKeyForMint, getReferralAccountPublicKey, getReferralStatePublicKey,
+  getUshMintPublicKey, getVaultSystemStatePublicKey
 } from '../Constants'
 
+import { Vault } from '../idl/vault'
 import { parseAnchorErrors } from '../utils/Errors'
 import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
-import { Vault } from '../idl/vault'
-
-
 
 /** @type {Function}  - Allows a referrer to claims their fees.
  * This checks the user has enough HDG in their wallet or staked and
@@ -36,30 +26,29 @@ import { Vault } from '../idl/vault'
  * - payer: Signer : who we are creating the referral account for
  * - poolPublicKey: PublicKey : a key to a pool position. If the position is closed or does not belong to the signer it will be ignored
  * - referrer: PublicKey
-*/
+ */
 export async function referralClaimFees(
   program: Program<Vault>,
   provider: Provider,
   payer: Signer,
   poolPosition: PublicKey
 ): Promise<PublicKey> {
-  // setup transaction  
+  // setup transaction
   const transaction = new Transaction()
   const signers = [payer]
 
   // General variables
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
 
   // Find referrer account
   const referrer = await payer.publicKey
-  const referralAccountPublicKey = await getReferralAccountPublicKey(referrer)
-
+  const referralAccountPublicKey = await getReferralAccountPublicKey(program.programId, referrer)
 
   // Derive the user referral account public key
-  const referallStatePublicKey = await getReferralStatePublicKey()
+  const referallStatePublicKey = await getReferralStatePublicKey(program.programId)
+
+  const hedgeMintPublicKey = await getHedgeMintPublicKey(program.programId)
 
-  const hedgeMintPublicKey = await getHedgeMintPublicKey()
-  
   // Get HDG account of signer
   const payerHdgAssociatedTokenAccount = await getOrCreateAssociatedTokenAccount(
     provider.connection,
@@ -68,7 +57,7 @@ export async function referralClaimFees(
     payer.publicKey
   )
 
-  const ushMintPublicKey = await getUshMintPublicKey()
+  const ushMintPublicKey = await getUshMintPublicKey(program.programId)
 
   // Get USH account of signer
   const payerUshAssociatedTokenAccount = await getOrCreateAssociatedTokenAccount(
@@ -87,8 +76,8 @@ export async function referralClaimFees(
     true
   )
 
-  const [feePoolPublicKey] = await getPoolPublicKeyForMint(hedgeMintPublicKey)
-  
+  const [feePoolPublicKey] = await getPoolPublicKeyForMint(program.programId, hedgeMintPublicKey)
+
   const feePoolAssociatedUshTokenAccount = await getOrCreateAssociatedTokenAccount(
     provider.connection,
     payer,
@@ -114,12 +103,10 @@ export async function referralClaimFees(
     )
   )
 
-
   await sendAndConfirmWithDebug(provider.connection, transaction, signers).catch(parseAnchorErrors)
   return referralAccountPublicKey
 }
 
-
 export async function referralClaimFeesInstruction(
   program: Program<Vault>,
   payerPublicKey: PublicKey,
@@ -135,7 +122,6 @@ export async function referralClaimFeesInstruction(
   feePoolPublicKey: PublicKey,
   feePoolAssociatedUshTokenAccountPublicKey: PublicKey
 ): Promise<TransactionInstruction> {
-
   return await program.methods
     .referralClaimFees()
     .accounts({

package/src/instructions/refreshOraclePrice.ts

@@ -1,17 +1,15 @@
 import { BN, Program, Provider } from '@project-serum/anchor'
 import {
   LAMPORTS_PER_SOL,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
+  PublicKey, Signer,
   SystemProgram,
   Transaction,
   TransactionInstruction,
-  TransactionSignature,
+  TransactionSignature
 } from '@solana/web3.js'
-import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
+import { CHAINLINK_PROGRAM_ID } from '../Constants'
 import { Vault } from '../idl/vault'
-import { HEDGE_PROGRAM_PUBLICKEY, CHAINLINK_PROGRAM_ID } from '../Constants'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 
 export async function refreshOraclePrice(
   program: Program<Vault>,
@@ -38,11 +36,11 @@ export async function refreshOraclePriceInstruction(
   const enc = new TextEncoder()
   const [oracleInfoAccount] = await PublicKey.findProgramAddress(
     [enc.encode(collateralType), enc.encode('Oracle')],
-    HEDGE_PROGRAM_PUBLICKEY
+    program.programId
   )
   const [vaultTypeAccount] = await PublicKey.findProgramAddress(
     [enc.encode(collateralType), enc.encode('State')],
-    HEDGE_PROGRAM_PUBLICKEY
+    program.programId
   )
   const oracleInfo = await program.account.oracleInfoForCollateralType.fetch(oracleInfoAccount)
 

package/src/instructions/repayVault.ts

@@ -2,24 +2,20 @@ import { BN, Program, Provider } from '@project-serum/anchor'
 import { getOrCreateAssociatedTokenAccount, TOKEN_PROGRAM_ID } from '@solana/spl-token'
 import {
   Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
+  PublicKey, Signer,
   SystemProgram,
   Transaction,
-  TransactionInstruction,
+  TransactionInstruction
 } from '@solana/web3.js'
-import { getLinkedListAccounts } from '../utils/getLinkedListAccounts'
 import {
   findAssociatedTokenAddress,
   getHedgeMintPublicKey,
-  getPoolPublicKeyForMint,
-  getVaultTypeAccountPublicKey,
-  getUshMintPublicKey,
-  getVaultSystemStatePublicKey,
+  getPoolPublicKeyForMint, getUshMintPublicKey,
+  getVaultSystemStatePublicKey
 } from '../Constants'
-import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 import { Vault } from '../idl/vault'
+import { getLinkedListAccounts } from '../utils/getLinkedListAccounts'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 
 export async function repayVault(
   program: Program<Vault>,
@@ -29,7 +25,7 @@ export async function repayVault(
   repayAmount: number,
   overrideTime?: number
 ): Promise<PublicKey> {
-  const ushMintPublickey = await getUshMintPublicKey()
+  const ushMintPublickey = await getUshMintPublicKey(program.programId)
 
   // Prep the user to get USH back out at some point
   const payerUshAccount = await getOrCreateAssociatedTokenAccount(
@@ -42,10 +38,15 @@ export async function repayVault(
 
   const vaultTypeAccount = await program.account.vaultType.fetch(vaultAccount.vaultType)
   const vaultTypeAssociatedTokenAccount = await findAssociatedTokenAddress(
+    program.programId,
     vaultAccount.vaultType,
     vaultTypeAccount.collateralMint
   )
-  const vaultAssociatedTokenAccount = await findAssociatedTokenAddress(vaultPublicKey, vaultTypeAccount.collateralMint)
+  const vaultAssociatedTokenAccount = await findAssociatedTokenAddress(
+    program.programId,
+    vaultPublicKey,
+    vaultTypeAccount.collateralMint
+  )
 
   const [oldSmallerPublicKey, newSmallerPublicKey, newLargerPublicKey] = await getLinkedListAccounts(
     program,
@@ -96,11 +97,12 @@ export async function repayVaultInstruction(
   repayAmount: BN,
   overrideTime?: number
 ): Promise<TransactionInstruction> {
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
-  const ushMintPublickey = await getUshMintPublicKey()
-  const hedgeMintPublickey = await getHedgeMintPublicKey()
-  const [hedgeStakingPoolPublicKey] = await getPoolPublicKeyForMint(hedgeMintPublickey)
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
+  const ushMintPublickey = await getUshMintPublicKey(program.programId)
+  const hedgeMintPublickey = await getHedgeMintPublicKey(program.programId)
+  const [hedgeStakingPoolPublicKey] = await getPoolPublicKeyForMint(program.programId, hedgeMintPublickey)
   const hedgeStakingPoolAssociatedUshTokenAccount = await findAssociatedTokenAddress(
+    program.programId,
     hedgeStakingPoolPublicKey,
     ushMintPublickey
   )

package/src/instructions/setHalted.ts

@@ -1,29 +1,10 @@
-import { BN, Program, Provider } from '@project-serum/anchor'
-import { TokenInstructions } from '@project-serum/serum'
-import { ASSOCIATED_TOKEN_PROGRAM_ID, TOKEN_PROGRAM_ID, getOrCreateAssociatedTokenAccount } from '@solana/spl-token'
-import {
-  Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
-  SystemProgram,
-  SYSVAR_RENT_PUBKEY,
-  Transaction,
-  TransactionInstruction,
-} from '@solana/web3.js'
-import {
-  findAssociatedTokenAddress,
-  findVaultAddress,
-  getVaultTypeAccountPublicKey,
-  getUshMintPublicKey,
-  getVaultSystemStatePublicKey,
-  getPoolPublicKeyForMint,
-  getHedgeMintPublicKey,
-} from '../Constants'
+import { Program, Provider } from '@project-serum/anchor'
+import { PublicKey, Signer, Transaction, TransactionInstruction } from '@solana/web3.js'
+import { getVaultSystemStatePublicKey } from '../Constants'
 
+import { Vault } from '../idl/vault'
 import { parseAnchorErrors } from '../utils/Errors'
 import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
-import { Vault } from '../idl/vault'
 
 export async function setHalted(
   program: Program<Vault>,
@@ -31,7 +12,7 @@ export async function setHalted(
   payer: Signer,
   halted: boolean
 ): Promise<PublicKey> {
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
 
   const transaction = new Transaction().add(
     await setHaltedInstruction(program, vaultSystemStatePublicKey, payer.publicKey, halted)

package/src/instructions/transferVault.ts

@@ -1,17 +1,12 @@
-import { BN, Program, Provider } from '@project-serum/anchor'
+import { Program, Provider } from '@project-serum/anchor'
 import {
   Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
-  SystemProgram,
-  SYSVAR_RENT_PUBKEY,
-  Transaction,
-  TransactionInstruction,
+  PublicKey, Signer, Transaction,
+  TransactionInstruction
 } from '@solana/web3.js'
 
-import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 import { Vault } from '../idl/vault'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 
 export async function transferVault(
   program: Program<Vault>,

package/src/instructions/updateReferralAccount.ts

@@ -1,22 +1,15 @@
 import { BN, Program, Provider } from '@project-serum/anchor'
 import {
-  Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
-  SystemProgram,
-  SYSVAR_RENT_PUBKEY,
-  Transaction,
-  TransactionInstruction,
+  PublicKey, Signer,
+  SystemProgram, Transaction,
+  TransactionInstruction
 } from '@solana/web3.js'
 import {
-  getVaultSystemStatePublicKey,
-  getReferralAccountPublicKey,
+  getReferralAccountPublicKey, getVaultSystemStatePublicKey
 } from '../Constants'
 
-import { parseAnchorErrors } from '../utils/Errors'
-import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 import { Vault } from '../idl/vault'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 
 export interface ReferralAccountConfig {
   setupCut?: BN
@@ -34,8 +27,8 @@ export async function updateReferralAccount(
   referrer: PublicKey,
   config: ReferralAccountConfig
 ): Promise<PublicKey> {
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
-  const referralAccountPublicKey = await getReferralAccountPublicKey(referrer)
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId, )
+  const referralAccountPublicKey = await getReferralAccountPublicKey(program.programId, referrer)
   const transaction = new Transaction().add(
     await updateReferralAccountInstruction(
       program,

package/src/instructions/updateReferralState.ts

@@ -1,22 +1,15 @@
 import { BN, Program, Provider } from '@project-serum/anchor'
 import {
-  Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
-  SystemProgram,
-  SYSVAR_RENT_PUBKEY,
-  Transaction,
-  TransactionInstruction,
+  PublicKey, Signer,
+  SystemProgram, Transaction,
+  TransactionInstruction
 } from '@solana/web3.js'
 import {
-  getVaultSystemStatePublicKey,
-  getReferralStatePublicKey,
+  getReferralStatePublicKey, getVaultSystemStatePublicKey
 } from '../Constants'
 
-import { parseAnchorErrors } from '../utils/Errors'
-import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 import { Vault } from '../idl/vault'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 
 export interface ReferralStateConfig {
   setupCut?: BN
@@ -43,8 +36,8 @@ export async function updateReferralState(
   payer: Signer,
   config: ReferralStateConfig
 ): Promise<PublicKey> {
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
-  const referralStatePublicKey = await getReferralStatePublicKey()
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId, )
+  const referralStatePublicKey = await getReferralStatePublicKey(program.programId, )
   const transaction = new Transaction().add(
     await updateReferralStateInstruction(
       program,

package/src/instructions/updateVaultType.ts

@@ -1,44 +1,30 @@
 import { BN, Program, Provider } from '@project-serum/anchor'
 import {
-  Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
-  SystemProgram,
-  SYSVAR_RENT_PUBKEY,
-  Transaction,
-  TransactionInstruction,
+  PublicKey, Signer, Transaction,
+  TransactionInstruction
 } from '@solana/web3.js'
 import {
-  findAssociatedTokenAddress,
-  findVaultAddress,
-  getVaultTypeAccountPublicKey,
-  getUshMintPublicKey,
-  getVaultSystemStatePublicKey,
-  getPoolPublicKeyForMint,
-  getHedgeMintPublicKey,
+  getVaultSystemStatePublicKey
 } from '../Constants'
 
-import { v4 as uuidv4 } from 'uuid'
-import { parseAnchorErrors } from '../utils/Errors'
-import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 import { Vault } from '../idl/vault'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 
 export interface VaultTypeConfig {
   maxDebtExtended?: BN
   minDebtPerVault?: BN
   loanInitFee?: BN
   interestRatePerSecond?: BN
-  
+
   emergencyModeThreshold?: BN
 
   oracleChainlink?: PublicKey
   oraclePyth?: PublicKey
   oracleSwitchboard?: PublicKey
 
-  priorityPyth?: number,
-  priorityChainlink?: number,
-  prioritySwitchboard?: number,
+  priorityPyth?: number
+  priorityChainlink?: number
+  prioritySwitchboard?: number
 
   deprecated?: boolean
 }
@@ -51,7 +37,7 @@ export async function updateVaultType(
   oracleInfoAccount: PublicKey,
   config: VaultTypeConfig
 ): Promise<PublicKey> {
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
   const transaction = new Transaction().add(
     await updateVaultTypeStatusInstruction(
       program,

package/src/instructions/withdrawStakingPool.ts

@@ -1,25 +1,21 @@
 import { BN, Program, Provider } from '@project-serum/anchor'
-import { ASSOCIATED_TOKEN_PROGRAM_ID, TOKEN_PROGRAM_ID } from '@solana/spl-token'
+import { TOKEN_PROGRAM_ID } from '@solana/spl-token'
 import {
   Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
-  SystemProgram,
-  SYSVAR_RENT_PUBKEY,
-  Transaction,
-  TransactionInstruction,
+  PublicKey, Signer,
+  SystemProgram, Transaction,
+  TransactionInstruction
 } from '@solana/web3.js'
-import { parseAnchorErrors } from '../utils/Errors'
 import {
   findAssociatedTokenAddress,
   getHedgeMintPublicKey,
   getPoolPublicKeyForMint,
   getUshMintPublicKey,
-  getVaultSystemStatePublicKey,
+  getVaultSystemStatePublicKey
 } from '../Constants'
-import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 import { Vault } from '../idl/vault'
+import { parseAnchorErrors } from '../utils/Errors'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
 
 export async function withdrawStakingPool(
   program: Program<Vault>,
@@ -51,16 +47,16 @@ export async function withdrawStakingPoolInstruction(
   stakedTokenMintPublicKey: PublicKey,
   overrideStartTime?: number
 ): Promise<TransactionInstruction> {
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
-  const ushMintPublickey = await getUshMintPublicKey()
-  const hedgeMintPublickey = await getHedgeMintPublicKey()
-  const [poolPublickey] = await getPoolPublicKeyForMint(stakedTokenMintPublicKey)
-  const poolAssociatedStakedTokenAccount = await findAssociatedTokenAddress(poolPublickey, stakedTokenMintPublicKey)
-  const poolAssociatedUshTokenAccount = await findAssociatedTokenAddress(poolPublickey, ushMintPublickey)
-  const payerAssociatedStakedTokenAccount = await findAssociatedTokenAddress(payerPublicKey, stakedTokenMintPublicKey)
-  const payerAssociatedHedgeAccount = await findAssociatedTokenAddress(payerPublicKey, hedgeMintPublickey)
-  const payerAssociatedUshAccount = await findAssociatedTokenAddress(payerPublicKey, ushMintPublickey)
-  const communityHedgeTokenAccount = await findAssociatedTokenAddress(vaultSystemStatePublicKey, hedgeMintPublickey)
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId, )
+  const ushMintPublickey = await getUshMintPublicKey(program.programId, )
+  const hedgeMintPublickey = await getHedgeMintPublicKey(program.programId, )
+  const [poolPublickey] = await getPoolPublicKeyForMint(program.programId, stakedTokenMintPublicKey)
+  const poolAssociatedStakedTokenAccount = await findAssociatedTokenAddress(program.programId, poolPublickey, stakedTokenMintPublicKey)
+  const poolAssociatedUshTokenAccount = await findAssociatedTokenAddress(program.programId, poolPublickey, ushMintPublickey)
+  const payerAssociatedStakedTokenAccount = await findAssociatedTokenAddress(program.programId, payerPublicKey, stakedTokenMintPublicKey)
+  const payerAssociatedHedgeAccount = await findAssociatedTokenAddress(program.programId, payerPublicKey, hedgeMintPublickey)
+  const payerAssociatedUshAccount = await findAssociatedTokenAddress(program.programId, payerPublicKey, ushMintPublickey)
+  const communityHedgeTokenAccount = await findAssociatedTokenAddress(program.programId, vaultSystemStatePublicKey, hedgeMintPublickey)
 
   return await program.methods
     .withdrawStakingPool(

package/src/instructions/withdrawVault.ts

@@ -1,26 +1,20 @@
 import { BN, Program, Provider } from '@project-serum/anchor'
-import { getOrCreateAssociatedTokenAccount, TOKEN_PROGRAM_ID } from '@solana/spl-token'
 import { TokenInstructions } from '@project-serum/serum'
+import { getOrCreateAssociatedTokenAccount, TOKEN_PROGRAM_ID } from '@solana/spl-token'
 import {
   Keypair,
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
+  PublicKey, Signer,
   SystemProgram,
   Transaction,
-  TransactionInstruction,
+  TransactionInstruction
 } from '@solana/web3.js'
 import {
-  findAssociatedTokenAddress,
-  getVaultTypeAccountPublicKey,
-  getUshMintPublicKey,
-  getVaultSystemStatePublicKey,
-  getPoolPublicKeyForMint,
-  getHedgeMintPublicKey,
+  findAssociatedTokenAddress, getHedgeMintPublicKey, getPoolPublicKeyForMint, getUshMintPublicKey,
+  getVaultSystemStatePublicKey
 } from '../Constants'
+import { Vault } from '../idl/vault'
 import { getLinkedListAccounts } from '../utils/getLinkedListAccounts'
 import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
-import { Vault } from '../idl/vault'
 
 export async function withdrawVault(
   program: Program<Vault>,
@@ -30,13 +24,13 @@ export async function withdrawVault(
   withdrawAmount: number,
   overrideTime?: number
 ): Promise<PublicKey> {
-  const ushMintPublickey = await getUshMintPublicKey()
+  const ushMintPublickey = await getUshMintPublicKey(program.programId, )
 
   // Prep the user to get USH back out at some point
   await getOrCreateAssociatedTokenAccount(provider.connection, payer, ushMintPublickey, payer.publicKey)
 
   const history = Keypair.generate()
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId, )
   const vaultAccount = await program.account.vault.fetch(vaultPublicKey)
   const vaultAssociatedCollateralAccount = await getOrCreateAssociatedTokenAccount(
     provider.connection,
@@ -61,8 +55,8 @@ export async function withdrawVault(
     vaultTypeAccountInfo.collateralMint,
     payer.publicKey
   )
-  const [hedgeStakingPoolPublicKey] = await getPoolPublicKeyForMint(await getHedgeMintPublicKey())
-  const hedgeStakingPoolAssociatedUshTokenAccount = await findAssociatedTokenAddress(
+  const [hedgeStakingPoolPublicKey] = await getPoolPublicKeyForMint(program.programId, await getHedgeMintPublicKey(program.programId, ))
+  const hedgeStakingPoolAssociatedUshTokenAccount = await findAssociatedTokenAddress(program.programId, 
     hedgeStakingPoolPublicKey,
     ushMintPublickey
   )