o2o-negosium-original/agent/negotiation/chat/service/script_repository.py

"""ScriptRepository — 테넌트 대화 스크립트 로드 + 치환 (P7 대화엔진의 데이터 계층).

- rq_type("재협상"|"재견적")별 스크립트 + 와일드카드 스크립트 + step 매핑 + 변수 매핑 로드.
- 탐색 순서: tenants/<id>/resources/ → 없으면 tenants/_base/resources/ 폴백.
- 브랜드({company_name}/{service_name})는 TenantConfig.resources 에서 주입(클린룸: 특정사 브랜드 비포함).
- 협상 변수({input_price}, {target}, {offer_1pct} 등)는 format_script 에서 치환.
- 표현 규칙: 스크립트는 의미(텍스트)만 소유한다. 문장 내 강조는 `**굵게**` 경량 마커만 허용
  — 렌더(굵기·색)는 프론트 소유(frontend emphasis.tsx). 선행 Chat_server 의 리치텍스트 JSON
  이중 관리(response_scripts.json)는 채택하지 않는다.

Chat_server 구조 참고, 동적 import 해킹/특정사 표현은 제거(CLEANROOM.md).
"""

import json
import os
import re
from typing import Any, Dict, Optional

from common.database.db_session_manager import DB_SESSION_MNG
from common.enums import DBType, DBWRType, ErrorType
from common.logger import LOG
from negotiation.cards.adapters.card_script_db import CardScriptDbRepository
from negotiation.cards.ports.card_script_port import ICardScriptRepository
from tenancy.config import TenantConfig

_RQ_FILES = {"재협상": "scripts_renegotiation.json", "재견적": "scripts_requote.json"}
_CARD_SOURCE_DB = "backoffice_db"
# `{name}` 단일 토큰만 매칭(색 마커 `{{강조|...}}` 는 내부 `|` 로 인해 비매칭 → 보존).
_VAR_TOKEN_RE = re.compile(r"\{(\w+)\}")


class ScriptRepository:
    def __init__(self, config: TenantConfig, tenants_dir: str,
                 card_repo: Optional[ICardScriptRepository] = None):
        self._config = config
        self._tenants_dir = tenants_dir
        self._cache: Dict[str, Any] = {}
        # 카드 멘트 DB 소스(backoffice_db). file 모드면 미사용.
        self._card_repo: ICardScriptRepository = card_repo or CardScriptDbRepository()

    @property
    def config(self) -> TenantConfig:
        """테넌트 config 노출 — ChatEngine 이 협상 규칙(negotiation.*)을 읽는다."""
        return self._config

    # ---- 경로 해석 (_base 폴백) ---------------------------------------
    def _resource_path(self, filename: str) -> Optional[str]:
        scripts_dir = self._config.resources.scripts_dir
        candidates = [
            os.path.join(self._tenants_dir, self._config.tenant_id, scripts_dir, filename),
            os.path.join(self._tenants_dir, "_base", scripts_dir, filename),
        ]
        for p in candidates:
            if os.path.exists(p):
                return p
        return None

    def _load_json(self, filename: str) -> dict:
        if filename in self._cache:
            return self._cache[filename]
        path = self._resource_path(filename)
        if not path:
            LOG.w(f"[ScriptRepository] resource not found: {filename} (tenant={self._config.tenant_id})")
            self._cache[filename] = {}
            return {}
        with open(path, "r", encoding="utf-8") as f:
            data = json.load(f)
        data.pop("_comment", None)
        self._cache[filename] = data
        return data

    # ---- 로드 ----------------------------------------------------------
    def load_scripts(self, rq_type: str = "재협상") -> dict:
        """rq_type 스크립트 + 와일드카드 병합(가격협상_와일드 진입용)."""
        filename = _RQ_FILES.get(rq_type, _RQ_FILES["재협상"])
        scripts = dict(self._load_json(filename))
        # 와일드카드는 재협상 흐름에 병합 (Chat_server 와 동일 동작).
        if rq_type == "재협상":
            for card_id, cfg in self._load_json("scripts_wildcard.json").items():
                scripts[card_id] = cfg
        return scripts

    def wildcard_scripts(self) -> dict:
        return self._load_json("scripts_wildcard.json")

    def card_scripts(self) -> dict:
        """가격협상 카드 스크립트(action_id → 멘트). scripts_cards.json (테넌트→_base 폴백)."""
        return self._load_json("scripts_cards.json")

    def card_script(self, action_id: int, variables: Optional[Dict[str, Any]] = None) -> Optional[str]:
        """선택된 카드(action_id)의 파일 스크립트를 변수 치환해서 반환. 없으면 None(호출부가 기본 문구 유지)."""
        text = self.card_scripts().get(str(action_id))
        if not text:
            return None
        return self.format_script(text, variables)

    async def resolve_card_template(self, action_id: int, card_id: Optional[str],
                                    prefer_db: bool = False) -> tuple:
        """카드 멘트의 **치환 전 템플릿**과 메타를 해석 → (template, tone, strategy_type).

        cards.source_type == 'backoffice_db' 면 card.nego_cards(DB, tone/strategy 포함) 우선,
        없거나 file 모드면 scripts_cards.json(파일, 메타 None) 폴백. 둘 다 없으면 (None, None, None).
        치환 전 템플릿을 그대로 주는 이유: LLM 표현층이 placeholder 를 유지한 채 재작성한 뒤
        format_script 로 치환해야 숫자를 LLM 이 절대 만지지 않기 때문.
        """
        if (prefer_db or self._config.cards.source_type == _CARD_SOURCE_DB) and card_id:
            card = await self._fetch_card_db(card_id)
            if card:
                return card  # (script, tone, strategy)
        return self.card_scripts().get(str(action_id)), None, None  # 파일 폴백(메타 없음)

    async def resolve_card_script(self, action_id: int, card_id: Optional[str],
                                  variables: Optional[Dict[str, Any]] = None,
                                  prefer_db: bool = False) -> Optional[str]:
        """카드 멘트 해석(템플릿 + 변수 치환). DB(정본) 우선 → 파일 폴백. 마커는 불투명 텍스트."""
        template, _, _ = await self.resolve_card_template(action_id, card_id, prefer_db=prefer_db)
        if not template:
            return None
        return self.format_script(template, variables)

    async def resolve_wild_card_template(self, number: str) -> Optional[str]:
        """선택형 와일드카드(WC-*) 멘트 템플릿 — card.wild_cards(DB, negodata 편집 정본) 조회.
        없거나 DB 불가면 None(호출부가 스텝 기본 멘트 폴백)."""
        async def _q(s):
            _, script = await self._card_repo.get_wild_card_by_number(s, number)
            return script

        try:
            return await DB_SESSION_MNG.execute_lambda(DBType.MAIN.value, DBWRType.DB_READ.value, _q)
        except Exception as ex:
            LOG.e_no_callstack(f"[ScriptRepository] 와일드카드 멘트 DB 조회 실패 number={number}: {ex}")
            return None

    async def _fetch_card_db(self, card_id: str) -> Optional[tuple]:
        async def _q(s):
            _, card = await self._card_repo.get_card_by_number(s, card_id)
            return card

        try:
            return await DB_SESSION_MNG.execute_lambda(DBType.MAIN.value, DBWRType.DB_READ.value, _q)
        except Exception as ex:  # DB 불가 등 — 파일 폴백으로 협상은 계속되어야 함
            LOG.e_no_callstack(f"[ScriptRepository] 카드 멘트 DB 조회 실패 card_id={card_id}: {ex}")
            return None

    def client_step_mapping(self) -> dict:
        return self._load_json("client_step_mapping.json")

    def variable_mapping(self) -> dict:
        return self._load_json("variable_mapping.json")

    # ---- 치환 ----------------------------------------------------------
    def _brand_vars(self) -> Dict[str, str]:
        return {
            "company_name": self._config.resources.company_name,
            "service_name": self._config.resources.service_name,
        }

    def format_script(self, text: str, variables: Optional[Dict[str, Any]] = None) -> str:
        """{company_name}/{service_name} + 협상 변수 치환. 누락 변수는 원형 유지.

        정규식으로 `{name}` 단일 토큰만 치환한다 — str.format_map 은 `{{`·`}}` 를 이스케이프로
        해석해 색 마커 `{{강조|...}}` 를 `{강조|...}` 로 붕괴시키므로 쓰지 않는다.
        색 마커는 내부에 `|`(비-\\w)가 있어 `\\{(\\w+)\\}` 에 매칭되지 않아 그대로 보존된다.
        """
        if not text:
            return text
        ctx = self._brand_vars()
        if variables:
            ctx.update({k: v for k, v in variables.items() if v is not None})

        def _repl(m):
            key = m.group(1)
            return str(ctx[key]) if key in ctx else m.group(0)  # 미등록 토큰은 원형

        return _VAR_TOKEN_RE.sub(_repl, text)

    def get_step(self, step: str, rq_type: str = "재협상", variables: Optional[Dict[str, Any]] = None) -> Optional[dict]:
        """step 정의를 반환하되 script 를 치환해서 돌려준다."""
        scripts = self.load_scripts(rq_type)
        node = scripts.get(step)
        if node is None:
            return None
        node = dict(node)
        node["script"] = self.format_script(node.get("script", ""), variables)
        return node