sdk: add settle pnl option to cost basis calc

drift-labs · Jan 22, 2024 · 098b2ca · 098b2ca
1 parent 807c037
commit 098b2ca
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sdk/src/math/position.ts b/sdk/src/math/position.ts
@@ -237,12 +237,16 @@ export function calculateEntryPrice(userPosition: PerpPosition): BN {
  * @param userPosition
  * @returns Precision: PRICE_PRECISION (10^10)
  */
-export function calculateCostBasis(userPosition: PerpPosition): BN {
+export function calculateCostBasis(
+	userPosition: PerpPosition,
+	includeSettledPnl = false
+): BN {
 	if (userPosition.baseAssetAmount.eq(ZERO)) {
 		return ZERO;
 	}
 
 	return userPosition.quoteAssetAmount
+		.add(includeSettledPnl ? userPosition.settledPnl : ZERO)
 		.mul(PRICE_PRECISION)
 		.mul(AMM_TO_QUOTE_PRECISION_RATIO)
 		.div(userPosition.baseAssetAmount)