import os
import requests
from typing import Dict

from symbol_details import SymbolDetails

HEADER_FOLDER = 'include'
XMAP_PATH_ARM7 = os.path.join('sub', 'build', 'arm7.nef.xMAP')
MAIN_LSF_PATH = 'main.lsf'

# Symbols with duplicate addresses that should be ignored.
XMAP_SYMBOL_BLACKLIST = set([
    'AI_POTENTIAL_ATTACK_TARGET_WEIGHTS_2'
])

"""
Dictionary format:
{
    <language>: {
        <codeRegion>: {
            <symbolAddress>: {
                Symbol details
            }
        }
    }
}
"""
def read_xmap_symbols() -> Dict[str, Dict[str, Dict[int, SymbolDetails]]]:
    xmap_symbols: Dict[str, Dict[str, Dict[int, SymbolDetails]]] = {}
    for language in ['us', 'eu', 'jp']:
        xmap_symbols[language] = read_xmap_symbols_for_language(language)
    return xmap_symbols

def line_is_data(line: str) -> bool:
    return '.rodata' in line or '.data' in line or '.bss' in line

def read_xmap_symbols_for_language(language: str) -> Dict[str, Dict[int, SymbolDetails]]:
    xmap_path = os.path.join('build', f'pmdsky.{language}', 'main.nef.xMAP')
    remote_xmap_url = f'https://raw.githubusercontent.com/pret/pmd-sky/xmap/pmdsky{language}.xMAP'
    xmap_symbols: Dict[str, Dict[int, SymbolDetails]] = {}
    overlay_names: Dict[str, int] = {}
    OVERLAY_START = 'Overlay '
    with open(MAIN_LSF_PATH, 'r') as main_lsf_file:
        for line in main_lsf_file.readlines():
            if line.startswith(OVERLAY_START):
                overlay_names[line[len(OVERLAY_START) : -1]] = len(overlay_names)

    SECTION_START = '# .'

    if os.path.exists(xmap_path):
        print(f'Using local xMAP file for {language}.')
        with open(xmap_path, 'r') as xmap_file:
            xmap_lines = xmap_file.readlines()
    else:
        print(f'No local xMAP file found for {language}. Using remote xMAP file from upstream repo.')
        remote_xmap_response = requests.get(remote_xmap_url)
        if remote_xmap_response.status_code == 200:
            xmap_lines = remote_xmap_response.text.split('\n')
        else:
            raise Exception(f'Received error {remote_xmap_response.status_code} from {remote_xmap_url}')

    def read_xmap_file(xmap_lines):
        current_section = None
        for line in xmap_lines:
            if line.startswith(SECTION_START):
                section_name = line[len(SECTION_START) : -1]
                # The corresponding BSS sections can contain RAM data symbols
                is_ram = section_name.endswith('.bss')
                section_name = section_name.removesuffix('.bss')
                if section_name in overlay_names:
                    current_section = str(overlay_names[section_name])
                elif section_name == 'main' or section_name == 'ITCM' or section_name == 'arm7':
                    current_section = section_name
                elif section_name == 'MAIN' or section_name == 'WRAM':
                    current_section = 'arm7'
                else:
                    current_section = None
                if current_section is not None and is_ram:
                    current_section = 'ram'
                if current_section is not None and current_section not in xmap_symbols:
                    xmap_symbols[current_section] = {}

            elif current_section is not None and line.startswith('  ') and ('.text' in line or '.itcm' in line or line_is_data(line)) and len(line) > 28 and line[28] not in NON_FUNCTION_SYMBOLS:
                symbol_split = line[28:-1].split('\t')
                symbol_name: str = symbol_split[0]

                if symbol_name in XMAP_SYMBOL_BLACKLIST or '$' in symbol_name or symbol_name.startswith('@'):
                    continue

                symbol_address = int(line[2:10], 16)
                if line_is_data(line):
                    if not symbol_name.startswith('$'):
                        xmap_symbols[current_section][symbol_address] = SymbolDetails(symbol_name, symbol_split[1][1:-1], True)
                else:
                    xmap_symbols[current_section][symbol_address] = SymbolDetails(symbol_name, symbol_split[1][1:-1], '.rodata' in line)

    NON_FUNCTION_SYMBOLS = set(['$', '.'])
    read_xmap_file(xmap_lines)

    if os.path.exists(XMAP_PATH_ARM7):
        print('Using local ARM7 xMAP file.')
        with open(XMAP_PATH_ARM7, 'r') as xmap_file:
            xmap_lines = xmap_file.readlines()
            read_xmap_file(xmap_lines)

    return xmap_symbols