release_hashes.py

#!/usr/bin/env python3

# Copyright (C) 2024, Mark Qvist

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

import os
import sys
import json
import hashlib

major_version = None
minor_version = None
target_version = None

file = open("Config.h", "rb")
config_data = file.read().splitlines()
for line in config_data:
    dline = line.decode("utf-8").strip()
    components = dline.split()
    if dline.startswith("#define MAJ_VERS"):
        major_version = "%01d" % ord(bytes.fromhex(dline.split()[2].split("x")[1]))
    if dline.startswith("#define MIN_VERS"):
        minor_version = "%02d" % ord(bytes.fromhex(dline.split()[2].split("x")[1]))

target_version = major_version+"."+minor_version

release_hashes = {}

if len(sys.argv) > 1:
    # Process a single file specified on the command line.
    filepath = sys.argv[1]

    if not os.path.isfile(filepath):
        print(f"Error: '{filepath}' is not a file.", file=sys.stderr)
        sys.exit(1)

    with open(filepath, "rb") as file:
        release_hashes[os.path.basename(filepath)] = {
            "hash": hashlib.sha256(file.read()).hexdigest(),
            "version": target_version,
        }
else:
    target_dir = "./Release"
    files = os.listdir(target_dir)
    for filename in files:
        if os.path.isfile(os.path.join(target_dir, filename)):
            if filename.startswith("rnode_firmware"):
                file = open(os.path.join(target_dir, filename), "rb")
                release_hashes[filename] = {
                    "hash": hashlib.sha256(file.read()).hexdigest(),
                    "version": target_version
                }

print(json.dumps(release_hashes))