add back gauge check functions

Author: jalbrekt85

fee_allocator/bribe_platforms/paladin.py

@@ -1,9 +1,11 @@
 from typing import Dict, Optional, Tuple, Any, List
 import pandas as pd
+from web3 import Web3
 from .base import BribePlatform
 from bal_tools.safe_tx_builder import SafeContract
 import json
 from pathlib import Path
+from fee_allocator.logger import logger
 
 
 class PaladinPlatform(BribePlatform):
@@ -80,17 +82,69 @@ def process_bribes(self, bribes_df: pd.DataFrame, builder: Any, usdc: Any) -> No
 
 
     def validate_gauge_requirements(self, gauge_address: str) -> Tuple[bool, Optional[str]]:
-        """No validation needed for ROLLOVER close type"""
+        base_dir = Path(__file__).parent.parent
+        with open(f"{base_dir}/abi/gauge.json", "r") as f:
+            gauge_abi = json.load(f)
+
+        w3 = self.run_config.mainnet.web3
+        usdc = Web3.to_checksum_address(self.usdc_address)
+        gauge = Web3.to_checksum_address(gauge_address)
+        contract = w3.eth.contract(address=gauge, abi=gauge_abi)
+
+        reward_tokens = [contract.functions.reward_tokens(i).call() for i in range(8)]
+        if usdc not in reward_tokens:
+            return False, f"USDC ({usdc}) not found in gauge reward tokens"
+
+        distributor = contract.functions.reward_data(usdc).call()[1]
+        if distributor.lower() not in [self.bal_quest_board.lower(), self.aura_quest_board.lower()]:
+            return False, f"Incorrect distributor {distributor}. Expected: {self.bal_quest_board} or {self.aura_quest_board}"
+
         return True, None
 
     @property
     def platform_name(self) -> str:
-        """Platform identifier for reporting"""
         return "paladin"
 
     @property
     def supported_markets(self) -> List[str]:
         return ["aura", "balancer"]
 
     def get_platform_for_market(self, market: str, voting_pool_override: Optional[str]) -> str:
-        return "paladin"
+        return "paladin"
+
+    def check_all_gauge_requirements(self, pools: List[Any]) -> List[Dict]:
+        """Check all Paladin gauges for requirements and return issues"""
+
+        gauges_with_issues = []
+
+        for pool in pools:
+            if pool.market_override != "paladin":
+                continue
+
+            valid, error_msg = self.validate_gauge_requirements(pool.gauge_address)
+            if not valid:
+                action_needed = []
+
+                if pool.to_bal_incentives_usd > 0:
+                    action_needed.append(f"Balancer distributor ({self.bal_quest_board})")
+                if pool.to_aura_incentives_usd > 0:
+                    action_needed.append(f"Aura distributor ({self.aura_quest_board})")
+
+                if action_needed:
+                    if "not found in gauge reward tokens" in error_msg:
+                        action_msg = f"Add USDC ({self.usdc_address}) as reward token and set {' and '.join(action_needed)}"
+                    elif "Incorrect distributor" in error_msg:
+                        action_msg = f"Set {' and '.join(action_needed)}"
+                    else:
+                        action_msg = error_msg
+
+                    logger.warning(f"Paladin gauge {pool.gauge_address} missing requirements: {action_msg}")
+                    gauges_with_issues.append({
+                        "gauge": pool.gauge_address,
+                        "pool_id": pool.pool_id,
+                        "chain": pool.chain.name,
+                        "action": action_msg,
+                        "amount": float(pool.total_to_incentives_usd)
+                    })
+
+        return gauges_with_issues