#!/usr/bin/env vpython3
# Copyright 2025 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse
import os
import pathlib
import shlex
import subprocess
import sys
import logging

_SRC_ROOT = pathlib.Path(__file__).resolve().parents[2]
sys.path.append(str(_SRC_ROOT / 'build/android'))
from pylib import constants


def main():
  parser = argparse.ArgumentParser(
    description='Continuously build and apply GN deps fixes.'
  )
  parser.add_argument(
    '-C', '--output-directory', help='Path to the root build directory.'
  )
  parser.add_argument(
    '--quiet', action='store_true', help='Pass --quiet to autoninja'
  )
  parser.add_argument('targets', nargs='*', help='Targets to build')

  logging.basicConfig(
    level=logging.INFO, format='\033[33mStatus:\033[0m %(message)s'
  )
  args = parser.parse_args()
  if args.output_directory:
    constants.SetOutputDirectory(args.output_directory)
  constants.CheckOutputDirectory()
  output_dir = constants.GetOutDirectory()

  cmd = ['autoninja', '-C', output_dir, '-config', 'no-remote-javac']
  if args.quiet:
    cmd += ['--quiet']
  cmd += args.targets

  siso_output_path = os.path.join(output_dir, 'siso_output')

  for iteration_count in range(1, 10000):
    logging.info('Building Iteration %d', iteration_count)
    # Need to delete file since it is not cleared if "gn gen" fails.
    if os.path.exists(siso_output_path):
      os.unlink(siso_output_path)
    result = subprocess.run(cmd)

    if result.returncode == 0:
      logging.info('Build successful after %d iteration(s).', iteration_count)
      return 0

    if not os.path.exists(siso_output_path):
      logging.error('siso_output not found at %s', siso_output_path)
      return 1

    with open(siso_output_path, 'r') as f:
      commands = [
        l.strip() for l in f if l.strip().startswith('build/gn_editor')
      ]

    if not commands:
      logging.error('Build failed and no gn_editor hints were found.')
      return 1

    for command in commands:
      logging.info('Running: %s', command)
      cmd_parts = shlex.split(command)
      subprocess.run(cmd_parts, check=True, cwd=_SRC_ROOT)

  logging.info(
    'Still not fixed after %d iterations. Giving up.', iteration_count
  )
  return 1


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