mirror of
https://github.com/github/codeql.git
synced 2025-12-16 16:53:25 +01:00
56 lines
1.4 KiB
Python
Executable File
56 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# Copyright (c) 2024 GitHub, Inc.
|
|
|
|
"""
|
|
Fix metadata in overridden registry, updating `metadata.json` to list correct versions and `source.json`
|
|
to list correct patches with sha256 hashes.
|
|
"""
|
|
|
|
import pathlib
|
|
import json
|
|
import base64
|
|
import hashlib
|
|
import re
|
|
|
|
this_dir = pathlib.Path(__file__).resolve().parent
|
|
|
|
|
|
def sha256(file):
|
|
with open(file, 'rb') as input:
|
|
hash = hashlib.sha256(input.read()).digest()
|
|
hash = base64.b64encode(hash).decode()
|
|
return f"sha256-{hash}"
|
|
|
|
|
|
def patch_file(file, f):
|
|
try:
|
|
data = file.read_text()
|
|
except FileNotFoundError:
|
|
data = None
|
|
file.write_text(f(data))
|
|
|
|
|
|
def patch_json(file, **kwargs):
|
|
def update(data):
|
|
data = json.loads(data) if data else {}
|
|
data.update(kwargs)
|
|
return json.dumps(data, indent=4) + "\n"
|
|
|
|
patch_file(file, update)
|
|
|
|
|
|
for entry in this_dir.joinpath("modules").iterdir():
|
|
if not entry.is_dir():
|
|
continue
|
|
versions = [e for e in entry.iterdir() if e.is_dir()]
|
|
|
|
patch_json(entry / "metadata.json", versions=[v.name for v in versions])
|
|
|
|
for version in versions:
|
|
patch_json(version / "source.json", patches={
|
|
p.name: sha256(p) for p in version.joinpath("patches").iterdir()
|
|
})
|
|
patch_file(version / "MODULE.bazel",
|
|
lambda s: re.sub(r'''version\s*=\s*['"].*['"]''', f'version = "{version.name}"', s, 1))
|