electron: support updating chromedriver with update.py

pkgs/development/tools/electron/update.py

@@ -73,6 +73,9 @@ SOURCE_INFO_JSON = "info.json"
 # Relatice path to the electron-bin info.json
 BINARY_INFO_JSON = "binary/info.json"
 
+# Relative path the the electron-chromedriver info.json
+CHROMEDRIVER_INFO_JSON = "chromedriver/info.json"
+
 # Number of spaces used for each indentation level
 JSON_INDENT = 4
 
@@ -85,6 +88,7 @@ click_log.basic_config(logger)
 
 nixpkgs_path = os.path.dirname(os.path.realpath(__file__)) + "/../../../.."
 
+
 class Repo:
     fetcher: str
     args: dict
@@ -277,6 +281,42 @@ class ElectronBinRepo(GitHubRepo):
         return out
 
 
+class ElectronChromedriverRepo(ElectronBinRepo):
+    def __init__(self, rev: str) -> None:
+        super().__init__("electron", "electron", rev)
+        self.systems = {
+            "i686-linux": "linux-ia32",
+            "x86_64-linux": "linux-x64",
+            "armv7l-linux": "linux-armv7l",
+            "aarch64-linux": "linux-arm64",
+            "x86_64-darwin": "darwin-x64",
+            "aarch64-darwin": "darwin-arm64",
+        }
+
+    def get_hashes(self, major_version: str) -> dict:
+        """Returns a dictionary of hashes for a given major version"""
+        m, _ = get_latest_version(major_version)
+        version: str = m["version"]
+
+        out = {}
+        out[major_version] = {
+            "hashes": {},
+            "version": version,
+        }
+
+        hashes: list = self.get_shasums256(version)
+
+        for nix_system, electron_system in self.systems.items():
+            filename = f"*chromedriver-v{version}-{electron_system}.zip"
+            if any([x.endswith(filename) for x in hashes]):
+                out[major_version]["hashes"][nix_system] = [
+                    x.split(" ")[0] for x in hashes if x.endswith(filename)
+                ][0]
+                out[major_version]["hashes"]["headers"] = self.get_headers(version)
+
+        return out
+
+
 # Releases that have reached end-of-life no longer receive any updates
 # and it is rather pointless trying to update those.
 #
@@ -305,10 +345,10 @@ def supported_version_range() -> range:
 
 @memory.cache
 def get_repo_hash(fetcher: str, args: dict) -> str:
-    expr = f'with import {nixpkgs_path} {{}};{fetcher}{{'
+    expr = f"with import {nixpkgs_path} {{}};{fetcher}{{"
     for key, val in args.items():
         expr += f'{key}="{val}";'
-    expr += '}'
+    expr += "}"
     cmd = ["nurl", "-H", "--expr", expr]
     print(" ".join(cmd), file=sys.stderr)
     out = subprocess.check_output(cmd)
@@ -412,6 +452,15 @@ def get_electron_bin_info(major_version: str) -> Tuple[str, str, ElectronBinRepo
     return (major_version, m, electron_repo)
 
 
+def get_electron_chromedriver_info(
+    major_version: str,
+) -> Tuple[str, str, ElectronChromedriverRepo]:
+    m, rev = get_latest_version(major_version)
+
+    electron_repo: ElectronChromedriverRepo = ElectronChromedriverRepo(rev)
+    return (major_version, m, electron_repo)
+
+
 def get_electron_info(major_version: str) -> Tuple[str, str, GitHubRepo]:
     m, rev = get_latest_version(major_version)
 
@@ -523,6 +572,36 @@ def update_bin(major_version: str, commit: bool) -> None:
         commit_result(package_name, old_version, new_version, BINARY_INFO_JSON)
 
 
+def update_chromedriver(major_version: str, commit: bool) -> None:
+    """Update a given electron-chromedriver release
+
+    Args:
+        major_version: The major version number, e.g. '27'
+        commit: Whether the updater should commit the result
+    """
+    package_name = f"electron-chromedriver_{major_version}"
+    print(f"Updating {package_name}")
+
+    electron_chromedriver_info = get_electron_chromedriver_info(major_version)
+    (_major_version, _version, repo) = electron_chromedriver_info
+
+    old_info = load_info_json(CHROMEDRIVER_INFO_JSON)
+    new_info = repo.get_hashes(major_version)
+
+    out = old_info | new_info
+
+    save_info_json(CHROMEDRIVER_INFO_JSON, out)
+
+    old_version = (
+        old_info[major_version]["version"] if major_version in old_info else None
+    )
+    new_version = new_info[major_version]["version"]
+    if old_version == new_version:
+        print(f"{package_name} is up-to-date")
+    elif commit:
+        commit_result(package_name, old_version, new_version, CHROMEDRIVER_INFO_JSON)
+
+
 def update_source(major_version: str, commit: bool) -> None:
     """Update a given electron-source release
 
@@ -643,7 +722,11 @@ def commit_result(
         )
         init_msg = f"init at {new_version}"
         update_msg = f"{old_version} -> {new_version}"
-        diff = f"- Diff: https://github.com/electron/electron/compare/refs/tags/v{old_version}...v{new_version}\n" if old_version != None else ""
+        diff = (
+            f"- Diff: https://github.com/electron/electron/compare/refs/tags/v{old_version}...v{new_version}\n"
+            if old_version != None
+            else ""
+        )
         commit_message = f"""{package_name}: {update_msg if old_version != None else init_msg}
 
 - Changelog: https://github.com/electron/electron/releases/tag/v{new_version}
@@ -682,6 +765,13 @@ def eval(version):
     print(json.dumps(tree, indent=JSON_INDENT, default=vars, sort_keys=True))
 
 
+@cli.command("update-chromedriver", help="Update a single major release")
+@click.option("-v", "--version", help="The major version, e.g. '23'")
+@click.option("-c", "--commit", is_flag=True, default=False, help="Commit the result")
+def update_chromedriver_cmd(version: str, commit: bool) -> None:
+    update_chromedriver(version, commit)
+
+
 @cli.command("update", help="Update a single major release")
 @click.option("-v", "--version", help="The major version, e.g. '23'")
 @click.option(
@@ -719,6 +809,8 @@ def update(version: str, bin_only: bool, source_only: bool, commit: bool) -> Non
         update_bin(version, commit)
         update_source(version, commit)
 
+    update_chromedriver(version, commit)
+
 
 @cli.command("update-all", help="Update all releases at once")
 @click.option(
@@ -765,6 +857,9 @@ def update_all(bin_only: bool, source_only: bool, commit: bool) -> None:
 
         update_all_source(commit)
 
+    for major_version, _ in filtered_bin_info.items():
+        update_chromedriver(major_version, commit)
+
 
 if __name__ == "__main__":
     cli()