#!/usr/bin/env python3
# Copyright 2023 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""This script is a wrapper around the //buildtools/reclient/reclientreport
binary that populates the log paths correctly for builds run via autoninja
Call this script with the same -C argument used for the autoninja build
Example usage:
$ reclientreport -C out/my-ninja-out
"""

import argparse
import os
import subprocess
import sys
import tarfile
import tempfile

import reclient_helper


# TODO(b/296402157): Remove once reclientreport binary saves all logs on windows
def temp_win_impl__b_296402157(out_dir):
  '''Temporary implementation until b/296402157 is fixed'''
  log_dir = os.path.abspath(os.path.join(out_dir, '.reproxy_tmp', 'logs'))
  with tempfile.NamedTemporaryFile(prefix='reclientreport',
                                   suffix='.tar.gz',
                                   delete=False) as f:
    with tarfile.open(fileobj=f, mode='w:gz') as tar:
      tar.add(log_dir, arcname=os.path.basename(log_dir))
    print(
        f'Created log file at {f.name}. Please attach this to your bug report!')


def main():
  parser = argparse.ArgumentParser(description=__doc__)
  parser.add_argument("--ninja_out",
                      "-C",
                      required=True,
                      help="ninja out directory used for the autoninja build")
  parser.add_argument('args', nargs=argparse.REMAINDER)

  args, extras = parser.parse_known_args()
  if sys.platform.startswith('win'):
    temp_win_impl__b_296402157(args.ninja_out)
    return
  if args.args and args.args[0] == '--':
    args.args.pop(0)
  if extras:
    args.args = extras + args.args

  reclient_helper.set_reproxy_path_flags(args.ninja_out, make_dirs=False)
  reclient_bin_dir = reclient_helper.find_reclient_bin_dir()
  code = subprocess.call([os.path.join(reclient_bin_dir, 'reclientreport')] +
                         args.args)
  if code != 0:
    print("Failed to collect logs, make sure that %s/.reproxy_tmp exists" %
          args.ninja_out,
          file=sys.stderr)


if __name__ == '__main__':
  sys.exit(main())