"""
factorio_rcon.py
팩토리오 서버에 RCON으로 연결하고 Lua 명령을 실행하는 모듈
"""
import socket
import struct
import time


class FactorioRCON:
    """팩토리오 RCON 클라이언트"""

    def __init__(self, host="127.0.0.1", port=25575, password="factorio_ai"):
        self.host = host
        self.port = port
        self.password = password
        self.sock = None
        self._req_id = 1

    def connect(self):
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.sock.settimeout(10)
        self.sock.connect((self.host, self.port))
        # 인증
        self._send_packet(3, self.password)  # type 3 = AUTH
        resp = self._recv_packet()
        if resp["id"] == -1:
            raise Exception("RCON 인증 실패: 비밀번호를 확인하세요")
        print(f"[RCON] 팩토리오 서버 연결 성공 ({self.host}:{self.port})")

    def disconnect(self):
        if self.sock:
            self.sock.close()
            self.sock = None

    def run(self, lua_command: str) -> str:
        """Lua 명령 실행 후 결과 반환"""
        # /c 없이 순수 Lua로 전송 (RCON은 자동으로 콘솔 명령으로 처리)
        cmd = f"/c {lua_command}"
        req_id = self._send_packet(2, cmd)  # type 2 = EXECCOMMAND
        result = self._recv_packet()
        return result.get("body", "").strip()

    def lua(self, code: str) -> str:
        """여러 줄 Lua 코드 실행"""
        return self.run(code)

    def _send_packet(self, ptype: int, body: str) -> int:
        req_id = self._req_id
        self._req_id += 1
        encoded = body.encode("utf-8") + b"\x00"
        size = 4 + 4 + len(encoded) + 1
        data = struct.pack("<iii", size, req_id, ptype) + encoded + b"\x00"
        self.sock.sendall(data)
        return req_id

    def _recv_packet(self) -> dict:
        # 패킷 크기 읽기
        raw = self._recv_bytes(4)
        size = struct.unpack("<i", raw)[0]
        data = self._recv_bytes(size)
        req_id, ptype = struct.unpack("<ii", data[:8])
        body = data[8:].rstrip(b"\x00").decode("utf-8", errors="replace")
        return {"id": req_id, "type": ptype, "body": body}

    def _recv_bytes(self, n: int) -> bytes:
        buf = b""
        while len(buf) < n:
            chunk = self.sock.recv(n - len(buf))
            if not chunk:
                raise Exception("RCON 연결이 끊어졌습니다")
            buf += chunk
        return buf

    def __enter__(self):
        self.connect()
        return self

    def __exit__(self, *args):
        self.disconnect()