81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
"""
|
|
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()
|