1"""
2Twelve Data Quote Tools — Real-time market quotes for stocks and forex.
3 
4Provides tools for fetching current market data including price, volume, and 52-week metrics.
5"""
6 
7import logging
8from typing import List
9 
10from core.tool import BaseTool, ToolContext, ToolResult
11from .client import get_client, handle_api_error
12 
13logger = logging.getLogger(__name__)
14 
15 
16class TwelveDataQuoteTool(BaseTool):
17    """Get real-time quote for a stock or forex pair."""
18 
19    @property
20    def name(self) -> str:
21        return "twelvedata_quote"
22 
23    @property
24    def description(self) -> str:
25        return """Get real-time market quote for a stock or forex pair.
26 
27Returns comprehensive current market data including:
28- Current price, open, high, low, close
29- Volume and trading data
30- Price change and percent change
31- 52-week high and low
32- Previous close and timestamp
33 
34Use this for current market analysis and live monitoring.
35 
36Parameters:
37- symbol: Stock symbol (e.g., AAPL, MSFT, TSLA) or forex pair (e.g., EUR/USD, GBP/JPY)
38- prepost: (optional) Include pre/post-market data when available (US/Cboe Europe, Pro+ only)
39 
40Returns: Real-time quote with all current market metrics"""
41 
42    @property
43    def parameters(self) -> dict:
44        return {
45            "type": "object",
46            "properties": {
47                "symbol": {
48                    "type": "string",
49                    "description": "Stock symbol (AAPL, MSFT) or forex pair (EUR/USD, GBP/JPY)",
50                },
51                "prepost": {
52                    "type": "boolean",
53                    "description": "Include pre/post-market data when available (US/Cboe Europe, Pro+ only)",
54                },
55            },
56            "required": ["symbol"],
57        }
58 
59    async def execute(self, ctx: ToolContext, symbol: str = "", prepost: bool = False, **kwargs) -> ToolResult:
60        if not symbol:
61            return ToolResult(success=False, error="'symbol' is required")
62        try:
63            data = await get_client().get_quote(symbol=symbol, prepost=prepost)
64            if data.get("status") == "error":
65                return ToolResult(success=False, error=f"API Error: {data.get('message', 'Unknown error')}")
66            return ToolResult(success=True, output=data)
67        except Exception as e:
68            return handle_api_error(e)
69 
70 
71class TwelveDataQuoteBatchTool(BaseTool):
72    """Get real-time quotes for multiple stocks or forex pairs at once."""
73 
74    @property
75    def name(self) -> str:
76        return "twelvedata_quote_batch"
77 
78    @property
79    def description(self) -> str:
80        return """Get real-time quotes for multiple stocks or forex pairs in a single API call.
81 
82Efficient way to fetch market data for multiple symbols simultaneously. Maximum 120 symbols per request.
83 
84Parameters:
85- symbols: Array of stock symbols or forex pairs (max 120)
86- prepost: (optional) Include pre/post-market data when available (US/Cboe Europe, Pro+ only)
87 
88Returns: Quotes for all requested symbols"""
89 
90    @property
91    def parameters(self) -> dict:
92        return {
93            "type": "object",
94            "properties": {
95                "symbols": {
96                    "type": "array",
97                    "items": {"type": "string"},
98                    "description": "Array of stock symbols or forex pairs (max 120)",
99                    "minItems": 1,
100                    "maxItems": 120,
101                },
102                "prepost": {
103                    "type": "boolean",
104                    "description": "Include pre/post-market data when available (US/Cboe Europe, Pro+ only)",
105                },
106            },
107            "required": ["symbols"],
108        }
109 
110    async def execute(self, ctx: ToolContext, symbols: List[str] = None, prepost: bool = False, **kwargs) -> ToolResult:
111        if not symbols:
112            return ToolResult(success=False, error="'symbols' array is required and must not be empty")
113        try:
114            data = await get_client().get_quote_batch(symbols=symbols, prepost=prepost)
115            if data.get("status") == "error":
116                return ToolResult(success=False, error=f"API Error: {data.get('message', 'Unknown error')}")
117            return ToolResult(success=True, output=data)
118        except Exception as e:
119            return handle_api_error(e)
120 
121 
122class TwelveDataPriceBatchTool(BaseTool):
123    """Get latest prices for multiple stocks or forex pairs at once."""
124 
125    @property
126    def name(self) -> str:
127        return "twelvedata_price_batch"
128 
129    @property
130    def description(self) -> str:
131        return """Get latest trading prices for multiple stocks or forex pairs in a single API call.
132 
133Lightweight endpoint for quick price checks on multiple symbols. Maximum 120 symbols per request.
134 
135Parameters:
136- symbols: Array of stock symbols or forex pairs (max 120)
137- prepost: (optional) Include pre/post-market data when available (US/Cboe Europe, Pro+ only)
138 
139Returns: Current prices for all requested symbols"""
140 
141    @property
142    def parameters(self) -> dict:
143        return {
144            "type": "object",
145            "properties": {
146                "symbols": {
147                    "type": "array",
148                    "items": {"type": "string"},
149                    "description": "Array of stock symbols or forex pairs (max 120)",
150                    "minItems": 1,
151                    "maxItems": 120,
152                },
153                "prepost": {
154                    "type": "boolean",
155                    "description": "Include pre/post-market data when available (US/Cboe Europe, Pro+ only)",
156                },
157            },
158            "required": ["symbols"],
159        }
160 
161    async def execute(self, ctx: ToolContext, symbols: List[str] = None, prepost: bool = False, **kwargs) -> ToolResult:
162        if not symbols:
163            return ToolResult(success=False, error="'symbols' array is required and must not be empty")
164        try:
165            data = await get_client().get_price_batch(symbols=symbols, prepost=prepost)
166            if data.get("status") == "error":
167                return ToolResult(success=False, error=f"API Error: {data.get('message', 'Unknown error')}")
168            return ToolResult(success=True, output=data)
169        except Exception as e:
170            return handle_api_error(e)
171