1#!/usr/bin/env python3
2"""
3CoinGecko Contract/Token Tools
4 
5Tools for fetching token data by contract address.
6"""
7 
8import os
9from dotenv import load_dotenv
10import json
11import argparse
12from typing import Dict, Any
13 
14from core.http_client import proxied_get
15 
16# Load environment variables
17load_dotenv()
18 
19 
20def get_api_key() -> str:
21    """Get CoinGecko API key from environment."""
22    api_key = os.getenv("COINGECKO_API_KEY")
23    if not api_key:
24        raise ValueError("COINGECKO_API_KEY environment variable is required")
25    return api_key
26 
27 
28def get_token_price(
29    platform: str,
30    contract_addresses: str,
31    vs_currencies: str = "usd",
32    include_market_cap: bool = False,
33    include_24hr_vol: bool = False,
34    include_24hr_change: bool = False,
35    include_last_updated_at: bool = False
36) -> Dict[str, Any]:
37    """
38    Get token prices by contract address.
39 
40    Args:
41        platform: Asset platform id (ethereum, solana, polygon-pos, etc.)
42        contract_addresses: Comma-separated contract addresses
43        vs_currencies: Comma-separated target currencies (usd, eur, btc)
44        include_market_cap: Include market cap
45        include_24hr_vol: Include 24h volume
46        include_24hr_change: Include 24h price change
47        include_last_updated_at: Include last updated timestamp
48 
49    Returns:
50        Dictionary with token prices
51    """
52    api_key = get_api_key()
53 
54    url = f"https://pro-api.coingecko.com/api/v3/simple/token_price/{platform}"
55    headers = {"x-cg-pro-api-key": api_key}
56    params = {
57        "contract_addresses": contract_addresses,
58        "vs_currencies": vs_currencies,
59        "include_market_cap": str(include_market_cap).lower(),
60        "include_24hr_vol": str(include_24hr_vol).lower(),
61        "include_24hr_change": str(include_24hr_change).lower(),
62        "include_last_updated_at": str(include_last_updated_at).lower()
63    }
64 
65    response = proxied_get(url, headers=headers, params=params, timeout=30)
66    response.raise_for_status()
67    data = response.json()
68 
69    # Format response
70    tokens = {}
71    for address, price_data in data.items():
72        tokens[address.lower()] = price_data
73 
74    return {
75        "platform": platform,
76        "vs_currencies": vs_currencies.split(","),
77        "tokens": tokens,
78        "count": len(tokens)
79    }
80 
81 
82def get_coin_by_contract(
83    platform: str,
84    contract_address: str
85) -> Dict[str, Any]:
86    """
87    Get coin data by contract address.
88 
89    Args:
90        platform: Asset platform id (ethereum, solana, polygon-pos, etc.)
91        contract_address: Token contract address
92 
93    Returns:
94        Dictionary with coin data
95    """
96    api_key = get_api_key()
97 
98    url = f"https://pro-api.coingecko.com/api/v3/coins/{platform}/contract/{contract_address}"
99    headers = {"x-cg-pro-api-key": api_key}
100 
101    response = proxied_get(url, headers=headers, timeout=30)
102    response.raise_for_status()
103    data = response.json()
104 
105    result = {
106        "id": data.get("id", ""),
107        "symbol": data.get("symbol", "").upper(),
108        "name": data.get("name", ""),
109        "asset_platform_id": data.get("asset_platform_id"),
110        "platforms": data.get("platforms", {}),
111        "block_time_in_minutes": data.get("block_time_in_minutes"),
112        "hashing_algorithm": data.get("hashing_algorithm"),
113        "categories": data.get("categories", []),
114        "description": data.get("description", {}).get("en", ""),
115        "links": {
116            "homepage": data.get("links", {}).get("homepage", []),
117            "blockchain_site": data.get("links", {}).get("blockchain_site", []),
118            "twitter_screen_name": data.get("links", {}).get("twitter_screen_name"),
119            "telegram_channel_identifier": data.get("links", {}).get("telegram_channel_identifier"),
120            "subreddit_url": data.get("links", {}).get("subreddit_url")
121        },
122        "image": data.get("image", {}),
123        "country_origin": data.get("country_origin"),
124        "genesis_date": data.get("genesis_date"),
125        "contract_address": data.get("contract_address"),
126        "sentiment_votes_up_percentage": data.get("sentiment_votes_up_percentage"),
127        "sentiment_votes_down_percentage": data.get("sentiment_votes_down_percentage"),
128        "market_cap_rank": data.get("market_cap_rank"),
129        "coingecko_rank": data.get("coingecko_rank"),
130        "coingecko_score": data.get("coingecko_score"),
131        "developer_score": data.get("developer_score"),
132        "community_score": data.get("community_score"),
133        "liquidity_score": data.get("liquidity_score"),
134        "public_interest_score": data.get("public_interest_score"),
135        "last_updated": data.get("last_updated")
136    }
137 
138    # Include market data if available
139    if "market_data" in data:
140        md = data["market_data"]
141        result["market_data"] = {
142            "current_price": md.get("current_price", {}),
143            "ath": md.get("ath", {}),
144            "ath_change_percentage": md.get("ath_change_percentage", {}),
145            "ath_date": md.get("ath_date", {}),
146            "atl": md.get("atl", {}),
147            "atl_change_percentage": md.get("atl_change_percentage", {}),
148            "atl_date": md.get("atl_date", {}),
149            "market_cap": md.get("market_cap", {}),
150            "market_cap_rank": md.get("market_cap_rank"),
151            "fully_diluted_valuation": md.get("fully_diluted_valuation", {}),
152            "total_volume": md.get("total_volume", {}),
153            "high_24h": md.get("high_24h", {}),
154            "low_24h": md.get("low_24h", {}),
155            "price_change_24h": md.get("price_change_24h"),
156            "price_change_percentage_24h": md.get("price_change_percentage_24h"),
157            "price_change_percentage_7d": md.get("price_change_percentage_7d"),
158            "price_change_percentage_14d": md.get("price_change_percentage_14d"),
159            "price_change_percentage_30d": md.get("price_change_percentage_30d"),
160            "price_change_percentage_60d": md.get("price_change_percentage_60d"),
161            "price_change_percentage_200d": md.get("price_change_percentage_200d"),
162            "price_change_percentage_1y": md.get("price_change_percentage_1y"),
163            "market_cap_change_24h": md.get("market_cap_change_24h"),
164            "market_cap_change_percentage_24h": md.get("market_cap_change_percentage_24h"),
165            "total_supply": md.get("total_supply"),
166            "max_supply": md.get("max_supply"),
167            "circulating_supply": md.get("circulating_supply")
168        }
169 
170    return result
171 
172 
173def main():
174    """CLI interface for contract tools."""
175    parser = argparse.ArgumentParser(
176        description="CoinGecko Contract/Token Tools",
177        formatter_class=argparse.RawDescriptionHelpFormatter
178    )
179 
180    subparsers = parser.add_subparsers(dest="command")
181 
182    # Token price command
183    price_parser = subparsers.add_parser("price", help="Get token price by contract")
184    price_parser.add_argument("platform", help="Asset platform (ethereum, solana, etc.)")
185    price_parser.add_argument("addresses", help="Contract addresses (comma-separated)")
186    price_parser.add_argument("--currencies", default="usd", help="Target currencies")
187    price_parser.add_argument("--market-cap", action="store_true", help="Include market cap")
188    price_parser.add_argument("--volume", action="store_true", help="Include 24h volume")
189    price_parser.add_argument("--change", action="store_true", help="Include 24h change")
190 
191    # Coin data command
192    data_parser = subparsers.add_parser("data", help="Get coin data by contract")
193    data_parser.add_argument("platform", help="Asset platform (ethereum, solana, etc.)")
194    data_parser.add_argument("address", help="Contract address")
195 
196    args = parser.parse_args()
197 
198    try:
199        if args.command == "price":
200            result = get_token_price(
201                platform=args.platform,
202                contract_addresses=args.addresses,
203                vs_currencies=args.currencies,
204                include_market_cap=args.market_cap,
205                include_24hr_vol=args.volume,
206                include_24hr_change=args.change
207            )
208        elif args.command == "data":
209            result = get_coin_by_contract(args.platform, args.address)
210        else:
211            parser.print_help()
212            return 0
213 
214        print(json.dumps(result, indent=2, ensure_ascii=False))
215        return 0
216 
217    except Exception as e:
218        print(json.dumps({"error": str(e)}, indent=2))
219        return 1
220 
221 
222if __name__ == "__main__":
223    exit(main())
224