from __future__ import annotations

import argparse
import shutil
from pathlib import Path

from generator.arrange.generate import generate_performance_scores
from generator.practice.generate import generate_practice_scores
from generator.score import count_json_files, validate_generated_tree
from generator.summary import validate_expected_performance_summary, validate_practice_variants

RUNTIME_FONT_FILES = ("Petaluma.otf", "PetalumaScript.otf", "OFL.txt")


def copy_runtime_fonts(static_assets_dir: Path, output_dir: Path) -> None:
  source = static_assets_dir / "fonts" / "petaluma"
  destination = output_dir / "fonts" / "petaluma"
  destination.mkdir(parents=True, exist_ok=True)
  for name in RUNTIME_FONT_FILES:
    shutil.copy2(source / name, destination / name)


def parse_args() -> argparse.Namespace:
  parser = argparse.ArgumentParser(description="Generate Product Vamp package score assets.")
  parser.add_argument("--data-dir", type=Path, required=True)
  parser.add_argument("--output-dir", type=Path, required=True)
  parser.add_argument("--static-assets-dir", type=Path, default=Path("package/assets"))
  parser.add_argument("--clean", action="store_true")
  parser.add_argument("--check", action="store_true")
  return parser.parse_args()


def main() -> int:
  args = parse_args()
  data_dir = args.data_dir.resolve()
  output_dir = args.output_dir.resolve()
  static_assets_dir = args.static_assets_dir.resolve()

  if args.clean and output_dir.exists():
    shutil.rmtree(output_dir)
  output_dir.mkdir(parents=True, exist_ok=True)

  copy_runtime_fonts(static_assets_dir, output_dir)

  performance_count = generate_performance_scores(data_dir / "songs", output_dir / "songs")
  practice_count = generate_practice_scores(data_dir / "practice", output_dir / "practice")

  validate_generated_tree(output_dir)
  print(
    "Generated Product Vamp assets: "
    f"{performance_count} performance scores, "
    f"{practice_count} practice score variants, "
    f"{count_json_files(output_dir)} total JSON files."
  )

  if args.check:
    validate_expected_performance_summary(data_dir / "expected" / "performance_summary.json", output_dir / "songs")
    validate_practice_variants(output_dir / "practice")
    print("Generator check passed.")
  return 0


if __name__ == "__main__":
  raise SystemExit(main())