Merge pull request #28 from drift-labs/ahzam/add-get-claimable-pnl

expose claimable PNL calculation functions

Author: 0xahzam

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "drift-ffi-sys"
-version = "2.154.0"
+version = "2.154.2"
 edition = "2021"
 description = "FFI bindings to drift program"
 

Cargo.toml

@@ -15,11 +15,11 @@ use drift_program::{
         oracle::{get_oracle_price as get_oracle_price_, OraclePriceData, OracleSource},
         oracle_map::OracleMap,
         order_params::PlaceOrderOptions,
-        perp_market::{ContractType, PerpMarket},
+        perp_market::{ContractType, PerpMarket, AMM},
         perp_market_map::PerpMarketMap,
         protected_maker_mode_config::ProtectedMakerParams,
         revenue_share::RevenueShareOrder,
-        spot_market::SpotMarket,
+        spot_market::{SpotBalanceType, SpotMarket},
         spot_market_map::SpotMarketMap,
         state::{FeeTier, State, ValidityGuardRails},
         user::{Order, PerpPosition, SpotPosition, User},
@@ -149,6 +149,16 @@ pub extern "C" fn math_calculate_margin_requirement_and_total_collateral_and_lia
     to_ffi_result(m)
 }
 
+#[no_mangle]
+pub extern "C" fn math_calculate_net_user_pnl(
+    amm: &AMM,
+    oracle_price: i64,
+) -> FfiResult<compat::i128> {
+    to_ffi_result(
+        drift_program::math::amm::calculate_net_user_pnl(amm, oracle_price).map(compat::i128),
+    )
+}
+
 #[no_mangle]
 pub extern "C" fn simulate_update_amm(
     market: &mut PerpMarket,
@@ -413,6 +423,19 @@ pub extern "C" fn perp_position_get_unrealized_pnl(
     to_ffi_result(position.get_unrealized_pnl(oracle_price).map(compat::i128))
 }
 
+#[no_mangle]
+pub extern "C" fn perp_position_get_claimable_pnl(
+    position: &PerpPosition,
+    oracle_price: i64,
+    pnl_pool_excess: compat::i128,
+) -> FfiResult<compat::i128> {
+    to_ffi_result(
+        position
+            .get_claimable_pnl(oracle_price, pnl_pool_excess.0)
+            .map(compat::i128),
+    )
+}
+
 #[no_mangle]
 pub extern "C" fn perp_position_is_available(position: &PerpPosition) -> bool {
     position.is_available()
@@ -481,6 +504,18 @@ pub extern "C" fn spot_position_get_token_amount(
     to_ffi_result(position.get_token_amount(market).map(compat::u128))
 }
 
+#[no_mangle]
+pub extern "C" fn spot_balance_get_token_amount(
+    balance: compat::u128,
+    spot_market: &SpotMarket,
+    balance_type: &SpotBalanceType,
+) -> FfiResult<compat::u128> {
+    to_ffi_result(
+        drift_program::math::spot_balance::get_token_amount(balance.0, spot_market, balance_type)
+            .map(compat::u128),
+    )
+}
+
 #[no_mangle]
 pub extern "C" fn user_get_spot_position(
     user: &User,