process_assets.py

import os
import argparse
import hashlib
import zipfile
import shutil
import glob
from pathlib import Path

def sha256sum(filename):
    h = hashlib.sha256()
    with open(filename, "rb") as f:
        for block in iter(lambda: f.read(4096), b""):
            h.update(block)
    return h.hexdigest()

def find_file(directory, pattern):
    matches = list(Path(directory).rglob(pattern))
    return matches[0] if matches else None

def process_artifacts(input_dir, output_dir, version):
    input_dir = Path(input_dir)
    output_dir = Path(output_dir)
    output_dir.mkdir(parents=True, exist_ok=True)

    # Define expected files and their source locations here
    assets = [
        ('phase_cli_linux_amd64_{version}.apk', 'phase-apk', '*.apk'),
        ('phase_cli_linux_amd64_{version}.deb', 'phase-deb', '*.deb'),
        ('phase_cli_linux_amd64_{version}.rpm', 'phase-rpm', '*.rpm'),
        ('phase_cli_linux_amd64_{version}.zip', 'Linux-binary', None),
        ('phase_cli_linux_arm64_{version}.zip', 'Linux-binary-arm64', None),
        ('phase_cli_macos_amd64_{version}.zip', 'macOS-amd64-binary', None),
        ('phase_cli_macos_arm64_{version}.zip', 'macOS-arm64-binary', None),
        ('phase_cli_windows_amd64_{version}.zip', 'Windows-binary', None),
    ]

    for output_file, source_dir, file_pattern in assets:
        output_file = output_file.format(version=version)
        source_path = input_dir / source_dir
        output_path = output_dir / output_file

        if file_pattern:
            # For APK, DEB, and RPM files
            file_path = find_file(input_dir, file_pattern)
            if file_path:
                shutil.copy2(file_path, output_path)
            else:
                print(f"Warning: No file matching '{file_pattern}' found in {input_dir}")
                continue
        elif source_path.is_dir():
            # For directories that need to be zipped
            with zipfile.ZipFile(output_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
                for file in source_path.rglob('*'):
                    if file.is_file():
                        # Adjust the archive structure
                        arcname = Path(source_dir) / file.relative_to(source_path)
                        zipf.write(file, arcname)
        else:
            print(f"Warning: Source not found for {output_file}")
            continue

        # Generate SHA256 hash
        hash_file = output_path.with_name(f"{output_path.name}.sha256")
        hash_value = sha256sum(output_path)
        hash_file.write_text(f'{hash_value} {output_file}\n')

def main():
    parser = argparse.ArgumentParser(description='Process Phase CLI artifacts')
    parser.add_argument('input_dir', help='Input directory containing artifacts')
    parser.add_argument('output_dir', help='Output directory for processed artifacts')
    parser.add_argument('--version', required=True, help='Version number for file names')
    
    args = parser.parse_args()
    
    process_artifacts(args.input_dir, args.output_dir, args.version)
    print(f"Artifacts processed and saved to {args.output_dir}")

if __name__ == "__main__":
    main()