hedge-web3 0.2.21 → 0.2.25

package/src/index.ts

@@ -1,37 +1,41 @@
-
-export * from './instructions/createStakingPool'
-export * from './instructions/depositStakingPool'
-export * from './instructions/withdrawStakingPool'
-export * from './instructions/claimStakingPoolPosition'
-export * from './instructions/depositLiquidationPool'
-export * from './instructions/closeLiquidationPoolPosition'
+export * from './Constants'
+export * from './HedgeDecimal'
+export * from './idl/vault'
 export * from './instructions/claimLiquidationPoolPosition'
+export * from './instructions/claimStakingPoolPosition'
 export * from './instructions/closeClaimedLiquidationPoolPosition'
+export * from './instructions/closeLiquidationPoolPosition'
+export * from './instructions/createReferralAccount'
+export * from './instructions/createStakingPool'
+export * from './instructions/createUserReferralAccount'
 export * from './instructions/createVault'
-export * from './instructions/psmMintUsh'
-export * from './instructions/psmRedeemUsh'
+export * from './instructions/depositLiquidationPool'
+export * from './instructions/depositStakingPool'
 export * from './instructions/depositVault'
-export * from './instructions/withdrawVault'
+export * from './instructions/initHedgeFoundation'
+export * from './instructions/initHedgeFoundationTokens'
+export * from './instructions/liquidateVault'
 export * from './instructions/loanVault'
-export * from './instructions/repayVault'
+export * from './instructions/psmMintUsh'
+export * from './instructions/psmRedeemUsh'
 export * from './instructions/redeemVault'
-export * from './instructions/liquidateVault'
+export * from './instructions/referralClaimFees'
 export * from './instructions/refreshOraclePrice'
-export * from './instructions/initHedgeFoundation'
-export * from './instructions/initHedgeFoundationTokens'
+export * from './instructions/repayVault'
 export * from './instructions/setHalted'
+export * from './instructions/transferVault'
+export * from './instructions/updateReferralAccount'
+export * from './instructions/updateReferralState'
 export * from './instructions/updateVaultType'
-
-export * from './HedgeDecimal'
-export * from './Constants'
-
-export * from './state/VaultAccount'
-export * from './state/VaultHistoryEvent'
-export * from './state/StakingPool'
-export * from './state/StakingPoolPosition'
+export * from './instructions/withdrawStakingPool'
+export * from './instructions/withdrawVault'
 export * from './state/LiquidationPoolEra'
 export * from './state/LiquidationPoolState'
 export * from './state/LiquidationPosition'
-
+export * from './state/StakingPool'
+export * from './state/StakingPoolPosition'
+export * from './state/VaultAccount'
+export * from './state/VaultHistoryEvent'
 export * from './utils/getLinkedListAccounts'
-export * from './idl/vault'
+
+

package/src/instructions/claimLiquidationPoolPosition.ts

@@ -1,23 +1,19 @@
 import { Program, Provider } from '@project-serum/anchor'
 import { ASSOCIATED_TOKEN_PROGRAM_ID, getOrCreateAssociatedTokenAccount, TOKEN_PROGRAM_ID } from '@solana/spl-token'
 import {
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
+  PublicKey, Signer,
   SystemProgram,
   SYSVAR_RENT_PUBKEY,
   Transaction,
-  TransactionInstruction,
+  TransactionInstruction
 } from '@solana/web3.js'
-import { parseAnchorErrors } from '../utils/Errors'
 import {
-  findAssociatedTokenAddress,
-  getVaultTypeAccountPublicKey,
-  getLiquidationPoolStatePublicKey,
-  getVaultSystemStatePublicKey,
+  findAssociatedTokenAddress, getLiquidationPoolStatePublicKey,
+  getVaultSystemStatePublicKey, getVaultTypeAccountPublicKey
 } 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 claimLiquidationPoolPosition(
   program: Program<Vault>,
@@ -27,7 +23,7 @@ export async function claimLiquidationPoolPosition(
   collateralType: string,
   overrideStartTime?: number
 ): Promise<PublicKey> {
-  const vaultTypeAccountPublicKey = await getVaultTypeAccountPublicKey(collateralType)
+  const vaultTypeAccountPublicKey = await getVaultTypeAccountPublicKey(program.programId, collateralType)
   const vaultTypeAccountInfo = await program.account.vaultType.fetch(vaultTypeAccountPublicKey)
   const collateralMintPublicKey = vaultTypeAccountInfo.collateralMint
 
@@ -62,23 +58,29 @@ export async function claimLiquidationPoolPositionInstruction(
   payerAssociatedTokenAccount: PublicKey,
   overrideStartTime?: number
 ): Promise<TransactionInstruction> {
+  const vaultSystemState = await getVaultSystemStatePublicKey(program.programId)
+  const poolStatePublicKey = await getLiquidationPoolStatePublicKey(program.programId)
+  const poolAssociatedTokenAccount = await findAssociatedTokenAddress(
+    program.programId,
+    poolStatePublicKey,
+    collateralMint
+  )
 
-  const vaultSystemState = await getVaultSystemStatePublicKey()
-  const poolStatePublicKey = await getLiquidationPoolStatePublicKey()
-  const poolAssociatedTokenAccount = await findAssociatedTokenAddress(poolStatePublicKey, collateralMint)
-
-  return await program.methods.claimLiquidationPoolPosition().accounts({
-    vaultSystemState: vaultSystemState,
-    poolState: poolStatePublicKey,
-    poolAssociatedTokenAccount: poolAssociatedTokenAccount,
-    vaultTypeAccount: vaultTypeAccount,
-    collateralMint: collateralMint,
-    poolPosition: poolPosition,
-    payer: payer,
-    payerAssociatedTokenAccount: payerAssociatedTokenAccount,
-    associatedTokenProgram: ASSOCIATED_TOKEN_PROGRAM_ID,
-    tokenProgram: TOKEN_PROGRAM_ID,
-    systemProgram: SystemProgram.programId,
-    rent: SYSVAR_RENT_PUBKEY,
-  }).instruction()
+  return await program.methods
+    .claimLiquidationPoolPosition()
+    .accounts({
+      vaultSystemState: vaultSystemState,
+      poolState: poolStatePublicKey,
+      poolAssociatedTokenAccount: poolAssociatedTokenAccount,
+      vaultTypeAccount: vaultTypeAccount,
+      collateralMint: collateralMint,
+      poolPosition: poolPosition,
+      payer: payer,
+      payerAssociatedTokenAccount: payerAssociatedTokenAccount,
+      associatedTokenProgram: ASSOCIATED_TOKEN_PROGRAM_ID,
+      tokenProgram: TOKEN_PROGRAM_ID,
+      systemProgram: SystemProgram.programId,
+      rent: SYSVAR_RENT_PUBKEY,
+    })
+    .instruction()
 }

package/src/instructions/claimStakingPoolPosition.ts

@@ -1,24 +1,19 @@
 import { Program, Provider } from '@project-serum/anchor'
 import { ASSOCIATED_TOKEN_PROGRAM_ID, getOrCreateAssociatedTokenAccount, TOKEN_PROGRAM_ID } from '@solana/spl-token'
 import {
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
+  PublicKey, Signer,
   SystemProgram,
   SYSVAR_RENT_PUBKEY,
   Transaction,
-  TransactionInstruction,
+  TransactionInstruction
 } from '@solana/web3.js'
-import { parseAnchorErrors } from '../utils/Errors'
 import {
-  getHedgeMintPublicKey,
-  findAssociatedTokenAddress,
-  getVaultTypeAccountPublicKey,
-  getPoolPublicKeyForMint,
-  getVaultSystemStatePublicKey,
+  getHedgeMintPublicKey, getPoolPublicKeyForMint,
+  getVaultSystemStatePublicKey, getVaultTypeAccountPublicKey
 } 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 claimStakingPoolPosition(
   program: Program<Vault>,
@@ -27,11 +22,11 @@ export async function claimStakingPoolPosition(
   payer: Signer,
   collateralType: string
 ): Promise<PublicKey> {
-  const vaultTypeAccount = await getVaultTypeAccountPublicKey(collateralType)
+  const vaultTypeAccount = await getVaultTypeAccountPublicKey(program.programId, collateralType)
   const vaultTypeAccountInfo = await program.account.vaultType.fetch(vaultTypeAccount)
   const collateralMint = vaultTypeAccountInfo.collateralMint
-  const stakedTokenMint = await getHedgeMintPublicKey()
-  const [feePool] = await getPoolPublicKeyForMint(stakedTokenMint)
+  const stakedTokenMint = await getHedgeMintPublicKey(program.programId)
+  const [feePool] = await getPoolPublicKeyForMint(program.programId, stakedTokenMint)
   const feePoolAssociatedTokenAccount = await getOrCreateAssociatedTokenAccount(
     provider.connection,
     payer,
@@ -74,7 +69,7 @@ export async function claimStakingPoolPositionInstruction(
   payer: PublicKey,
   payerAssociatedTokenAccount: PublicKey
 ): Promise<TransactionInstruction> {
-  const vaultSystemState = await getVaultSystemStatePublicKey()
+  const vaultSystemState = await getVaultSystemStatePublicKey(program.programId)
 
   return await program.methods
     .claimStakingPoolPosition()

package/src/instructions/closeClaimedLiquidationPoolPosition.ts

@@ -10,7 +10,9 @@ export async function closeClaimedLiquidationPoolPosition(
   poolPosition: PublicKey,
   payer: Signer
 ): Promise<PublicKey> {
-  const transaction = new Transaction().add(await closeClaimedLiquidationPoolPositionInstruction(program, poolPosition, payer.publicKey))
+  const transaction = new Transaction().add(
+    await closeClaimedLiquidationPoolPositionInstruction(program, poolPosition, payer.publicKey)
+  )
 
   await sendAndConfirmWithDebug(provider.connection, transaction, [payer]).catch(parseAnchorErrors)
   return poolPosition
@@ -25,7 +27,7 @@ export async function closeClaimedLiquidationPoolPositionInstruction(
     .closeClaimedLiquidationPoolPosition()
     .accounts({
       poolPosition: poolPosition,
-      payer: payer
+      payer: payer,
     })
     .instruction()
 }

package/src/instructions/closeLiquidationPoolPosition.ts

@@ -1,44 +1,43 @@
 import { BN, Program, Provider } from '@project-serum/anchor'
 import { ASSOCIATED_TOKEN_PROGRAM_ID, getOrCreateAssociatedTokenAccount, TOKEN_PROGRAM_ID } from '@solana/spl-token'
 import {
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
+  PublicKey, Signer,
   SystemProgram,
   SYSVAR_RENT_PUBKEY,
   Transaction,
-  TransactionInstruction,
+  TransactionInstruction
 } from '@solana/web3.js'
-import { parseAnchorErrors } from '../utils/Errors'
 import {
   getHedgeMintPublicKey,
   getLiquidationPoolStatePublicKey,
-  getLiquidationPoolUshAccountPublicKey,
-  getUshMintPublicKey,
-  getVaultSystemStatePublicKey,
+  getLiquidationPoolUshAccountPublicKey, getReferralAccountPublicKey, getReferralStatePublicKey, getUserReferralAccountPublicKey, getUshMintPublicKey,
+  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 closeLiquidationPoolPosition(
   program: Program<Vault>,
   provider: Provider,
   poolPosition: PublicKey,
   payer: Signer,
-  overrideStartTime?: number
+  overrideStartTime?: number,
+  referrer?: PublicKey
 ): Promise<PublicKey> {
-  const ushMintPublickey = await getUshMintPublicKey()
+  const ushMintPublickey = await getUshMintPublicKey(program.programId)
   const payerUshAccount = await getOrCreateAssociatedTokenAccount(
     provider.connection,
     payer,
     ushMintPublickey,
     payer.publicKey
   )
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
+  const referralStatePublicKey = await getReferralStatePublicKey(program.programId)
   const liquidationPositionAccount = await program.account.liquidationPosition.fetch(poolPosition)
   const poolEra = liquidationPositionAccount.era
 
-  const hedgeMint = await getHedgeMintPublicKey()
+  const hedgeMint = await getHedgeMintPublicKey(program.programId)
   const payerAssociatedHedgeAccount = await getOrCreateAssociatedTokenAccount(
     provider.connection,
     payer,
@@ -53,6 +52,16 @@ export async function closeLiquidationPoolPosition(
     true
   )
 
+  let referralAccountPublicKey = null
+  if (typeof referrer === 'undefined') {
+    referralAccountPublicKey = await getReferralAccountPublicKey(program.programId, vaultSystemStatePublicKey)
+  } else {
+    referralAccountPublicKey = await getReferralAccountPublicKey(program.programId, referrer)
+  }
+
+  // Derive the user referral account public key
+  const userReferralAccountPublicKey = await getUserReferralAccountPublicKey(program.programId, payer.publicKey)
+
   const transaction = new Transaction().add(
     await closeLiquidationPoolPositionInstruction(
       program,
@@ -62,6 +71,9 @@ export async function closeLiquidationPoolPosition(
       payerUshAccount.address,
       payerAssociatedHedgeAccount.address,
       communityAssociatedHedgeTokenAccount.address,
+      userReferralAccountPublicKey,
+      referralAccountPublicKey,
+      referralStatePublicKey,
       overrideStartTime
     )
   )
@@ -77,13 +89,16 @@ export async function closeLiquidationPoolPositionInstruction(
   payerUshAccount: PublicKey,
   payerAssociatedHedgeAccount: PublicKey,
   communityAssociatedHedgeTokenAccount: PublicKey,
+  userReferralAccountPublicKey: PublicKey,
+  referralAccountPublicKey: PublicKey,
+  referralStatePublicKey: PublicKey,
   overrideStartTime?: number
 ): Promise<TransactionInstruction> {
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
-  const ushMint = await getUshMintPublicKey()
-  const hedgeMint = await getHedgeMintPublicKey()
-  const poolUshAccount = await getLiquidationPoolUshAccountPublicKey()
-  const poolState = await getLiquidationPoolStatePublicKey()
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
+  const ushMint = await getUshMintPublicKey(program.programId)
+  const hedgeMint = await getHedgeMintPublicKey(program.programId)
+  const poolUshAccount = await getLiquidationPoolUshAccountPublicKey(program.programId)
+  const poolState = await getLiquidationPoolStatePublicKey(program.programId)
 
   return await program.methods
     .closeLiquidationPoolPosition(
@@ -101,6 +116,9 @@ export async function closeLiquidationPoolPositionInstruction(
       ushMint: ushMint,
       payerAssociatedHedgeAccount: payerAssociatedHedgeAccount,
       communityAssociatedHedgeTokenAccount: communityAssociatedHedgeTokenAccount,
+      userReferralAccount: userReferralAccountPublicKey,
+      referralAccount: referralAccountPublicKey,
+      referralState: referralStatePublicKey,
       associatedTokenProgram: ASSOCIATED_TOKEN_PROGRAM_ID,
       tokenProgram: TOKEN_PROGRAM_ID,
       systemProgram: SystemProgram.programId,

package/src/instructions/createReferralAccount.ts

@@ -0,0 +1,115 @@
+import { BN, Program, Provider } from '@project-serum/anchor'
+import { getOrCreateAssociatedTokenAccount, TOKEN_PROGRAM_ID } from '@solana/spl-token'
+import {
+  PublicKey, Signer,
+  SystemProgram,
+  SYSVAR_RENT_PUBKEY,
+  Transaction,
+  TransactionInstruction
+} from '@solana/web3.js'
+import {
+  getHedgeMintPublicKey, getReferralAccountPublicKey, getReferralStatePublicKey,
+  getUserReferralAccountPublicKey, getVaultSystemStatePublicKey
+} from '../Constants'
+
+import { Vault } from '../idl/vault'
+import { parseAnchorErrors } from '../utils/Errors'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
+
+/** @type {Function}  - Creates a new referral account.
+ * This checks the user has enough HDG in their wallet or staked and
+ * then allows them to create a referral account. If they meet the PSM referral threshold,
+ * they will be eligible for PSM cut.
+ * Params:
+ * - program: Program<Vault> : The program instance of Hedge Vault program
+ * - provider: Provider : Current connection
+ * - 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
+ * - overrideTime?: number
+ */
+export async function createReferralAccount(
+  program: Program<Vault>,
+  provider: Provider,
+  payer: Signer,
+  poolPosition: PublicKey,
+  overrideTime?: number
+): Promise<PublicKey> {
+  // setup transaction
+  const transaction = new Transaction()
+  const signers = [payer]
+  const referrer = payer.publicKey
+
+  // Setup public keys
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
+
+  // Find referral account
+  let referralAccountPublicKey = await getReferralAccountPublicKey(program.programId, referrer)
+
+  // Get the referral state PDA
+  const referallStatePublicKey = await getReferralStatePublicKey(program.programId)
+
+  // Get HDG mint public key
+  const hedgeMintPublicKey = await getHedgeMintPublicKey(program.programId)
+
+  // Get HDG account of signer
+  const payerHdgAssociatedTokenAccount = await getOrCreateAssociatedTokenAccount(
+    provider.connection,
+    payer,
+    hedgeMintPublicKey,
+    payer.publicKey
+  )
+
+  // Derive the user referral account public key
+  const userReferralAccountPublicKey = await getUserReferralAccountPublicKey(program.programId, payer.publicKey)
+
+  transaction.add(
+    await createReferralAccountInstruction(
+      program,
+      payer.publicKey,
+      poolPosition,
+      referralAccountPublicKey,
+      referallStatePublicKey,
+      hedgeMintPublicKey,
+      payerHdgAssociatedTokenAccount.address,
+      userReferralAccountPublicKey,
+      overrideTime
+    )
+  )
+
+  await sendAndConfirmWithDebug(provider.connection, transaction, signers).catch(parseAnchorErrors)
+  return referralAccountPublicKey
+}
+
+export async function createReferralAccountInstruction(
+  program: Program<Vault>,
+  payerPublicKey: PublicKey,
+  poolPositionPublicKey: PublicKey,
+  referralAccountPublicKey: PublicKey,
+  referralStatePublicKey: PublicKey,
+  hedgeMintPublicKey: PublicKey,
+  hdgAssociatedTokenAccountPublicKey: PublicKey,
+  userReferralAccountPublicKey: PublicKey,
+  overrideTime?: number
+): Promise<TransactionInstruction> {
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
+
+  return await program.methods
+    .createReferralAccount(
+      new BN(overrideTime ?? Math.floor(Date.now() / 1000)) // override override times
+    )
+    .accounts({
+      signer: payerPublicKey,
+      vaultSystemState: vaultSystemStatePublicKey,
+      referralState: referralStatePublicKey,
+      referralAccount: referralAccountPublicKey,
+      userReferralAccount: userReferralAccountPublicKey,
+      poolPosition: poolPositionPublicKey,
+      hedgeMint: hedgeMintPublicKey,
+      hdgAta: hdgAssociatedTokenAccountPublicKey,
+      systemProgram: SystemProgram.programId,
+      tokenProgram: TOKEN_PROGRAM_ID,
+      rent: SYSVAR_RENT_PUBKEY,
+    })
+    .instruction()
+}

package/src/instructions/createStakingPool.ts

@@ -1,23 +1,21 @@
 import { BN, Program, Provider } from '@project-serum/anchor'
 import { ASSOCIATED_TOKEN_PROGRAM_ID, TOKEN_PROGRAM_ID } from '@solana/spl-token'
 import {
-  PublicKey,
-  sendAndConfirmTransaction,
-  Signer,
+  PublicKey, Signer,
   SystemProgram,
   SYSVAR_RENT_PUBKEY,
   Transaction,
-  TransactionInstruction,
+  TransactionInstruction
 } from '@solana/web3.js'
-import { parseAnchorErrors } from '../utils/Errors'
 import {
   findAssociatedTokenAddress,
   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 createStakingPool(
   program: Program<Vault>,
@@ -31,7 +29,7 @@ export async function createStakingPool(
     await createStakingPoolInstruction(program, payer.publicKey, mintPublicKey, hedgeTokensToBeMinted, overrideStartTime)
   )
   await sendAndConfirmWithDebug(provider.connection, transaction, [payer]).catch(parseAnchorErrors)
-  const [poolPublickey] = await getPoolPublicKeyForMint(mintPublicKey)
+  const [poolPublickey] = await getPoolPublicKeyForMint(program.programId, mintPublicKey)
   return poolPublickey
 }
 
@@ -42,12 +40,12 @@ export async function createStakingPoolInstruction(
   hedgeTokensToBeMinted: number,
   overrideStartTime?: number
 ): Promise<TransactionInstruction> {
-  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey()
-  const ushMintPublickey = await getUshMintPublicKey()
-  const [poolPublickey, poolBump] = await getPoolPublicKeyForMint(mintPublicKey)
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
+  const ushMintPublickey = await getUshMintPublicKey(program.programId)
+  const [poolPublickey, poolBump] = await getPoolPublicKeyForMint(program.programId, mintPublicKey)
 
-  const poolAssociatedStakedTokenAccount = await findAssociatedTokenAddress(poolPublickey, mintPublicKey)
-  const poolAssociatedUshTokenAccount = await findAssociatedTokenAddress(poolPublickey, ushMintPublickey)
+  const poolAssociatedStakedTokenAccount = await findAssociatedTokenAddress(program.programId, poolPublickey, mintPublicKey)
+  const poolAssociatedUshTokenAccount = await findAssociatedTokenAddress(program.programId, poolPublickey, ushMintPublickey)
   return await program.methods
     .createStakingPool(poolBump, new BN(hedgeTokensToBeMinted), new BN(overrideStartTime ?? Math.floor(Date.now() / 1000)))
     .accounts({

package/src/instructions/createUserReferralAccount.ts

@@ -0,0 +1,75 @@
+import { BN, Program, Provider } from '@project-serum/anchor'
+import { TOKEN_PROGRAM_ID } from '@solana/spl-token'
+import {
+  PublicKey, Signer,
+  SystemProgram,
+  SYSVAR_RENT_PUBKEY,
+  Transaction,
+  TransactionInstruction
+} from '@solana/web3.js'
+import { getReferralAccountPublicKey, getUserReferralAccountPublicKey, getVaultSystemStatePublicKey } from '../Constants'
+
+import { Vault } from '../idl/vault'
+import { parseAnchorErrors } from '../utils/Errors'
+import sendAndConfirmWithDebug from '../utils/sendAndConfirmWithDebug'
+
+export async function createUserReferralAccount(
+  program: Program<Vault>,
+  provider: Provider,
+  payer: Signer,
+  referrer?: PublicKey,
+  overrideTime?: number
+): Promise<PublicKey> {
+  // setup transaction
+  const transaction = new Transaction()
+  const signers = [payer]
+
+  // Find referrer account
+  let referralAccountPublicKey = null
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
+  if (typeof referrer === 'undefined') {
+    referralAccountPublicKey = await getReferralAccountPublicKey(program.programId, vaultSystemStatePublicKey)
+  } else {
+    referralAccountPublicKey = await getReferralAccountPublicKey(program.programId, referrer)
+  }
+
+  // Derive the user referral account public key
+  const userReferralAccountPublicKey = await getUserReferralAccountPublicKey(program.programId, payer.publicKey)
+
+  transaction.add(
+    await createUserReferralAccountInstruction(
+      program,
+      payer.publicKey,
+      referralAccountPublicKey,
+      userReferralAccountPublicKey,
+      overrideTime
+    )
+  )
+
+  await sendAndConfirmWithDebug(provider.connection, transaction, signers).catch(parseAnchorErrors)
+  return userReferralAccountPublicKey
+}
+
+export async function createUserReferralAccountInstruction(
+  program: Program<Vault>,
+  payerPublicKey: PublicKey,
+  referralAccount: PublicKey,
+  userReferralAccount: PublicKey,
+  overrideTime?: number
+): Promise<TransactionInstruction> {
+  const vaultSystemStatePublicKey = await getVaultSystemStatePublicKey(program.programId)
+
+  return await program.methods
+    .createUserReferralAccount(
+      new BN(overrideTime ?? Math.floor(Date.now() / 1000)) // override override time
+    )
+    .accounts({
+      signer: payerPublicKey,
+      userReferralAccount: userReferralAccount,
+      referralAccount: referralAccount,
+      systemProgram: SystemProgram.programId,
+      tokenProgram: TOKEN_PROGRAM_ID,
+      rent: SYSVAR_RENT_PUBKEY,
+    })
+    .instruction()
+}