"""Asset class detection utilities for the CLI."""

# Known commodities from Alpha Vantage
KNOWN_COMMODITIES = {
    "WTI",
    "BRENT",
    "NATURAL_GAS",
    "COPPER",
    "ALUMINUM",
    "WHEAT",
    "CORN",
    "SUGAR",
    "COTTON",
    "COFFEE",
}

# Known cryptocurrencies supported by Alpha Vantage
KNOWN_CRYPTOS = {
    "BTC",
    "ETH",
    "BNB",
    "XRP",
    "ADA",
    "SOL",
    "DOGE",
    "DOT",
    "MATIC",
    "AVAX",
    "LINK",
    "UNI",
    "ATOM",
    "LTC",
    "BCH",
    "XLM",
    "ALGO",
    "VET",
    "ICP",
    "FIL",
}


def detect_asset_class(symbol: str) -> str:
    """
    Automatically detect if a symbol is a commodity, crypto, or equity.
    
    Args:
        symbol: The ticker symbol (e.g., "BRENT", "BTC", "AAPL")
    
    Returns:
        "commodity", "crypto", or "equity" based on the symbol
    """
    symbol_upper = symbol.upper()
    if symbol_upper in KNOWN_COMMODITIES:
        return "commodity"
    elif symbol_upper in KNOWN_CRYPTOS:
        return "crypto"
    else:
        return "equity"


def get_asset_class_display_name(asset_class: str) -> str:
    """Get a human-friendly display name for the asset class."""
    return asset_class.capitalize()