feat: Add execution for curve
- Add CurveSwapStructEncoder and tests - Add CurveSwapExecutorExposed and tests - Add needed interfaces #time 0m #time 0m #time 0m
This commit is contained in:
Diana Carvalho
98
propeller-swap-encoders/propeller_swap_encoders/curve.py
Normal file
98
propeller-swap-encoders/propeller_swap_encoders/curve.py
Normal file
@@ -0,0 +1,98 @@
|
||||
import enum
|
||||
from typing import Any
|
||||
|
||||
from core.encoding.interface import EncodingContext, SwapStructEncoder
|
||||
from core.type_aliases import Address
|
||||
from eth_abi.packed import encode_abi_packed
|
||||
from eth_utils import to_checksum_address
|
||||
|
||||
curve_config = {
|
||||
# curve pool type 4
|
||||
"eth_stable_pools": [
|
||||
"0xA96A65c051bF88B4095Ee1f2451C2A9d43F53Ae2",
|
||||
"0xF9440930043eb3997fc70e1339dBb11F341de7A8",
|
||||
"0xa1F8A6807c402E4A15ef4EBa36528A3FED24E577",
|
||||
"0xBfAb6FA95E0091ed66058ad493189D2cB29385E6",
|
||||
"0x94B17476A93b3262d87B9a326965D1E91f9c13E7",
|
||||
],
|
||||
# curve pool type 7
|
||||
"v2_eth_pools": [
|
||||
"0x9409280DC1e6D33AB7A8C6EC03e5763FB61772B5",
|
||||
"0x5FAE7E604FC3e24fd43A72867ceBaC94c65b404A",
|
||||
"0x0f3159811670c117c372428D4E69AC32325e4D0F",
|
||||
"0x838af967537350D2C44ABB8c010E49E32673ab94",
|
||||
"0xC26b89A667578ec7b3f11b2F98d6Fd15C07C54ba",
|
||||
"0x6bfE880Ed1d639bF80167b93cc9c56a39C1Ba2dC",
|
||||
"0x0E9B5B092caD6F1c5E6bc7f89Ffe1abb5c95F1C2",
|
||||
"0x21410232B484136404911780bC32756D5d1a9Fa9",
|
||||
"0xfB8814D005C5f32874391e888da6eB2fE7a27902",
|
||||
"0xe0e970a99bc4F53804D8145beBBc7eBc9422Ba7F",
|
||||
"0x6e314039f4C56000F4ebb3a7854A84cC6225Fb92",
|
||||
"0xf861483fa7E511fbc37487D91B6FAa803aF5d37c",
|
||||
],
|
||||
}
|
||||
|
||||
|
||||
class CurvePoolType(enum.IntEnum):
|
||||
"""
|
||||
Represents different swap logics of curve pools. For more details, please see
|
||||
CurveSwapMethodV1 in defibot-contracts repository.
|
||||
"""
|
||||
|
||||
simple = 0
|
||||
simple_no_amount = 1
|
||||
tricrypto = 3
|
||||
eth_stableswap = 4
|
||||
underlying = 5
|
||||
underlying_no_amount = 6
|
||||
crypto_v2 = 7
|
||||
crypto_v2_2_tokens_not_factory = 8
|
||||
|
||||
|
||||
curve_v2_pool_type_mapping: dict[str, CurvePoolType] = {
|
||||
"tricrypto2_non_factory": CurvePoolType.tricrypto,
|
||||
"two_token_factory": CurvePoolType.crypto_v2,
|
||||
"two_token_non_factory": CurvePoolType.crypto_v2_2_tokens_not_factory,
|
||||
}
|
||||
|
||||
|
||||
class CurveSwapStructEncoder(SwapStructEncoder):
|
||||
eth_stable_pools: list[str] = curve_config["eth_stable_pools"]
|
||||
v2_eth_pools = curve_config["v2_eth_pools"]
|
||||
|
||||
def encode_swap_struct(
|
||||
self, swap: dict[str, Any], receiver: Address, encoding_context: EncodingContext
|
||||
) -> bytes:
|
||||
|
||||
pool_type = swap["pool_type"]
|
||||
if pool_type == "CurveSimulatedPoolState":
|
||||
curve_pool_type = (
|
||||
CurvePoolType.tricrypto
|
||||
if swap["protocol_specific_attrs"]["is_curve_tricrypto"]
|
||||
else CurvePoolType.simple_no_amount
|
||||
)
|
||||
elif to_checksum_address(swap["pool_id"]) in self.v2_eth_pools:
|
||||
curve_pool_type = CurvePoolType.crypto_v2
|
||||
elif to_checksum_address(swap["pool_id"]) in self.eth_stable_pools:
|
||||
curve_pool_type = CurvePoolType.eth_stableswap
|
||||
else:
|
||||
curve_pool_type = (
|
||||
curve_v2_pool_type_mapping[
|
||||
swap["protocol_specific_attrs"]["curve_v2_pool_type"]
|
||||
]
|
||||
if pool_type == "CurveV2PoolState"
|
||||
else CurvePoolType.simple_no_amount
|
||||
)
|
||||
|
||||
return encode_abi_packed(
|
||||
["address", "address", "address", "uint8", "uint8", "uint8", "bool"],
|
||||
[
|
||||
swap["buy_token"].address,
|
||||
swap["pool_id"],
|
||||
receiver,
|
||||
curve_pool_type,
|
||||
swap["pool_tokens"].index(swap["sell_token"]),
|
||||
swap["pool_tokens"].index(swap["buy_token"]),
|
||||
swap["token_approval_needed"],
|
||||
],
|
||||
)
|
||||
@@ -0,0 +1,148 @@
|
||||
from core.encoding.interface import EncodingContext
|
||||
from core.models.evm.ethereum_token import EthereumToken
|
||||
|
||||
from propeller_swap_encoders.curve import CurveSwapStructEncoder
|
||||
|
||||
WETH = EthereumToken(
|
||||
symbol="WETH",
|
||||
address="0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2",
|
||||
decimals=18,
|
||||
gas=0,
|
||||
)
|
||||
USDT = EthereumToken(
|
||||
symbol="USDT", address="0xdAC17F958D2ee523a2206206994597C13D831ec7", decimals=6
|
||||
)
|
||||
WBTC = EthereumToken(
|
||||
symbol="WBTC", address="0x2260fac5e5542a773aa44fbcfedf7c193bc2c599", decimals=8
|
||||
)
|
||||
|
||||
|
||||
def test_encode_curve_v2():
|
||||
bob = "0x000000000000000000000000000000000000007B"
|
||||
|
||||
swap = {
|
||||
"pool_id": "0xD51a44d3FaE010294C616388b506AcdA1bfAAE46",
|
||||
"sell_token": USDT,
|
||||
"buy_token": WETH,
|
||||
"split": 0,
|
||||
"sell_amount": 0,
|
||||
"buy_amount": 100,
|
||||
"token_approval_needed": False,
|
||||
"pool_tokens": (USDT, WBTC, WETH),
|
||||
"pool_type": "CurveV2PoolState",
|
||||
"protocol_specific_attrs": {
|
||||
"curve_v2_pool_type": "tricrypto2_non_factory",
|
||||
"is_curve_tricrypto": None,
|
||||
"quote": None,
|
||||
"pool_fee": None,
|
||||
},
|
||||
}
|
||||
|
||||
curve_encoder = CurveSwapStructEncoder()
|
||||
encoded = curve_encoder.encode_swap_struct(
|
||||
swap, receiver=bob, encoding_context=EncodingContext()
|
||||
)
|
||||
assert (
|
||||
encoded.hex()
|
||||
==
|
||||
# buy token
|
||||
"c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2"
|
||||
# pool address
|
||||
"d51a44d3fae010294c616388b506acda1bfaae46"
|
||||
# receiver
|
||||
"000000000000000000000000000000000000007b"
|
||||
# pool type (tricrypto = 3)
|
||||
"03"
|
||||
# i (sell token index)
|
||||
"00"
|
||||
# j (buy token index)
|
||||
"02"
|
||||
# token_approval_needed
|
||||
"00"
|
||||
)
|
||||
|
||||
|
||||
def test_encode_curve_v1():
|
||||
bob = "0x000000000000000000000000000000000000007B"
|
||||
swap = {
|
||||
"pool_id": "bebc44782c7db0a1a60cb6fe97d0b483032ff1c7",
|
||||
"sell_token": USDT,
|
||||
"buy_token": WETH,
|
||||
"split": 0,
|
||||
"sell_amount": 0,
|
||||
"buy_amount": 100,
|
||||
"token_approval_needed": False,
|
||||
"pool_tokens": (USDT, WBTC, WETH),
|
||||
"pool_type": "CurveV1PoolState",
|
||||
"protocol_specific_attrs": {
|
||||
"curve_v2_pool_type": None,
|
||||
"is_curve_tricrypto": None,
|
||||
"quote": None,
|
||||
"pool_fee": 1000000,
|
||||
},
|
||||
}
|
||||
curve_encoder = CurveSwapStructEncoder()
|
||||
encoded = curve_encoder.encode_swap_struct(
|
||||
swap, receiver=bob, encoding_context=EncodingContext()
|
||||
)
|
||||
assert (
|
||||
encoded.hex()
|
||||
==
|
||||
# buy token
|
||||
"c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2"
|
||||
# pool address
|
||||
"bebc44782c7db0a1a60cb6fe97d0b483032ff1c7"
|
||||
# receiver
|
||||
"000000000000000000000000000000000000007b"
|
||||
# pool type (simple_no_amount = 1)
|
||||
"01"
|
||||
# i (sell token index)
|
||||
"00"
|
||||
# j (buy token index)
|
||||
"02"
|
||||
# token_approval_needed
|
||||
"00"
|
||||
)
|
||||
|
||||
|
||||
def test_encode_curve_evm_crypto_pool():
|
||||
bob = "0x000000000000000000000000000000000000007B"
|
||||
swap = {
|
||||
"pool_id": "bebc44782c7db0a1a60cb6fe97d0b483032ff1c7",
|
||||
"sell_token": USDT,
|
||||
"buy_token": WETH,
|
||||
"split": 0,
|
||||
"sell_amount": 0,
|
||||
"buy_amount": 100,
|
||||
"token_approval_needed": False,
|
||||
"pool_tokens": (USDT, WBTC, WETH),
|
||||
"pool_type": "CurveSimulatedPoolState",
|
||||
"protocol_specific_attrs": {
|
||||
"curve_v2_pool_type": None,
|
||||
"is_curve_tricrypto": True,
|
||||
"quote": None,
|
||||
"pool_fee": None,
|
||||
},
|
||||
}
|
||||
curve_encoder = CurveSwapStructEncoder()
|
||||
encoded = curve_encoder.encode_swap_struct(
|
||||
swap, receiver=bob, encoding_context=EncodingContext()
|
||||
)
|
||||
assert (
|
||||
encoded.hex()
|
||||
==
|
||||
# buy token
|
||||
"c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2"
|
||||
# pool address
|
||||
"bebc44782c7db0a1a60cb6fe97d0b483032ff1c7"
|
||||
# receiver
|
||||
"000000000000000000000000000000000000007b"
|
||||
# pool type (tricrypto = 3)
|
||||
"03"
|
||||
# i (sell token index)
|
||||
"00"
|
||||
# j (buy token index)
|
||||
"02"
|
||||
# token_approval_needed
|
||||
"00"
|
||||
)
|
||||
Reference in New Issue
Block a user