src/test/tokenhelper.sol

1// SPDX-License-Identifier: UNLICENSED
2 
3pragma solidity ^0.8.0;
4 
5library TokenHelper {
6    function callTokenFunction(
7        address tokenAddress,
8        bytes memory data,
9        bool staticCall
10    ) private returns (bytes memory) {
11        (bool success, bytes memory result) = staticCall ? tokenAddress.staticcall(data) : tokenAddress.call(data);
12        require(success, "Failed to call token function");
13        return result;
14    }
15 
16    function getTokenBalance(address tokenAddress, address targetAddress) internal returns (uint256) {
17        bytes memory result =
18            callTokenFunction(tokenAddress, abi.encodeWithSignature("balanceOf(address)", targetAddress), true);
19        return abi.decode(result, (uint256));
20    }
21 
22    function getTokenDecimals(
23        address tokenAddress
24    ) internal returns (uint8) {
25        bytes memory result = callTokenFunction(tokenAddress, abi.encodeWithSignature("decimals()"), true);
26        return abi.decode(result, (uint8));
27    }
28 
29    function getTokenSymbol(
30        address tokenAddress
31    ) internal returns (string memory) {
32        bytes memory result = callTokenFunction(tokenAddress, abi.encodeWithSignature("symbol()"), true);
33        return abi.decode(result, (string));
34    }
35 
36    function approveToken(address token, address spender, uint256 spendAmount) internal returns (bool) {
37        bytes memory result =
38            callTokenFunction(token, abi.encodeWithSignature("approve(address,uint256)", spender, spendAmount), false);
39        return abi.decode(result, (bool));
40    }
41 
42    function transferToken(address token, address receiver, uint256 amount) internal returns (bool) {
43        bytes memory result =
44            callTokenFunction(token, abi.encodeWithSignature("transfer(address,uint256)", receiver, amount), false);
45        return abi.decode(result, (bool));
46    }
47}