dbx-main/app/store.py

"""JSON 기반 사용자/권한 저장소.

- app/data/users.json 에 사용자 레코드 저장
- 동시성: 파일락 대신 process-내 threading.Lock + 원자적 쓰기(temp → rename)
- 운영시 컨테이너 한 대 가정. 다중 인스턴스 필요해지면 DB 로 교체.
"""
from __future__ import annotations

import json
import os
import tempfile
import threading
from datetime import datetime, timezone
from pathlib import Path
from typing import Any, Iterable

# 슈퍼 관리자 — 강등/삭제 불가
SUPER_ADMIN_EMAIL = "king@dbxcorp.co.kr"

# 시스템에서 지원하는 모듈 키 — 신규 추가시 여기 + 메뉴/템플릿 동시 갱신
MODULE_KEYS: tuple[str, ...] = ("corm", "order", "expense", "vacation")


def _now_iso() -> str:
    return datetime.now(timezone.utc).isoformat(timespec="seconds")


class UserStore:
    def __init__(self, path: Path):
        self._path = path
        self._lock = threading.Lock()
        self._path.parent.mkdir(parents=True, exist_ok=True)
        if not self._path.exists():
            self._write_atomic({"users": {}})

    def _read(self) -> dict[str, Any]:
        try:
            with self._path.open("r", encoding="utf-8") as f:
                data = json.load(f)
        except (FileNotFoundError, json.JSONDecodeError):
            data = {"users": {}}
        if "users" not in data or not isinstance(data["users"], dict):
            data["users"] = {}
        return data

    def _write_atomic(self, data: dict[str, Any]) -> None:
        fd, tmp = tempfile.mkstemp(
            prefix=".users.", suffix=".json.tmp", dir=str(self._path.parent)
        )
        try:
            with os.fdopen(fd, "w", encoding="utf-8") as f:
                json.dump(data, f, ensure_ascii=False, indent=2)
            os.replace(tmp, self._path)
        except Exception:
            try:
                os.unlink(tmp)
            except OSError:
                pass
            raise

    @staticmethod
    def _default_modules(is_admin: bool) -> dict[str, bool]:
        return {key: is_admin for key in MODULE_KEYS}

    @staticmethod
    def _normalize_modules(modules: dict[str, Any] | None, *, is_admin: bool) -> dict[str, bool]:
        base = UserStore._default_modules(is_admin)
        if isinstance(modules, dict):
            for key in MODULE_KEYS:
                if key in modules:
                    base[key] = bool(modules[key])
        if is_admin:
            for key in MODULE_KEYS:
                base[key] = True
        return base

    def get(self, email: str) -> dict[str, Any] | None:
        email = email.lower().strip()
        with self._lock:
            data = self._read()
            user = data["users"].get(email)
            if user is None:
                return None
            return self._enrich(email, user)

    def list_all(self) -> list[dict[str, Any]]:
        with self._lock:
            data = self._read()
            return [self._enrich(email, rec) for email, rec in data["users"].items()]

    def upsert_login(self, *, email: str, name: str, picture: str) -> dict[str, Any]:
        """로그인 시 호출. 신규면 생성, 기존이면 last_login/name/picture 갱신."""
        email = email.lower().strip()
        is_super = email == SUPER_ADMIN_EMAIL
        with self._lock:
            data = self._read()
            now = _now_iso()
            existing = data["users"].get(email)
            if existing is None:
                rec = {
                    "email": email,
                    "name": name,
                    "picture": picture,
                    "role": "admin" if is_super else "user",
                    "modules": self._default_modules(is_admin=is_super),
                    "created_at": now,
                    "last_login": now,
                }
            else:
                rec = dict(existing)
                rec["name"] = name or rec.get("name", email)
                rec["picture"] = picture or rec.get("picture", "")
                rec["last_login"] = now
                # 슈퍼 관리자는 항상 admin + 모든 모듈
                if is_super:
                    rec["role"] = "admin"
                    rec["modules"] = self._default_modules(is_admin=True)
                else:
                    rec["modules"] = self._normalize_modules(
                        rec.get("modules"), is_admin=(rec.get("role") == "admin")
                    )
            data["users"][email] = rec
            self._write_atomic(data)
            return self._enrich(email, rec)

    def update_permissions(
        self, *, email: str, role: str | None, modules: dict[str, Any] | None
    ) -> dict[str, Any]:
        """관리자가 다른 사용자의 권한을 수정."""
        email = email.lower().strip()
        if email == SUPER_ADMIN_EMAIL:
            raise PermissionError("슈퍼 관리자 계정은 수정할 수 없습니다.")
        if role is not None and role not in ("admin", "user"):
            raise ValueError(f"role 값이 잘못되었습니다: {role}")
        with self._lock:
            data = self._read()
            rec = data["users"].get(email)
            if rec is None:
                raise KeyError(f"사용자를 찾을 수 없습니다: {email}")
            if role is not None:
                rec["role"] = role
            new_is_admin = rec.get("role") == "admin"
            rec["modules"] = self._normalize_modules(
                modules if modules is not None else rec.get("modules"),
                is_admin=new_is_admin,
            )
            data["users"][email] = rec
            self._write_atomic(data)
            return self._enrich(email, rec)

    def delete(self, email: str) -> None:
        email = email.lower().strip()
        if email == SUPER_ADMIN_EMAIL:
            raise PermissionError("슈퍼 관리자 계정은 삭제할 수 없습니다.")
        with self._lock:
            data = self._read()
            if email in data["users"]:
                del data["users"][email]
                self._write_atomic(data)

    @staticmethod
    def _enrich(email: str, rec: dict[str, Any]) -> dict[str, Any]:
        out = dict(rec)
        out["email"] = email
        out["is_super_admin"] = email == SUPER_ADMIN_EMAIL
        out["modules"] = UserStore._normalize_modules(
            out.get("modules"), is_admin=(out.get("role") == "admin")
        )
        return out


def is_admin(user_rec: dict[str, Any] | None) -> bool:
    return bool(user_rec and user_rec.get("role") == "admin")


def has_module(user_rec: dict[str, Any] | None, module: str) -> bool:
    if not user_rec:
        return False
    if is_admin(user_rec):
        return True
    mods = user_rec.get("modules") or {}
    return bool(mods.get(module))


def allowed_modules(user_rec: dict[str, Any] | None) -> set[str]:
    return {m for m in MODULE_KEYS if has_module(user_rec, m)}