1#!/usr/bin/env python3
2"""
3Coinglass Open Interest Module
4 
5Fetch aggregate open interest data across major cryptocurrency exchanges.
6"""
7 
8import sys
9import json
10import argparse
11from typing import Dict, Any, Optional, List
12 
13from ._api import cg_request
14 
15# MCP Tool Schema
16MCP_OPEN_INTEREST_SCHEMA = {
17    "name": "cg_open_interest",
18    "title": "Coinglass Open Interest",
19    "description": (
20        "Get aggregate open interest across exchanges for a symbol."
21    ),
22    "inputSchema": {
23        "type": "object",
24        "properties": {
25            "symbol": {
26                "type": "string",
27                "description": "Symbol (BTC, ETH, SOL, etc.)"
28            },
29            "interval": {
30                "type": "string",
31                "description": "Time interval: 0 (all), h1, h4, h12, h24",
32                "default": "0",
33                "enum": ["0", "h1", "h4", "h12", "h24"]
34            }
35        },
36        "required": ["symbol"],
37        "additionalProperties": False
38    }
39}
40 
41 
42def get_open_interest(
43    symbol: str = "BTC",
44    interval: str = "0"
45) -> Optional[Dict[str, Any]]:
46    """
47    Get aggregate open interest across exchanges for a symbol.
48 
49    Uses v2 API for basic OI data.
50 
51    Args:
52        symbol: Coin symbol (BTC, ETH, SOL, etc.)
53        interval: Time interval (0=all, h1, h4, h12, h24).
54 
55    Returns:
56        Dict with open interest data by exchange.
57 
58    Raises:
59        CoinglassError: On API failure.
60    """
61    return cg_request(
62        "open_interest",
63        params={"symbol": symbol, "interval": interval},
64        version="v2"
65    )
66 
67 
68def get_open_interest_history(
69    symbol: str = "BTC",
70    interval: str = "h4",
71    exchange: Optional[str] = None
72) -> Optional[List[Dict[str, Any]]]:
73    """
74    Get open interest history (aggregated across exchanges).
75 
76    Uses v4 API for historical data.
77 
78    Args:
79        symbol: Coin symbol.
80        interval: Time interval (h1, h4, h12, h24).
81        exchange: Optional exchange filter.
82 
83    Returns:
84        List of historical OI data points.
85    """
86    params = {"symbol": symbol, "interval": interval}
87    if exchange:
88        params["exchange"] = exchange
89    return cg_request(
90        "api/futures/open-interest/aggregated-history",
91        params=params
92    )
93 
94 
95def main():
96    """CLI entry point."""
97    parser = argparse.ArgumentParser(
98        description="Coinglass Open Interest Tools"
99    )
100    parser.add_argument("--symbol", "-s", default="BTC")
101    parser.add_argument("--interval", "-i", default="h4")
102    parser.add_argument("--exchange", "-e", default=None)
103    parser.add_argument("--history", action="store_true",
104                        help="Show OI history")
105    parser.add_argument("--json", "-j", action="store_true")
106    args = parser.parse_args()
107 
108    try:
109        if args.history:
110            result = get_open_interest_history(
111                args.symbol, args.interval, args.exchange
112            )
113        else:
114            result = get_open_interest(args.symbol, args.interval)
115 
116        if result:
117            print(json.dumps(result, indent=2))
118        else:
119            print("No data found", file=sys.stderr)
120    except Exception as e:
121        print(f"Error: {e}", file=sys.stderr)
122        sys.exit(1)
123 
124 
125if __name__ == "__main__":
126    main()
127