From 1af0a3e9481be6a6cb6abec0aa486cf347030161 Mon Sep 17 00:00:00 2001
From: Mikko Ohtamaa <mikko@redinnovation.com>
Date: Sun, 24 Nov 2024 14:45:19 +0100
Subject: [PATCH] Lagoon initial work

---
 eth_defi/lagoon/__init__.py |   1 +
 eth_defi/lagoon/vault.py    | 153 ++++++++++++++++++++++++++++++++++++
 2 files changed, 154 insertions(+)
 create mode 100644 eth_defi/lagoon/__init__.py
 create mode 100644 eth_defi/lagoon/vault.py

diff --git a/eth_defi/lagoon/__init__.py b/eth_defi/lagoon/__init__.py
new file mode 100644
index 00000000..3c04617d
--- /dev/null
+++ b/eth_defi/lagoon/__init__.py
@@ -0,0 +1 @@
+"""Lagoon vault protocol integration."""
\ No newline at end of file
diff --git a/eth_defi/lagoon/vault.py b/eth_defi/lagoon/vault.py
new file mode 100644
index 00000000..72454bad
--- /dev/null
+++ b/eth_defi/lagoon/vault.py
@@ -0,0 +1,153 @@
+from web3 import Web3
+
+from eth_defi.vault.base import VaultBase, VaultSpec, VaultInfo
+
+
+class LagoonVaultInfo(VaultInfo):
+    """TODO: Add Lagoon vault info query"""
+    pass
+
+
+class LagoonVault(VaultBase):
+    """Python interface for interacting with Velvet Capital vaults."""
+
+    def __init__(
+        self,
+        web3: Web3,
+        spec: VaultSpec,
+    ):
+        """
+        :param spec:
+            Address must be Velvet portfolio address (not vault address)
+        """
+        assert isinstance(web3, Web3)
+        assert isinstance(spec, VaultSpec)
+        self.web3 = web3
+
+    def has_block_range_event_support(self):
+        return True
+
+    def get_flow_manager(self):
+        raise NotImplementedError("Velvet does not support individual deposit/redemption events yet")
+
+    def fetch_info(self) -> LagoonVaultInfo:
+        """Read vault parameters from the chain."""
+
+
+
+        raise NotImplementedError("Velvet does not support fetching info yet")
+
+    @cached_property
+    def info(self) -> VelvetVaultInfo:
+        return self.fetch_info()
+
+    @property
+    def vault_address(self) -> HexAddress:
+        return self.info["vaultAddress"]
+
+    @property
+    def owner_address(self) -> HexAddress:
+        return self.info["owner"]
+
+    @property
+    def portfolio_address(self) -> HexAddress:
+        return self.info["portfolio"]
+
+    @property
+    def name(self) -> str:
+        return self.info["name"]
+
+    @property
+    def token_symbol(self) -> str:
+        return self.info["symbol"]
+
+    def fetch_portfolio(
+        self,
+        universe: TradingUniverse,
+        block_identifier: BlockIdentifier | None = None,
+    ) -> VaultPortfolio:
+        """Read the current token balances of a vault.
+
+        - SHould be supported by all implementations
+        """
+
+        vault_address = self.info["vaultAddress"]
+
+        erc20_balances = fetch_erc20_balances_multicall(
+            self.web3,
+            vault_address,
+            universe.spot_token_addresses,
+            block_identifier=block_identifier,
+            decimalise=True,
+        )
+        return VaultPortfolio(
+            spot_erc20=erc20_balances,
+        )
+
+    def prepare_swap_with_enso(
+        self,
+        token_in: HexAddress | str,
+        token_out: HexAddress | str,
+        swap_amount: int,
+        slippage: float,
+        remaining_tokens: set,
+        swap_all=False,
+        from_: HexAddress | str | None = None,
+    ) -> dict:
+        """Prepare a swap transaction using Enso intent engine and Vevlet API.
+
+        :param from_:
+            Fill int the from field for the tx data.
+
+            Used with Anvil and unlocked accounts.
+        """
+
+        if swap_all:
+            remaining_tokens.remove(token_in)
+
+        tx_data = swap_with_velvet_and_enso(
+            rebalance_address=self.info["rebalancing"],
+            owner_address=self.owner_address,
+            token_in=token_in,
+            token_out=token_out,
+            swap_amount=swap_amount,
+            slippage=slippage,
+            remaining_tokens=remaining_tokens,
+            chain_id=self.web3.eth.chain_id,
+        )
+
+        if from_:
+            tx_data["from"] = Web3.to_checksum_address(from_)
+
+        return tx_data
+
+    def prepare_deposit_with_enso(
+        self,
+        from_: HexAddress | str,
+        deposit_token_address: HexAddress | str,
+        amount: int,
+    ):
+        """Prepare a deposit transaction with Enso intents.
+
+        - Velvet trades any incoming assets and distributes them on open positions
+        """
+        tx_data = deposit_to_velvet(
+            portfolio=self.portfolio_address,
+            from_address=from_,
+            deposit_token_address=deposit_token_address,
+            amount=amount,
+            chain_id=self.web3.eth.chain_id,
+        )
+        return tx_data
+
+    def _make_api_request(
+        self,
+        endpoint: str,
+        params: dict | None = None,
+    ) -> dict:
+        url = f"{self.api_url}/{endpoint}"
+        resp = self.session.get(url)
+        resp.raise_for_status()
+        data = resp.json()
+        return data
+