1"""
2Twelve Data Reference Data Tools — Search and discover stocks and forex pairs.
3 
4Provides tools for searching symbols, listing available stocks, forex pairs, and exchanges.
5"""
6 
7import logging
8 
9from core.tool import BaseTool, ToolContext, ToolResult
10from .client import get_client, handle_api_error
11 
12logger = logging.getLogger(__name__)
13 
14 
15class TwelveDataSearchTool(BaseTool):
16    """Search for stocks or forex pairs by name or symbol."""
17 
18    @property
19    def name(self) -> str:
20        return "twelvedata_search"
21 
22    @property
23    def description(self) -> str:
24        return """Search for stocks or forex pairs by company name, symbol, or currency.
25 
26Use this to find the correct symbol before fetching quotes or time series data.
27 
28Examples:
29- Search for "Apple" to find AAPL
30- Search for "EUR" to find EUR/USD and other EUR pairs
31- Search for "MSFT" to get Microsoft details
32 
33Parameters:
34- query: Search query (company name, stock symbol, or currency pair)
35 
36Returns: Array of matching symbols with name, exchange, type, and currency info"""
37 
38    @property
39    def parameters(self) -> dict:
40        return {
41            "type": "object",
42            "properties": {
43                "query": {
44                    "type": "string",
45                    "description": "Search query (company name, stock symbol, or currency)",
46                },
47            },
48            "required": ["query"],
49        }
50 
51    async def execute(self, ctx: ToolContext, query: str = "", **kwargs) -> ToolResult:
52        if not query:
53            return ToolResult(success=False, error="'query' is required")
54        try:
55            data = await get_client().search_symbol(query=query)
56            if data.get("status") == "error":
57                return ToolResult(success=False, error=f"API Error: {data.get('message', 'Unknown error')}")
58            return ToolResult(success=True, output=data)
59        except Exception as e:
60            return handle_api_error(e)
61 
62 
63class TwelveDataStocksTool(BaseTool):
64    """Get list of available stocks, optionally filtered by exchange or country."""
65 
66    @property
67    def name(self) -> str:
68        return "twelvedata_stocks"
69 
70    @property
71    def description(self) -> str:
72        return """Get list of available stocks on Twelve Data.
73 
74Can filter by exchange (NASDAQ, NYSE, etc.) or country (US, GB, etc.) to narrow results.
75 
76Parameters:
77- exchange: (optional) Filter by exchange code (e.g., NASDAQ, NYSE, LSE)
78- country: (optional) Filter by country code (e.g., US, GB, JP)
79 
80Returns: Array of stocks with symbol, name, currency, exchange, and type"""
81 
82    @property
83    def parameters(self) -> dict:
84        return {
85            "type": "object",
86            "properties": {
87                "exchange": {
88                    "type": "string",
89                    "description": "Filter by exchange code (NASDAQ, NYSE, etc.) - optional",
90                },
91                "country": {
92                    "type": "string",
93                    "description": "Filter by country code (US, GB, JP, etc.) - optional",
94                },
95            },
96        }
97 
98    async def execute(self, ctx: ToolContext, exchange: str = "", country: str = "", **kwargs) -> ToolResult:
99        try:
100            data = await get_client().get_stocks(
101                exchange=exchange if exchange else None,
102                country=country if country else None,
103            )
104            if data.get("status") == "error":
105                return ToolResult(success=False, error=f"API Error: {data.get('message', 'Unknown error')}")
106            return ToolResult(success=True, output=data)
107        except Exception as e:
108            return handle_api_error(e)
109 
110 
111class TwelveDataForexPairsTool(BaseTool):
112    """Get list of available forex pairs."""
113 
114    @property
115    def name(self) -> str:
116        return "twelvedata_forex_pairs"
117 
118    @property
119    def description(self) -> str:
120        return """Get list of all available forex (currency) pairs on Twelve Data.
121 
122Returns major, minor, and exotic forex pairs including:
123- Major pairs: EUR/USD, GBP/USD, USD/JPY, etc.
124- Minor pairs: EUR/GBP, GBP/JPY, etc.
125- Exotic pairs: USD/TRY, EUR/HUF, etc.
126 
127No parameters required.
128 
129Returns: Array of forex pairs with symbol, currency base, currency quote"""
130 
131    @property
132    def parameters(self) -> dict:
133        return {"type": "object", "properties": {}}
134 
135    async def execute(self, ctx: ToolContext, **kwargs) -> ToolResult:
136        try:
137            data = await get_client().get_forex_pairs()
138            if data.get("status") == "error":
139                return ToolResult(success=False, error=f"API Error: {data.get('message', 'Unknown error')}")
140            return ToolResult(success=True, output=data)
141        except Exception as e:
142            return handle_api_error(e)
143 
144 
145class TwelveDataExchangesTool(BaseTool):
146    """Get list of supported stock exchanges."""
147 
148    @property
149    def name(self) -> str:
150        return "twelvedata_exchanges"
151 
152    @property
153    def description(self) -> str:
154        return """Get list of all supported stock exchanges on Twelve Data.
155 
156Returns global exchanges including NASDAQ, NYSE, LSE, TSE, and many more.
157 
158No parameters required.
159 
160Returns: Array of exchanges with name, code, country, and timezone"""
161 
162    @property
163    def parameters(self) -> dict:
164        return {"type": "object", "properties": {}}
165 
166    async def execute(self, ctx: ToolContext, **kwargs) -> ToolResult:
167        try:
168            data = await get_client().get_exchanges()
169            if data.get("status") == "error":
170                return ToolResult(success=False, error=f"API Error: {data.get('message', 'Unknown error')}")
171            return ToolResult(success=True, output=data)
172        except Exception as e:
173            return handle_api_error(e)
174