1#!/usr/bin/env python3
2"""
3CoinGecko NFT Tools
4 
5Tools for fetching NFT collection data including lists, details, and contract lookups.
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_nfts_list(
29    order: str = "market_cap_usd_desc",
30    per_page: int = 100,
31    page: int = 1
32) -> Dict[str, Any]:
33    """
34    Get all NFT collections with IDs and contract addresses.
35 
36    Args:
37        order: Sort order (market_cap_usd_desc, h24_volume_usd_desc, etc.)
38        per_page: Results per page (max 250)
39        page: Page number
40 
41    Returns:
42        Dictionary with list of NFT collections
43    """
44    api_key = get_api_key()
45 
46    url = "https://pro-api.coingecko.com/api/v3/nfts/list"
47    headers = {"x-cg-pro-api-key": api_key}
48    params = {
49        "order": order,
50        "per_page": min(per_page, 250),
51        "page": page
52    }
53 
54    response = proxied_get(url, headers=headers, params=params, timeout=30)
55    response.raise_for_status()
56    data = response.json()
57 
58    nfts = []
59    for nft in data:
60        nfts.append({
61            "id": nft.get("id", ""),
62            "contract_address": nft.get("contract_address", ""),
63            "name": nft.get("name", ""),
64            "asset_platform_id": nft.get("asset_platform_id"),
65            "symbol": nft.get("symbol", "")
66        })
67 
68    return {
69        "nfts": nfts,
70        "count": len(nfts),
71        "page": page,
72        "per_page": per_page,
73        "order": order
74    }
75 
76 
77def get_nft(nft_id: str) -> Dict[str, Any]:
78    """
79    Get NFT collection data including floor price, volume, and market cap.
80 
81    Args:
82        nft_id: CoinGecko NFT collection id
83 
84    Returns:
85        Dictionary with NFT collection data
86    """
87    api_key = get_api_key()
88 
89    url = f"https://pro-api.coingecko.com/api/v3/nfts/{nft_id}"
90    headers = {"x-cg-pro-api-key": api_key}
91 
92    response = proxied_get(url, headers=headers, timeout=30)
93    response.raise_for_status()
94    data = response.json()
95 
96    return {
97        "id": data.get("id", ""),
98        "contract_address": data.get("contract_address", ""),
99        "asset_platform_id": data.get("asset_platform_id"),
100        "name": data.get("name", ""),
101        "symbol": data.get("symbol", ""),
102        "image": data.get("image", {}),
103        "description": data.get("description"),
104        "native_currency": data.get("native_currency"),
105        "native_currency_symbol": data.get("native_currency_symbol"),
106        "floor_price": data.get("floor_price", {}),
107        "market_cap": data.get("market_cap", {}),
108        "volume_24h": data.get("volume_24h", {}),
109        "floor_price_in_usd_24h_percentage_change": data.get("floor_price_in_usd_24h_percentage_change"),
110        "floor_price_24h_percentage_change": data.get("floor_price_24h_percentage_change", {}),
111        "market_cap_24h_percentage_change": data.get("market_cap_24h_percentage_change", {}),
112        "volume_24h_percentage_change": data.get("volume_24h_percentage_change", {}),
113        "number_of_unique_addresses": data.get("number_of_unique_addresses"),
114        "number_of_unique_addresses_24h_percentage_change": data.get("number_of_unique_addresses_24h_percentage_change"),
115        "volume_in_usd_24h_percentage_change": data.get("volume_in_usd_24h_percentage_change"),
116        "total_supply": data.get("total_supply"),
117        "one_day_sales": data.get("one_day_sales"),
118        "one_day_sales_24h_percentage_change": data.get("one_day_sales_24h_percentage_change"),
119        "one_day_average_sale_price": data.get("one_day_average_sale_price"),
120        "one_day_average_sale_price_24h_percentage_change": data.get("one_day_average_sale_price_24h_percentage_change"),
121        "links": data.get("links", {}),
122        "floor_price_7d_percentage_change": data.get("floor_price_7d_percentage_change", {}),
123        "floor_price_14d_percentage_change": data.get("floor_price_14d_percentage_change", {}),
124        "floor_price_30d_percentage_change": data.get("floor_price_30d_percentage_change", {}),
125        "floor_price_60d_percentage_change": data.get("floor_price_60d_percentage_change", {}),
126        "floor_price_1y_percentage_change": data.get("floor_price_1y_percentage_change", {}),
127        "explorers": data.get("explorers", [])
128    }
129 
130 
131def get_nft_by_contract(
132    asset_platform: str,
133    contract_address: str
134) -> Dict[str, Any]:
135    """
136    Get NFT collection data by contract address.
137 
138    Args:
139        asset_platform: Asset platform id (e.g., ethereum, solana)
140        contract_address: NFT contract address
141 
142    Returns:
143        Dictionary with NFT collection data
144    """
145    api_key = get_api_key()
146 
147    url = f"https://pro-api.coingecko.com/api/v3/nfts/{asset_platform}/contract/{contract_address}"
148    headers = {"x-cg-pro-api-key": api_key}
149 
150    response = proxied_get(url, headers=headers, timeout=30)
151    response.raise_for_status()
152    data = response.json()
153 
154    return {
155        "id": data.get("id", ""),
156        "contract_address": data.get("contract_address", ""),
157        "asset_platform_id": data.get("asset_platform_id"),
158        "name": data.get("name", ""),
159        "symbol": data.get("symbol", ""),
160        "image": data.get("image", {}),
161        "description": data.get("description"),
162        "native_currency": data.get("native_currency"),
163        "native_currency_symbol": data.get("native_currency_symbol"),
164        "floor_price": data.get("floor_price", {}),
165        "market_cap": data.get("market_cap", {}),
166        "volume_24h": data.get("volume_24h", {}),
167        "floor_price_in_usd_24h_percentage_change": data.get("floor_price_in_usd_24h_percentage_change"),
168        "floor_price_24h_percentage_change": data.get("floor_price_24h_percentage_change", {}),
169        "market_cap_24h_percentage_change": data.get("market_cap_24h_percentage_change", {}),
170        "volume_24h_percentage_change": data.get("volume_24h_percentage_change", {}),
171        "number_of_unique_addresses": data.get("number_of_unique_addresses"),
172        "number_of_unique_addresses_24h_percentage_change": data.get("number_of_unique_addresses_24h_percentage_change"),
173        "volume_in_usd_24h_percentage_change": data.get("volume_in_usd_24h_percentage_change"),
174        "total_supply": data.get("total_supply"),
175        "one_day_sales": data.get("one_day_sales"),
176        "one_day_sales_24h_percentage_change": data.get("one_day_sales_24h_percentage_change"),
177        "one_day_average_sale_price": data.get("one_day_average_sale_price"),
178        "one_day_average_sale_price_24h_percentage_change": data.get("one_day_average_sale_price_24h_percentage_change"),
179        "links": data.get("links", {}),
180        "floor_price_7d_percentage_change": data.get("floor_price_7d_percentage_change", {}),
181        "floor_price_14d_percentage_change": data.get("floor_price_14d_percentage_change", {}),
182        "floor_price_30d_percentage_change": data.get("floor_price_30d_percentage_change", {}),
183        "floor_price_60d_percentage_change": data.get("floor_price_60d_percentage_change", {}),
184        "floor_price_1y_percentage_change": data.get("floor_price_1y_percentage_change", {}),
185        "explorers": data.get("explorers", [])
186    }
187 
188 
189def main():
190    """CLI interface for NFT tools."""
191    parser = argparse.ArgumentParser(
192        description="CoinGecko NFT Tools",
193        formatter_class=argparse.RawDescriptionHelpFormatter
194    )
195 
196    subparsers = parser.add_subparsers(dest="command")
197 
198    # List command
199    list_parser = subparsers.add_parser("list", help="Get all NFT collections")
200    list_parser.add_argument("--order", default="market_cap_usd_desc")
201    list_parser.add_argument("--per-page", type=int, default=100)
202    list_parser.add_argument("--page", type=int, default=1)
203 
204    # Get command
205    get_parser = subparsers.add_parser("get", help="Get NFT collection details")
206    get_parser.add_argument("nft_id", help="NFT collection ID")
207 
208    # Contract command
209    contract_parser = subparsers.add_parser("contract", help="Get NFT by contract")
210    contract_parser.add_argument("platform", help="Asset platform (ethereum, solana)")
211    contract_parser.add_argument("address", help="Contract address")
212 
213    args = parser.parse_args()
214 
215    try:
216        if args.command == "list":
217            result = get_nfts_list(args.order, args.per_page, args.page)
218        elif args.command == "get":
219            result = get_nft(args.nft_id)
220        elif args.command == "contract":
221            result = get_nft_by_contract(args.platform, args.address)
222        else:
223            parser.print_help()
224            return 0
225 
226        print(json.dumps(result, indent=2, ensure_ascii=False))
227        return 0
228 
229    except Exception as e:
230        print(json.dumps({"error": str(e)}, indent=2))
231        return 1
232 
233 
234if __name__ == "__main__":
235    exit(main())
236