@drift-labs/sdk 2.86.0-beta.0 → 2.86.0-beta.10

package/src/math/fuel.ts

@@ -0,0 +1,56 @@
+import { BN } from '@coral-xyz/anchor';
+import { SpotMarketAccount, PerpMarketAccount } from '..';
+import {
+	QUOTE_PRECISION,
+	ZERO,
+	FUEL_WINDOW,
+} from '../constants/numericConstants';
+
+export function calculateSpotFuelBonus(
+	spotMarket: SpotMarketAccount,
+	signedTokenValue: BN,
+	fuelBonusNumerator: BN
+): BN {
+	let result: BN;
+
+	if (signedTokenValue.abs().lt(new BN(1))) {
+		result = ZERO;
+	} else if (signedTokenValue.gt(new BN(0))) {
+		result = signedTokenValue
+			.abs()
+			.mul(fuelBonusNumerator)
+			.mul(new BN(spotMarket.fuelBoostDeposits))
+			.div(FUEL_WINDOW)
+			.div(QUOTE_PRECISION.div(new BN(10)));
+	} else {
+		result = signedTokenValue
+			.abs()
+			.mul(fuelBonusNumerator)
+			.mul(new BN(spotMarket.fuelBoostBorrows))
+			.div(FUEL_WINDOW)
+			.div(QUOTE_PRECISION.div(new BN(10)));
+	}
+
+	return result;
+}
+
+export function calculatePerpFuelBonus(
+	perpMarket: PerpMarketAccount,
+	baseAssetValue: BN,
+	fuelBonusNumerator: BN
+): BN {
+	let result: BN;
+
+	if (baseAssetValue.abs().lte(QUOTE_PRECISION)) {
+		result = new BN(0);
+	} else {
+		result = baseAssetValue
+			.abs()
+			.mul(fuelBonusNumerator)
+			.mul(new BN(perpMarket.fuelBoostPosition))
+			.div(FUEL_WINDOW)
+			.div(QUOTE_PRECISION.div(new BN(10)));
+	}
+
+	return result;
+}

package/src/types.ts

@@ -636,6 +636,10 @@ export type PerpMarketAccount = {
 	quoteSpotMarketIndex: number;
 	feeAdjustment: number;
 	pausedOperations: number;
+
+	fuelBoostPosition: number;
+	fuelBoostMaker: number;
+	fuelBoostTaker: number;
 };
 
 export type HistoricalOracleData = {
@@ -734,6 +738,11 @@ export type SpotMarketAccount = {
 
 	maxTokenBorrowsFraction: number;
 	minBorrowRate: number;
+
+	fuelBoostDeposits: number;
+	fuelBoostBorrows: number;
+	fuelBoostMaker: number;
+	fuelBoostTaker: number;
 };
 
 export type PoolBalance = {
@@ -876,6 +885,14 @@ export type UserStatsAccount = {
 	isReferrer: boolean;
 	authority: PublicKey;
 	ifStakedQuoteAssetAmount: BN;
+
+	fuelDeposits: number;
+	fuelBorrows: number;
+	fuelPositions: number;
+	fuelTaker: number;
+	fuelMaker: number;
+
+	ifStakedGovTokenAmount: BN;
 };
 
 export type UserAccount = {
@@ -905,6 +922,8 @@ export type UserAccount = {
 	hasOpenOrder: boolean;
 	openAuctions: number;
 	hasOpenAuction: boolean;
+
+	lastFuelBonusUpdateTs: BN;
 };
 
 export type SpotPosition = {

package/src/user.ts

@@ -35,6 +35,7 @@ import {
 	TEN_THOUSAND,
 	TWO,
 	ZERO,
+	FUEL_START_TS,
 } from './constants/numericConstants';
 import {
 	DataAndSlot,
@@ -88,6 +89,8 @@ import { calculateLiveOracleTwap } from './math/oracles';
 import { getPerpMarketTierNumber, getSpotMarketTierNumber } from './math/tiers';
 import { StrictOraclePrice } from './oracles/strictOraclePrice';
 
+import { calculateSpotFuelBonus, calculatePerpFuelBonus } from './math/fuel';
+
 export class User {
 	driftClient: DriftClient;
 	userAccountPublicKey: PublicKey;
@@ -873,6 +876,123 @@ export class User {
 			}, ZERO);
 	}
 
+	public getFuelBonus(
+		now: BN,
+		includeSettled = true,
+		includeUnsettled = true
+	): {
+		depositFuel;
+		borrowFuel;
+		positionFuel;
+		takerFuel;
+		makerFuel;
+	} {
+		const userAccount: UserAccount = this.getUserAccount();
+
+		const result = {
+			takerFuel: ZERO,
+			makerFuel: ZERO,
+			depositFuel: ZERO,
+			borrowFuel: ZERO,
+			positionFuel: ZERO,
+		};
+
+		if (includeSettled) {
+			const userStats: UserStatsAccount = this.driftClient
+				.getUserStats()
+				.getAccount();
+			result.takerFuel = result.takerFuel.add(new BN(userStats.fuelTaker));
+			result.makerFuel = result.makerFuel.add(new BN(userStats.fuelMaker));
+			result.depositFuel = result.depositFuel.add(
+				new BN(userStats.fuelDeposits)
+			);
+			result.borrowFuel = result.borrowFuel.add(new BN(userStats.fuelBorrows));
+			result.positionFuel = result.positionFuel.add(
+				new BN(userStats.fuelPositions)
+			);
+		}
+
+		if (includeUnsettled) {
+			const fuelBonusNumerator = BN.max(
+				now.sub(BN.max(userAccount.lastFuelBonusUpdateTs, FUEL_START_TS)),
+				ZERO
+			);
+
+			if (fuelBonusNumerator.gt(ZERO)) {
+				for (const spotPosition of this.getActiveSpotPositions()) {
+					const spotMarketAccount: SpotMarketAccount =
+						this.driftClient.getSpotMarketAccount(spotPosition.marketIndex);
+
+					const tokenAmount = this.getTokenAmount(spotPosition.marketIndex);
+					const oraclePriceData = this.getOracleDataForSpotMarket(
+						spotPosition.marketIndex
+					);
+
+					const twap5min = calculateLiveOracleTwap(
+						spotMarketAccount.historicalOracleData,
+						oraclePriceData,
+						now,
+						FIVE_MINUTE // 5MIN
+					);
+					const strictOraclePrice = new StrictOraclePrice(
+						oraclePriceData.price,
+						twap5min
+					);
+
+					const signedTokenValue = getStrictTokenValue(
+						tokenAmount,
+						spotMarketAccount.decimals,
+						strictOraclePrice
+					);
+
+					if (signedTokenValue.gt(ZERO)) {
+						result.depositFuel = result.depositFuel.add(
+							calculateSpotFuelBonus(
+								spotMarketAccount,
+								signedTokenValue,
+								fuelBonusNumerator
+							)
+						);
+					} else {
+						result.borrowFuel = result.borrowFuel.add(
+							calculateSpotFuelBonus(
+								spotMarketAccount,
+								signedTokenValue,
+								fuelBonusNumerator
+							)
+						);
+					}
+				}
+
+				for (const perpPosition of this.getActivePerpPositions()) {
+					const oraclePriceData = this.getOracleDataForPerpMarket(
+						perpPosition.marketIndex
+					);
+
+					const perpMarketAccount = this.driftClient.getPerpMarketAccount(
+						perpPosition.marketIndex
+					);
+
+					const baseAssetValue = this.getPerpPositionValue(
+						perpPosition.marketIndex,
+						oraclePriceData,
+						false
+					);
+
+					result.positionFuel = result.positionFuel.add(
+						calculatePerpFuelBonus(
+							perpMarketAccount,
+							baseAssetValue,
+							fuelBonusNumerator
+						)
+					);
+				}
+			}
+		}
+
+		return result;
+	}
+
 	public getSpotMarketAssetAndLiabilityValue(
 		marketIndex?: number,
 		marginCategory?: MarginCategory,