#!/usr/bin/env python3

# SPDX-FileCopyrightText: © 2024 ZeldaRET
# SPDX-License-Identifier: CC0-1.0

import argparse
import csv
from pathlib import Path

from file_addresses import parse_file_addresses, get_z_name_for_overlay


def main():
    parser = argparse.ArgumentParser(
        description="List generated files for the Makefile."
    )
    parser.add_argument(
        "-o", "--output-dir", help="Output directory", type=Path, required=True
    )
    parser.add_argument(
        "--config-dir", help="Config directory", type=Path, required=True
    )

    args = parser.parse_args()

    if not args.config_dir.exists():  # disassembly is not set up for this version
        return

    file_addresses = parse_file_addresses(args.config_dir / "file_addresses.csv")

    generated_files = set()
    for dma_file in file_addresses:
        file_splits_path = args.config_dir / f"files_{dma_file.name}.csv"
        if file_splits_path.exists():
            with open(file_splits_path) as f:
                reader = csv.reader(f)
                for row in reader:
                    if (
                        not row  # blank line
                        or row[2].startswith(".")  # section name or .end
                        or row[2].endswith(
                            "_reloc.s"
                        )  # TODO: disassemble overlay reloc sections?
                    ):
                        continue
                    generated_files.add(row[2])
        elif dma_file.overlay_dir is not None:
            z_name = get_z_name_for_overlay(dma_file.name)
            generated_files.add(
                f"src/overlays/{dma_file.overlay_dir}/{dma_file.name}/{z_name}.s"
            )
        else:
            raise Exception(
                f"DMA file {dma_file.name} is not an overlay but has no file splits"
            )

    for filename in sorted(generated_files):
        print(args.output_dir / filename)


if __name__ == "__main__":
    main()