@drift-labs/sdk 2.97.0-beta.2 → 2.97.0-beta.21

package/src/idl/switchboard_on_demand_30.json

@@ -443,10 +443,14 @@
           }
         },
         {
-          "name": "queue"
+          "name": "queue",
+          "relations": [
+            "oracle"
+          ]
         },
         {
-          "name": "authority"
+          "name": "authority",
+          "signer": true
         },
         {
           "name": "program_state",
@@ -503,6 +507,9 @@
         },
         {
           "name": "stake_pool"
+        },
+        {
+          "name": "delegation_group"
         }
       ],
       "args": [
@@ -955,10 +962,7 @@
         },
         {
           "name": "authority",
-          "signer": true,
-          "relations": [
-            "queue"
-          ]
+          "signer": true
         },
         {
           "name": "program_authority"
@@ -1038,6 +1042,13 @@
         {
           "name": "oracle",
           "writable": true
+        },
+        {
+          "name": "authority",
+          "signer": true
+        },
+        {
+          "name": "state"
         }
       ],
       "args": [
@@ -1294,10 +1305,7 @@
         },
         {
           "name": "authority",
-          "signer": true,
-          "relations": [
-            "queue"
-          ]
+          "signer": true
         },
         {
           "name": "program_authority"
@@ -1336,10 +1344,7 @@
         },
         {
           "name": "authority",
-          "signer": true,
-          "relations": [
-            "queue"
-          ]
+          "signer": true
         },
         {
           "name": "state"
@@ -1387,6 +1392,13 @@
         {
           "name": "recent_slothashes",
           "address": "SysvarS1otHashes111111111111111111111111111"
+        },
+        {
+          "name": "authority",
+          "signer": true,
+          "relations": [
+            "randomness"
+          ]
         }
       ],
       "args": [
@@ -1860,6 +1872,19 @@
     }
   ],
   "events": [
+    {
+      "name": "CostWhitelistEvent",
+      "discriminator": [
+        56,
+        107,
+        191,
+        127,
+        116,
+        6,
+        138,
+        149
+      ]
+    },
     {
       "name": "GarbageCollectionEvent",
       "discriminator": [
@@ -2235,9 +2260,84 @@
     {
       "code": 6047,
       "name": "InvalidInstruction"
+    },
+    {
+      "code": 6048,
+      "name": "OracleAlreadyVerified"
+    },
+    {
+      "code": 6049,
+      "name": "GuardianNotVerified"
+    },
+    {
+      "code": 6050,
+      "name": "InvalidConstraint"
+    },
+    {
+      "code": 6051,
+      "name": "InvalidDelegationGroup"
+    },
+    {
+      "code": 6052,
+      "name": "OracleKeyNotFound"
+    },
+    {
+      "code": 6053,
+      "name": "GuardianReregisterAttempt"
     }
   ],
   "types": [
+    {
+      "name": "CompactResult",
+      "serialization": "bytemuck",
+      "repr": {
+        "kind": "c"
+      },
+      "type": {
+        "kind": "struct",
+        "fields": [
+          {
+            "name": "std_dev",
+            "docs": [
+              "The standard deviation of the submissions needed for quorom size"
+            ],
+            "type": "f32"
+          },
+          {
+            "name": "mean",
+            "docs": [
+              "The mean of the submissions needed for quorom size"
+            ],
+            "type": "f32"
+          },
+          {
+            "name": "slot",
+            "docs": [
+              "The slot at which this value was signed."
+            ],
+            "type": "u64"
+          }
+        ]
+      }
+    },
+    {
+      "name": "CostWhitelistEvent",
+      "type": {
+        "kind": "struct",
+        "fields": [
+          {
+            "name": "feeds",
+            "type": {
+              "vec": "pubkey"
+            }
+          },
+          {
+            "name": "reward",
+            "type": "u32"
+          }
+        ]
+      }
+    },
     {
       "name": "CurrentResult",
       "serialization": "bytemuck",
@@ -2384,7 +2484,7 @@
             }
           },
           {
-            "name": "idx",
+            "name": "_reserved1",
             "type": "u32"
           },
           {
@@ -2437,12 +2537,7 @@
       "name": "GuardianUnregisterParams",
       "type": {
         "kind": "struct",
-        "fields": [
-          {
-            "name": "idx",
-            "type": "u32"
-          }
-        ]
+        "fields": []
       }
     },
     {
@@ -2948,7 +3043,7 @@
         "kind": "struct",
         "fields": [
           {
-            "name": "recent_slot",
+            "name": "_reserved1",
             "type": "u64"
           }
         ]
@@ -3068,14 +3163,18 @@
             }
           },
           {
-            "name": "_padding1",
+            "name": "padding1",
             "type": {
               "array": [
                 "u8",
-                3
+                2
               ]
             }
           },
+          {
+            "name": "historical_result_idx",
+            "type": "u8"
+          },
           {
             "name": "min_sample_size",
             "type": "u8"
@@ -3089,7 +3188,7 @@
             "type": "u64"
           },
           {
-            "name": "ipfs_hash",
+            "name": "_reserved1",
             "type": {
               "array": [
                 "u8",
@@ -3110,38 +3209,51 @@
             "type": "u32"
           },
           {
-            "name": "_ebuf4",
+            "name": "padding2",
             "type": {
               "array": [
                 "u8",
-                20
+                12
               ]
             }
           },
           {
-            "name": "_ebuf3",
+            "name": "historical_results",
+            "type": {
+              "array": [
+                {
+                  "defined": {
+                    "name": "CompactResult"
+                  }
+                },
+                32
+              ]
+            }
+          },
+          {
+            "name": "_ebuf4",
             "type": {
               "array": [
                 "u8",
-                24
+                8
               ]
             }
           },
           {
-            "name": "_ebuf2",
+            "name": "_ebuf3",
             "type": {
               "array": [
                 "u8",
-                256
+                24
               ]
             }
           },
           {
-            "name": "_ebuf1",
+            "name": "_ebuf2",
             "type": {
               "array": [
                 "u8",
-                512
+                256
               ]
             }
           }
@@ -4237,12 +4349,43 @@
             "name": "lut_slot",
             "type": "u64"
           },
+          {
+            "name": "base_reward",
+            "type": "u32"
+          },
+          {
+            "name": "_ebuf6",
+            "type": {
+              "array": [
+                "u8",
+                28
+              ]
+            }
+          },
+          {
+            "name": "_ebuf5",
+            "type": {
+              "array": [
+                "u8",
+                32
+              ]
+            }
+          },
+          {
+            "name": "_ebuf4",
+            "type": {
+              "array": [
+                "u8",
+                64
+              ]
+            }
+          },
           {
             "name": "_ebuf3",
             "type": {
               "array": [
                 "u8",
-                256
+                128
               ]
             }
           },
@@ -4256,11 +4399,14 @@
             }
           },
           {
-            "name": "_ebuf1",
+            "name": "cost_whitelist",
+            "docs": [
+              "Cost whitelist by authority"
+            ],
             "type": {
               "array": [
-                "u8",
-                1024
+                "pubkey",
+                32
               ]
             }
           }
@@ -4346,6 +4492,18 @@
           {
             "name": "subsidy_amount",
             "type": "u32"
+          },
+          {
+            "name": "base_reward",
+            "type": "u32"
+          },
+          {
+            "name": "add_cost_wl",
+            "type": "pubkey"
+          },
+          {
+            "name": "rm_cost_wl",
+            "type": "pubkey"
           }
         ]
       }

package/src/math/margin.ts

@@ -187,20 +187,20 @@ export function calculateWorstCasePerpLiabilityValue(
 
 export function calculatePerpLiabilityValue(
 	baseAssetAmount: BN,
-	oraclePrice: BN,
+	price: BN,
 	isPredictionMarket: boolean
 ): BN {
 	if (isPredictionMarket) {
 		if (baseAssetAmount.gt(ZERO)) {
-			return baseAssetAmount.mul(oraclePrice).div(BASE_PRECISION);
+			return baseAssetAmount.mul(price).div(BASE_PRECISION);
 		} else {
 			return baseAssetAmount
 				.abs()
-				.mul(MAX_PREDICTION_PRICE.sub(oraclePrice))
+				.mul(MAX_PREDICTION_PRICE.sub(price))
 				.div(BASE_PRECISION);
 		}
 	} else {
-		return baseAssetAmount.abs().mul(oraclePrice).div(BASE_PRECISION);
+		return baseAssetAmount.abs().mul(price).div(BASE_PRECISION);
 	}
 }
 
@@ -214,16 +214,19 @@ export function calculateMarginUSDCRequiredForTrade(
 	driftClient: DriftClient,
 	targetMarketIndex: number,
 	baseSize: BN,
-	userMaxMarginRatio?: number
+	userMaxMarginRatio?: number,
+	userHighLeverageMode?: boolean,
+	entryPrice?: BN
 ): BN {
 	const targetMarket = driftClient.getPerpMarketAccount(targetMarketIndex);
-	const oracleData = driftClient.getOracleDataForPerpMarket(
-		targetMarket.marketIndex
-	);
+
+	const price =
+		entryPrice ??
+		driftClient.getOracleDataForPerpMarket(targetMarket.marketIndex).price;
 
 	const perpLiabilityValue = calculatePerpLiabilityValue(
 		baseSize,
-		oracleData.price,
+		price,
 		isVariant(targetMarket.contractType, 'prediction')
 	);
 
@@ -232,7 +235,8 @@ export function calculateMarginUSDCRequiredForTrade(
 			targetMarket,
 			baseSize.abs(),
 			'Initial',
-			userMaxMarginRatio
+			userMaxMarginRatio,
+			userHighLeverageMode
 		)
 	)
 		.mul(perpLiabilityValue)
@@ -251,13 +255,17 @@ export function calculateCollateralDepositRequiredForTrade(
 	targetMarketIndex: number,
 	baseSize: BN,
 	collateralIndex: number,
-	userMaxMarginRatio?: number
+	userMaxMarginRatio?: number,
+	userHighLeverageMode?: boolean,
+	estEntryPrice?: BN
 ): BN {
 	const marginRequiredUsdc = calculateMarginUSDCRequiredForTrade(
 		driftClient,
 		targetMarketIndex,
 		baseSize,
-		userMaxMarginRatio
+		userMaxMarginRatio,
+		userHighLeverageMode,
+		estEntryPrice
 	);
 
 	const collateralMarket = driftClient.getSpotMarketAccount(collateralIndex);

package/src/math/market.ts

@@ -130,8 +130,24 @@ export function calculateMarketMarginRatio(
 	market: PerpMarketAccount,
 	size: BN,
 	marginCategory: MarginCategory,
-	customMarginRatio = 0
+	customMarginRatio = 0,
+	userHighLeverageMode = false
 ): number {
+	let marginRationInitial;
+	let marginRatioMaintenance;
+
+	if (
+		userHighLeverageMode &&
+		market.highLeverageMarginRatioInitial > 0 &&
+		market.highLeverageMarginRatioMaintenance
+	) {
+		marginRationInitial = market.highLeverageMarginRatioInitial;
+		marginRatioMaintenance = market.highLeverageMarginRatioMaintenance;
+	} else {
+		marginRationInitial = market.marginRatioInitial;
+		marginRatioMaintenance = market.marginRatioMaintenance;
+	}
+
 	let marginRatio;
 	switch (marginCategory) {
 		case 'Initial': {
@@ -140,7 +156,7 @@ export function calculateMarketMarginRatio(
 				calculateSizePremiumLiabilityWeight(
 					size,
 					new BN(market.imfFactor),
-					new BN(market.marginRatioInitial),
+					new BN(marginRationInitial),
 					MARGIN_PRECISION
 				).toNumber(),
 				customMarginRatio
@@ -151,7 +167,7 @@ export function calculateMarketMarginRatio(
 			marginRatio = calculateSizePremiumLiabilityWeight(
 				size,
 				new BN(market.imfFactor),
-				new BN(market.marginRatioMaintenance),
+				new BN(marginRatioMaintenance),
 				MARGIN_PRECISION
 			).toNumber();
 			break;

package/src/types.ts

@@ -68,6 +68,11 @@ export enum UserStatus {
 	ADVANCED_LP = 8,
 }
 
+export class MarginMode {
+	static readonly DEFAULT = { default: {} };
+	static readonly HIGH_LEVERAGE = { highLeverage: {} };
+}
+
 export class ContractType {
 	static readonly PERPETUAL = { perpetual: {} };
 	static readonly FUTURE = { future: {} };
@@ -661,6 +666,9 @@ export type PerpMarketAccount = {
 	fuelBoostTaker: number;
 	fuelBoostMaker: number;
 	fuelBoostPosition: number;
+
+	highLeverageMarginRatioInitial: number;
+	highLeverageMarginRatioMaintenance: number;
 };
 
 export type HistoricalOracleData = {
@@ -906,7 +914,7 @@ export type UserStatsAccount = {
 		current_epoch_referrer_reward: BN;
 	};
 	referrer: PublicKey;
-	isReferrer: boolean;
+	referrerStatus: boolean;
 	authority: PublicKey;
 	ifStakedQuoteAssetAmount: BN;
 
@@ -950,6 +958,7 @@ export type UserAccount = {
 	openAuctions: number;
 	hasOpenAuction: boolean;
 	lastFuelBonusUpdateTs: number;
+	marginMode: MarginMode;
 };
 
 export type SpotPosition = {
@@ -1093,6 +1102,11 @@ export type ReferrerInfo = {
 	referrerStats: PublicKey;
 };
 
+export enum ReferrerStatus {
+	IsReferrer = 1,
+	IsReferred = 2,
+}
+
 export enum PlaceAndTakeOrderSuccessCondition {
 	PartialFill = 1,
 	FullFill = 2,
@@ -1301,3 +1315,9 @@ export type SignedTxData = {
 	lastValidBlockHeight?: number;
 	blockHash: string;
 };
+
+export type HighLeverageModeConfig = {
+	maxUsers: number;
+	currentUsers: number;
+	reduceOnly: boolean;
+};

package/src/user.ts

@@ -676,7 +676,8 @@ export class User {
 			this.driftClient.getPerpMarketAccount(marketIndex),
 			baseAssetAmount,
 			'Initial',
-			this.getUserAccount().maxMarginRatio
+			this.getUserAccount().maxMarginRatio,
+			this.isHighLeverageMode()
 		);
 
 		return freeCollateral.mul(MARGIN_PRECISION).div(new BN(marginRatio));
@@ -1507,7 +1508,8 @@ export class User {
 					market,
 					baseAssetAmount.abs(),
 					marginCategory,
-					this.getUserAccount().maxMarginRatio
+					this.getUserAccount().maxMarginRatio,
+					this.isHighLeverageMode()
 				)
 			);
 
@@ -1972,7 +1974,8 @@ export class User {
 			market,
 			maxSize,
 			marginCategory,
-			this.getUserAccount().maxMarginRatio
+			this.getUserAccount().maxMarginRatio,
+			this.isHighLeverageMode()
 		);
 
 		// use more fesible size since imf factor activated
@@ -1993,7 +1996,8 @@ export class User {
 				market,
 				targetSize,
 				marginCategory,
-				this.getUserAccount().maxMarginRatio
+				this.getUserAccount().maxMarginRatio,
+				this.isHighLeverageMode()
 			);
 			attempts += 1;
 		}
@@ -2154,6 +2158,10 @@ export class User {
 		return (this.getUserAccount().status & UserStatus.BANKRUPT) > 0;
 	}
 
+	public isHighLeverageMode(): boolean {
+		return isVariant(this.getUserAccount().marginMode, 'highLeverage');
+	}
+
 	/**
 	 * Checks if any user position cumulative funding differs from respective market cumulative funding
 	 * @returns
@@ -2443,7 +2451,9 @@ export class User {
 			const marginRatio = calculateMarketMarginRatio(
 				market,
 				baseAssetAmount.abs(),
-				'Maintenance'
+				'Maintenance',
+				this.getUserAccount().maxMarginRatio,
+				this.isHighLeverageMode()
 			);
 
 			return liabilityValue.mul(new BN(marginRatio)).div(MARGIN_PRECISION);
@@ -2488,7 +2498,8 @@ export class User {
 			market,
 			proposedBaseAssetAmount.abs(),
 			marginCategory,
-			this.getUserAccount().maxMarginRatio
+			this.getUserAccount().maxMarginRatio,
+			this.isHighLeverageMode()
 		);
 		const marginRatioQuotePrecision = new BN(marginRatio)
 			.mul(QUOTE_PRECISION)
@@ -2602,13 +2613,16 @@ export class User {
 
 	public getMarginUSDCRequiredForTrade(
 		targetMarketIndex: number,
-		baseSize: BN
+		baseSize: BN,
+		estEntryPrice?: BN
 	): BN {
 		return calculateMarginUSDCRequiredForTrade(
 			this.driftClient,
 			targetMarketIndex,
 			baseSize,
-			this.getUserAccount().maxMarginRatio
+			this.getUserAccount().maxMarginRatio,
+			undefined,
+			estEntryPrice
 		);
 	}
 
@@ -2622,7 +2636,8 @@ export class User {
 			targetMarketIndex,
 			baseSize,
 			collateralIndex,
-			this.getUserAccount().maxMarginRatio
+			this.getUserAccount().maxMarginRatio,
+			false // assume user cant be high leverage if they havent created user account ?
 		);
 	}
 
@@ -3771,7 +3786,8 @@ export class User {
 				perpMarket,
 				worstCaseBaseAmount.abs(),
 				marginCategory,
-				this.getUserAccount().maxMarginRatio
+				this.getUserAccount().maxMarginRatio,
+				this.isHighLeverageMode()
 			)
 		);