"""Semantic news scanner for early catalyst detection."""

from typing import Any, Dict, List

from tradingagents.dataflows.discovery.scanner_registry import SCANNER_REGISTRY, BaseScanner
from tradingagents.dataflows.discovery.utils import Priority
from tradingagents.utils.logger import get_logger

logger = get_logger(__name__)


class SemanticNewsScanner(BaseScanner):
    """Scan news for early catalysts using semantic analysis."""

    name = "semantic_news"
    pipeline = "news"

    def __init__(self, config: Dict[str, Any]):
        super().__init__(config)
        self.sources = self.scanner_config.get("sources", ["google_news"])
        self.lookback_hours = self.scanner_config.get("lookback_hours", 6)
        self.min_importance = self.scanner_config.get("min_news_importance", 5)
        self.min_similarity = self.scanner_config.get("min_similarity", 0.5)

    def scan(self, state: Dict[str, Any]) -> List[Dict[str, Any]]:
        if not self.is_enabled():
            return []

        logger.info("📰 Scanning news catalysts...")

        try:
            from datetime import datetime

            from tradingagents.tools.executor import execute_tool

            # Get recent global news
            date_str = datetime.now().strftime("%Y-%m-%d")
            result = execute_tool("get_global_news", date=date_str)

            if not result or not isinstance(result, str):
                return []

            # Extract tickers mentioned in news
            import re

            ticker_pattern = r"\b([A-Z]{2,5})\b|\$([A-Z]{2,5})"
            matches = re.findall(ticker_pattern, result)

            tickers = list(set([t[0] or t[1] for t in matches if t[0] or t[1]]))
            stop_words = {
                "NYSE",
                "NASDAQ",
                "CEO",
                "CFO",
                "IPO",
                "ETF",
                "USA",
                "SEC",
                "NEWS",
                "STOCK",
                "MARKET",
            }
            tickers = [t for t in tickers if t not in stop_words]

            candidates = []
            for ticker in tickers[: self.limit]:
                candidates.append(
                    {
                        "ticker": ticker,
                        "source": self.name,
                        "context": "Mentioned in recent market news",
                        "priority": Priority.MEDIUM.value,
                        "strategy": "news_catalyst",
                    }
                )

            logger.info(f"Found {len(candidates)} news mentions")
            return candidates

        except Exception as e:
            logger.warning(f"⚠️  News scan failed: {e}")
            return []


SCANNER_REGISTRY.register(SemanticNewsScanner)