import re
from collections import defaultdict


def parse_defaults(path):
    pattern = re.compile(r"(DESKTOP_DEFAULT|ANDROID_DEFAULT)\((.+)\)")

    contents = ""
    with open(path, "r") as f:
        contents = f.read()

    defaults = defaultdict(lambda: [])
    for match in pattern.finditer(contents):
        platform = match.group(1)
        target = match.group(2)
        defaults[platform].append(target)

    return defaults


def write_defaults(output, defaults):
    output.write("export const DefaultTargets = {\n")
    for platform in defaults:
        output.write(f'\t"{platform}": [')
        for target in defaults[platform]:
            output.write(f'"{target}",')
        output.write("],\n")
    output.write("}\n")


def parse_targets(path):
    pattern = re.compile(r"ITEM_VALUE\((.+),[\s]+(.+)\)")

    contents = ""
    with open(path, "r") as f:
        contents = f.read()

    targets = {}
    for match in pattern.finditer(contents):
        target = match.group(1)
        value = match.group(2).replace("llu", "").replace("'", "")
        value = value.split(" << ")
        targets[target] = value

    return targets


def write_targets(output, targets):
    output.write("export const Targets = {\n")
    for target, value in targets.items():
        target_w_padding = f'\t"{target}":'.ljust(45)
        if len(value) == 2:
            # value is in the format 1 << n = ["1", "n"]
            output.write(
                f"{target_w_padding} BigInt({value[0]}) << BigInt({value[1]}),\n"
            )
        else:
            # value is in the format 0xFFFF... = ["0xffff"]
            output.write(f'{target_w_padding} BigInt("{value[0]}"),\n')

    output.write("}\n")


def main(output, targets_path, defaults_path):
    output.write("// This is a generated file. Please do not edit.\n")
    output.write(
        "// See extract_rfp_targets.py, RFPTargets.inc, and RFPTargetsDefault.inc files instead.\n"
    )

    write_targets(output, parse_targets(targets_path))
    output.write("\n")
    write_defaults(output, parse_defaults(defaults_path))