#!/usr/bin/env python3
# source syntax callbacks execute in the host while actor code is emitted separately.
from pathlib import Path
import sys
import argparse
import subprocess
import tempfile
sys.dont_write_bytecode = True  # importing imagekit must not leave __pycache__ in the tree
from imagekit import micro_image

parser = argparse.ArgumentParser()
parser.add_argument('--host', default='build/bin/zy')
parser.add_argument('--mkgenome', default='build/bin/mkgenome')
parser.add_argument('--case', default='')
args = parser.parse_args()
prefix = b''.join(Path(f).read_bytes() + b'\n' for f in ('src/runtime/rt.zc', 'src/core/core.zc', 'src/runtime/run.zc'))
cases = [('declaration', Path('tests/actor/accept/declaration.zc').read_text(), b'5 7 0 0\n'), ('generic', Path('tests/actor/accept/generic_declaration.zc').read_text(), b'5 7 4 0\n')]
for name, source, expected in cases:
    if args.case and args.case not in name:
        continue
    with tempfile.TemporaryDirectory(prefix='tinyvm-actor-image-') as work:
        directory = Path(work)
        compiled = subprocess.run([args.host], input=Path('src/tools/build.zc').read_bytes() + b'`' + prefix + source.encode(), capture_output=True, timeout=30)
        assert compiled.returncode == 0 and len(compiled.stdout) >= 56 and not compiled.stdout.startswith(b'zc:'), (name, compiled.stdout[:2000], compiled.stderr)
        image = subprocess.run([args.mkgenome], input=compiled.stdout, capture_output=True, check=True)
        binary = micro_image(directory, image.stdout)
        result = subprocess.run([str(binary)], capture_output=True, timeout=12)
        assert result.returncode == 0 and result.stdout == expected, (name, result.returncode, result.stdout, result.stderr)
        print('ok   actor compiler-free image: ' + name)