#!/usr/bin/env python3
import sys
from pathlib import Path

sys.path.insert(0, str(Path.cwd()))

from gx_cli.main import main


def test_compile_basic():
    """Test basic compilation: .py → .gx"""
    output_path = Path("/tmp/test_compile_basic.gx")

    # Clean up
    if output_path.exists():
        output_path.unlink()

    # Compile
    exit_code = main(["compile", "/home/dave/sample_code.py", "-o", str(output_path)])

    if exit_code != 0:
        print("FAIL: compile returned non-zero exit code")
        return False

    if not output_path.exists():
        print("FAIL: output .gx file was not created")
        return False

    size = output_path.stat().st_size
    if size <= 0:
        print("FAIL: output .gx file is empty")
        return False

    # Verify magic bytes
    data = output_path.read_bytes()
    if not data.startswith(b'XIC'):
        print("FAIL: output .gx file does not have XIC magic bytes")
        return False

    print(f"PASS: Compiled to .gx ({size} bytes, magic=XIC)")
    return True


def test_compile_auto_output():
    """Test compilation with auto output naming"""
    source = Path("/home/dave/sample_code.py")
    expected_output = source.with_suffix(".gx")

    # Clean up
    if expected_output.exists():
        expected_output.unlink()

    # Compile without -o (should use sample_code.gx)
    exit_code = main(["compile", str(source)])

    if exit_code != 0:
        print("FAIL: compile with auto output returned non-zero")
        return False

    if not expected_output.exists():
        print(f"FAIL: expected output {expected_output} was not created")
        return False

    size = expected_output.stat().st_size
    print(f"PASS: Auto output naming worked ({size} bytes)")
    return True


def main_test():
    print("[TEST SUITE] test_compile.py")
    print()

    tests = [
        ("Basic compile", test_compile_basic),
        ("Auto output naming", test_compile_auto_output),
    ]

    passed = 0
    failed = 0

    for name, test_func in tests:
        print(f"Running: {name}...", end=" ")
        try:
            if test_func():
                passed += 1
            else:
                failed += 1
        except Exception as e:
            print(f"FAIL: {e}")
            failed += 1

    print()
    print(f"Results: {passed} passed, {failed} failed")

    return 0 if failed == 0 else 1


if __name__ == "__main__":
    sys.exit(main_test())