Kodiak integration

constants/kodiak.py

@@ -0,0 +1,11 @@
+from enum import Enum
+
+KODIAK_API_URL = "https://backend.kodiak.finance"
+
+class KodiakIslandAddress(Enum):
+    USDE = "0xE5A2ab5D2fb268E5fF43A5564e44c3309609aFF9"
+    SUSDE = "0xD5B6EA3544a51BfdDa7E6926BdF778339801dFe8"
+
+class Tokens(Enum):
+    USDE = "0x5d3a1Ff2b6BAb83b63cd9AD0787074081a52ef34"
+    SUSDE = "0x211Cc4DD073734dA055fbF44a2b4667d5E5fE5d2"

constants/summary_columns.py

@@ -114,6 +114,10 @@ class SummaryColumn(Enum):
     STRATA_MONEY_SENIOR = ("strata_srusde_pts", SummaryColumnType.ETHENA_PTS)
     STRATA_MONEY_JUNIOR = ("strata_jrusde_pts", SummaryColumnType.ETHENA_PTS)
 
+    # Kodiak
+    KODIAK_USDE_PTS = ("kodiak_usde_pts", SummaryColumnType.ETHENA_PTS)
+    KODIAK_SUSDE_PTS = ("kodiak_susde_pts", SummaryColumnType.ETHENA_PTS)
+
     def __init__(self, column_name: str, col_type: SummaryColumnType):
         self.column_name = column_name
         self.col_type = col_type

integrations/integration_ids.py

@@ -580,6 +580,19 @@ class IntegrationID(Enum):
         Token.USDE,
     )
 
+    # Kodiak
+    KODIAK_USDE = (
+        "kodiak_usde",
+        "Kodiak USDe",
+        Token.USDE
+    )
+
+    KODIAK_SUSDE = (
+        "kodiak_susde",
+        "Kodiak sUSDe",
+        Token.SUSDE
+    )
+
     # Uniswap V4
     UNISWAP_V4_POOL = (
         "uniswap_v4_pool",

integrations/kodiak_susde_integration.py

@@ -0,0 +1,16 @@
+from constants.chains import Chain
+from integrations.integration_ids import IntegrationID
+from utils.kodiak import KodiakIntegration
+
+if __name__ == "__main__":
+    integration = KodiakIntegration(
+        IntegrationID.KODIAK_SUSDE,
+        1,
+        Chain.BERACHAIN
+    )
+
+    print(integration.get_block_balances({
+        1: {
+            "0x0000000000000000000000000000000000000000": 100.0
+        }
+    }, [12886286, 1]))

integrations/kodiak_usde_integration.py

@@ -0,0 +1,16 @@
+from constants.chains import Chain
+from integrations.integration_ids import IntegrationID
+from utils.kodiak import KodiakIntegration
+
+if __name__ == "__main__":
+    integration = KodiakIntegration(
+        IntegrationID.KODIAK_USDE,
+        1,
+        Chain.BERACHAIN
+    )
+
+    print(integration.get_block_balances({
+        1: {
+            "0x0000000000000000000000000000000000000000": 100.0
+        }
+    }, [12886286, 1]))

utils/kodiak.py

@@ -0,0 +1,73 @@
+from copy import deepcopy
+import json
+from typing import Dict, List
+from eth_typing import ChecksumAddress
+import requests
+from constants.chains import Chain
+from integrations.integration import Integration
+from integrations.integration_ids import IntegrationID
+from constants.kodiak import KODIAK_API_URL, KodiakIslandAddress, Tokens
+
+BASE_URL = f"{KODIAK_API_URL}/balances"
+
+class KodiakIntegration(Integration):
+
+    def __init__(
+        self,
+        integration_id: IntegrationID,
+        start_block: int,
+        chain: Chain,
+    ):
+        if chain not in [Chain.BERACHAIN]:
+            raise ValueError(f"KodiakIntegration not supported on chain {chain}")
+
+        if integration_id is IntegrationID.KODIAK_USDE:
+            self.island = KodiakIslandAddress.USDE
+            self.token = Tokens.USDE
+        elif integration_id is IntegrationID.KODIAK_SUSDE:
+            self.island = KodiakIslandAddress.SUSDE
+            self.token = Tokens.SUSDE
+        else:
+            raise ValueError(f"KodiakIntegration does not support integration ID {integration_id}")
+
+        super().__init__(
+            integration_id,
+            start_block,
+            chain
+        )
+
+    def get_block_balances(
+        self, cached_data: Dict[int, Dict[ChecksumAddress, float]], blocks: List[int]
+    ) -> Dict[int, Dict[ChecksumAddress, float]]:
+        """
+        Get the balances of all users at the given blocks
+        """
+        base_url = f"{BASE_URL}/{self.token.value.lower()}"
+
+        result = {}
+        for block in blocks:
+            if block in cached_data:
+                result[block] = deepcopy(cached_data[block])
+                continue
+
+            params = {"excludeSources": "balance", "blockNumber": block}
+            response = requests.get(base_url, params=params)
+            data = response.json()
+
+            block_data = {}
+            for entry in data:
+                user_address = ChecksumAddress(entry["id"])
+                value = 0.0
+                for balance_entry in entry["_sources"]:
+                    if "extraData" in balance_entry:
+                        extra_data = balance_entry["extraData"]
+                        if extra_data.get("islandId") == self.island.value.lower():
+                            value += float(balance_entry["value"])
+                    elif balance_entry["source"] == "v3-position":
+                        value += float(balance_entry["value"])
+
+                if value > 0:
+                    block_data[user_address] = value
+            result[block] = block_data
+
+        return result