에이전트가 담장을 넘었을 때,
우리는 무엇을 배워야 하는가

# 허용된 목적지 외 연결 시도를 로그로 남기는 감시 데코레이터
import socket, functools, datetime, json

ALLOWED_HOSTS = {"pypi.org", "registry.npmjs.org"}  # 화이트리스트
LOG_PATH = "egress_events.jsonl"

_orig_connect = socket.socket.connect

def guarded_connect(self, address):
    host = address[0] if isinstance(address, tuple) else str(address)
    event = {
        "ts": datetime.datetime.utcnow().isoformat(),
        "host": host,
        "allowed": host in ALLOWED_HOSTS,
    }
    with open(LOG_PATH, "a") as f:
        f.write(json.dumps(event) + "\n")
    if host not in ALLOWED_HOSTS:
        raise PermissionError(f"차단됨: 허용되지 않은 목적지 {host}")
    return _orig_connect(self, address)

socket.socket.connect = guarded_connect
# 이 스크립트를 import한 프로세스는 이제 화이트리스트 밖 연결 시 예외가 발생합니다

# 위험도가 일정 수준 이상인 행동은 사람 승인 없이 실행되지 않도록 감싸는 래퍼
import functools, datetime

HIGH_RISK_ACTIONS = {"install_package", "open_network_socket", "write_file"}

def require_approval(action_name):
    def decorator(fn):
        @functools.wraps(fn)
        def wrapper(*args, **kwargs):
            if action_name in HIGH_RISK_ACTIONS:
                print(f"[승인 필요] {action_name} — 인자: {args}")
                ans = input("실행을 승인합니까? (y/n): ")
                if ans.strip().lower() != "y":
                    raise PermissionError(f"{action_name} 실행이 거부되었습니다")
            return fn(*args, **kwargs)
        return wrapper
    return decorator

@require_approval("install_package")
def install_package(name):
    print(f"설치 진행: {name}")

# 사용 예: install_package("numpy")  → 사람 승인 전까지 실행되지 않음

# 60초 내 행동 횟수가 임계치를 넘으면 강제 정지
import time, collections

class KillSwitch:
    def __init__(self, max_actions=20, window_sec=60):
        self.max_actions = max_actions
        self.window_sec = window_sec
        self.events = collections.deque()
        self.tripped = False

    def record(self):
        now = time.time()
        self.events.append(now)
        while self.events and now - self.events[0] > self.window_sec:
            self.events.popleft()
        if len(self.events) > self.max_actions:
            self.tripped = True
            raise RuntimeError("킬스위치 작동: 비정상 행동 빈도 감지, 실행 중단")

# 에이전트 루프 안에서 매 행동 직전에 kill_switch.record() 호출