7b6b4e31b6
Co-authored-by: Andrew Bradshaw <andrew.bradshaw@mongodb.com> GitOrigin-RevId: ac3a9ec25fee78b30cd4af625b3ebd8f615ef80e
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
import base64
|
|
import hashlib
|
|
import json
|
|
import os
|
|
import socket
|
|
import sys
|
|
|
|
import git
|
|
|
|
|
|
def write_workstation_bazelrc(args):
|
|
workstation_file = ".bazelrc.common_bes"
|
|
existing_hash = ""
|
|
if os.path.exists(workstation_file):
|
|
with open(workstation_file) as f:
|
|
existing_hash = hashlib.md5(f.read().encode()).hexdigest()
|
|
|
|
try:
|
|
repo = git.Repo()
|
|
except Exception:
|
|
print(
|
|
"Unable to setup git repo, skipping workstation file generation. This will result in incomplete telemetry data being uploaded."
|
|
)
|
|
else:
|
|
try:
|
|
status = "clean" if repo.head.commit.diff(None) is None else "modified"
|
|
except Exception:
|
|
status = "Unknown"
|
|
|
|
try:
|
|
remote = repo.branches.master.repo.remote().url
|
|
except Exception:
|
|
try:
|
|
remote = repo.remotes[0].url
|
|
except Exception:
|
|
remote = "Unknown"
|
|
|
|
try:
|
|
branch = repo.active_branch.name
|
|
except Exception:
|
|
branch = "Unknown"
|
|
|
|
try:
|
|
commit = repo.commit("HEAD")
|
|
except Exception:
|
|
commit = "Unknown"
|
|
|
|
try:
|
|
reader = repo.config_reader()
|
|
user = reader.get_value("user", "email")
|
|
except Exception:
|
|
user = "Unknown"
|
|
|
|
if os.environ.get("CI") is not None:
|
|
user = os.environ.get("author_email", "Unknown")
|
|
|
|
try:
|
|
hostname = socket.gethostname()
|
|
except Exception:
|
|
hostname = "Unknown"
|
|
|
|
developer_build = os.environ.get("CI") is None
|
|
b64_cmd_line = base64.b64encode(json.dumps(args[1:]).encode()).decode()
|
|
normalized_os = sys.platform.replace("win32", "windows").replace("darwin", "macos")
|
|
bazelrc_contents = f"""\
|
|
# Generated file, do not modify
|
|
common --bes_keywords=developerBuild={developer_build}
|
|
common --bes_keywords=user_email={user}
|
|
common --bes_keywords=operating_system={normalized_os}
|
|
common --bes_keywords=engflow:BuildScmRemote={remote}
|
|
common --bes_keywords=engflow:BuildScmBranch={branch}
|
|
common --bes_keywords=engflow:BuildScmRevision={commit}
|
|
common --bes_keywords=engflow:BuildScmStatus={status}
|
|
common --bes_keywords=rawCommandLineBase64={b64_cmd_line}
|
|
"""
|
|
|
|
if developer_build:
|
|
bazelrc_contents += f"common --bes_keywords=workstation={hostname}{os.linesep}"
|
|
|
|
current_hash = hashlib.md5(bazelrc_contents.encode()).hexdigest()
|
|
if existing_hash != current_hash:
|
|
with open(workstation_file, "w") as f:
|
|
f.write(bazelrc_contents)
|