campaign.py

from typing import List
from constants.example_integrations import (
    ACTIVE_ENA_START_BLOCK_EXAMPLE,
    BEEFY_ARBITRUM_START_BLOCK_EXAMPLE,
)
from integrations.beefy_cached_balance_example_integration import (
    BeefyCachedBalanceIntegration,
)
from integrations.claimed_ena_example_integration import ClaimedEnaIntegration
from utils import pendle
from web3 import Web3

from constants.chains import Chain
from constants.pendle import PENDLE_USDE_JULY_DEPLOYMENT_BLOCK
from constants.summary_columns import SummaryColumn
from integrations.integration import Integration
from integrations.integration_ids import IntegrationID
from integrations.pendle_lpt_integration import PendleLPTIntegration
from integrations.pendle_yt_integration import PendleYTIntegration
from integrations.template import ProtocolNameIntegration

# TODO: Add your integration here
INTEGRATIONS: List[Integration] = [
    # Template integration
    ProtocolNameIntegration(
        integration_id=IntegrationID.EXAMPLE,
        start_block=20000000,
        summary_cols=[SummaryColumn.TEMPLATE_PTS],
        chain=Chain.ETHEREUM,
        reward_multiplier=20,
        excluded_addresses={
            Web3.to_checksum_address("0x0000000000000000000000000000000000000000")
        },
        end_block=40000000,
    ),
    # Example integration using cached user balances for improved performance,
    # reads from previous balance snapshots
    ClaimedEnaIntegration(
        integration_id=IntegrationID.CLAIMED_ENA_EXAMPLE,
        start_block=ACTIVE_ENA_START_BLOCK_EXAMPLE,
        summary_cols=[SummaryColumn.CLAIMED_ENA_PTS_EXAMPLE],
        reward_multiplier=1,
    ),
    # Cached balances integration example, based on API calls
    BeefyCachedBalanceIntegration(
        integration_id=IntegrationID.BEEFY_CACHED_BALANCE_EXAMPLE,
        start_block=BEEFY_ARBITRUM_START_BLOCK_EXAMPLE,
        summary_cols=[SummaryColumn.BEEFY_CACHED_BALANCE_EXAMPLE],
        chain=Chain.ARBITRUM,
        reward_multiplier=1,
    ),
    # Simple Integration class examples (outdated),
    # don't use these anymore
    PendleLPTIntegration(
        integration_id=IntegrationID.PENDLE_USDE_LPT,
        start_block=PENDLE_USDE_JULY_DEPLOYMENT_BLOCK,
        sy_contract=pendle.sy_contract,
        lp_contract=pendle.lpt_contract,
        summary_cols=[SummaryColumn.PENDLE_SHARDS],
        chain=Chain.ETHEREUM,
        reward_multiplier=20,
    ),
    PendleLPTIntegration(
        integration_id=IntegrationID.PENDLE_ARBITRUM_USDE_LPT,
        start_block=PENDLE_USDE_JULY_DEPLOYMENT_BLOCK,
        sy_contract=pendle.usde_arb_SY_contract,
        lp_contract=pendle.usde_arb_LPT_contract,
        summary_cols=[SummaryColumn.PENDLE_ARBITRUM_SHARDS],
        chain=Chain.ARBITRUM,
        reward_multiplier=20,
    ),
    PendleYTIntegration(
        integration_id=IntegrationID.PENDLE_USDE_YT,
        start_block=PENDLE_USDE_JULY_DEPLOYMENT_BLOCK,
        summary_cols=[SummaryColumn.PENDLE_SHARDS],
        yt_contract=pendle.yt_contract,
        chain=Chain.ETHEREUM,
        reward_multiplier=20,
    ),
    PendleYTIntegration(
        integration_id=IntegrationID.PENDLE_ARBITRUM_USDE_YT,
        start_block=PENDLE_USDE_JULY_DEPLOYMENT_BLOCK,
        summary_cols=[SummaryColumn.PENDLE_ARBITRUM_SHARDS],
        yt_contract=pendle.usde_arb_YT_contract,
        chain=Chain.ARBITRUM,
        reward_multiplier=20,
    ),
]


def get_integrations() -> List[Integration]:
    return INTEGRATIONS