@affluent-org/sdk 0.0.1

package/dist/contracts/vault/share-vault/index.js

@@ -0,0 +1,373 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ShareVault = void 0;
+const core_1 = require("@ton/core");
+const jetton_minter_1 = require("../../wton/jetton-minter");
+class ShareVault {
+    address;
+    init;
+    constructor(address, init) {
+        this.address = address;
+        this.init = init;
+    }
+    static Op = {
+        // owner
+        SetWhitelistPools: 0x59292de8,
+        SetTargetWeight: 0x73ffc8,
+        SetGasFee: 0x50d87d1d,
+        ChangeManager: 0x6780b0d9,
+        UpgradeCode: 0x61bddf8b,
+        ChangeAdmin: 0x4840664f,
+        // manager
+        SupplyToPool: 0x996bd32d,
+        WithdrawFromPool: 0x188047fa,
+        // user
+        Supply: 0x480074d8,
+        Withdraw: 0xcb03bfaf,
+        RequestAssetWalletAddress: 0x16359e7d,
+        ActionNotification: 0xa1b21e8b,
+        ProvideAggregatedPool: 0xb2a4c51b,
+        TakeAggregatedPool: 0x75a40ce7,
+    };
+    static Error = {
+        UnknownOp: 0xffff,
+        InvalidWallet: 1002,
+        InvalidAddress: 1003,
+        NotFoundPoolState: 1004,
+        ActionTimeout: 1005,
+        InvalidBurnRequest: 1006,
+        NotWhitelistedPool: 1007,
+        NotEnoughCash: 1008,
+        NotEnoughBalance: 1009,
+    };
+    static Gas = {
+        baseFee: (0, core_1.toNano)(0.1),
+    };
+    static createFromAddress(address) {
+        return new ShareVault(address);
+    }
+    async sendMsgBody(provider, via, value, body) {
+        await provider.internal(via, {
+            value,
+            sendMode: core_1.SendMode.PAY_GAS_SEPARATELY,
+            body: body,
+        });
+        return body;
+    }
+    async sendSetTargetWeight(provider, via, value, opts) {
+        const body = (0, core_1.beginCell)()
+            .storeUint(ShareVault.Op.SetTargetWeight, 32)
+            .storeUint(opts.queryID ?? 0, 64)
+            .storeAddress(opts.pool)
+            .storeUint(opts.targetWeight, 14)
+            .endCell();
+        await provider.internal(via, {
+            value,
+            sendMode: core_1.SendMode.PAY_GAS_SEPARATELY,
+            body: body,
+        });
+        return body;
+    }
+    async sendSetGasFee(provider, via, value, opts) {
+        const body = ShareVault.createSetGasConfigMsgBody(opts);
+        await provider.internal(via, {
+            value,
+            sendMode: core_1.SendMode.PAY_GAS_SEPARATELY,
+            body: body,
+        });
+        return body;
+    }
+    async sendSupplyToPool(provider, via, value, opts) {
+        const body = (0, core_1.beginCell)()
+            .storeUint(ShareVault.Op.SupplyToPool, 32)
+            .storeUint(opts.queryID ?? 0, 64)
+            .storeAddress(opts.pool)
+            .storeCoins(opts.amount)
+            .endCell();
+        await provider.internal(via, {
+            value,
+            sendMode: core_1.SendMode.PAY_GAS_SEPARATELY,
+            body: body,
+        });
+        return body;
+    }
+    async sendWithdrawFromPool(provider, via, value, opts) {
+        const body = (0, core_1.beginCell)()
+            .storeUint(ShareVault.Op.WithdrawFromPool, 32)
+            .storeUint(opts.queryID ?? 0, 64)
+            .storeAddress(opts.pool)
+            .storeUint(opts.isShare ? 1 : 0, 1)
+            .storeCoins(opts.amount)
+            .endCell();
+        await provider.internal(via, {
+            value,
+            sendMode: core_1.SendMode.PAY_GAS_SEPARATELY,
+            body: body,
+        });
+        return body;
+    }
+    async sendChangeManager(provider, via, value, opts) {
+        await provider.internal(via, {
+            value: value,
+            sendMode: core_1.SendMode.PAY_GAS_SEPARATELY,
+            body: (0, core_1.beginCell)()
+                .storeUint(ShareVault.Op.ChangeManager, 32)
+                .storeUint(opts.queryID ?? 0, 64)
+                .storeAddress(opts.address)
+                .endCell(),
+        });
+    }
+    async sendChangeAdmin(provider, via, value, opts) {
+        await provider.internal(via, {
+            value: value,
+            sendMode: core_1.SendMode.PAY_GAS_SEPARATELY,
+            body: ShareVault.createChangeAdminMsgBody(opts),
+        });
+    }
+    static createChangeAdminMsgBody(opts) {
+        return (0, core_1.beginCell)()
+            .storeUint(ShareVault.Op.ChangeAdmin, 32)
+            .storeUint(opts.queryID ?? 0, 64)
+            .storeAddress(opts.address)
+            .endCell();
+    }
+    static parseChangeAdminMsgBody(cell) {
+        const slice = cell.beginParse();
+        const op = slice.loadUint(32);
+        const queryID = slice.loadUint(64);
+        const address = slice.loadAddress();
+        return { op, queryID, address };
+    }
+    static createSetGasConfigMsgBody(params) {
+        const gasFeeCell = (0, core_1.beginCell)()
+            .storeCoins(params.gasFee.supply)
+            .storeCoins(params.gasFee.withdraw)
+            .storeCoins(params.gasFee.factorialSupply)
+            .storeCoins(params.gasFee.factorialWithdraw)
+            .storeCoins(params.gasFee.mintVaultShareTokens)
+            .endCell();
+        return (0, core_1.beginCell)()
+            .storeUint(ShareVault.Op.SetGasFee, 32)
+            .storeUint(params.queryID ?? 0, 64)
+            .storeRef(gasFeeCell)
+            .endCell();
+    }
+    static createSetWhitelistPoolsMsgBody(params) {
+        return (0, core_1.beginCell)()
+            .storeUint(ShareVault.Op.SetWhitelistPools, 32)
+            .storeUint(params.queryID ?? 0, 64)
+            .storeRef(ShareVault.createWhitelistsCell(params.whitelists))
+            .endCell();
+    }
+    static createChangeManagerMsgBody(params) {
+        return (0, core_1.beginCell)()
+            .storeUint(ShareVault.Op.ChangeManager, 32)
+            .storeUint(params.queryID ?? 0, 64)
+            .storeAddress(params.manager)
+            .endCell();
+    }
+    static createWhitelistsCell(whitelists) {
+        const chunkedWhitelists = [];
+        for (let i = 0; i < whitelists.length; i += 3) {
+            chunkedWhitelists.push(whitelists.slice(i, i + 3));
+        }
+        const cells = chunkedWhitelists.map((chunk) => {
+            let cell = (0, core_1.beginCell)();
+            cell = cell.storeUint(chunk.length, 2);
+            chunk.forEach(([address, flag]) => {
+                cell = cell.storeAddress(address).storeUint(flag, 1);
+            });
+            return cell;
+        });
+        const finalCell = cells.reduce((acc, cur) => {
+            return acc ? cur.storeRef(acc) : cur;
+        }, null);
+        return finalCell?.endCell() || (0, core_1.beginCell)().endCell();
+    }
+    async getWalletAddress(provider, address) {
+        const result = await provider.get("get_wallet_address", [
+            {
+                type: "slice",
+                cell: (0, core_1.beginCell)().storeAddress(address).endCell(),
+            },
+        ]);
+        return result.stack.readAddress();
+    }
+    async getPoolAggregatorAddress(provider, address) {
+        const result = await provider.get("get_pool_aggregator_address", [
+            {
+                type: "slice",
+                cell: (0, core_1.beginCell)().storeAddress(address).endCell(),
+            },
+        ]);
+        return result.stack.readAddress();
+    }
+    static calculateGasFee(actionOp, baseFee, poolCount, gas, isWTON = false) {
+        if (actionOp === ShareVault.Op.Supply) {
+            let totalFee = baseFee;
+            totalFee += gas.supply;
+            totalFee += gas.factorialSupply;
+            if (isWTON)
+                totalFee += jetton_minter_1.WTON_UNWRAP_GAS;
+            return totalFee;
+        }
+        else if (actionOp === ShareVault.Op.Withdraw) {
+            let totalFee = baseFee;
+            totalFee += gas.withdraw;
+            totalFee += BigInt(poolCount) * gas.factorialWithdraw;
+            if (isWTON)
+                totalFee += BigInt(poolCount) * jetton_minter_1.WTON_UNWRAP_GAS;
+            return totalFee;
+        }
+        else if (actionOp === ShareVault.Op.SupplyToPool) {
+            let totalFee = baseFee;
+            totalFee += gas.supply;
+            totalFee += gas.factorialSupply;
+            return totalFee;
+        }
+        else if (actionOp === ShareVault.Op.WithdrawFromPool) {
+            let totalFee = baseFee;
+            totalFee += gas.withdraw;
+            totalFee += gas.factorialWithdraw;
+            return totalFee;
+        }
+        throw "invalid opcode";
+    }
+    async getJettonData(provider) {
+        const _result = await provider.get("get_jetton_data", []);
+        const totalSupply = _result.stack.readBigNumber();
+        _result.stack.readBigNumber();
+        const admin = _result.stack.readAddress();
+        const content = _result.stack.readStringOpt();
+        return {
+            totalSupply,
+            admin,
+            content,
+        };
+    }
+    async sendUpgradeCode(provider, via, value, opts) {
+        await provider.internal(via, {
+            value: value,
+            sendMode: core_1.SendMode.PAY_GAS_SEPARATELY,
+            body: ShareVault.createUpgradeCodeMsgBody(opts),
+        });
+    }
+    async getVaultData(provider) {
+        const state = await provider.getState();
+        if (state.state.type !== "active")
+            throw "not active";
+        if (!state.state.data)
+            throw "not active";
+        const data = core_1.Cell.fromBoc(state.state.data)[0];
+        return {
+            code: core_1.Cell.fromBoc(state.state.code)[0],
+            address: this.address,
+            ...ShareVault.parseVaultData(data),
+        };
+    }
+    static parseVaultData(_data) {
+        const data = _data.beginParse();
+        const totalSupply = data.loadCoins();
+        const ownerAddress = data.loadMaybeAddress();
+        const managerAddress = data.loadMaybeAddress();
+        const assetBalance = data.loadCoins();
+        const cash = data.loadCoins();
+        const totalTargetWeight = data.loadCoins();
+        const nextAggregatorIndex = data.loadCoins();
+        const isWtonVault = data.loadBoolean();
+        const _whitelistedPools = data.loadDict(core_1.Dictionary.Keys.Address(), DictioryValueAccountWhitelistPool);
+        const whitelistedPools = _whitelistedPools.keys().reduce((acc, cur) => {
+            acc[cur.toString()] = {
+                address: cur,
+                ..._whitelistedPools.get(cur),
+            };
+            return acc;
+        }, {});
+        const config = data.loadRef().beginParse();
+        const assetAddress = config.loadAddress();
+        const assetWalletAddress = config.loadMaybeAddress();
+        const content = config.loadRef();
+        const walletCode = config.loadRef();
+        const pool_aggregator_code = config.loadRef();
+        const gasConfig = ShareVault.parseGasConfig(data.loadRef());
+        return {
+            totalSupply,
+            owner: ownerAddress,
+            manager: managerAddress,
+            asset: assetAddress,
+            balance: assetBalance,
+            cash,
+            totalTargetWeight,
+            nextAggregatorIndex,
+            isWtonVault,
+            assetWallet: assetWalletAddress,
+            whitelistedPools,
+            minimumGasFee: gasConfig,
+        };
+    }
+    static parseGasConfig(cell) {
+        const slice = cell.beginParse();
+        return {
+            supply: slice.loadCoins(),
+            withdraw: slice.loadCoins(),
+            factorialSupply: slice.loadCoins(),
+            factorialWithdraw: slice.loadCoins(),
+            mintVaultShareTokens: slice.loadCoins(),
+        };
+    }
+    static createSupplyForwardPayload(opts) {
+        return (0, core_1.beginCell)()
+            .storeAddress(opts.responseAddress)
+            .storeCoins(opts.jettonForwardTonAmount ?? 0n)
+            .storeMaybeRef(opts.jettonForwardPayload)
+            .storeCoins(opts.actionForwardTonAmount ?? 0n)
+            .storeMaybeRef(opts.actionForwardPayload)
+            .endCell();
+    }
+    static createWithdrawForwardPayload(opts = {}) {
+        return (0, core_1.beginCell)()
+            .storeCoins(opts.jettonForwardTonAmount ?? 0n)
+            .storeMaybeRef(opts.jettonForwardPayload)
+            .storeCoins(opts.actionForwardTonAmount ?? 0n)
+            .storeMaybeRef(opts.actionForwardPayload)
+            .endCell();
+    }
+    static createSetGasFeeMsgBody(opts) {
+        return (0, core_1.beginCell)()
+            .storeUint(ShareVault.Op.SetGasFee, 32)
+            .storeUint(opts.queryID ?? 0, 64)
+            .storeRef(opts.gasFee)
+            .endCell();
+    }
+    static parseSetGasFeeMsgBody(cell) {
+        const slice = cell.beginParse();
+        const op = slice.loadUint(32);
+        const queryID = slice.loadUint(64);
+        const gasFee = slice.loadRef();
+        return { op, queryID, gasFee };
+    }
+    async getState(provider) {
+        return await provider.getState();
+    }
+    static createUpgradeCodeMsgBody(opts) {
+        return (0, core_1.beginCell)()
+            .storeUint(ShareVault.Op.UpgradeCode, 32)
+            .storeUint(opts.queryID ?? 0, 64)
+            .storeRef(opts.newCode)
+            .endCell();
+    }
+}
+exports.ShareVault = ShareVault;
+const DictioryValueAccountWhitelistPool = {
+    serialize: (src, builder) => { },
+    parse: (src) => {
+        const isWhitelisted = src.loadBit();
+        const targetWeight = BigInt(src.loadUint(14));
+        const supply = src.loadCoins();
+        return {
+            isWhitelisted,
+            targetWeight,
+            supply,
+        };
+    },
+};