@drift-labs/sdk 2.21.0-beta.2 → 2.22.0-beta.0

package/lib/driftClient.d.ts

@@ -104,6 +104,7 @@ export declare class DriftClient {
     fetchAllUserAccounts(includeIdle?: boolean): Promise<ProgramAccount<UserAccount>[]>;
     getUserAccountsForDelegate(delegate: PublicKey): Promise<UserAccount[]>;
     getUserAccountsForAuthority(authority: PublicKey): Promise<UserAccount[]>;
+    getReferrerNameAccountsForAuthority(authority: PublicKey): Promise<ReferrerNameAccount[]>;
     deleteUser(subAccountId?: number, txParams?: TxParams): Promise<TransactionSignature>;
     getUser(subAccountId?: number): User;
     getUsers(): User[];

package/lib/driftClient.js

@@ -361,7 +361,7 @@ class DriftClient {
     async initializeReferrerName(name) {
         const userAccountPublicKey = (0, pda_1.getUserAccountPublicKeySync)(this.program.programId, this.wallet.publicKey, 0);
         const nameBuffer = (0, userName_1.encodeName)(name);
-        const referrerNameAccountPublicKey = await (0, pda_1.getReferrerNamePublicKeySync)(this.program.programId, nameBuffer);
+        const referrerNameAccountPublicKey = (0, pda_1.getReferrerNamePublicKeySync)(this.program.programId, nameBuffer);
         const tx = await this.program.transaction.initializeReferrerName(nameBuffer, {
             accounts: {
                 referrerName: referrerNameAccountPublicKey,
@@ -462,6 +462,18 @@ class DriftClient {
         ]);
         return programAccounts.map((programAccount) => programAccount.account);
     }
+    async getReferrerNameAccountsForAuthority(authority) {
+        const programAccounts = await this.program.account.referrerName.all([
+            {
+                memcmp: {
+                    offset: 8,
+                    /** data to match, as base-58 encoded string and limited to less than 129 bytes */
+                    bytes: bs58_1.default.encode(authority.toBuffer()),
+                },
+            },
+        ]);
+        return programAccounts.map((programAccount) => programAccount.account);
+    }
     async deleteUser(subAccountId = 0, txParams) {
         var _a;
         const userAccountPublicKey = (0, pda_1.getUserAccountPublicKeySync)(this.program.programId, this.wallet.publicKey, subAccountId);

package/lib/idl/drift.json

@@ -1,5 +1,5 @@
 {
-    "version": "2.21.0-beta.2",
+    "version": "2.22.0-beta.0",
     "name": "drift",
     "instructions": [
         {

package/lib/index.d.ts

@@ -66,4 +66,5 @@ export * from './dlob/DLOBOrders';
 export * from './dlob/NodeList';
 export * from './userMap/userMap';
 export * from './userMap/userStatsMap';
+export * from './math/bankruptcy';
 export { BN, PublicKey, pyth };

package/lib/index.js

@@ -90,3 +90,4 @@ __exportStar(require("./dlob/DLOBOrders"), exports);
 __exportStar(require("./dlob/NodeList"), exports);
 __exportStar(require("./userMap/userMap"), exports);
 __exportStar(require("./userMap/userStatsMap"), exports);
+__exportStar(require("./math/bankruptcy"), exports);

package/lib/math/bankruptcy.d.ts

@@ -0,0 +1,2 @@
+import { User } from '../user';
+export declare function isUserBankrupt(user: User): boolean;

package/lib/math/bankruptcy.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isUserBankrupt = void 0;
+const __1 = require("..");
+function isUserBankrupt(user) {
+    const userAccount = user.getUserAccount();
+    let hasLiability = false;
+    for (const position of userAccount.spotPositions) {
+        if (position.scaledBalance.gt(__1.ZERO)) {
+            if ((0, __1.isVariant)(position.balanceType, 'deposit')) {
+                return false;
+            }
+            if ((0, __1.isVariant)(position.balanceType, 'borrow')) {
+                hasLiability = true;
+            }
+        }
+    }
+    for (const position of userAccount.perpPositions) {
+        if (!position.baseAssetAmount.eq(__1.ZERO) ||
+            position.quoteAssetAmount.gt(__1.ZERO) ||
+            (0, __1.hasOpenOrders)(position) ||
+            position.lpShares.gt(__1.ZERO)) {
+            return false;
+        }
+        if (position.quoteAssetAmount.lt(__1.ZERO)) {
+            hasLiability = true;
+        }
+    }
+    return hasLiability;
+}
+exports.isUserBankrupt = isUserBankrupt;

package/lib/math/position.d.ts

@@ -51,3 +51,4 @@ export declare function calculateCostBasis(userPosition: PerpPosition): BN;
 export declare function findDirectionToClose(userPosition: PerpPosition): PositionDirection;
 export declare function positionCurrentDirection(userPosition: PerpPosition): PositionDirection;
 export declare function isEmptyPosition(userPosition: PerpPosition): boolean;
+export declare function hasOpenOrders(position: PerpPosition): boolean;

package/lib/math/position.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isEmptyPosition = exports.positionCurrentDirection = exports.findDirectionToClose = exports.calculateCostBasis = exports.calculateEntryPrice = exports.calculateBreakEvenPrice = exports.positionIsAvailable = exports.calculatePositionFundingPNL = exports.calculateClaimablePnl = exports.calculatePositionPNL = exports.calculateBaseAssetValue = void 0;
+exports.hasOpenOrders = exports.isEmptyPosition = exports.positionCurrentDirection = exports.findDirectionToClose = exports.calculateCostBasis = exports.calculateEntryPrice = exports.calculateBreakEvenPrice = exports.positionIsAvailable = exports.calculatePositionFundingPNL = exports.calculateClaimablePnl = exports.calculatePositionPNL = exports.calculateBaseAssetValue = void 0;
 const __1 = require("../");
 const numericConstants_1 = require("../constants/numericConstants");
 const types_1 = require("../types");
@@ -190,3 +190,9 @@ function isEmptyPosition(userPosition) {
     return userPosition.baseAssetAmount.eq(numericConstants_1.ZERO) && userPosition.openOrders === 0;
 }
 exports.isEmptyPosition = isEmptyPosition;
+function hasOpenOrders(position) {
+    return (position.openOrders != 0 ||
+        !position.openBids.eq(numericConstants_1.ZERO) ||
+        !position.openAsks.eq(numericConstants_1.ZERO));
+}
+exports.hasOpenOrders = hasOpenOrders;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.21.0-beta.2",
+	"version": "2.22.0-beta.0",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",
@@ -60,4 +60,4 @@
 	"engines": {
 		"node": ">=12"
 	}
-}
+}

package/src/driftClient.ts

@@ -603,7 +603,7 @@ export class DriftClient {
 
 		const nameBuffer = encodeName(name);
 
-		const referrerNameAccountPublicKey = await getReferrerNamePublicKeySync(
+		const referrerNameAccountPublicKey = getReferrerNamePublicKeySync(
 			this.program.programId,
 			nameBuffer
 		);
@@ -774,6 +774,24 @@ export class DriftClient {
 		);
 	}
 
+	public async getReferrerNameAccountsForAuthority(
+		authority: PublicKey
+	): Promise<ReferrerNameAccount[]> {
+		const programAccounts = await this.program.account.referrerName.all([
+			{
+				memcmp: {
+					offset: 8,
+					/** data to match, as base-58 encoded string and limited to less than 129 bytes */
+					bytes: bs58.encode(authority.toBuffer()),
+				},
+			},
+		]);
+
+		return programAccounts.map(
+			(programAccount) => programAccount.account as ReferrerNameAccount
+		);
+	}
+
 	public async deleteUser(
 		subAccountId = 0,
 		txParams?: TxParams

package/src/idl/drift.json

@@ -1,5 +1,5 @@
 {
-  "version": "2.21.0-beta.2",
+  "version": "2.22.0-beta.0",
   "name": "drift",
   "instructions": [
     {

package/src/index.ts

@@ -68,5 +68,6 @@ export * from './dlob/DLOBOrders';
 export * from './dlob/NodeList';
 export * from './userMap/userMap';
 export * from './userMap/userStatsMap';
+export * from './math/bankruptcy';
 
 export { BN, PublicKey, pyth };

package/src/math/bankruptcy.ts

@@ -0,0 +1,34 @@
+import { ZERO, hasOpenOrders, isVariant } from '..';
+import { User } from '../user';
+
+export function isUserBankrupt(user: User): boolean {
+	const userAccount = user.getUserAccount();
+	let hasLiability = false;
+	for (const position of userAccount.spotPositions) {
+		if (position.scaledBalance.gt(ZERO)) {
+			if (isVariant(position.balanceType, 'deposit')) {
+				return false;
+			}
+			if (isVariant(position.balanceType, 'borrow')) {
+				hasLiability = true;
+			}
+		}
+	}
+
+	for (const position of userAccount.perpPositions) {
+		if (
+			!position.baseAssetAmount.eq(ZERO) ||
+			position.quoteAssetAmount.gt(ZERO) ||
+			hasOpenOrders(position) ||
+			position.lpShares.gt(ZERO)
+		) {
+			return false;
+		}
+
+		if (position.quoteAssetAmount.lt(ZERO)) {
+			hasLiability = true;
+		}
+	}
+
+	return hasLiability;
+}

package/src/math/position.ts

@@ -270,3 +270,11 @@ export function positionCurrentDirection(
 export function isEmptyPosition(userPosition: PerpPosition): boolean {
 	return userPosition.baseAssetAmount.eq(ZERO) && userPosition.openOrders === 0;
 }
+
+export function hasOpenOrders(position: PerpPosition): boolean {
+	return (
+		position.openOrders != 0 ||
+		!position.openBids.eq(ZERO) ||
+		!position.openAsks.eq(ZERO)
+	);
+}

package/src/assert/assert.js

@@ -1,9 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.assert = void 0;
-function assert(condition, error) {
-    if (!condition) {
-        throw new Error(error || 'Unspecified AssertionError');
-    }
-}
-exports.assert = assert;

package/src/token/index.js

@@ -1,38 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.parseTokenAccount = void 0;
-const spl_token_1 = require("@solana/spl-token");
-const web3_js_1 = require("@solana/web3.js");
-function parseTokenAccount(data) {
-    const accountInfo = spl_token_1.AccountLayout.decode(data);
-    accountInfo.mint = new web3_js_1.PublicKey(accountInfo.mint);
-    accountInfo.owner = new web3_js_1.PublicKey(accountInfo.owner);
-    accountInfo.amount = spl_token_1.u64.fromBuffer(accountInfo.amount);
-    if (accountInfo.delegateOption === 0) {
-        accountInfo.delegate = null;
-        // eslint-disable-next-line new-cap
-        accountInfo.delegatedAmount = new spl_token_1.u64(0);
-    }
-    else {
-        accountInfo.delegate = new web3_js_1.PublicKey(accountInfo.delegate);
-        accountInfo.delegatedAmount = spl_token_1.u64.fromBuffer(accountInfo.delegatedAmount);
-    }
-    accountInfo.isInitialized = accountInfo.state !== 0;
-    accountInfo.isFrozen = accountInfo.state === 2;
-    if (accountInfo.isNativeOption === 1) {
-        accountInfo.rentExemptReserve = spl_token_1.u64.fromBuffer(accountInfo.isNative);
-        accountInfo.isNative = true;
-    }
-    else {
-        accountInfo.rentExemptReserve = null;
-        accountInfo.isNative = false;
-    }
-    if (accountInfo.closeAuthorityOption === 0) {
-        accountInfo.closeAuthority = null;
-    }
-    else {
-        accountInfo.closeAuthority = new web3_js_1.PublicKey(accountInfo.closeAuthority);
-    }
-    return accountInfo;
-}
-exports.parseTokenAccount = parseTokenAccount;

package/src/util/computeUnits.js

@@ -1,27 +0,0 @@
-"use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.findComputeUnitConsumption = void 0;
-function findComputeUnitConsumption(programId, connection, txSignature, commitment = 'confirmed') {
-    return __awaiter(this, void 0, void 0, function* () {
-        const tx = yield connection.getTransaction(txSignature, { commitment });
-        const computeUnits = [];
-        const regex = new RegExp(`Program ${programId.toString()} consumed ([0-9]{0,6}) of ([0-9]{0,7}) compute units`);
-        tx.meta.logMessages.forEach((logMessage) => {
-            const match = logMessage.match(regex);
-            if (match && match[1]) {
-                computeUnits.push(match[1]);
-            }
-        });
-        return computeUnits;
-    });
-}
-exports.findComputeUnitConsumption = findComputeUnitConsumption;

package/src/util/getTokenAddress.js

@@ -1,9 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getTokenAddress = void 0;
-const spl_token_1 = require("@solana/spl-token");
-const web3_js_1 = require("@solana/web3.js");
-const getTokenAddress = (mintAddress, userPubKey) => {
-    return spl_token_1.Token.getAssociatedTokenAddress(spl_token_1.ASSOCIATED_TOKEN_PROGRAM_ID, spl_token_1.TOKEN_PROGRAM_ID, new web3_js_1.PublicKey(mintAddress), new web3_js_1.PublicKey(userPubKey));
-};
-exports.getTokenAddress = getTokenAddress;

package/src/util/promiseTimeout.js

@@ -1,14 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.promiseTimeout = void 0;
-function promiseTimeout(promise, timeoutMs) {
-    let timeoutId;
-    const timeoutPromise = new Promise((resolve) => {
-        timeoutId = setTimeout(() => resolve(null), timeoutMs);
-    });
-    return Promise.race([promise, timeoutPromise]).then((result) => {
-        clearTimeout(timeoutId);
-        return result;
-    });
-}
-exports.promiseTimeout = promiseTimeout;

package/src/util/tps.js

@@ -1,27 +0,0 @@
-"use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.estimateTps = void 0;
-function estimateTps(programId, connection, failed) {
-    return __awaiter(this, void 0, void 0, function* () {
-        let signatures = yield connection.getSignaturesForAddress(programId, undefined, 'finalized');
-        if (failed) {
-            signatures = signatures.filter((signature) => signature.err);
-        }
-        const numberOfSignatures = signatures.length;
-        if (numberOfSignatures === 0) {
-            return 0;
-        }
-        return (numberOfSignatures /
-            (signatures[0].blockTime - signatures[numberOfSignatures - 1].blockTime));
-    });
-}
-exports.estimateTps = estimateTps;