@haven-fi/solauto-sdk 1.0.25 → 1.0.27

package/src/utils/jupiterUtils.ts

@@ -62,6 +62,7 @@ export async function getJupSwapTransaction(
       : swapDetails.exactIn
         ? "ExactIn"
         : undefined,
+    slippageBps: 10
   });
 
   const finalPriceSlippageBps = Math.round(

package/src/utils/marginfiUtils.ts

@@ -88,7 +88,9 @@ export async function getAllMarginfiAccountsByAuthority(
   umi: Umi,
   authority: PublicKey,
   compatibleWithSolauto?: boolean
-): Promise<PublicKey[]> {
+): Promise<
+  { marginfiAccount: PublicKey; supplyMint?: PublicKey; debtMint?: PublicKey }[]
+> {
   const marginfiAccounts = await umi.rpc.getProgramAccounts(
     MARGINFI_PROGRAM_ID,
     {
@@ -123,9 +125,15 @@ export async function getAllMarginfiAccountsByAuthority(
     );
     return positionStates
       .filter((x) => x.state !== undefined)
-      .map((x) => toWeb3JsPublicKey(x.publicKey));
+      .map((x) => ({
+        marginfiAccount: toWeb3JsPublicKey(x.publicKey),
+        supplyMint: toWeb3JsPublicKey(x.state!.supply.mint),
+        debtMint: toWeb3JsPublicKey(x.state!.debt.mint),
+      }));
   } else {
-    return marginfiAccounts.map((x) => toWeb3JsPublicKey(x.publicKey));
+    return marginfiAccounts.map((x) => ({
+      marginfiAccount: toWeb3JsPublicKey(x.publicKey),
+    }));
   }
 }
 

package/src/utils/solauto/generalUtils.ts

@@ -27,6 +27,10 @@ import {
   getAllMarginfiAccountsByAuthority,
   getMarginfiAccountPositionState,
 } from "../marginfiUtils";
+import {
+  SelfManagedPositionDetails,
+  SolautoPositionDetails,
+} from "../../types/solauto";
 
 function newPeriodsPassed(
   automation: AutomationSettings,
@@ -278,12 +282,6 @@ export async function getReferralsByUser(
   return accounts.map((x) => toWeb3JsPublicKey(x.publicKey));
 }
 
-export interface SolautoPositionDetails {
-  positionId: number;
-  lendingPlatform: LendingPlatform;
-  protocolAccount?: PublicKey;
-}
-
 export async function getAllPositionsByAuthority(
   umi: Umi,
   user: PublicKey
@@ -295,16 +293,27 @@ export async function getAllPositionsByAuthority(
     ...solautoManagedPositions.map((x) => ({
       positionId: x.positionId,
       lendingPlatform: x.lendingPlatform,
+      selfManaged: false,
     }))
   );
 
-  const marginfiPositions = await getAllMarginfiAccountsByAuthority(umi, user, true);
+  const marginfiPositions = await getAllMarginfiAccountsByAuthority(
+    umi,
+    user,
+    true
+  );
   allPositions.push(
-    ...marginfiPositions.map((pubkey) => ({
-      positionId: 0,
-      lendingPlatform: LendingPlatform.Marginfi,
-      protocolAccount: pubkey,
-    }))
+    ...marginfiPositions.map(
+      (x) =>
+        ({
+          positionId: 0,
+          selfManaged: true,
+          lendingPlatform: LendingPlatform.Marginfi,
+          protocolAccount: x.marginfiAccount,
+          supplyMint: x.supplyMint,
+          debtMint: x.debtMint,
+        }) as SelfManagedPositionDetails
+    )
   );
 
   // TODO support other platforms

package/src/utils/solauto/rebalanceUtils.ts

@@ -1,6 +1,6 @@
 import { PublicKey } from "@solana/web3.js";
 import { SolautoClient } from "../../clients/solautoClient";
-import { PositionTokenUsage } from "../../generated";
+import { FeeType, PositionTokenUsage } from "../../generated";
 import {
   eligibleForNextAutomationPeriod,
   getAdjustedSettingsFromAutomation,
@@ -188,7 +188,7 @@ export function getRebalanceValues(
   if (increasingLeverage) {
     adjustmentFeeBps = getSolautoFeesBps(
       client.referredByState !== undefined,
-      client.solautoPositionData!.feeType
+      client.solautoPositionData?.feeType ?? FeeType.Small
     ).total;
   }
 

package/tests/transactions/solautoMarginfi.ts

@@ -25,7 +25,7 @@ describe("Solauto Marginfi tests", async () => {
 
   const payForTransactions = false;
   const useJitoBundle = false;
-  const positionId = 1;
+  const positionId = 0;
 
   it("open - deposit - borrow - rebalance to 0 - withdraw - close", async () => {
     const client = new SolautoMarginfiClient(process.env.HELIUS_API_KEY!, true);
@@ -45,6 +45,11 @@ describe("Solauto Marginfi tests", async () => {
       {
         signer,
         positionId,
+        marginfiAccount: new PublicKey(
+          "4nNvUXF5YqHFcH2nGweSiuvy1ct7V5FXfoCLKFYUN36z"
+        ),
+        supplyMint: NATIVE_MINT,
+        debtMint: new PublicKey(USDC_MINT),
       }
     );
 
@@ -58,29 +63,29 @@ describe("Solauto Marginfi tests", async () => {
       targetBoostToBps: none(),
     };
 
-    if (client.solautoPositionData === null) {
-      transactionItems.push(
-        new TransactionItem(async () => {
-          return {
-            tx: client.openPosition(settingParams),
-          };
-        }, "open position")
-      );
+    // if (client.solautoPositionData === null) {
+    //   transactionItems.push(
+    //     new TransactionItem(async () => {
+    //       return {
+    //         tx: client.openPosition(),
+    //       };
+    //     }, "open position")
+    //   );
 
-      const initialSupplyUsd = 50;
-      transactionItems.push(
-        new TransactionItem(async () => {
-          const [supplyPrice] = await getTokenPrices([supply]);
-          return {
-            tx: client.protocolInteraction(
-              solautoAction("Deposit", [
-                toBaseUnit(initialSupplyUsd / supplyPrice, supplyDecimals),
-              ])
-            ),
-          };
-        }, "deposit")
-      );
-    }
+    //   const initialSupplyUsd = 50;
+    //   transactionItems.push(
+    //     new TransactionItem(async () => {
+    //       const [supplyPrice] = await getTokenPrices([supply]);
+    //       return {
+    //         tx: client.protocolInteraction(
+    //           solautoAction("Deposit", [
+    //             toBaseUnit(initialSupplyUsd / supplyPrice, supplyDecimals),
+    //           ])
+    //         ),
+    //       };
+    //     }, "deposit")
+    //   );
+    // }
 
     // transactionItems.push(
     //   new TransactionItem(
@@ -101,10 +106,24 @@ describe("Solauto Marginfi tests", async () => {
     //   )
     // );
 
+    // const initialSupplyUsd = 50;
+    // transactionItems.push(
+    //   new TransactionItem(async () => {
+    //     const [supplyPrice] = await getTokenPrices([supply]);
+    //     return {
+    //       tx: client.protocolInteraction(
+    //         solautoAction("Deposit", [
+    //           toBaseUnit(initialSupplyUsd / supplyPrice, supplyDecimals),
+    //         ])
+    //       ),
+    //     };
+    //   }, "deposit")
+    // );
+
     transactionItems.push(
       new TransactionItem(
         async (attemptNum) =>
-          await buildSolautoRebalanceTransaction(client, undefined, attemptNum),
+          await buildSolautoRebalanceTransaction(client, 5000, attemptNum),
         "rebalance"
       )
     );
@@ -141,7 +160,7 @@ describe("Solauto Marginfi tests", async () => {
       transactionItems,
       undefined,
       !payForTransactions,
-      useJitoBundle,
+      useJitoBundle
     ).send();
   });
 });

package/tests/unit/lookupTables.ts

@@ -1,14 +1,15 @@
 import { describe, it } from 'mocha';
-import { PublicKey } from "@solana/web3.js";
+import { clusterApiUrl, Connection, PublicKey } from "@solana/web3.js";
 import {
   MARGINFI_ACCOUNTS,
   MARGINFI_ACCOUNTS_LOOKUP_TABLE,
 } from "../../src/constants/marginfiAccounts";
-import { CONNECTION } from "../../src/constants/solautoConstants";
+
+const conn = new Connection(clusterApiUrl("mainnet-beta"), "confirmed");
 
 describe("Assert lookup tables up-to-date", async () => {
   it("marginfi accounts LUT should have everything", async () => {
-    const lookupTable = await CONNECTION.getAddressLookupTable(
+    const lookupTable = await conn.getAddressLookupTable(
       new PublicKey(MARGINFI_ACCOUNTS_LOOKUP_TABLE)
     );
     if (lookupTable === null) {