sdk: add isUserBankrupt

CHANGELOG.md

@@ -10,6 +10,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Features
 
+- sdk: add isUserBankrupt ([#399](https://github.com/drift-labs/protocol-v2/pull/399))
+
 ### Fixes
 
 ### Breaking

sdk/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 };

sdk/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;
+}

sdk/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)
+	);
+}