@haven-fi/solauto-sdk 1.0.582 → 1.0.584

package/dist/services/flashLoans/flProviderAggregator.js

@@ -0,0 +1,46 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FlProviderAggregator = void 0;
+const flProviderBase_1 = require("./flProviderBase");
+const marginfiFlProvider_1 = require("./marginfiFlProvider");
+class FlProviderAggregator extends flProviderBase_1.FlProviderBase {
+    constructor(umi, signer, supplyMint, debtMint) {
+        super(umi, signer, supplyMint, debtMint);
+        this.marginfiFlProvider = new marginfiFlProvider_1.MarginfiFlProvider(umi, signer, supplyMint, debtMint);
+    }
+    async initialize() {
+        // TODO: PF
+        // Once we have more than one, set the right fl provider for each liquidity source
+        await this.marginfiFlProvider.initialize();
+    }
+    async flAccountPrereqIxs() {
+        return await this.marginfiFlProvider.initializeIMfiAccounts();
+    }
+    otherSigners() {
+        // TODO: PF
+        return [...this.flSigners, ...this.marginfiFlProvider.otherSigners()];
+    }
+    lutAccountsToAdd() {
+        return [
+            ...super.lutAccountsToAdd(),
+            ...this.marginfiFlProvider.lutAccountsToAdd(),
+        ];
+    }
+    flProvider(source) {
+        // TODO: PF
+        return this.marginfiFlProvider;
+    }
+    liquidityAvailable(source) {
+        return this.flProvider(source).liquidityAvailable(source);
+    }
+    flFeeBps(source, signerFlashLoan) {
+        return this.flProvider(source).flFeeBps(source);
+    }
+    flashBorrow(flashLoan, destTokenAccount) {
+        return this.flProvider(flashLoan.liquiditySource).flashBorrow(flashLoan, destTokenAccount);
+    }
+    flashRepay(flashLoan) {
+        return this.flProvider(flashLoan.liquiditySource).flashRepay(flashLoan);
+    }
+}
+exports.FlProviderAggregator = FlProviderAggregator;

package/dist/services/flashLoans/flProviderBase.d.ts

@@ -0,0 +1,23 @@
+import { Signer, TransactionBuilder, Umi } from "@metaplex-foundation/umi";
+import { PublicKey } from "@solana/web3.js";
+import { TokenType } from "../../generated";
+import { FlashLoanDetails } from "../../types";
+export declare abstract class FlProviderBase {
+    protected umi: Umi;
+    protected signer: Signer;
+    protected supplyMint: PublicKey;
+    protected debtMint: PublicKey;
+    protected flSigners: Signer[];
+    constructor(umi: Umi, signer: Signer, supplyMint: PublicKey, debtMint: PublicKey);
+    abstract initialize(): Promise<void>;
+    lutAccountsToAdd(): PublicKey[];
+    otherSigners(): Signer[];
+    mint(source: TokenType): PublicKey;
+    abstract liquidityAvailable(source: TokenType): bigint;
+    liquidityAvailableUsd(source: TokenType): number;
+    abstract flFeeBps(source: TokenType, signerFlashLoan?: boolean): number;
+    abstract flashBorrow(flashLoan: FlashLoanDetails, destTokenAccount: PublicKey): TransactionBuilder;
+    abstract flashRepay(flashLoan: FlashLoanDetails): TransactionBuilder;
+    protected signerFlashBorrow(flashLoan: FlashLoanDetails, destTokenAccount: PublicKey): TransactionBuilder;
+}
+//# sourceMappingURL=flProviderBase.d.ts.map

package/dist/services/flashLoans/flProviderBase.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"flProviderBase.d.ts","sourceRoot":"","sources":["../../../src/services/flashLoans/flProviderBase.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,MAAM,EAEN,kBAAkB,EAClB,GAAG,EACJ,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAQ5C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAG/C,8BAAsB,cAAc;IAIhC,SAAS,CAAC,GAAG,EAAE,GAAG;IAClB,SAAS,CAAC,MAAM,EAAE,MAAM;IACxB,SAAS,CAAC,UAAU,EAAE,SAAS;IAC/B,SAAS,CAAC,QAAQ,EAAE,SAAS;IAN/B,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,CAAM;gBAGvB,GAAG,EAAE,GAAG,EACR,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,SAAS,EACrB,QAAQ,EAAE,SAAS;IAG/B,QAAQ,CAAC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAEpC,gBAAgB,IAAI,SAAS,EAAE;IAI/B,YAAY,IAAI,MAAM,EAAE;IAIxB,IAAI,CAAC,MAAM,EAAE,SAAS;IAItB,QAAQ,CAAC,kBAAkB,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IACtD,qBAAqB,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAShD,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,EAAE,eAAe,CAAC,EAAE,OAAO,GAAG,MAAM;IACvE,QAAQ,CAAC,WAAW,CAClB,SAAS,EAAE,gBAAgB,EAC3B,gBAAgB,EAAE,SAAS,GAC1B,kBAAkB;IACrB,QAAQ,CAAC,UAAU,CAAC,SAAS,EAAE,gBAAgB,GAAG,kBAAkB;IAEpE,SAAS,CAAC,iBAAiB,CACzB,SAAS,EAAE,gBAAgB,EAC3B,gBAAgB,EAAE,SAAS,GAC1B,kBAAkB;CAyBtB"}

package/dist/services/flashLoans/flProviderBase.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FlProviderBase = void 0;
+const umi_1 = require("@metaplex-foundation/umi");
+const utils_1 = require("../../utils");
+const generated_1 = require("../../generated");
+const umi_web3js_adapters_1 = require("@metaplex-foundation/umi-web3js-adapters");
+class FlProviderBase {
+    constructor(umi, signer, supplyMint, debtMint) {
+        this.umi = umi;
+        this.signer = signer;
+        this.supplyMint = supplyMint;
+        this.debtMint = debtMint;
+        this.flSigners = [];
+    }
+    lutAccountsToAdd() {
+        return [];
+    }
+    otherSigners() {
+        return this.flSigners;
+    }
+    mint(source) {
+        return source === generated_1.TokenType.Supply ? this.supplyMint : this.debtMint;
+    }
+    liquidityAvailableUsd(source) {
+        return ((0, utils_1.fromBaseUnit)(this.liquidityAvailable(source), (0, utils_1.tokenInfo)(this.mint(source)).decimals) * ((0, utils_1.safeGetPrice)(this.mint(source)) ?? 0));
+    }
+    signerFlashBorrow(flashLoan, destTokenAccount) {
+        if (!destTokenAccount.equals((0, utils_1.getTokenAccount)((0, umi_web3js_adapters_1.toWeb3JsPublicKey)(this.signer.publicKey), flashLoan.mint))) {
+            return (0, umi_1.transactionBuilder)().add((0, utils_1.splTokenTransferUmiIx)(this.signer, (0, utils_1.getTokenAccount)((0, umi_web3js_adapters_1.toWeb3JsPublicKey)(this.signer.publicKey), flashLoan.mint), destTokenAccount, (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(this.signer.publicKey), flashLoan.baseUnitAmount));
+        }
+        else {
+            return (0, umi_1.transactionBuilder)();
+        }
+    }
+}
+exports.FlProviderBase = FlProviderBase;

package/dist/services/flashLoans/index.d.ts

@@ -0,0 +1,4 @@
+export * from "./flProviderAggregator";
+export * from "./flProviderBase";
+export * from "./marginfiFlProvider";
+//# sourceMappingURL=index.d.ts.map

package/dist/services/flashLoans/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/services/flashLoans/index.ts"],"names":[],"mappings":"AAAA,cAAc,wBAAwB,CAAC;AACvC,cAAc,kBAAkB,CAAC;AACjC,cAAc,sBAAsB,CAAC"}

package/dist/services/flashLoans/index.js

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./flProviderAggregator"), exports);
+__exportStar(require("./flProviderBase"), exports);
+__exportStar(require("./marginfiFlProvider"), exports);

package/dist/services/flashLoans/marginfiFlProvider.d.ts

@@ -0,0 +1,25 @@
+import { TransactionBuilder } from "@metaplex-foundation/umi";
+import { FlProviderBase } from "./flProviderBase";
+import { PublicKey } from "@solana/web3.js";
+import { TokenType } from "../../generated";
+import { FlashLoanDetails } from "../../types";
+export declare class MarginfiFlProvider extends FlProviderBase {
+    private existingMarginfiAccounts;
+    private supplyBankLiquiditySource;
+    private debtBankLiquiditySource;
+    private supplyImfiAccount;
+    private debtImfiAccount;
+    initialize(): Promise<void>;
+    private setAvailableBanks;
+    private setIntermediaryAccount;
+    initializeIMfiAccounts(): Promise<TransactionBuilder>;
+    lutAccountsToAdd(): PublicKey[];
+    private liquidityBank;
+    private iMfiAccount;
+    liquidityAvailable(source: TokenType): bigint;
+    flFeeBps(source: TokenType, signerFlashLoan?: boolean): number;
+    flashBorrow(flashLoan: FlashLoanDetails, destTokenAccount: PublicKey): TransactionBuilder;
+    flashRepay(flashLoan: FlashLoanDetails): TransactionBuilder;
+    closeBalance(marginfiAccount: PublicKey, bank: PublicKey, marginfiGroup: PublicKey): TransactionBuilder;
+}
+//# sourceMappingURL=marginfiFlProvider.d.ts.map

package/dist/services/flashLoans/marginfiFlProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"marginfiFlProvider.d.ts","sourceRoot":"","sources":["../../../src/services/flashLoans/marginfiFlProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAML,kBAAkB,EACnB,MAAM,0BAA0B,CAAC;AAalC,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,SAAS,EAA8B,MAAM,iBAAiB,CAAC;AAcxE,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAQ/C,qBAAa,kBAAmB,SAAQ,cAAc;IACpD,OAAO,CAAC,wBAAwB,CAAqB;IACrD,OAAO,CAAC,yBAAyB,CAAQ;IACzC,OAAO,CAAC,uBAAuB,CAAQ;IACvC,OAAO,CAAC,iBAAiB,CAAe;IACxC,OAAO,CAAC,eAAe,CAAe;IAEhC,UAAU;YAiBF,iBAAiB;IA2C/B,OAAO,CAAC,sBAAsB;IAqCxB,sBAAsB,IAAI,OAAO,CAAC,kBAAkB,CAAC;IAyC3D,gBAAgB,IAAI,SAAS,EAAE;IAY/B,OAAO,CAAC,aAAa;IAMrB,OAAO,CAAC,WAAW;IAMnB,kBAAkB,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAI7C,QAAQ,CAAC,MAAM,EAAE,SAAS,EAAE,eAAe,CAAC,EAAE,OAAO,GAAG,MAAM;IAW9D,WAAW,CACT,SAAS,EAAE,gBAAgB,EAC3B,gBAAgB,EAAE,SAAS,GAC1B,kBAAkB;IAoCrB,UAAU,CAAC,SAAS,EAAE,gBAAgB,GAAG,kBAAkB;IA6G3D,YAAY,CACV,eAAe,EAAE,SAAS,EAC1B,IAAI,EAAE,SAAS,EACf,aAAa,EAAE,SAAS;CAW3B"}

package/dist/services/flashLoans/marginfiFlProvider.js

@@ -0,0 +1,246 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MarginfiFlProvider = void 0;
+const umi_1 = require("@metaplex-foundation/umi");
+const constants_1 = require("../../constants");
+const marginfi_sdk_1 = require("../../marginfi-sdk");
+const flProviderBase_1 = require("./flProviderBase");
+const web3_js_1 = require("@solana/web3.js");
+const utils_1 = require("../../utils");
+const generated_1 = require("../../generated");
+const umi_web3js_adapters_1 = require("@metaplex-foundation/umi-web3js-adapters");
+class MarginfiFlProvider extends flProviderBase_1.FlProviderBase {
+    async initialize() {
+        await this.setAvailableBanks();
+        this.existingMarginfiAccounts = await (0, utils_1.getEmptyMarginfiAccountsByAuthority)(this.umi, (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(this.signer.publicKey));
+        if (this.liquidityBank(generated_1.TokenType.Supply).group.toString() !==
+            this.liquidityBank(generated_1.TokenType.Debt).group.toString()) {
+            this.setIntermediaryAccount([generated_1.TokenType.Supply]);
+            this.setIntermediaryAccount([generated_1.TokenType.Debt]);
+        }
+        else {
+            this.setIntermediaryAccount([generated_1.TokenType.Supply, generated_1.TokenType.Debt]);
+        }
+    }
+    async setAvailableBanks() {
+        const availableBanks = [];
+        const checkIfUsable = (group, mint) => {
+            if (Object.keys(constants_1.MARGINFI_ACCOUNTS[group]).includes(mint.toString())) {
+                availableBanks.push(constants_1.MARGINFI_ACCOUNTS[group][mint.toString()].bank);
+            }
+        };
+        for (const group of Object.keys(constants_1.MARGINFI_ACCOUNTS)) {
+            checkIfUsable(group, this.supplyMint);
+            checkIfUsable(group, this.debtMint);
+        }
+        const banks = await (0, marginfi_sdk_1.safeFetchAllBank)(this.umi, availableBanks.map((x) => (0, umi_1.publicKey)(x)));
+        if (!(0, utils_1.safeGetPrice)(this.supplyMint) || !(0, utils_1.safeGetPrice)(this.debtMint)) {
+            await (0, utils_1.fetchTokenPrices)([this.supplyMint, this.debtMint]);
+        }
+        const mapBanksAndBalances = (mint) => banks
+            .filter((x) => (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(x.mint).equals(mint))
+            .map((x) => {
+            return [
+                (0, utils_1.fromBaseUnit)((0, utils_1.getBankLiquidityAvailableBaseUnit)(x, false), (0, utils_1.tokenInfo)(mint).decimals) * (0, utils_1.safeGetPrice)(mint),
+                x,
+            ];
+        })
+            .sort((a, b) => b[0] - a[0]);
+        const supplyBanks = mapBanksAndBalances(this.supplyMint);
+        const debtBanks = mapBanksAndBalances(this.debtMint);
+        this.supplyBankLiquiditySource = supplyBanks[0][1];
+        this.debtBankLiquiditySource = debtBanks[0][1];
+    }
+    setIntermediaryAccount(sources) {
+        const compatibleMarginfiAccounts = this.existingMarginfiAccounts.filter((x) => x.group.toString() == this.liquidityBank(sources[0]).group);
+        const signer = compatibleMarginfiAccounts.length > 0
+            ? undefined
+            : (0, umi_1.createSignerFromKeypair)(this.umi, this.umi.eddsa.generateKeypair());
+        const accountPk = compatibleMarginfiAccounts.length > 0
+            ? (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(compatibleMarginfiAccounts[0].publicKey)
+            : (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(signer.publicKey);
+        const accountData = compatibleMarginfiAccounts.length > 0
+            ? compatibleMarginfiAccounts[0]
+            : undefined;
+        if (signer) {
+            this.flSigners.push(signer);
+        }
+        (0, utils_1.consoleLog)("Intermediary MF account:", accountPk.toString());
+        for (const s of sources) {
+            const data = {
+                signer,
+                accountPk,
+                accountData,
+            };
+            if (s === generated_1.TokenType.Supply) {
+                this.supplyImfiAccount = data;
+            }
+            else {
+                this.debtImfiAccount = data;
+            }
+        }
+    }
+    async initializeIMfiAccounts() {
+        const supplyImfiAccount = this.iMfiAccount(generated_1.TokenType.Supply);
+        const debtImfiAccount = this.iMfiAccount(generated_1.TokenType.Debt);
+        const [supplyImfiRpcAccount, debtImfiRpcAccount] = await this.umi.rpc.getAccounts([
+            (0, umi_1.publicKey)(supplyImfiAccount.accountPk),
+            (0, umi_1.publicKey)(debtImfiAccount.accountPk),
+        ]);
+        let tx = (0, umi_1.transactionBuilder)();
+        if (!(0, utils_1.rpcAccountCreated)(supplyImfiRpcAccount)) {
+            tx = tx.add((0, marginfi_sdk_1.marginfiAccountInitialize)(this.umi, {
+                marginfiAccount: supplyImfiAccount.signer,
+                marginfiGroup: this.supplyBankLiquiditySource.group,
+                authority: this.signer,
+                feePayer: this.signer,
+            }));
+        }
+        if (supplyImfiAccount.accountPk.toString() !==
+            debtImfiAccount.accountPk.toString() &&
+            !(0, utils_1.rpcAccountCreated)(debtImfiRpcAccount)) {
+            tx = tx.add((0, marginfi_sdk_1.marginfiAccountInitialize)(this.umi, {
+                marginfiAccount: debtImfiAccount.signer,
+                marginfiGroup: this.debtBankLiquiditySource.group,
+                authority: this.signer,
+                feePayer: this.signer,
+            }));
+        }
+        return tx;
+    }
+    lutAccountsToAdd() {
+        return [
+            ...super.lutAccountsToAdd(),
+            ...Array.from(new Set([
+                this.iMfiAccount(generated_1.TokenType.Supply).accountPk.toString(),
+                this.iMfiAccount(generated_1.TokenType.Debt).accountPk.toString(),
+            ])).map((x) => new web3_js_1.PublicKey(x)),
+        ];
+    }
+    liquidityBank(source) {
+        return source === generated_1.TokenType.Supply
+            ? this.supplyBankLiquiditySource
+            : this.debtBankLiquiditySource;
+    }
+    iMfiAccount(source) {
+        return source === generated_1.TokenType.Supply
+            ? this.supplyImfiAccount
+            : this.debtImfiAccount;
+    }
+    liquidityAvailable(source) {
+        return (0, utils_1.getBankLiquidityAvailableBaseUnit)(this.liquidityBank(source), false);
+    }
+    flFeeBps(source, signerFlashLoan) {
+        if (signerFlashLoan) {
+            return 0;
+        }
+        return (0, utils_1.bytesToI80F48)(this.liquidityBank(source).config.interestRateConfig
+            .protocolOriginationFee.value);
+    }
+    flashBorrow(flashLoan, destTokenAccount) {
+        if (flashLoan.signerFlashLoan) {
+            return this.signerFlashBorrow(flashLoan, destTokenAccount);
+        }
+        const bank = this.liquidityBank(flashLoan.liquiditySource);
+        const associatedBankAccs = (0, utils_1.findMarginfiAccounts)((0, umi_web3js_adapters_1.toWeb3JsPublicKey)(bank.publicKey));
+        const iMfiAccount = this.iMfiAccount(flashLoan.liquiditySource);
+        return (0, umi_1.transactionBuilder)()
+            .add((0, marginfi_sdk_1.lendingAccountStartFlashloan)(this.umi, {
+            endIndex: 0, // We set this after building the transaction
+            ixsSysvar: (0, umi_1.publicKey)(web3_js_1.SYSVAR_INSTRUCTIONS_PUBKEY),
+            marginfiAccount: (0, umi_1.publicKey)(iMfiAccount.accountPk),
+            signer: this.signer,
+        }))
+            .add((0, marginfi_sdk_1.lendingAccountBorrow)(this.umi, {
+            amount: flashLoan.baseUnitAmount,
+            bank: (0, umi_1.publicKey)(bank),
+            bankLiquidityVault: (0, umi_1.publicKey)(associatedBankAccs.liquidityVault),
+            bankLiquidityVaultAuthority: (0, umi_1.publicKey)(associatedBankAccs.vaultAuthority),
+            destinationTokenAccount: (0, umi_1.publicKey)(destTokenAccount),
+            marginfiAccount: (0, umi_1.publicKey)(iMfiAccount.accountPk),
+            marginfiGroup: this.liquidityBank(flashLoan.liquiditySource).group,
+            signer: this.signer,
+        }));
+    }
+    flashRepay(flashLoan) {
+        if (flashLoan.signerFlashLoan) {
+            return (0, umi_1.transactionBuilder)();
+        }
+        const bank = this.liquidityBank(flashLoan.liquiditySource);
+        const associatedBankAccs = (0, utils_1.findMarginfiAccounts)((0, umi_web3js_adapters_1.toWeb3JsPublicKey)(bank.publicKey));
+        const marginfiGroup = (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(this.liquidityBank(flashLoan.liquiditySource).group);
+        const iMfiAccount = this.iMfiAccount(flashLoan.liquiditySource);
+        const remainingAccounts = [];
+        let includedFlashLoanToken = false;
+        if (iMfiAccount?.accountData) {
+            iMfiAccount.accountData.lendingAccount.balances.forEach(async (x) => {
+                if (x.active) {
+                    if (x.bankPk.toString() === bank.publicKey.toString()) {
+                        includedFlashLoanToken = true;
+                    }
+                    // TODO: Don't dynamically pull from bank until Marginfi sorts out their price oracle issues.
+                    // const bankData = await safeFetchBank(this.umi, publicKey(accounts.data.bank));
+                    // const priceOracle = bankData!.config.oracleKeys[0];
+                    const priceOracle = (0, umi_1.publicKey)((0, utils_1.findMarginfiAccounts)((0, umi_web3js_adapters_1.toWeb3JsPublicKey)(x.bankPk)).priceOracle);
+                    remainingAccounts.push(...[
+                        {
+                            pubkey: x.bankPk,
+                            isSigner: false,
+                            isWritable: false,
+                        },
+                        {
+                            pubkey: priceOracle,
+                            isSigner: false,
+                            isWritable: false,
+                        },
+                    ]);
+                }
+            });
+        }
+        if (!iMfiAccount.accountData || !includedFlashLoanToken) {
+            remainingAccounts.push(...[
+                {
+                    pubkey: bank.publicKey,
+                    isSigner: false,
+                    isWritable: false,
+                },
+                {
+                    pubkey: (0, umi_1.publicKey)(associatedBankAccs.priceOracle),
+                    isSigner: false,
+                    isWritable: false,
+                },
+            ]);
+        }
+        const banksRequiringBalanceClose = Array.from(new Set([
+            bank.publicKey.toString(),
+            ...(iMfiAccount.accountData?.lendingAccount.balances ?? [])
+                .filter((x) => x.active && (0, utils_1.bytesToI80F48)(x.liabilityShares.value) > 0)
+                .map((x) => x.bankPk.toString()),
+        ]));
+        return (0, umi_1.transactionBuilder)()
+            .add((0, marginfi_sdk_1.lendingAccountRepay)(this.umi, {
+            amount: flashLoan.baseUnitAmount,
+            repayAll: null,
+            bank: bank.publicKey,
+            bankLiquidityVault: (0, umi_1.publicKey)(associatedBankAccs.liquidityVault),
+            marginfiAccount: (0, umi_1.publicKey)(iMfiAccount.accountPk),
+            marginfiGroup: (0, umi_1.publicKey)(marginfiGroup),
+            signer: this.signer,
+            signerTokenAccount: (0, umi_1.publicKey)((0, utils_1.getTokenAccount)((0, umi_web3js_adapters_1.toWeb3JsPublicKey)(this.signer.publicKey), flashLoan.mint)),
+        }))
+            .add(banksRequiringBalanceClose.map((x) => this.closeBalance(iMfiAccount.accountPk, new web3_js_1.PublicKey(x), marginfiGroup)))
+            .add((0, marginfi_sdk_1.lendingAccountEndFlashloan)(this.umi, {
+            marginfiAccount: (0, umi_1.publicKey)(iMfiAccount.accountPk),
+            signer: this.signer,
+        }).addRemainingAccounts(remainingAccounts));
+    }
+    closeBalance(marginfiAccount, bank, marginfiGroup) {
+        return (0, umi_1.transactionBuilder)().add((0, marginfi_sdk_1.lendingAccountCloseBalance)(this.umi, {
+            signer: this.signer,
+            marginfiAccount: (0, umi_1.publicKey)(marginfiAccount),
+            bank: (0, umi_1.publicKey)(bank),
+            marginfiGroup: (0, umi_1.publicKey)(marginfiGroup),
+        }));
+    }
+}
+exports.MarginfiFlProvider = MarginfiFlProvider;

package/dist/services/index.d.ts

@@ -0,0 +1,6 @@
+export * from "./flashLoans";
+export * from "./rebalance";
+export * from "./solauto";
+export * from "./swap";
+export * from "./transactions";
+//# sourceMappingURL=index.d.ts.map

package/dist/services/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/services/index.ts"],"names":[],"mappings":"AAAA,cAAc,cAAc,CAAC;AAC7B,cAAc,aAAa,CAAC;AAC5B,cAAc,WAAW,CAAC;AAC1B,cAAc,QAAQ,CAAC;AACvB,cAAc,gBAAgB,CAAC"}

package/dist/services/index.js

@@ -0,0 +1,21 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./flashLoans"), exports);
+__exportStar(require("./rebalance"), exports);
+__exportStar(require("./solauto"), exports);
+__exportStar(require("./swap"), exports);
+__exportStar(require("./transactions"), exports);

package/dist/services/rebalance/index.d.ts

@@ -0,0 +1,3 @@
+export * from "./rebalanceTxBuilder";
+export * from "./rebalanceValues";
+//# sourceMappingURL=index.d.ts.map

package/dist/services/rebalance/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/services/rebalance/index.ts"],"names":[],"mappings":"AAAA,cAAc,sBAAsB,CAAC;AACrC,cAAc,mBAAmB,CAAC"}

package/dist/services/rebalance/index.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./rebalanceTxBuilder"), exports);
+__exportStar(require("./rebalanceValues"), exports);

package/dist/services/rebalance/rebalanceSwapManager.d.ts

@@ -0,0 +1,30 @@
+import { QuoteResponse } from "@jup-ag/api";
+import { FlashLoanRequirements } from "../../types";
+import { SolautoClient } from "../solauto";
+import { SwapParams } from "../swap";
+import { RebalanceValues } from "./rebalanceValues";
+export declare class RebalanceSwapManager {
+    private client;
+    private values;
+    private flRequirements?;
+    private targetLiqUtilizationRateBps?;
+    swapParams: SwapParams;
+    swapQuote?: QuoteResponse;
+    flBorrowAmount?: bigint;
+    private jupSwapManager;
+    constructor(client: SolautoClient, values: RebalanceValues, flRequirements?: FlashLoanRequirements | undefined, targetLiqUtilizationRateBps?: number | undefined);
+    private isBoost;
+    private usdToSwap;
+    private postRebalanceLiqUtilizationRateBps;
+    private findSufficientQuote;
+    private swapDetails;
+    private bigIntWithIncrement;
+    setSwapParams(attemptNum: number): Promise<void>;
+    getSwapTxData(): Promise<{
+        swapQuote: QuoteResponse;
+        lookupTableAddresses: string[];
+        setupInstructions: import("@metaplex-foundation/umi").TransactionBuilder;
+        swapIx: import("@metaplex-foundation/umi").TransactionBuilder;
+    }>;
+}
+//# sourceMappingURL=rebalanceSwapManager.d.ts.map

package/dist/services/rebalance/rebalanceSwapManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rebalanceSwapManager.d.ts","sourceRoot":"","sources":["../../../src/services/rebalance/rebalanceSwapManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAkB,UAAU,EAAa,MAAM,SAAS,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAapD,qBAAa,oBAAoB;IAQ7B,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,cAAc,CAAC;IACvB,OAAO,CAAC,2BAA2B,CAAC;IAV/B,UAAU,EAAG,UAAU,CAAC;IACxB,SAAS,CAAC,EAAE,aAAa,CAAC;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IAE/B,OAAO,CAAC,cAAc,CAAkB;gBAG9B,MAAM,EAAE,aAAa,EACrB,MAAM,EAAE,eAAe,EACvB,cAAc,CAAC,EAAE,qBAAqB,YAAA,EACtC,2BAA2B,CAAC,EAAE,MAAM,YAAA;IAK9C,OAAO,CAAC,OAAO;IAIf,OAAO,CAAC,SAAS;IAIjB,OAAO,CAAC,kCAAkC;YA6B5B,mBAAmB;IAoCjC,OAAO,CAAC,WAAW;IAoBnB,OAAO,CAAC,mBAAmB;IAIrB,aAAa,CAAC,UAAU,EAAE,MAAM;IA4EhC,aAAa;;;;;;CAWpB"}

package/dist/services/rebalance/rebalanceSwapManager.js

@@ -0,0 +1,144 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RebalanceSwapManager = void 0;
+const swap_1 = require("../swap");
+const generated_1 = require("../../generated");
+const utils_1 = require("../../utils");
+const umi_web3js_adapters_1 = require("@metaplex-foundation/umi-web3js-adapters");
+class RebalanceSwapManager {
+    constructor(client, values, flRequirements, targetLiqUtilizationRateBps) {
+        this.client = client;
+        this.values = values;
+        this.flRequirements = flRequirements;
+        this.targetLiqUtilizationRateBps = targetLiqUtilizationRateBps;
+        this.jupSwapManager = new swap_1.JupSwapManager(client.signer);
+    }
+    isBoost() {
+        return this.values.rebalanceDirection === generated_1.RebalanceDirection.Boost;
+    }
+    usdToSwap() {
+        return Math.abs(this.values.debtAdjustmentUsd);
+    }
+    postRebalanceLiqUtilizationRateBps(swapOutputAmount) {
+        let supplyUsd = this.client.solautoPosition.supplyUsd();
+        // TODO: add token balance change
+        let debtUsd = this.client.solautoPosition.debtUsd();
+        const outputToken = (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(this.isBoost()
+            ? this.client.solautoPosition.state().supply.mint
+            : this.client.solautoPosition.state().debt.mint);
+        const swapOutputUsd = swapOutputAmount
+            ? (0, utils_1.fromBaseUnit)(swapOutputAmount, (0, utils_1.tokenInfo)(outputToken).decimals) *
+                ((0, utils_1.safeGetPrice)(outputToken) ?? 0)
+            : this.usdToSwap();
+        supplyUsd = this.isBoost()
+            ? supplyUsd + swapOutputUsd
+            : supplyUsd - this.usdToSwap();
+        debtUsd = this.isBoost()
+            ? debtUsd + this.usdToSwap()
+            : debtUsd - swapOutputUsd;
+        return (0, utils_1.getLiqUtilzationRateBps)(supplyUsd, debtUsd, this.client.solautoPosition.state().liqThresholdBps ?? 0);
+    }
+    async findSufficientQuote(swapInput, criteria) {
+        let swapQuote;
+        let insufficient = false;
+        for (let i = 0; i < 10; i++) {
+            (0, utils_1.consoleLog)("Finding sufficient quote...");
+            swapQuote = await this.jupSwapManager.getQuote(swapInput);
+            const outputAmount = parseInt(swapQuote.outAmount);
+            const postRebalanceRate = this.postRebalanceLiqUtilizationRateBps(BigInt(outputAmount));
+            insufficient = criteria.minOutputAmount
+                ? outputAmount < Number(criteria.minOutputAmount)
+                : criteria.minLiqUtilizationRateBps
+                    ? postRebalanceRate < criteria.minLiqUtilizationRateBps
+                    : postRebalanceRate > criteria.maxLiqUtilizationRateBps;
+            if (insufficient) {
+                (0, utils_1.consoleLog)("Insufficient swap quote:", swapQuote);
+                swapInput.amount = this.bigIntWithIncrement(swapInput.amount, 0.01);
+            }
+            else {
+                break;
+            }
+        }
+        return swapQuote;
+    }
+    swapDetails() {
+        const input = this.isBoost()
+            ? this.client.solautoPosition.state().debt
+            : this.client.solautoPosition.state().supply;
+        const output = this.isBoost()
+            ? this.client.solautoPosition.state().supply
+            : this.client.solautoPosition.state().debt;
+        let inputAmount = (0, utils_1.toBaseUnit)(this.usdToSwap() / (0, utils_1.safeGetPrice)(input.mint), input.decimals);
+        return {
+            input,
+            output,
+            inputAmount,
+        };
+    }
+    bigIntWithIncrement(num, inc) {
+        return num + BigInt(Math.round(Number(num) * inc));
+    }
+    async setSwapParams(attemptNum) {
+        const rebalanceToZero = this.targetLiqUtilizationRateBps === 0;
+        let { input, output, inputAmount } = this.swapDetails();
+        let outputAmount = rebalanceToZero
+            ? output.amountUsed.baseUnit +
+                BigInt(Math.round(Number(output.amountUsed.baseUnit) *
+                    // Add this small percentage to account for the APR on the debt between now and the transaction
+                    0.0001))
+            : (0, utils_1.toBaseUnit)(this.usdToSwap() / (0, utils_1.safeGetPrice)(output.mint), output.decimals);
+        const flashLoanRepayFromDebt = !this.isBoost() &&
+            this.flRequirements &&
+            this.flRequirements.liquiditySource === generated_1.TokenType.Debt;
+        const exactOut = flashLoanRepayFromDebt && !rebalanceToZero;
+        const exactIn = !exactOut;
+        if (exactIn && (rebalanceToZero || this.values.repayingCloseToMaxLtv)) {
+            inputAmount = this.bigIntWithIncrement(inputAmount, 0.005);
+        }
+        const swapAmount = exactOut
+            ? this.flRequirements
+                ? this.bigIntWithIncrement(outputAmount, this.flRequirements.flFeeBps ?? 0)
+                : outputAmount
+            : inputAmount;
+        const swapInput = {
+            inputMint: (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(input.mint),
+            outputMint: (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(output.mint),
+            exactIn,
+            exactOut,
+            amount: swapAmount,
+        };
+        (0, utils_1.consoleLog)("Swap input:", swapInput);
+        if (exactIn && (rebalanceToZero || this.values.repayingCloseToMaxLtv)) {
+            this.swapQuote = await this.findSufficientQuote(swapInput, {
+                minOutputAmount: rebalanceToZero ? outputAmount : undefined,
+                maxLiqUtilizationRateBps: this.values.repayingCloseToMaxLtv
+                    ? (0, utils_1.maxRepayToBps)(this.client.solautoPosition.state().maxLtvBps ?? 0, this.client.solautoPosition.state().liqThresholdBps ?? 0) - 15
+                    : undefined,
+            });
+        }
+        if (this.flRequirements) {
+            this.flBorrowAmount = exactOut
+                ? outputAmount
+                : this.swapQuote
+                    ? BigInt(parseInt(this.swapQuote.inAmount))
+                    : inputAmount;
+        }
+        this.swapParams = {
+            ...swapInput,
+            destinationWallet: flashLoanRepayFromDebt
+                ? (0, umi_web3js_adapters_1.toWeb3JsPublicKey)(this.client.signer.publicKey)
+                : this.client.solautoPosition.publicKey,
+            slippageIncFactor: 0.2 + attemptNum * 0.25,
+        };
+    }
+    async getSwapTxData() {
+        const { jupQuote, lookupTableAddresses, setupInstructions, swapIx } = await this.jupSwapManager.getJupSwapTxData(this.swapParams);
+        return {
+            swapQuote: jupQuote,
+            lookupTableAddresses,
+            setupInstructions,
+            swapIx,
+        };
+    }
+}
+exports.RebalanceSwapManager = RebalanceSwapManager;