import os
import json
import logging
import requests
from hyperliquid.info import Info
from hyperliquid.utils import constants

from logging_utils import setup_logging

def update_coin_mapping():
    """
    Fetches all assets from Hyperliquid and all coins from CoinGecko,
    then creates and saves a mapping from the Hyperliquid symbol to the
    CoinGecko ID.
    """
    setup_logging('normal', 'CoinMapUpdater')
    logging.info("Starting coin mapping update process...")
    
    # --- 1. Fetch all assets from Hyperliquid ---
    try:
        logging.info("Fetching assets from Hyperliquid...")
        info = Info(constants.MAINNET_API_URL, skip_ws=True)
        # The meta object contains the 'universe' list with asset details
        meta, asset_contexts = info.meta_and_asset_ctxs()
        
        # --- FIX: The asset names are in the 'universe' list inside the meta object ---
        # The 'universe' is a list of dictionaries, each with a 'name'
        hyperliquid_assets = [asset['name'] for asset in meta['universe']]
        
        logging.info(f"Found {len(hyperliquid_assets)} assets on Hyperliquid.")
    except Exception as e:
        logging.error(f"Failed to fetch assets from Hyperliquid: {e}")
        return

    # --- 2. Fetch all coins from CoinGecko ---
    try:
        logging.info("Fetching coin list from CoinGecko...")
        response = requests.get("https://api.coingecko.com/api/v3/coins/list")
        response.raise_for_status()
        coingecko_coins = response.json()
        # Create a lookup table: {symbol: id}
        coingecko_lookup = {coin['symbol'].upper(): coin['id'] for coin in coingecko_coins}
        logging.info(f"Found {len(coingecko_coins)} coins on CoinGecko.")
    except requests.exceptions.RequestException as e:
        logging.error(f"Failed to fetch coin list from CoinGecko: {e}")
        return

    # --- 3. Create the mapping ---
    final_mapping = {}
    manual_overrides = {
        "HYPE": "hyperliquid",
        "PUMP": "pump-fun",
        "ASTER": "astar",
    }

    logging.info("Generating symbol-to-id mapping...")
    for asset_symbol in hyperliquid_assets:
        # Check for manual overrides first
        if asset_symbol in manual_overrides:
            final_mapping[asset_symbol] = manual_overrides[asset_symbol]
            continue
        
        # Try to find a direct match in the CoinGecko lookup table
        if asset_symbol in coingecko_lookup:
            final_mapping[asset_symbol] = coingecko_lookup[asset_symbol]
        else:
            logging.warning(f"No direct match found for '{asset_symbol}' on CoinGecko. It will be excluded.")

    # --- 4. Save the mapping to a file ---
    map_file_path = os.path.join("_data", "coin_id_map.json")
    try:
        with open(map_file_path, 'w', encoding='utf-8') as f:
            json.dump(final_mapping, f, indent=4, sort_keys=True)
        logging.info(f"Successfully saved new coin mapping with {len(final_mapping)} entries to '{map_file_path}'.")
    except IOError as e:
        logging.error(f"Failed to write coin mapping file: {e}")

if __name__ == "__main__":
    update_coin_mapping()