import os

from .constants import (
    PHASE0,
    ALTAIR,
    BELLATRIX,
    CAPELLA,
    DENEB,
    EIP6110,
    WHISK,
    EIP7002,
)


PREVIOUS_FORK_OF = {
    PHASE0: None,
    ALTAIR: PHASE0,
    BELLATRIX: ALTAIR,
    CAPELLA: BELLATRIX,
    DENEB: CAPELLA,
    EIP6110: DENEB,
    WHISK: CAPELLA,
    EIP7002: CAPELLA,
}

ALL_FORKS = list(PREVIOUS_FORK_OF.keys())

IGNORE_SPEC_FILES = [
    "specs/phase0/deposit-contract.md"
]

EXTRA_SPEC_FILES = {
    BELLATRIX: "sync/optimistic.md"
}


def is_post_fork(a, b) -> bool:
    """
    Returns true if fork a is after b, or if a == b
    """
    if a == b:
        return True

    prev_fork = PREVIOUS_FORK_OF[a]
    if prev_fork == b:
        return True
    elif prev_fork == None:
        return False
    else:
        return is_post_fork(prev_fork, b)


def get_fork_directory(fork):
    dir1 = f'specs/{fork}'
    if os.path.exists(dir1):
        return dir1
    dir2 = f'specs/_features/{fork}'
    if os.path.exists(dir2):
        return dir2
    raise FileNotFoundError(f"No directory found for fork: {fork}")


def get_md_doc_paths(spec_fork: str) -> str:
    md_doc_paths = ""

    for fork in ALL_FORKS:
        if is_post_fork(spec_fork, fork):
            # Append all files in fork directory recursively
            for root, dirs, files in os.walk(get_fork_directory(fork)):
                for filename in files:
                    filepath = os.path.join(root, filename)
                    if filepath.endswith('.md') and filepath not in IGNORE_SPEC_FILES:
                        md_doc_paths += filepath + "\n"
            # Append extra files if any
            if fork in EXTRA_SPEC_FILES:
                md_doc_paths += EXTRA_SPEC_FILES[fork] + "\n"

    return md_doc_paths