#!/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))