from __future__ import annotations

import subprocess
import sys
from dataclasses import dataclass
from pathlib import Path
from typing import Protocol

SCRIPT_TIMEOUT = 60
OUTPUT_LIMIT = 4000
SAFE_PATH = "/usr/local/bin:/usr/bin:/bin"
RADIO = "kvb-org-radio/scripts/radio.py"
RASCI = "kvb-org-rasci/scripts/rasci.py"
NIGHT = "kvb-org-sunset/scripts/night.py"
OPERATOR_SCRIPT = "kvb-org-operator/scripts/operator.py"


class Runner(Protocol):
    def run(self, script: str, args: list[str]) -> dict: ...


@dataclass(frozen=True)
class Scripts:
    skills: Path
    council: Path
    home: Path

    def run(self, script: str, args: list[str]) -> dict:
        command = " ".join([Path(script).name, *args])
        env = {"PATH": SAFE_PATH, "HOME": str(self.home), "AGENTS_COUNCIL_ROOT": str(self.council), "PYTHONDONTWRITEBYTECODE": "1"}
        try:
            done = subprocess.run(
                [sys.executable, str(self.skills / script), *args],
                capture_output=True,
                encoding="utf-8",
                errors="replace",
                timeout=SCRIPT_TIMEOUT,
                env=env,
                stdin=subprocess.DEVNULL,
                check=False,
            )
        except (OSError, subprocess.TimeoutExpired) as error:
            return {"command": command, "code": -1, "stdout": "", "stderr": str(error)}
        return {"command": command, "code": done.returncode, "stdout": done.stdout[-OUTPUT_LIMIT:], "stderr": done.stderr[-OUTPUT_LIMIT:]}