Revision control
Copy as Markdown
#!/usr/bin/env python3
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
import logging
import os
import sys
from pygit2 import Repository
from update_buildconfig_from_gradle import main as update_build_config
CURRENT_DIR = os.path.dirname(os.path.realpath(__file__))
PROJECT_DIR = os.path.realpath(os.path.join(CURRENT_DIR, "..", "..", ".."))
OUTPUT_DIR = os.path.join(PROJECT_DIR, "artifacts", "git")
BUILDCONFIG_DIFF_FILE_NAME = "buildconfig.diff"
BUILDCONFIG_DIFF_FILE = os.path.join(OUTPUT_DIR, BUILDCONFIG_DIFF_FILE_NAME)
BUILDCONFIG_FILE_NAME = ".buildconfig.yml"
logger = logging.getLogger(__name__)
def _are_buildconfig_files_changed(git_diff):
return any(
delta.new_file.path.endswith(BUILDCONFIG_FILE_NAME) for delta in git_diff.deltas
)
def _execute_taskcluster_steps(diff, task_id):
os.makedirs(OUTPUT_DIR, exist_ok=True)
with open(BUILDCONFIG_DIFF_FILE, mode="w") as f:
f.write(diff.patch)
tc_root_url = os.environ["TASKCLUSTER_ROOT_URL"]
artifact_url = f"{tc_root_url}/api/queue/v1/task/{task_id}/artifacts/public%2Fgit%2F{BUILDCONFIG_DIFF_FILE_NAME}" # noqa E501
message = f"""{BUILDCONFIG_FILE_NAME} file changed! Please update it by running:
curl --location --compressed {artifact_url} | git apply
Then commit and push!
"""
logger.error(message)
def _execute_local_steps():
logger.error(f"{BUILDCONFIG_FILE_NAME} file updated! Please commit these changes.")
def main():
update_build_config()
diff = Repository(PROJECT_DIR).diff()
if _are_buildconfig_files_changed(diff):
task_id = os.environ.get("TASK_ID")
if task_id:
_execute_taskcluster_steps(diff, task_id)
else:
_execute_local_steps()
sys.exit(1)
logger.info(f"All good! {BUILDCONFIG_FILE_NAME} is up-to-date with gradle.")
__name__ == "__main__" and main()