1#!/usr/bin/env python3
2"""
3CoinGecko Global Data Tools
4 
5Tools for fetching global crypto market statistics and DeFi data.
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 
20# MCP Tool Schemas
21MCP_GLOBAL_SCHEMA = {
22    "name": "cg_global",
23    "title": "CoinGecko Global Market Stats",
24    "description": "Get global cryptocurrency market statistics including total market cap, volume, BTC dominance.",
25    "inputSchema": {
26        "type": "object",
27        "properties": {},
28        "additionalProperties": False
29    }
30}
31 
32MCP_GLOBAL_DEFI_SCHEMA = {
33    "name": "cg_global_defi",
34    "title": "CoinGecko Global DeFi Stats",
35    "description": "Get global DeFi market statistics including TVL and DeFi dominance.",
36    "inputSchema": {
37        "type": "object",
38        "properties": {},
39        "additionalProperties": False
40    }
41}
42 
43 
44def get_api_key() -> str:
45    """Get CoinGecko API key from environment."""
46    api_key = os.getenv("COINGECKO_API_KEY")
47    if not api_key:
48        raise ValueError("COINGECKO_API_KEY environment variable is required")
49    return api_key
50 
51 
52def get_global() -> Dict[str, Any]:
53    """
54    Get global cryptocurrency market statistics.
55 
56    Returns:
57        Dictionary with market cap, volume, BTC dominance, etc.
58    """
59    api_key = get_api_key()
60 
61    url = "https://pro-api.coingecko.com/api/v3/global"
62    headers = {"x-cg-pro-api-key": api_key}
63 
64    response = proxied_get(url, headers=headers, timeout=15)
65    response.raise_for_status()
66    data = response.json().get("data", {})
67 
68    return {
69        "active_cryptocurrencies": data.get("active_cryptocurrencies"),
70        "upcoming_icos": data.get("upcoming_icos"),
71        "ongoing_icos": data.get("ongoing_icos"),
72        "ended_icos": data.get("ended_icos"),
73        "markets": data.get("markets"),
74        "total_market_cap": data.get("total_market_cap", {}),
75        "total_volume": data.get("total_volume", {}),
76        "market_cap_percentage": data.get("market_cap_percentage", {}),
77        "market_cap_change_percentage_24h_usd": data.get("market_cap_change_percentage_24h_usd"),
78        "updated_at": data.get("updated_at")
79    }
80 
81 
82def get_global_defi() -> Dict[str, Any]:
83    """
84    Get global DeFi market statistics.
85 
86    Returns:
87        Dictionary with DeFi market cap, TVL, dominance
88    """
89    api_key = get_api_key()
90 
91    url = "https://pro-api.coingecko.com/api/v3/global/decentralized_finance_defi"
92    headers = {"x-cg-pro-api-key": api_key}
93 
94    response = proxied_get(url, headers=headers, timeout=15)
95    response.raise_for_status()
96    data = response.json().get("data", {})
97 
98    return {
99        "defi_market_cap": data.get("defi_market_cap"),
100        "eth_market_cap": data.get("eth_market_cap"),
101        "defi_to_eth_ratio": data.get("defi_to_eth_ratio"),
102        "trading_volume_24h": data.get("trading_volume_24h"),
103        "defi_dominance": data.get("defi_dominance"),
104        "top_coin_name": data.get("top_coin_name"),
105        "top_coin_defi_dominance": data.get("top_coin_defi_dominance")
106    }
107 
108 
109def main():
110    """CLI interface for global data tools."""
111    parser = argparse.ArgumentParser(
112        description="CoinGecko Global Data Tools",
113        formatter_class=argparse.RawDescriptionHelpFormatter
114    )
115 
116    subparsers = parser.add_subparsers(dest="command")
117 
118    # Global command
119    subparsers.add_parser("global", help="Get global market stats")
120 
121    # DeFi command
122    subparsers.add_parser("defi", help="Get DeFi market stats")
123 
124    parser.add_argument("--schema", action="store_true", help="Output MCP schema")
125 
126    args = parser.parse_args()
127 
128    if args.schema:
129        schemas = {
130            "global": MCP_GLOBAL_SCHEMA,
131            "global_defi": MCP_GLOBAL_DEFI_SCHEMA
132        }
133        print(json.dumps(schemas, indent=2))
134        return 0
135 
136    try:
137        if args.command == "global":
138            result = get_global()
139        elif args.command == "defi":
140            result = get_global_defi()
141        else:
142            parser.print_help()
143            return 0
144 
145        print(json.dumps(result, indent=2, ensure_ascii=False))
146        return 0
147 
148    except Exception as e:
149        print(json.dumps({"error": str(e)}, indent=2))
150        return 1
151 
152 
153if __name__ == "__main__":
154    exit(main())
155