selenium/scripts/selenium_manager.py at trunk · SeleniumHQ/selenium · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#!/usr/bin/env python

import json
import os
import sys
from pathlib import Path

import urllib3

# Pins the selenium manager release given as the first argument, or the latest one,
# with sha256 values; updates `//common:selenium_manager.bzl`

RELEASES = "https://github.com/SeleniumHQ/selenium_manager_artifacts/releases"
RAW = "https://raw.githubusercontent.com/SeleniumHQ/selenium_manager_artifacts"

http = urllib3.PoolManager(
    retries=urllib3.Retry(total=5, backoff_factor=2, status_forcelist=[429, 500, 502, 503, 504], raise_on_status=False)
)


def get_latest_tag():
    r = http.request("GET", f"{RELEASES}/latest")
    return r.url.rsplit("/", 1)[-1]


# Each release tag points at the commit holding that release's hashes
def get_sha_json(tag):
    r = http.request("GET", f"{RAW}/{tag}/latest.json")
    if r.status != 200:
        raise RuntimeError(f"No hashes found for {tag} (HTTP {r.status})")
    return json.loads(r.data)


def print_linux_x86_64(base_url, sha):
    return """    http_file(
        name = "download_sm_linux_x86_64",
        executable = True,
        sha256 = "{}",
        url = "{}",
    )

""".format(sha, base_url + "/selenium-manager-linux-x86_64")


def print_linux_arm64(base_url, sha):
    return """    http_file(
        name = "download_sm_linux_arm64",
        executable = True,
        sha256 = "{}",
        url = "{}",
    )

""".format(sha, base_url + "/selenium-manager-linux-arm64")


def print_macos(base_url, sha):
    return """    http_file(
        name = "download_sm_macos",
        executable = True,
        sha256 = "{}",
        url = "{}",
    )

""".format(sha, base_url + "/selenium-manager-macos")


def print_windows(base_url, sha):
    return """    http_file(
        name = "download_sm_windows",
        executable = True,
        sha256 = "{}",
        url = "{}",
    )

""".format(sha, base_url + "/selenium-manager-windows.exe")


def print_sbom(base_url, sha):
    return """    http_file(
        name = "download_sm_sbom",
        sha256 = "{}",
        url = "{}",
    )

""".format(sha, base_url + "/selenium-manager.cdx.json")


def print_notice(base_url, sha):
    return """    http_file(
        name = "download_sm_notice",
        sha256 = "{}",
        url = "{}",
    )
""".format(sha, base_url + "/selenium-manager-THIRD-PARTY-NOTICES.txt")


if __name__ == "__main__":
    content = """# DO NOT EDIT! This file is generated by scripts/selenium_manager.py.
# Regenerate with: `bazel run //scripts:selenium_manager`

load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_file")

def selenium_manager():
"""
    tag = sys.argv[1] if len(sys.argv) > 1 else get_latest_tag()
    base_url = f"{RELEASES}/download/{tag}"
    sha_dict = get_sha_json(tag)
    content = content + print_linux_x86_64(base_url, sha_dict["linux-x86_64"])
    content = content + print_linux_arm64(base_url, sha_dict["linux-arm64"])
    content = content + print_macos(base_url, sha_dict["macos"])
    content = content + print_windows(base_url, sha_dict["windows"])
    content = content + print_sbom(base_url, sha_dict["sbom"])
    content = content + print_notice(base_url, sha_dict["notice"])
    content += """
def _selenium_manager_artifacts_impl(_ctx):
    selenium_manager()

selenium_manager_artifacts = module_extension(
    implementation = _selenium_manager_artifacts_impl,
)
"""

    current_script_dir = Path(os.path.realpath(__file__)).parent
    target_file_path = current_script_dir.parent / "common/selenium_manager.bzl"

    with open(target_file_path, "w") as file:
        file.write(content)