1#!/usr/bin/env python3
2"""
3CoinGecko Infrastructure Tools
4 
5Tools for fetching reference data: asset platforms, exchange rates, supported currencies, and categories list.
6"""
7 
8import os
9from dotenv import load_dotenv
10import json
11import argparse
12from typing import Dict, Any, Optional
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_asset_platforms(filter: Optional[str] = None) -> Dict[str, Any]:
29    """
30    Get all blockchain networks/asset platforms.
31 
32    Args:
33        filter: Filter platforms (e.g., "nft" for NFT-supporting platforms)
34 
35    Returns:
36        Dictionary with list of asset platforms
37    """
38    api_key = get_api_key()
39 
40    url = "https://pro-api.coingecko.com/api/v3/asset_platforms"
41    headers = {"x-cg-pro-api-key": api_key}
42    params = {}
43    if filter:
44        params["filter"] = filter
45 
46    response = proxied_get(url, headers=headers, params=params, timeout=30)
47    response.raise_for_status()
48    data = response.json()
49 
50    platforms = []
51    for platform in data:
52        platforms.append({
53            "id": platform.get("id", ""),
54            "chain_identifier": platform.get("chain_identifier"),
55            "name": platform.get("name", ""),
56            "shortname": platform.get("shortname"),
57            "native_coin_id": platform.get("native_coin_id"),
58            "image": platform.get("image", {})
59        })
60 
61    return {
62        "platforms": platforms,
63        "count": len(platforms),
64        "filter": filter
65    }
66 
67 
68def get_exchange_rates() -> Dict[str, Any]:
69    """
70    Get BTC exchange rates to all fiat currencies.
71 
72    Returns:
73        Dictionary with BTC exchange rates
74    """
75    api_key = get_api_key()
76 
77    url = "https://pro-api.coingecko.com/api/v3/exchange_rates"
78    headers = {"x-cg-pro-api-key": api_key}
79 
80    response = proxied_get(url, headers=headers, timeout=30)
81    response.raise_for_status()
82    data = response.json()
83 
84    rates = data.get("rates", {})
85    formatted_rates = {}
86    for currency, info in rates.items():
87        formatted_rates[currency] = {
88            "name": info.get("name", ""),
89            "unit": info.get("unit", ""),
90            "value": info.get("value"),
91            "type": info.get("type", "")
92        }
93 
94    return {
95        "base": "btc",
96        "rates": formatted_rates,
97        "count": len(formatted_rates)
98    }
99 
100 
101def get_vs_currencies() -> Dict[str, Any]:
102    """
103    Get list of supported quote currencies (vs_currencies).
104 
105    Returns:
106        Dictionary with list of supported currencies
107    """
108    api_key = get_api_key()
109 
110    url = "https://pro-api.coingecko.com/api/v3/simple/supported_vs_currencies"
111    headers = {"x-cg-pro-api-key": api_key}
112 
113    response = proxied_get(url, headers=headers, timeout=30)
114    response.raise_for_status()
115    data = response.json()
116 
117    return {
118        "currencies": data,
119        "count": len(data)
120    }
121 
122 
123def get_categories_list() -> Dict[str, Any]:
124    """
125    Get lightweight list of category names and IDs (no market data).
126 
127    Returns:
128        Dictionary with list of categories
129    """
130    api_key = get_api_key()
131 
132    url = "https://pro-api.coingecko.com/api/v3/coins/categories/list"
133    headers = {"x-cg-pro-api-key": api_key}
134 
135    response = proxied_get(url, headers=headers, timeout=30)
136    response.raise_for_status()
137    data = response.json()
138 
139    categories = []
140    for category in data:
141        categories.append({
142            "category_id": category.get("category_id", ""),
143            "name": category.get("name", "")
144        })
145 
146    return {
147        "categories": categories,
148        "count": len(categories)
149    }
150 
151 
152def main():
153    """CLI interface for infrastructure tools."""
154    parser = argparse.ArgumentParser(
155        description="CoinGecko Infrastructure Tools",
156        formatter_class=argparse.RawDescriptionHelpFormatter
157    )
158 
159    subparsers = parser.add_subparsers(dest="command")
160 
161    # Asset platforms command
162    platforms_parser = subparsers.add_parser("platforms", help="Get all asset platforms")
163    platforms_parser.add_argument("--filter", help="Filter platforms (e.g., nft)")
164 
165    # Exchange rates command
166    subparsers.add_parser("exchange-rates", help="Get BTC exchange rates")
167 
168    # Supported currencies command
169    subparsers.add_parser("currencies", help="Get supported vs_currencies")
170 
171    # Categories list command
172    subparsers.add_parser("categories", help="Get category names and IDs")
173 
174    args = parser.parse_args()
175 
176    try:
177        if args.command == "platforms":
178            result = get_asset_platforms(getattr(args, 'filter', None))
179        elif args.command == "exchange-rates":
180            result = get_exchange_rates()
181        elif args.command == "currencies":
182            result = get_vs_currencies()
183        elif args.command == "categories":
184            result = get_categories_list()
185        else:
186            parser.print_help()
187            return 0
188 
189        print(json.dumps(result, indent=2, ensure_ascii=False))
190        return 0
191 
192    except Exception as e:
193        print(json.dumps({"error": str(e)}, indent=2))
194        return 1
195 
196 
197if __name__ == "__main__":
198    exit(main())
199