from skills.sr_ranking.scripts.get_ranking import get_sr_ranking

# Get top 10 SRs by voting rewards
result = await get_sr_ranking(top_n=10, sort_by="reward_rate")

print(result)

# Get best ROI for voters
result = await get_sr_ranking(
    top_n=5, 
    sort_by="voter_apy"  # Focus on what voters earn
)

for sr in result['data']['rankings']:
    print(f"{sr['name']}: {sr['voter_apy']:.2f}% APY")

# Compare specific SR addresses
result = await compare_srs([
    "TLyqzVGLV1srkB7dToTAEqgDSfPtXRJZYH",  # BitTorrent
    "TMuA6YqfCeX8EhbfYEg5y7S4DqzSJireY9"   # JustLend DAO
])

{
  "success": true,
  "message": "Retrieved top 10 Super Representatives",
  "data": {
    "total_srs": 127,
    "top_srs": 27,
    "update_time": "2026-02-08T04:35:00Z",
    "rankings": [
      {
        "rank": 1,
        "address": "TLyqzVGLV1srkB7dToTAEqgDSfPtXRJZYH",
        "name": "BitTorrent",
        "url": "https://bittorrent.com",
        "total_votes": 45800000000,
        "vote_percentage": 9.45,
        "brokerage": 20,
        "voter_reward_rate": 80,
        "voter_apy": 4.32,
        "blocks_produced": 1234567,
        "blocks_missed": 12,
        "uptime": 99.999,
        "is_top_27": true
      },
      {
        "rank": 2,
        "address": "TMuA6YqfCeX8EhbfYEg5y7S4DqzSJireY9",
        "name": "JustLend DAO",
        "url": "https://justlend.org",
        "total_votes": 42100000000,
        "vote_percentage": 8.69,
        "brokerage": 15,
        "voter_reward_rate": 85,
        "voter_apy": 4.58,
        "blocks_produced": 1123456,
        "blocks_missed": 5,
        "uptime": 99.999,
        "is_top_27": true
      }
    ],
    "summary": {
      "avg_voter_apy": 4.12,
      "highest_apy": 4.58,
      "lowest_brokerage": 15,
      "most_voted": "BitTorrent"
    }
  },
  "recommendations": [
    "🏆 Best APY: JustLend DAO (4.58%)",
    "📊 Most voted: BitTorrent (9.45% of total)",
    "💰 Lowest brokerage: JustLend DAO (15%)"
  ]
}

Block Reward: 16 TRX
Brokerage: 20%

SR keeps:   16 TRX × 20% = 3.2 TRX
Voters get: 16 TRX × 80% = 12.8 TRX (shared among all voters)

# Estimate rewards for voting
from skills.sr_ranking.scripts.get_ranking import simulate_voting_rewards

result = await simulate_voting_rewards(
    sr_address="TLyqzVGLV1srkB7dToTAEqgDSfPtXRJZYH",
    vote_amount=10000  # Vote with 10,000 TRX
)

print(f"Daily rewards: {result['daily_trx']} TRX")
print(f"Annual APY: {result['apy']:.2f}%")

1. Freeze TRX → Get voting power
   ↓
2. Check SR rankings (this skill)
   ↓
3. Vote for SRs (vote_witness skill)
   ↓
4. Earn rewards (distributed every 6h)
   ↓
5. Claim rewards + Re-invest

# Available sort_by values:
- "voter_apy"      # Best returns for voters (recommended)
- "total_votes"    # Most popular SRs
- "brokerage"      # Lowest brokerage (best sharing)
- "uptime"         # Most reliable block producers
- "blocks_produced" # Most productive

@mcp.tool()
async def get_super_representative_ranking(
    top_n: int = 10,
    sort_by: str = "voter_apy"
) -> str:
    """
    Get TRON Super Representative rankings.
    
    Args:
        top_n: Number of top SRs to return (default: 10)
        sort_by: Sort criteria - voter_apy, total_votes, brokerage, uptime
        
    Returns:
        Ranked list of SRs with voting rewards comparison
    """
    result = await get_sr_ranking(top_n, sort_by)
    return format_sr_comparison(result)

User: "我想投票赚收益，帮我找最好的超级代表"

Agent:
1. [Calls get_sr_ranking(top_n=5, sort_by="voter_apy")]
   
2. "为您找到Top 5收益最高的超级代表：
   
   🏆 第1名: JustLend DAO
      - 年化收益: 4.58% APY
      - 佣金率: 15% (voters拿85%)
      - 可靠性: 99.999% uptime
   
   🥈 第2名: BitTorrent
      - 年化收益: 4.32% APY
      - 佣金率: 20% (voters拿80%)
      - 票数: 最多 (45.8B votes)
   
   ..."

3. User: "那选JustLend DAO吧"

4. Agent: "好的，投票10,000 TRX到JustLend DAO，预计：
   - 每日收益: ~1.25 TRX
   - 每月收益: ~37.6 TRX
   - 年收益: ~458 TRX (4.58% APY)
   
   是否确认投票?"

Error: Cannot fetch SR list from TronGrid
Solution: Check network connection or try again

Error: sort_by must be one of: voter_apy, total_votes, brokerage, uptime
Solution: Use valid sort criteria