#!/usr/bin/env python
"""
Combine coverage data from multiple jobs, keeping the data only from the most recent attempt from each job.
Coverage artifacts must be named using the format: "Coverage $(System.JobAttempt) {StableUniqueNameForEachJob}"
The recommended coverage artifact name format is: Coverage $(System.JobAttempt) $(System.StageDisplayName) $(System.JobDisplayName)
Keep in mind that Azure Pipelines does not enforce unique job display names (only names).
It is up to pipeline authors to avoid name collisions when deviating from the recommended format.
"""

from __future__ import annotations

import os
import re
import shutil
import sys


def main():
    """Main program entry point."""
    source_directory = sys.argv[1]

    if "/ansible_collections/" in os.getcwd():
        output_path = "tests/output"
    else:
        output_path = "test/results"

    destination_directory = os.path.join(output_path, "coverage")

    if not os.path.exists(destination_directory):
        os.makedirs(destination_directory)

    jobs = {}
    count = 0

    for name in os.listdir(source_directory):
        match = re.search("^Coverage (?P<attempt>[0-9]+) (?P<label>.+)$", name)
        label = match.group("label")
        attempt = int(match.group("attempt"))
        jobs[label] = max(attempt, jobs.get(label, 0))

    for label, attempt in jobs.items():
        name = f"Coverage {attempt} {label}"
        source = os.path.join(source_directory, name)
        source_files = os.listdir(source)

        for source_file in source_files:
            source_path = os.path.join(source, source_file)
            destination_path = os.path.join(destination_directory, source_file + "." + label)
            print(f'"{source_path}" -> "{destination_path}"')
            shutil.copyfile(source_path, destination_path)
            count += 1

    print("Coverage file count: %d" % count)
    print("##vso[task.setVariable variable=coverageFileCount]%d" % count)
    print("##vso[task.setVariable variable=outputPath]%s" % output_path)


if __name__ == "__main__":
    main()
