#!/usr/bin/env python3

# SPDX-FileCopyrightText: © 2023 ZeldaRET
# SPDX-License-Identifier: MIT

from __future__ import annotations

import argparse
import mapfile_parser
from pathlib import Path


ASMPATH = Path("asm")
NONMATCHINGS = "nonmatchings"


def getProgressFromMapFile(mapFile: mapfile_parser.MapFile, asmPath: Path, nonmatchings: Path, aliases: dict[str, str]=dict(), pathIndex: int=2) -> tuple[mapfile_parser.ProgressStats, dict[str, mapfile_parser.ProgressStats]]:
    totalStats = mapfile_parser.ProgressStats()
    progressPerFolder: dict[str, mapfile_parser.ProgressStats] = dict()

    for file in mapFile:
        if len(file.symbols) == 0:
            continue

        folder = file.filepath.parts[pathIndex]
        if folder in aliases:
            folder = aliases[folder]

        if folder not in progressPerFolder:
            progressPerFolder[folder] = mapfile_parser.ProgressStats()

        originalFilePath = Path(*file.filepath.parts[pathIndex:])
        fullAsmFile = asmPath / originalFilePath.with_suffix(".s")
        wholeFileIsUndecomped = fullAsmFile.exists()

        for func in file.symbols:
            if wholeFileIsUndecomped:
                totalStats.undecompedSize += func.size
                progressPerFolder[folder].undecompedSize += func.size
            elif mapFile.findSymbolByName(func.name) is not None:
                totalStats.undecompedSize += func.size
                progressPerFolder[folder].undecompedSize += func.size
            else:
                totalStats.decompedSize += func.size
                progressPerFolder[folder].decompedSize += func.size

    return totalStats, progressPerFolder


def getProgress(mapPath: Path, version: str) -> tuple[mapfile_parser.ProgressStats, dict[str, mapfile_parser.ProgressStats]]:
    mapFile = mapfile_parser.MapFile()
    mapFile.readMapFile(mapPath)

    for file in mapFile:
        if len(file.symbols) == 0:
            continue

        filepathParts = list(file.filepath.parts)
        if version in filepathParts:
            filepathParts.remove(version)
        file.filepath = Path(*filepathParts)

    nonMatchingsPath = ASMPATH / version / NONMATCHINGS

    return mapFile.filterBySegmentType(".text").getProgress(ASMPATH / version, nonMatchingsPath, aliases={"ultralib": "libultra"})

def progressMain():
    parser = argparse.ArgumentParser()
    parser.add_argument("-v", "--version", help="version to process", default="jp")

    args = parser.parse_args()

    mapPath = Path("build") / f"animalforest.{args.version}.map"

    totalStats, progressPerFolder = getProgress(mapPath, args.version)

    mapfile_parser.progress_stats.printStats(totalStats, progressPerFolder)

if __name__ == "__main__":
    progressMain()