"""Fetch only pinned TYGER EARS sources and public inference files, not vLLM/TTS."""
import argparse
from pathlib import Path
import shutil
import subprocess
import paths


def main():
    parser=argparse.ArgumentParser();parser.add_argument('--source-only',action='store_true');args=parser.parse_args()
    if not (paths.TYGER/'.git').exists():
        if paths.TYGER.exists() and any(paths.TYGER.iterdir()):raise SystemExit('Refuse to overwrite a nonempty TYGER source directory')
        paths.TYGER.parent.mkdir(parents=True,exist_ok=True)
        subprocess.run(['git','clone','https://github.com/nineninesix-ai/tyger-s2s.git',str(paths.TYGER)],check=True)
    status=subprocess.check_output(['git','-C',str(paths.TYGER),'status','--porcelain'],text=True)
    if status.strip():raise SystemExit('TYGER has local changes; preserve them and resolve the source pin manually')
    subprocess.run(['git','-C',str(paths.TYGER),'checkout','--detach',paths.TYGER_REVISION],check=True)
    if not args.source_only:
        paths.CACHE.mkdir(parents=True,exist_ok=True)
        if not (paths.BUNDLE/'model.safetensors').exists() and shutil.disk_usage(paths.CACHE).free<6*1024**3:
            raise SystemExit('Need at least 6 GiB free before downloading the 2.6 GB inference bundle')
        from huggingface_hub import snapshot_download
        result=snapshot_download('nineninesix/tyger',revision=paths.BUNDLE_REVISION,cache_dir=paths.CACHE,token=False,
            allow_patterns=['model.safetensors','config.json','manifest.json','nemotron/*','LICENSE*','NOTICE*','licenses/*'])
        print('Pinned EARS bundle:',result)
    print('Pinned TYGER EARS source:',paths.TYGER_REVISION)


if __name__=='__main__':main()