1#!/usr/bin/env python3
2"""
3CoinGecko Search Tools
4 
5Tools for searching coins, exchanges, categories, and NFTs.
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 search(query: str) -> Dict[str, Any]:
29    """
30    Search for coins, exchanges, categories, and NFTs.
31 
32    Args:
33        query: Search query string
34 
35    Returns:
36        Dictionary with search results across all categories
37    """
38    api_key = get_api_key()
39 
40    url = "https://pro-api.coingecko.com/api/v3/search"
41    headers = {"x-cg-pro-api-key": api_key}
42    params = {"query": query}
43 
44    response = proxied_get(url, headers=headers, params=params, timeout=30)
45    response.raise_for_status()
46    data = response.json()
47 
48    # Format coins
49    coins = []
50    for coin in data.get("coins", []):
51        coins.append({
52            "id": coin.get("id", ""),
53            "name": coin.get("name", ""),
54            "api_symbol": coin.get("api_symbol", ""),
55            "symbol": coin.get("symbol", "").upper(),
56            "market_cap_rank": coin.get("market_cap_rank"),
57            "thumb": coin.get("thumb"),
58            "large": coin.get("large")
59        })
60 
61    # Format exchanges
62    exchanges = []
63    for exchange in data.get("exchanges", []):
64        exchanges.append({
65            "id": exchange.get("id", ""),
66            "name": exchange.get("name", ""),
67            "market_type": exchange.get("market_type"),
68            "thumb": exchange.get("thumb"),
69            "large": exchange.get("large")
70        })
71 
72    # Format categories
73    categories = []
74    for category in data.get("categories", []):
75        categories.append({
76            "id": category.get("id"),
77            "name": category.get("name", "")
78        })
79 
80    # Format NFTs
81    nfts = []
82    for nft in data.get("nfts", []):
83        nfts.append({
84            "id": nft.get("id", ""),
85            "name": nft.get("name", ""),
86            "symbol": nft.get("symbol", ""),
87            "thumb": nft.get("thumb")
88        })
89 
90    # Format ICOs (if present)
91    icos = data.get("icos", [])
92 
93    return {
94        "query": query,
95        "coins": coins,
96        "coins_count": len(coins),
97        "exchanges": exchanges,
98        "exchanges_count": len(exchanges),
99        "categories": categories,
100        "categories_count": len(categories),
101        "nfts": nfts,
102        "nfts_count": len(nfts),
103        "icos": icos,
104        "icos_count": len(icos)
105    }
106 
107 
108def main():
109    """CLI interface for search tools."""
110    parser = argparse.ArgumentParser(
111        description="CoinGecko Search Tools",
112        formatter_class=argparse.RawDescriptionHelpFormatter
113    )
114 
115    parser.add_argument("query", help="Search query")
116 
117    args = parser.parse_args()
118 
119    try:
120        result = search(args.query)
121        print(json.dumps(result, indent=2, ensure_ascii=False))
122        return 0
123 
124    except Exception as e:
125        print(json.dumps({"error": str(e)}, indent=2))
126        return 1
127 
128 
129if __name__ == "__main__":
130    exit(main())
131