from pathlib import Path
import argparse

from serato_doctor.crate_parser import parse_crates
from serato_doctor.models.library import Library
from serato_doctor.scanner import scan_audio
from serato_doctor.report import write_csv, write_missing_report


def main():
    parser = argparse.ArgumentParser(prog="serato-doctor")
    parser.add_argument("--serato", default=str(Path.home() / "Music/_Serato_"))
    parser.add_argument("--music", default=str(Path.home() / "Library/CloudStorage/OneDrive-Personal/Jukebox"))
    parser.add_argument("--out", default=str(Path.home() / "Desktop/serato_doctor_scan.csv"))
    parser.add_argument("--report", default=str(Path.home() / "Desktop/serato_doctor_missing_report.txt"))
    args = parser.parse_args()

    serato = Path(args.serato)
    music = Path(args.music)
    out = Path(args.out)
    report = Path(args.report)

    library = Library.build(
        references=parse_crates(serato / "Subcrates"),
        tracks=scan_audio(music),
    )
    results = library.reconcile_by_filename()
    missing_count = sum(1 for result in results if not result.exists_by_filename)

    write_csv(results, out)
    write_missing_report(results, report)

    print(f"Crate references: {len(library.references)}")
    print(f"Disk tracks: {len(library.tracks)}")
    print(f"Missing by filename: {missing_count}")
    print(f"CSV: {out}")
    print(f"Report: {report}")


if __name__ == "__main__":
    main()