"""Check the actual wheel and sdist for isolation and required runtime assets."""
import tarfile
import struct
import zipfile
from pathlib import Path

root = Path(__file__).resolve().parents[1]
required = {'pompom/main.py', 'pompom/static/index.html', 'pompom/static/main.js', 'pompom/assets/reminder-1.wav', 'pompom/assets/reminder-2.wav', 'pompom/speech.py', 'pompom/assets/speech-slouch.wav', 'pompom/assets/speech-custom.wav', 'pompom/assets/speech-phone-slouch.wav', 'pompom/assets/speech-phone-custom.wav', 'pompom/assets/speech-slouch-custom.wav', 'pompom/assets/speech-phone-slouch-custom.wav'}

def check_wav(data, name):
    fmt = data.index(b'fmt ') + 8
    encoding, channels, rate, byte_rate, alignment, bits = struct.unpack_from('<HHIIHH', data, fmt)
    assert encoding == 1 and alignment == channels * bits // 8, name
    assert byte_rate == rate * alignment, f'{name}: invalid WAV byte rate'

import tomllib
version = tomllib.loads((root / 'pyproject.toml').read_text())['project']['version']
for path in [*sorted((root / 'dist').glob(f'pompom-{version}-*.whl')), *sorted((root / 'dist').glob(f'pompom-{version}.tar.gz'))]:
    if path.suffix == '.whl':
        with zipfile.ZipFile(path) as archive:
            names = archive.namelist()
            for name in names:
                if name.endswith('.wav'):
                    check_wav(archive.read(name), name)
        assert all(n.startswith('pompom/') or '.dist-info/' in n for n in names), names
    else:
        with tarfile.open(path) as archive:
            names = [n.split('/', 1)[1].removeprefix('src/') for n in archive.getnames() if '/' in n]
            for member in archive.getmembers():
                if member.isfile() and member.name.endswith('.wav'):
                    check_wav(archive.extractfile(member).read(), member.name)
        assert not any(n.split('/')[0] in {'api','apps','clients','devices','docs','tests','scripts','.venv'} for n in names)
    assert required <= set(names), required - set(names)
    assert not any('__pycache__' in n or n.endswith(('.pyc','.env','settings.json')) for n in names)
    print(f'OK: {path.name} — Reachy-only package with UI and audio')
assert list((root / 'dist').glob('*.whl')) and list((root / 'dist').glob('*.tar.gz')), 'Build first: python -m build'