@drift-labs/sdk 2.0.8 → 2.0.10

package/lib/adminClient.d.ts

@@ -42,6 +42,7 @@ export declare class AdminClient extends DriftClient {
     updateSpotMarketMinOrderSize(spotMarketIndex: number, orderSize: BN): Promise<TransactionSignature>;
     updatePerpMarketExpiry(perpMarketIndex: number, expiryTs: BN): Promise<TransactionSignature>;
     updateSpotMarketOracle(spotMarketIndex: number, oracle: PublicKey, oracleSource: OracleSource): Promise<TransactionSignature>;
+    updateSpotMarketOrdersEnabled(spotMarketIndex: number, ordersEnabled: boolean): Promise<TransactionSignature>;
     updateSpotMarketExpiry(spotMarketIndex: number, expiryTs: BN): Promise<TransactionSignature>;
     updateWhitelistMint(whitelistMint?: PublicKey): Promise<TransactionSignature>;
     updateDiscountMint(discountMint: PublicKey): Promise<TransactionSignature>;

package/lib/adminClient.js

@@ -449,7 +449,7 @@ class AdminClient extends driftClient_1.DriftClient {
         });
     }
     async updateSpotMarketOracle(spotMarketIndex, oracle, oracleSource) {
-        return await this.program.rpc.updatePerpMarketOracle(oracle, oracleSource, {
+        return await this.program.rpc.updateSpotMarketOracle(oracle, oracleSource, {
             accounts: {
                 admin: this.wallet.publicKey,
                 state: await this.getStatePublicKey(),
@@ -458,6 +458,15 @@ class AdminClient extends driftClient_1.DriftClient {
             },
         });
     }
+    async updateSpotMarketOrdersEnabled(spotMarketIndex, ordersEnabled) {
+        return await this.program.rpc.updateSpotMarketOrdersEnabled(ordersEnabled, {
+            accounts: {
+                admin: this.wallet.publicKey,
+                state: await this.getStatePublicKey(),
+                spotMarket: await (0, pda_1.getSpotMarketPublicKey)(this.program.programId, spotMarketIndex),
+            },
+        });
+    }
     async updateSpotMarketExpiry(spotMarketIndex, expiryTs) {
         return await this.program.rpc.updateSpotMarketExpiry(expiryTs, {
             accounts: {

package/lib/driftClient.d.ts

@@ -82,6 +82,7 @@ export declare class DriftClient {
     getInitializeUserStatsIx(): Promise<TransactionInstruction>;
     updateUserName(name: string, subAccountId?: number): Promise<TransactionSignature>;
     updateUserCustomMarginRatio(marginRatio: number, subAccountId?: number): Promise<TransactionSignature>;
+    updateUserMarginTradingEnabled(marginTradingEnabled: boolean, subAccountId?: number): Promise<TransactionSignature>;
     updateUserDelegate(delegate: PublicKey, subAccountId?: number): Promise<TransactionSignature>;
     getUserAccountsForDelegate(delegate: PublicKey): Promise<UserAccount[]>;
     getUserAccountsForAuthority(authority: PublicKey): Promise<UserAccount[]>;

package/lib/driftClient.js

@@ -333,6 +333,14 @@ class DriftClient {
             },
         });
     }
+    async updateUserMarginTradingEnabled(marginTradingEnabled, subAccountId = 0) {
+        return await this.program.rpc.updateUserMarginTradingEnabled(subAccountId, marginTradingEnabled, {
+            accounts: {
+                user: await this.getUserAccountPublicKey(),
+                authority: this.wallet.publicKey,
+            },
+        });
+    }
     async updateUserDelegate(delegate, subAccountId = 0) {
         return await this.program.rpc.updateUserDelegate(subAccountId, delegate, {
             accounts: {

package/lib/factory/bigNum.js

@@ -316,19 +316,15 @@ class BigNum {
         let val = usingCustomPrecision
             ? this.prettyPrint(useTradePrecision, precisionOverride)
             : BigNum.fromPrint(this.toFixed(2), new anchor_1.BN(2)).prettyPrint();
-        // Append two trailing zeroes if not using custom precision
+        // Append trailing zeroes out to 2 decimal places if not using custom precision
         if (!usingCustomPrecision) {
             const [_, rightSide] = val.split(BigNum.delim);
             const trailingLength = (_a = rightSide === null || rightSide === void 0 ? void 0 : rightSide.length) !== null && _a !== void 0 ? _a : 0;
-            if (trailingLength < 2) {
-                const numHasDecimals = this.print().includes(BigNum.delim);
-                // Handle case where pretty print won't include the decimal point
-                if (trailingLength === 0 && numHasDecimals) {
-                    val = `${val}.00`;
-                }
-                else {
-                    val = `${val}${new Array(2 - trailingLength).fill('0').join('')}`;
-                }
+            if (trailingLength === 0) {
+                val = `${val}${BigNum.delim}00`;
+            }
+            else if (trailingLength === 1) {
+                val = `${val}0`;
             }
         }
         return `${prefix}${val.replace('-', '')}`;

package/lib/idl/drift.json

@@ -733,6 +733,31 @@
                 }
             ]
         },
+        {
+            "name": "updateUserMarginTradingEnabled",
+            "accounts": [
+                {
+                    "name": "user",
+                    "isMut": true,
+                    "isSigner": false
+                },
+                {
+                    "name": "authority",
+                    "isMut": false,
+                    "isSigner": true
+                }
+            ],
+            "args": [
+                {
+                    "name": "subAccountId",
+                    "type": "u16"
+                },
+                {
+                    "name": "marginTradingEnabled",
+                    "type": "bool"
+                }
+            ]
+        },
         {
             "name": "updateUserDelegate",
             "accounts": [

package/lib/math/amm.js

@@ -218,7 +218,7 @@ exports.calculateMaxSpread = calculateMaxSpread;
 function calculateSpreadBN(baseSpread, lastOracleReservePriceSpreadPct, lastOracleConfPct, maxSpread, quoteAssetReserve, terminalQuoteAssetReserve, pegMultiplier, netBaseAssetAmount, reservePrice, totalFeeMinusDistributions, baseAssetReserve, minBaseAssetReserve, maxBaseAssetReserve) {
     let longSpread = baseSpread / 2;
     let shortSpread = baseSpread / 2;
-    if (lastOracleReservePriceSpreadPct.gt(numericConstants_1.ZERO)) {
+    if (lastOracleReservePriceSpreadPct.gte(numericConstants_1.ZERO)) {
         shortSpread = Math.max(shortSpread, lastOracleReservePriceSpreadPct.abs().toNumber() +
             lastOracleConfPct.toNumber());
     }

package/lib/userMap/userMap.d.ts

@@ -38,4 +38,5 @@ export declare class UserMap implements UserMapInterface {
     getUserAuthority(key: string): PublicKey | undefined;
     updateWithOrderRecord(record: OrderRecord): Promise<void>;
     values(): IterableIterator<User>;
+    size(): number;
 }

package/lib/userMap/userMap.js

@@ -83,5 +83,8 @@ class UserMap {
     values() {
         return this.userMap.values();
     }
+    size() {
+        return this.userMap.size;
+    }
 }
 exports.UserMap = UserMap;

package/lib/userMap/userStatsMap.d.ts

@@ -16,4 +16,5 @@ export declare class UserStatsMap {
     get(authorityPublicKey: string): UserStats;
     mustGet(authorityPublicKey: string): Promise<UserStats>;
     values(): IterableIterator<UserStats>;
+    size(): number;
 }

package/lib/userMap/userStatsMap.js

@@ -64,5 +64,8 @@ class UserStatsMap {
     values() {
         return this.userStatsMap.values();
     }
+    size() {
+        return this.userStatsMap.size;
+    }
 }
 exports.UserStatsMap = UserStatsMap;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.0.8",
+	"version": "2.0.10",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",

package/src/adminClient.ts

@@ -852,7 +852,7 @@ export class AdminClient extends DriftClient {
 		oracle: PublicKey,
 		oracleSource: OracleSource
 	): Promise<TransactionSignature> {
-		return await this.program.rpc.updatePerpMarketOracle(oracle, oracleSource, {
+		return await this.program.rpc.updateSpotMarketOracle(oracle, oracleSource, {
 			accounts: {
 				admin: this.wallet.publicKey,
 				state: await this.getStatePublicKey(),
@@ -865,6 +865,22 @@ export class AdminClient extends DriftClient {
 		});
 	}
 
+	public async updateSpotMarketOrdersEnabled(
+		spotMarketIndex: number,
+		ordersEnabled: boolean
+	): Promise<TransactionSignature> {
+		return await this.program.rpc.updateSpotMarketOrdersEnabled(ordersEnabled, {
+			accounts: {
+				admin: this.wallet.publicKey,
+				state: await this.getStatePublicKey(),
+				spotMarket: await getSpotMarketPublicKey(
+					this.program.programId,
+					spotMarketIndex
+				),
+			},
+		});
+	}
+
 	public async updateSpotMarketExpiry(
 		spotMarketIndex: number,
 		expiryTs: BN

package/src/driftClient.ts

@@ -556,6 +556,22 @@ export class DriftClient {
 		);
 	}
 
+	public async updateUserMarginTradingEnabled(
+		marginTradingEnabled: boolean,
+		subAccountId = 0
+	): Promise<TransactionSignature> {
+		return await this.program.rpc.updateUserMarginTradingEnabled(
+			subAccountId,
+			marginTradingEnabled,
+			{
+				accounts: {
+					user: await this.getUserAccountPublicKey(),
+					authority: this.wallet.publicKey,
+				},
+			}
+		);
+	}
+
 	public async updateUserDelegate(
 		delegate: PublicKey,
 		subAccountId = 0

package/src/factory/bigNum.ts

@@ -481,21 +481,15 @@ export class BigNum {
 			? this.prettyPrint(useTradePrecision, precisionOverride)
 			: BigNum.fromPrint(this.toFixed(2), new BN(2)).prettyPrint();
 
-		// Append two trailing zeroes if not using custom precision
+		// Append trailing zeroes out to 2 decimal places if not using custom precision
 		if (!usingCustomPrecision) {
 			const [_, rightSide] = val.split(BigNum.delim);
-
 			const trailingLength = rightSide?.length ?? 0;
 
-			if (trailingLength < 2) {
-				const numHasDecimals = this.print().includes(BigNum.delim);
-
-				// Handle case where pretty print won't include the decimal point
-				if (trailingLength === 0 && numHasDecimals) {
-					val = `${val}.00`;
-				} else {
-					val = `${val}${new Array(2 - trailingLength).fill('0').join('')}`;
-				}
+			if (trailingLength === 0) {
+				val = `${val}${BigNum.delim}00`;
+			} else if (trailingLength === 1) {
+				val = `${val}0`;
 			}
 		}
 

package/src/idl/drift.json

@@ -733,6 +733,31 @@
         }
       ]
     },
+    {
+      "name": "updateUserMarginTradingEnabled",
+      "accounts": [
+        {
+          "name": "user",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "authority",
+          "isMut": false,
+          "isSigner": true
+        }
+      ],
+      "args": [
+        {
+          "name": "subAccountId",
+          "type": "u16"
+        },
+        {
+          "name": "marginTradingEnabled",
+          "type": "bool"
+        }
+      ]
+    },
     {
       "name": "updateUserDelegate",
       "accounts": [

package/src/math/amm.ts

@@ -410,7 +410,7 @@ export function calculateSpreadBN(
 	let longSpread = baseSpread / 2;
 	let shortSpread = baseSpread / 2;
 
-	if (lastOracleReservePriceSpreadPct.gt(ZERO)) {
+	if (lastOracleReservePriceSpreadPct.gte(ZERO)) {
 		shortSpread = Math.max(
 			shortSpread,
 			lastOracleReservePriceSpreadPct.abs().toNumber() +

package/src/userMap/userMap.ts

@@ -123,4 +123,8 @@ export class UserMap implements UserMapInterface {
 	public values(): IterableIterator<User> {
 		return this.userMap.values();
 	}
+
+	public size(): number {
+		return this.userMap.size;
+	}
 }

package/src/userMap/userStatsMap.ts

@@ -105,4 +105,8 @@ export class UserStatsMap {
 	public values(): IterableIterator<UserStats> {
 		return this.userStatsMap.values();
 	}
+
+	public size(): number {
+		return this.userStatsMap.size;
+	}
 }