@chorus-one/ethereum/dist/mjs/lib/methods/buildUnstakeTx.js

All-in-one toolkit for building staking dApps on Ethereum network

import { encodeFunctionData } from 'viem';
import { keeperABI } from '../contracts/keeperAbi';
import { VaultABI } from '../contracts/vaultAbi';
import { getHarvestParams } from '../utils/getHarvestParams';
export async function buildUnstakeTx(request) {
    const { userAccount, connector, vault, amount } = request;
    const isCollateralized = await connector.eth.readContract({
        abi: keeperABI,
        address: connector.keeper,
        functionName: 'isCollateralized',
        args: [vault]
    });
    const shares = await connector.eth.readContract({
        abi: VaultABI,
        address: vault,
        functionName: 'convertToShares',
        args: [amount]
    });
    let tx;
    if (isCollateralized) {
        // This branch of logic is invoked when the stake locked in vault
        // is collateralized in form of beacon chain validators
        const canHarvest = await connector.eth.readContract({
            abi: keeperABI,
            address: connector.keeper,
            functionName: 'canHarvest',
            args: [vault]
        });
        if (canHarvest) {
            const harvestParams = await getHarvestParams(connector, vault);
            const updateStateTx = encodeFunctionData({
                abi: VaultABI,
                functionName: 'updateState',
                args: [
                    {
                        proof: harvestParams.proof,
                        rewardsRoot: harvestParams.rewardsRoot,
                        reward: harvestParams.reward,
                        unlockedMevReward: harvestParams.unlockedMevReward
                    }
                ]
            });
            const enterExitQueueTx = encodeFunctionData({
                abi: VaultABI,
                functionName: 'enterExitQueue',
                args: [shares, userAccount]
            });
            tx = encodeFunctionData({
                abi: VaultABI,
                functionName: 'multicall',
                args: [[updateStateTx, enterExitQueueTx]]
            });
        }
        else {
            tx = encodeFunctionData({
                abi: VaultABI,
                functionName: 'enterExitQueue',
                args: [shares, userAccount]
            });
        }
    }
    else {
        // This branch of logic is invoked when the stake locked in vault
        // does not actually power any validators
        tx = encodeFunctionData({
            abi: VaultABI,
            functionName: 'redeem',
            args: [shares, userAccount]
        });
    }
    return {
        to: vault,
        data: tx
    };
}