build/autoroll/extract_and_commit_util.py

# 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.
"""Extracts the zipfile from the cipd instance and commits it to the repo."""

import argparse
import os
import pathlib
import shutil
import subprocess
import sys
from typing import List
import zipfile

_SRC_PATH = pathlib.Path(__file__).resolve().parents[2]
_TO_COMMIT_ZIP_NAME = 'to_commit.zip'
_CHROMIUM_SRC_PREFIX = 'CHROMIUM_SRC'


def _HasChanges(repo):
  output = subprocess.check_output(
      ['git', '-C', repo, 'status', '--porcelain=v1'])
  return bool(output)


def main(committed_dir_path):
  parser = argparse.ArgumentParser(
      description=__doc__, formatter_class=argparse.RawTextHelpFormatter)
  parser.add_argument(
      '--cipd-package-path', required=True, help='Path to cipd package.')
  parser.add_argument(
      '--no-git-add',
      action='store_false',
      dest='git_add',
      help='Whether to git add the extracted files.')
  options = parser.parse_args()

  cipd_package_path = pathlib.Path(options.cipd_package_path)
  to_commit_zip_path = cipd_package_path / _TO_COMMIT_ZIP_NAME
  if not to_commit_zip_path.exists():
    print(f'No zipfile found at {to_commit_zip_path}', file=sys.stderr)
    print('Doing nothing', file=sys.stderr)
    return
  if os.path.exists(committed_dir_path):
    # Delete original contents.
    shutil.rmtree(committed_dir_path)
  os.makedirs(committed_dir_path)
  # Replace with the contents of the zip.
  with zipfile.ZipFile(to_commit_zip_path) as z:
    z.extractall(committed_dir_path)
  changed_file_paths: List[str] = [
      str(committed_dir_path.relative_to(_SRC_PATH))
  ]

  committed_chromium_src_dir = committed_dir_path / _CHROMIUM_SRC_PREFIX
  for root, _, files in os.walk(committed_chromium_src_dir):
    for file in files:
      file_path = os.path.join(root, file)
      path_relative_to_src = os.path.relpath(file_path,
                                             committed_chromium_src_dir)
      full_src_path = _SRC_PATH / path_relative_to_src
      if full_src_path.exists():
        full_src_path.unlink()
      shutil.move(file_path, full_src_path)
      changed_file_paths.append(path_relative_to_src)

  if not _HasChanges(_SRC_PATH):
    print("No changes found after extracting zip. Did you run this script "
          "before?")
    return

  if options.git_add:
    git_add_cmd = ['git', '-C', _SRC_PATH, 'add'] + changed_file_paths
    subprocess.check_call(git_add_cmd)