#!/usr/bin/env python3
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
""" Merge all the profdata files in PROFDATA_DIR, utilizing the binaries in
    BINARIES_DIR, to create a unified coverage report in REPORT_DIR. (If no
    REPORT_DIR is provided, defaults to `out/report`.)

  * Example usage: merge_all_profdata.py --profdata-dir [PROFDATA_DIR]
    --binaries-dir [BINARIES_DIR] [--report-dir [REPORT_DIR]]
"""

import argparse
import logging
import os
import subprocess


def _ParseCommandArguments():
  """Adds and parses relevant arguments for tool commands.

  Returns:
    A dictionary representing the arguments.
  """
  arg_parser = argparse.ArgumentParser()
  arg_parser.usage = __doc__

  arg_parser.add_argument('--llvm-profdata',
                          required=True,
                          type=str,
                          help='Location of LLVM profdata tool')
  arg_parser.add_argument('--profdata-dir',
                          required=True,
                          type=str,
                          help='Directory in which profdata files are stored.')
  arg_parser.add_argument(
      '--outfile',
      type=str,
      required=True,
      help=('Directory where the coverage report should go. '
            'Default is out/report.'))

  args = arg_parser.parse_args()

  return args


args = _ParseCommandArguments()
targets_to_cover = []
for profdata_file in os.listdir(args.profdata_dir):
  targets_to_cover.append(os.path.join(args.profdata_dir, profdata_file))

subprocess_cmd = [args.llvm_profdata, 'merge', '-o', args.outfile]

for target in targets_to_cover:
  subprocess_cmd.append(target)
try:
  subprocess.check_call(subprocess_cmd)
except:
  logging.error("An error occured while merging the profdata.")
  exit(1)