@drift-labs/sdk 0.2.0-master.24 → 0.2.0-master.26

package/src/admin.ts

@@ -13,13 +13,14 @@ import { BN } from '@project-serum/anchor';
 import * as anchor from '@project-serum/anchor';
 import {
 	getClearingHouseStateAccountPublicKeyAndNonce,
-	getBankVaultAuthorityPublicKey,
-	getBankPublicKey,
-	getBankVaultPublicKey,
+	getSpotMarketPublicKey,
+	getSpotMarketVaultPublicKey,
 	getMarketPublicKey,
 	getInsuranceFundVaultPublicKey,
-	getInsuranceFundVaultAuthorityPublicKey,
+	getSerumOpenOrdersPublicKey,
+	getSerumFulfillmentConfigPublicKey,
 } from './addresses/pda';
+import { squareRootBN } from './math/utils';
 import { TOKEN_PROGRAM_ID } from '@solana/spl-token';
 import { ClearingHouse } from './clearingHouse';
 import { PEG_PRECISION, ZERO } from './constants/numericConstants';
@@ -38,20 +39,16 @@ export class Admin extends ClearingHouse {
 			throw new Error('Clearing house already initialized');
 		}
 
+		const [clearingHouseStatePublicKey] =
+			await getClearingHouseStateAccountPublicKeyAndNonce(
+				this.program.programId
+			);
+
 		const [insuranceVaultPublicKey] = await PublicKey.findProgramAddress(
 			[Buffer.from(anchor.utils.bytes.utf8.encode('insurance_vault'))],
 			this.program.programId
 		);
 
-		const [insuranceVaultAuthority] = await PublicKey.findProgramAddress(
-			[insuranceVaultPublicKey.toBuffer()],
-			this.program.programId
-		);
-
-		const [clearingHouseStatePublicKey] =
-			await getClearingHouseStateAccountPublicKeyAndNonce(
-				this.program.programId
-			);
 		const initializeTx = await this.program.transaction.initialize(
 			adminControlsPrices,
 			{
@@ -59,9 +56,9 @@ export class Admin extends ClearingHouse {
 					admin: this.wallet.publicKey,
 					state: clearingHouseStatePublicKey,
 					quoteAssetMint: usdcMint,
-					insuranceVault: insuranceVaultPublicKey,
-					insuranceVaultAuthority: insuranceVaultAuthority,
 					rent: SYSVAR_RENT_PUBKEY,
+					insuranceVault: insuranceVaultPublicKey,
+					clearingHouseSigner: this.getSignerPublicKey(),
 					systemProgram: anchor.web3.SystemProgram.programId,
 					tokenProgram: TOKEN_PROGRAM_ID,
 				},
@@ -77,7 +74,7 @@ export class Admin extends ClearingHouse {
 		return [initializeTxSig];
 	}
 
-	public async initializeBank(
+	public async initializeSpotMarket(
 		mint: PublicKey,
 		optimalUtilization: BN,
 		optimalRate: BN,
@@ -91,31 +88,23 @@ export class Admin extends ClearingHouse {
 		imfFactor = new BN(0),
 		liquidationFee = ZERO
 	): Promise<TransactionSignature> {
-		const bankIndex = this.getStateAccount().numberOfBanks;
-		const bank = await getBankPublicKey(this.program.programId, bankIndex);
-
-		const bankVault = await getBankVaultPublicKey(
+		const spotMarketIndex = this.getStateAccount().numberOfSpotMarkets;
+		const spotMarket = await getSpotMarketPublicKey(
 			this.program.programId,
-			bankIndex
+			spotMarketIndex
 		);
 
-		const bankVaultAuthority = await getBankVaultAuthorityPublicKey(
+		const spotMarketVault = await getSpotMarketVaultPublicKey(
 			this.program.programId,
-			bankIndex
+			spotMarketIndex
 		);
 
 		const insuranceFundVault = await getInsuranceFundVaultPublicKey(
 			this.program.programId,
-			bankIndex
+			spotMarketIndex
 		);
 
-		const insuranceFundVaultAuthority =
-			await getInsuranceFundVaultAuthorityPublicKey(
-				this.program.programId,
-				bankIndex
-			);
-
-		const initializeTx = await this.program.transaction.initializeBank(
+		const initializeTx = await this.program.transaction.initializeSpotMarket(
 			optimalUtilization,
 			optimalRate,
 			maxRate,
@@ -130,12 +119,11 @@ export class Admin extends ClearingHouse {
 				accounts: {
 					admin: this.wallet.publicKey,
 					state: await this.getStatePublicKey(),
-					bank,
-					bankVault,
-					bankVaultAuthority,
+					spotMarket,
+					spotMarketVault,
 					insuranceFundVault,
-					insuranceFundVaultAuthority,
-					bankMint: mint,
+					clearingHouseSigner: this.getSignerPublicKey(),
+					spotMarketMint: mint,
 					oracle,
 					rent: SYSVAR_RENT_PUBKEY,
 					systemProgram: anchor.web3.SystemProgram.programId,
@@ -146,7 +134,7 @@ export class Admin extends ClearingHouse {
 
 		const { txSig } = await this.txSender.send(initializeTx, [], this.opts);
 
-		await this.accountSubscriber.addBank(bankIndex);
+		await this.accountSubscriber.addSpotMarket(spotMarketIndex);
 		await this.accountSubscriber.addOracle({
 			source: oracleSource,
 			publicKey: oracle,
@@ -155,6 +143,41 @@ export class Admin extends ClearingHouse {
 		return txSig;
 	}
 
+	public async initializeSerumFulfillmentConfig(
+		marketIndex: BN,
+		serumMarket: PublicKey,
+		serumProgram: PublicKey
+	): Promise<TransactionSignature> {
+		const serumOpenOrders = getSerumOpenOrdersPublicKey(
+			this.program.programId,
+			serumMarket
+		);
+
+		const serumFulfillmentConfig = getSerumFulfillmentConfigPublicKey(
+			this.program.programId,
+			serumMarket
+		);
+
+		return await this.program.rpc.initializeSerumFulfillmentConfig(
+			marketIndex,
+			{
+				accounts: {
+					admin: this.wallet.publicKey,
+					state: await this.getStatePublicKey(),
+					baseSpotMarket: this.getSpotMarketAccount(marketIndex).pubkey,
+					quoteSpotMarket: this.getQuoteSpotMarketAccount().pubkey,
+					clearingHouseSigner: this.getSignerPublicKey(),
+					serumProgram,
+					serumMarket,
+					serumOpenOrders,
+					rent: SYSVAR_RENT_PUBKEY,
+					systemProgram: anchor.web3.SystemProgram.programId,
+					serumFulfillmentConfig,
+				},
+			}
+		);
+	}
+
 	public async initializeMarket(
 		priceOracle: PublicKey,
 		baseAssetReserve: BN,
@@ -197,7 +220,7 @@ export class Admin extends ClearingHouse {
 			this.opts
 		);
 
-		await this.accountSubscriber.addMarket(
+		await this.accountSubscriber.addPerpMarket(
 			this.getStateAccount().numberOfMarkets
 		);
 		await this.accountSubscriber.addOracle({
@@ -209,23 +232,29 @@ export class Admin extends ClearingHouse {
 	}
 
 	public async moveAmmPrice(
+		marketIndex: BN,
 		baseAssetReserve: BN,
 		quoteAssetReserve: BN,
-		marketIndex: BN
+		sqrtK?: BN
 	): Promise<TransactionSignature> {
 		const marketPublicKey = await getMarketPublicKey(
 			this.program.programId,
 			marketIndex
 		);
 
+		if (sqrtK == undefined) {
+			sqrtK = squareRootBN(baseAssetReserve.mul(quoteAssetReserve));
+		}
+
 		return await this.program.rpc.moveAmmPrice(
 			baseAssetReserve,
 			quoteAssetReserve,
+			sqrtK,
 			{
 				accounts: {
 					state: await this.getStatePublicKey(),
 					admin: this.wallet.publicKey,
-					market: marketPublicKey,
+					perpMarket: marketPublicKey,
 				},
 			}
 		);
@@ -240,16 +269,16 @@ export class Admin extends ClearingHouse {
 				state: await this.getStatePublicKey(),
 				admin: this.wallet.publicKey,
 				market: await getMarketPublicKey(this.program.programId, marketIndex),
-				oracle: this.getMarketAccount(marketIndex).amm.oracle,
+				oracle: this.getPerpMarketAccount(marketIndex).amm.oracle,
 			},
 		});
 	}
 
 	public async moveAmmToPrice(
-		marketIndex: BN,
+		perpMarketIndex: BN,
 		targetPrice: BN
 	): Promise<TransactionSignature> {
-		const market = this.getMarketAccount(marketIndex);
+		const market = this.getPerpMarketAccount(perpMarketIndex);
 
 		const [direction, tradeSize, _] = calculateTargetPriceTrade(
 			market,
@@ -269,17 +298,18 @@ export class Admin extends ClearingHouse {
 
 		const marketPublicKey = await getMarketPublicKey(
 			this.program.programId,
-			marketIndex
+			perpMarketIndex
 		);
 
 		return await this.program.rpc.moveAmmPrice(
 			newBaseAssetAmount,
 			newQuoteAssetAmount,
+			market.amm.sqrtK,
 			{
 				accounts: {
 					state: await this.getStatePublicKey(),
 					admin: this.wallet.publicKey,
-					market: marketPublicKey,
+					perpMarket: marketPublicKey,
 				},
 			}
 		);
@@ -293,7 +323,7 @@ export class Admin extends ClearingHouse {
 			this.program.programId,
 			marketIndex
 		);
-		const ammData = this.getMarketAccount(marketIndex).amm;
+		const ammData = this.getPerpMarketAccount(marketIndex).amm;
 
 		return await this.program.rpc.repegAmmCurve(newPeg, {
 			accounts: {
@@ -308,7 +338,7 @@ export class Admin extends ClearingHouse {
 	public async updateAmmOracleTwap(
 		marketIndex: BN
 	): Promise<TransactionSignature> {
-		const ammData = this.getMarketAccount(marketIndex).amm;
+		const ammData = this.getPerpMarketAccount(marketIndex).amm;
 		const marketPublicKey = await getMarketPublicKey(
 			this.program.programId,
 			marketIndex
@@ -327,7 +357,7 @@ export class Admin extends ClearingHouse {
 	public async resetAmmOracleTwap(
 		marketIndex: BN
 	): Promise<TransactionSignature> {
-		const ammData = this.getMarketAccount(marketIndex).amm;
+		const ammData = this.getPerpMarketAccount(marketIndex).amm;
 		const marketPublicKey = await getMarketPublicKey(
 			this.program.programId,
 			marketIndex
@@ -348,14 +378,14 @@ export class Admin extends ClearingHouse {
 		recipient: PublicKey
 	): Promise<TransactionSignature> {
 		const state = await this.getStateAccount();
-		const bank = this.getQuoteAssetBankAccount();
+		const spotMarket = this.getQuoteSpotMarketAccount();
 		return await this.program.rpc.withdrawFromInsuranceVault(amount, {
 			accounts: {
 				admin: this.wallet.publicKey,
 				state: await this.getStatePublicKey(),
-				bank: bank.pubkey,
+				spotMarket: spotMarket.pubkey,
 				insuranceVault: state.insuranceVault,
-				insuranceVaultAuthority: state.insuranceVaultAuthority,
+				clearingHouseSigner: this.getSignerPublicKey(),
 				recipient: recipient,
 				tokenProgram: TOKEN_PROGRAM_ID,
 			},
@@ -371,15 +401,15 @@ export class Admin extends ClearingHouse {
 			this.program.programId,
 			marketIndex
 		);
-		const bank = this.getQuoteAssetBankAccount();
+		const spotMarket = this.getQuoteSpotMarketAccount();
 		return await this.program.rpc.withdrawFromMarketToInsuranceVault(amount, {
 			accounts: {
 				admin: this.wallet.publicKey,
 				state: await this.getStatePublicKey(),
 				market: marketPublicKey,
-				bank: bank.pubkey,
-				bankVault: bank.vault,
-				bankVaultAuthority: bank.vaultAuthority,
+				spotMarket: spotMarket.pubkey,
+				spotMarketVault: spotMarket.vault,
+				clearingHouseSigner: this.getSignerPublicKey(),
 				recipient: recipient,
 				tokenProgram: TOKEN_PROGRAM_ID,
 			},
@@ -391,7 +421,7 @@ export class Admin extends ClearingHouse {
 		amount: BN
 	): Promise<TransactionSignature> {
 		const state = await this.getStateAccount();
-		const bank = this.getQuoteAssetBankAccount();
+		const spotMarket = this.getQuoteSpotMarketAccount();
 
 		return await this.program.rpc.withdrawFromInsuranceVaultToMarket(amount, {
 			accounts: {
@@ -399,10 +429,9 @@ export class Admin extends ClearingHouse {
 				state: await this.getStatePublicKey(),
 				market: await getMarketPublicKey(this.program.programId, marketIndex),
 				insuranceVault: state.insuranceVault,
-				insuranceVaultAuthority: state.insuranceVaultAuthority,
-				bank: bank.pubkey,
-				bankVault: bank.vault,
-				bankVaultAuthority: bank.vaultAuthority,
+				clearingHouseSigner: this.getSignerPublicKey(),
+				quoteSpotMarket: spotMarket.pubkey,
+				spotMarketVault: spotMarket.vault,
 				tokenProgram: TOKEN_PROGRAM_ID,
 			},
 		});
@@ -603,30 +632,33 @@ export class Admin extends ClearingHouse {
 		});
 	}
 
-	public async updateBankWithdrawGuardThreshold(
-		bankIndex: BN,
+	public async updateWithdrawGuardThreshold(
+		marketIndex: BN,
 		withdrawGuardThreshold: BN
 	): Promise<TransactionSignature> {
-		return await this.program.rpc.updateBankWithdrawGuardThreshold(
+		return await this.program.rpc.updateWithdrawGuardThreshold(
 			withdrawGuardThreshold,
 			{
 				accounts: {
 					admin: this.wallet.publicKey,
 					state: await this.getStatePublicKey(),
-					bank: await getBankPublicKey(this.program.programId, bankIndex),
+					spotMarket: await getSpotMarketPublicKey(
+						this.program.programId,
+						marketIndex
+					),
 				},
 			}
 		);
 	}
 
-	public async updateBankIfFactor(
-		bankIndex: BN,
+	public async updateSpotMarketIfFactor(
+		marketIndex: BN,
 		userIfFactor: BN,
 		totalIfFactor: BN,
 		liquidationIfFactor: BN
 	): Promise<TransactionSignature> {
-		return await this.program.rpc.updateBankIfFactor(
-			bankIndex,
+		return await this.program.rpc.updateSpotMarketIfFactor(
+			marketIndex,
 			userIfFactor,
 			totalIfFactor,
 			liquidationIfFactor,
@@ -634,23 +666,29 @@ export class Admin extends ClearingHouse {
 				accounts: {
 					admin: this.wallet.publicKey,
 					state: await this.getStatePublicKey(),
-					bank: await getBankPublicKey(this.program.programId, bankIndex),
+					spotMarket: await getSpotMarketPublicKey(
+						this.program.programId,
+						marketIndex
+					),
 				},
 			}
 		);
 	}
 
-	public async updateBankInsuranceWithdrawEscrowPeriod(
-		bankIndex: BN,
+	public async updateInsuranceWithdrawEscrowPeriod(
+		marketIndex: BN,
 		insuranceWithdrawEscrowPeriod: BN
 	): Promise<TransactionSignature> {
-		return await this.program.rpc.updateBankInsuranceWithdrawEscrowPeriod(
+		return await this.program.rpc.updateInsuranceWithdrawEscrowPeriod(
 			insuranceWithdrawEscrowPeriod,
 			{
 				accounts: {
 					admin: this.wallet.publicKey,
 					state: await this.getStatePublicKey(),
-					bank: await getBankPublicKey(this.program.programId, bankIndex),
+					spotMarket: await getSpotMarketPublicKey(
+						this.program.programId,
+						marketIndex
+					),
 				},
 			}
 		);
@@ -715,6 +753,22 @@ export class Admin extends ClearingHouse {
 		);
 	}
 
+	public async updateMarketExpiry(
+		perpMarketIndex: BN,
+		expiryTs: BN
+	): Promise<TransactionSignature> {
+		return await this.program.rpc.updateMarketExpiry(expiryTs, {
+			accounts: {
+				admin: this.wallet.publicKey,
+				state: await this.getStatePublicKey(),
+				perpMarket: await getMarketPublicKey(
+					this.program.programId,
+					perpMarketIndex
+				),
+			},
+		});
+	}
+
 	public async updateWhitelistMint(
 		whitelistMint?: PublicKey
 	): Promise<TransactionSignature> {
@@ -802,7 +856,7 @@ export class Admin extends ClearingHouse {
 				accounts: {
 					admin: this.wallet.publicKey,
 					state: await this.getStatePublicKey(),
-					market: this.getMarketAccount(marketIndex).pubkey,
+					market: this.getPerpMarketAccount(marketIndex).pubkey,
 				},
 			}
 		);
@@ -816,8 +870,28 @@ export class Admin extends ClearingHouse {
 			accounts: {
 				admin: this.wallet.publicKey,
 				state: await this.getStatePublicKey(),
-				market: this.getMarketAccount(marketIndex).pubkey,
+				market: this.getPerpMarketAccount(marketIndex).pubkey,
 			},
 		});
 	}
+
+	public async updateMarketMaxImbalances(
+		marketIndex: BN,
+		unrealizedMaxImbalance: BN,
+		maxRevenueWithdrawPerPeriod: BN,
+		quoteMaxInsurance: BN
+	): Promise<TransactionSignature> {
+		return await this.program.rpc.updateMarketMaxImbalances(
+			unrealizedMaxImbalance,
+			maxRevenueWithdrawPerPeriod,
+			quoteMaxInsurance,
+			{
+				accounts: {
+					admin: this.wallet.publicKey,
+					state: await this.getStatePublicKey(),
+					market: await getMarketPublicKey(this.program.programId, marketIndex),
+				},
+			}
+		);
+	}
 }

package/src/assert/assert.js

@@ -1,5 +1,5 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
+exports.__esModule = true;
 exports.assert = void 0;
 function assert(condition, error) {
     if (!condition) {