import subprocess
import sys
import tempfile
import os
import time


class CodeExecutor:
    def execute(self, code: str, test_cases: str = "", timeout: int = 10) -> dict:
        full_code = code
        if test_cases.strip():
            full_code = code + "\n\n" + test_cases

        tmp = None
        try:
            with tempfile.NamedTemporaryFile(
                mode="w", suffix=".py", delete=False, encoding="utf-8"
            ) as f:
                f.write(full_code)
                tmp = f.name

            env = os.environ.copy()
            env["PYTHONIOENCODING"] = "utf-8"
            env["PYTHONUTF8"] = "1"

            t0 = time.time()
            result = subprocess.run(
                [sys.executable, "-X", "utf8", tmp],
                capture_output=True,
                text=True,
                encoding="utf-8",
                errors="replace",
                timeout=timeout,
                env=env,
            )
            elapsed = time.time() - t0

            return {
                "success": result.returncode == 0,
                "stdout": result.stdout,
                "stderr": result.stderr,
                "return_code": result.returncode,
                "elapsed": round(elapsed, 3),
                "timed_out": False,
                "full_code": full_code,
            }
        except subprocess.TimeoutExpired:
            return {
                "success": False,
                "stdout": "",
                "stderr": f"TimeoutError: Execution exceeded {timeout} seconds.",
                "return_code": -1,
                "elapsed": float(timeout),
                "timed_out": True,
                "full_code": full_code,
            }
        except Exception as e:
            return {
                "success": False,
                "stdout": "",
                "stderr": str(e),
                "return_code": -1,
                "elapsed": 0.0,
                "timed_out": False,
                "full_code": full_code,
            }
        finally:
            if tmp:
                try:
                    os.unlink(tmp)
                except OSError:
                    pass