import hashlib import argparse import sys import os FLATID = "app.zen_browser.zen" def get_sha256sum(filename): """Calculate the SHA256 checksum of a file.""" sha256 = hashlib.sha256() try: with open(filename, "rb") as f: for byte_block in iter(lambda: f.read(4096), b""): sha256.update(byte_block) except FileNotFoundError: print(f"File {filename} not found.") sys.exit(1) return sha256.hexdigest() def build_template(template, linux_sha256, flatpak_sha256, version, linux_aarch64_sha256): """Build the template with the provided hashes and version.""" print(f"Building template with version {version}") print(f"\tLinux archive sha256: {linux_sha256}") print(f"\tLinux aarch64 archive sha256: {linux_aarch64_sha256}") print(f"\tFlatpak archive sha256: {flatpak_sha256}") return template.format(linux_sha256=linux_sha256, flatpak_sha256=flatpak_sha256, version=version, linux_aarch64_sha256=linux_aarch64_sha256) def get_template(template_root): """Get the template content from the specified root directory.""" file = os.path.join(template_root, f"{FLATID}.yml.template") print(f"Reading template {file}") try: with open(file, "r") as f: return f.read() except FileNotFoundError: print(f"Template {file} not found.") sys.exit(1) def main(): """Main function to parse arguments and process files.""" parser = argparse.ArgumentParser(description="Prepare flatpak release") parser.add_argument("--version", help="Version of the release", required=True) parser.add_argument("--linux-archive", help="Linux archive", required=True) parser.add_argument("--linux-aarch64-archive", help="Linux aarch64 archive", required=True) parser.add_argument("--flatpak-archive", help="Flatpak archive", required=True) parser.add_argument("--output", help="Output file", default=f"{FLATID}.yml") parser.add_argument("--template-root", help="Template root", default="flatpak") args = parser.parse_args() linux_sha256 = get_sha256sum(args.linux_archive) linux_aarch64_sha256 = get_sha256sum(args.linux_aarch64_archive) flatpak_sha256 = get_sha256sum(args.flatpak_archive) template = build_template(get_template(args.template_root), linux_sha256, flatpak_sha256, args.version, linux_aarch64_sha256) print(f"Writing output to {args.output}") with open(args.output, "w") as f: f.write(template) if __name__ == "__main__": main()