mirror of
https://github.com/mongodb/mongo.git
synced 2024-11-24 00:17:37 +01:00
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
import argparse
|
|
import os
|
|
import platform
|
|
import stat
|
|
import urllib.request
|
|
|
|
BUILDIFIER_VERSION = "v6.4.0"
|
|
RELEASE_URL = (
|
|
f"https://mdb-build-public.s3.amazonaws.com/bazel-buildifier-binaries/{BUILDIFIER_VERSION}/"
|
|
)
|
|
|
|
|
|
def determine_platform():
|
|
syst = platform.system()
|
|
pltf = None
|
|
if syst == "Darwin":
|
|
pltf = "darwin"
|
|
elif syst == "Windows":
|
|
pltf = "windows"
|
|
elif syst == "Linux":
|
|
pltf = "linux"
|
|
else:
|
|
raise RuntimeError("Platform cannot be inferred.")
|
|
return pltf
|
|
|
|
|
|
def determine_architecture():
|
|
arch = None
|
|
machine = platform.machine()
|
|
if machine in ("AMD64", "x86_64"):
|
|
arch = "amd64"
|
|
elif machine in ("arm", "arm64", "aarch64"):
|
|
arch = "arm64"
|
|
else:
|
|
raise RuntimeError(f"Detected architecture is not supported: {machine}")
|
|
|
|
return arch
|
|
|
|
|
|
def download(download_location: str = "./"):
|
|
operating_system = determine_platform()
|
|
architechture = determine_architecture()
|
|
if operating_system == "windows" and architechture == "arm64":
|
|
raise RuntimeError("There are no published arm windows releases for buildifier.")
|
|
|
|
extension = ".exe" if operating_system == "windows" else ""
|
|
binary_name = f"buildifier-{operating_system}-{architechture}{extension}"
|
|
url = f"{RELEASE_URL}{binary_name}"
|
|
|
|
file_location = os.path.join(download_location, f"buildifier{extension}")
|
|
urllib.request.urlretrieve(url, file_location)
|
|
print(f"Downloaded buildifier from {url} to {file_location}")
|
|
os.chmod(file_location, stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR)
|
|
print(f"Set user executable permissions on {file_location}")
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
prog="DownloadBuildifier",
|
|
description="This downloads buildifier, it is intended for use in evergreen."
|
|
"This is our temperary solution to get bazel linting while we determine a "
|
|
"long-term solution for getting buildifier on evergreen/development hosts.",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--download-location",
|
|
"-f",
|
|
help="Name of directory to download the buildifier binary to.",
|
|
default="./",
|
|
)
|
|
|
|
args = parser.parse_known_args()
|
|
|
|
download(download_location=args[0].download_location)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|