파일을 쓰는 도구, 그리고 백업

"""쓰기 도구 — 읽기 도구와 파일을 나눈다.

트랙 3 의 "read-only tool 을 write tool 로 몰래 변경 금지" 규칙을
파일 분리로 강제한다. 쓰기 기능을 늘리려면 이 파일을 고쳐야 하므로
변경이 리뷰에서 눈에 띈다.

경로 검사 규칙은 읽기 도구와 같은 것을 쓴다. 규칙이 두 벌이 되면
한쪽만 고쳐지는 사고가 난다.
"""
from datetime import datetime

from sec_tools import ALLOWED_ROOT, ToolPermissionError
from sec_tools import _resolve_inside_allowed_root as resolve_inside_allowed_root

BACKUP_DIR = ALLOWED_ROOT / ".backup"


def _guard_not_backup(target, relative_path):
    """백업 폴더를 건드리면 되돌리기가 무력해진다."""
    if target == BACKUP_DIR or BACKUP_DIR in target.parents:
        raise ToolPermissionError("백업 폴더에는 쓸 수 없습니다: %s" % relative_path)


def _make_backup(target, relative_path):
    """덮어쓰기 전 원본을 시각을 붙여 보관하고 파일 이름을 돌려준다."""
    BACKUP_DIR.mkdir(parents=True, exist_ok=True)
    stamp = datetime.now().strftime("%Y%m%d-%H%M%S-%f")
    flat = relative_path.replace("/", "_")
    destination = BACKUP_DIR / ("%s.%s" % (flat, stamp))
    destination.write_text(target.read_text(encoding="utf-8"), encoding="utf-8")
    return destination.name


def write_workspace_file(relative_path, content):
    """Write a UTF-8 text file inside the workspace. Backs up any existing file first."""
    target = resolve_inside_allowed_root(relative_path)
    _guard_not_backup(target, relative_path)

    backup = _make_backup(target, relative_path) if target.is_file() else None
    target.parent.mkdir(parents=True, exist_ok=True)
    target.write_text(content, encoding="utf-8")
    return {"written": relative_path, "backup": backup}


def list_backups(relative_path):
    """이 경로의 백업 파일 이름을 최신순으로 돌려준다."""
    if not BACKUP_DIR.is_dir():
        return []
    prefix = relative_path.replace("/", "_") + "."
    names = [p.name for p in BACKUP_DIR.iterdir() if p.name.startswith(prefix)]
    return sorted(names, reverse=True)


def restore_from_backup(relative_path, backup_name):
    """백업에서 되돌린다. 이 동작도 고위험이므로 승인 대상이다."""
    if backup_name not in list_backups(relative_path):
        raise ToolPermissionError("모르는 백업입니다: %s" % backup_name)

    target = resolve_inside_allowed_root(relative_path)
    _guard_not_backup(target, relative_path)
    source = BACKUP_DIR / backup_name

    target.parent.mkdir(parents=True, exist_ok=True)
    target.write_text(source.read_text(encoding="utf-8"), encoding="utf-8")
    return {"restored": relative_path, "from": backup_name}


# 쓰기 도구 목록. 읽기 목록(sec_tools.READ_ONLY_TOOLS)과 섞지 않는다.
WRITE_TOOLS = [write_workspace_file, restore_from_backup]

from sec_tools import _resolve_inside_allowed_root as resolve_inside_allowed_root

    backup = _make_backup(target, relative_path) if target.is_file() else None

actions_TODO.md.20260815-061341-208193

def _guard_not_backup(target, relative_path):
    """백업 폴더를 건드리면 되돌리기가 무력해진다."""
    if target == BACKUP_DIR or BACKUP_DIR in target.parents:
        raise ToolPermissionError("백업 폴더에는 쓸 수 없습니다: %s" % relative_path)

def restore_from_backup(relative_path, backup_name):
    """백업에서 되돌린다. 이 동작도 고위험이므로 승인 대상이다."""
    if backup_name not in list_backups(relative_path):
        raise ToolPermissionError("모르는 백업입니다: %s" % backup_name)

# 쓰기 도구 목록. 읽기 목록(sec_tools.READ_ONLY_TOOLS)과 섞지 않는다.
WRITE_TOOLS = [write_workspace_file, restore_from_backup]

    def test_write_tools_are_separate_from_read_tools(self):
        read_names = {fn.__name__ for fn in sec_tools.READ_ONLY_TOOLS}
        write_names = {fn.__name__ for fn in sec_write_tools.WRITE_TOOLS}
        self.assertEqual(read_names & write_names, set())