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

package/src/admin.ts

@@ -13,11 +13,14 @@ import { BN } from '@project-serum/anchor';
 import * as anchor from '@project-serum/anchor';
 import {
 	getClearingHouseStateAccountPublicKeyAndNonce,
-	getBankPublicKey,
-	getBankVaultPublicKey,
+	getSpotMarketPublicKey,
+	getSpotMarketVaultPublicKey,
 	getMarketPublicKey,
 	getInsuranceFundVaultPublicKey,
+	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';
@@ -71,7 +74,7 @@ export class Admin extends ClearingHouse {
 		return [initializeTxSig];
 	}
 
-	public async initializeBank(
+	public async initializeSpotMarket(
 		mint: PublicKey,
 		optimalUtilization: BN,
 		optimalRate: BN,
@@ -85,20 +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 spotMarketIndex = this.getStateAccount().numberOfSpotMarkets;
+		const spotMarket = await getSpotMarketPublicKey(
+			this.program.programId,
+			spotMarketIndex
+		);
 
-		const bankVault = await getBankVaultPublicKey(
+		const spotMarketVault = await getSpotMarketVaultPublicKey(
 			this.program.programId,
-			bankIndex
+			spotMarketIndex
 		);
 
 		const insuranceFundVault = await getInsuranceFundVaultPublicKey(
 			this.program.programId,
-			bankIndex
+			spotMarketIndex
 		);
 
-		const initializeTx = await this.program.transaction.initializeBank(
+		const initializeTx = await this.program.transaction.initializeSpotMarket(
 			optimalUtilization,
 			optimalRate,
 			maxRate,
@@ -113,11 +119,11 @@ export class Admin extends ClearingHouse {
 				accounts: {
 					admin: this.wallet.publicKey,
 					state: await this.getStatePublicKey(),
-					bank,
-					bankVault,
+					spotMarket,
+					spotMarketVault,
 					insuranceFundVault,
 					clearingHouseSigner: this.getSignerPublicKey(),
-					bankMint: mint,
+					spotMarketMint: mint,
 					oracle,
 					rent: SYSVAR_RENT_PUBKEY,
 					systemProgram: anchor.web3.SystemProgram.programId,
@@ -128,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,
@@ -137,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,
@@ -179,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({
@@ -191,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,
 				},
 			}
 		);
@@ -222,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,
@@ -251,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,
 				},
 			}
 		);
@@ -275,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: {
@@ -290,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
@@ -309,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
@@ -330,12 +378,12 @@ 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,
 				clearingHouseSigner: this.getSignerPublicKey(),
 				recipient: recipient,
@@ -353,14 +401,14 @@ 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,
+				spotMarket: spotMarket.pubkey,
+				spotMarketVault: spotMarket.vault,
 				clearingHouseSigner: this.getSignerPublicKey(),
 				recipient: recipient,
 				tokenProgram: TOKEN_PROGRAM_ID,
@@ -373,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: {
@@ -382,8 +430,8 @@ export class Admin extends ClearingHouse {
 				market: await getMarketPublicKey(this.program.programId, marketIndex),
 				insuranceVault: state.insuranceVault,
 				clearingHouseSigner: this.getSignerPublicKey(),
-				bank: bank.pubkey,
-				bankVault: bank.vault,
+				quoteSpotMarket: spotMarket.pubkey,
+				spotMarketVault: spotMarket.vault,
 				tokenProgram: TOKEN_PROGRAM_ID,
 			},
 		});
@@ -584,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,
@@ -615,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
+					),
 				},
 			}
 		);
@@ -696,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> {
@@ -783,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,
 				},
 			}
 		);
@@ -797,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) {