"""Database-backed sys_config and audit adapters for客服 Agent runtime."""
from __future__ import annotations

from repositories.sys_config import SysConfigRepo
from service.customer_agent.audit import write_anonymous_sensitive_audit


class DatabaseConfigProvider:
    def __init__(self, *, session_factory, repo_factory=SysConfigRepo):
        self.session_factory = session_factory
        self.repo_factory = repo_factory

    async def get(self, key: str, default: str | None = None) -> str | None:
        async with self.session_factory() as session:
            return await self.repo_factory(session).get_value(key, default)

    async def write_audit(self, *, action: str, trace_id: str, session_id: str):
        if action != "anon_sensitive_input":
            raise ValueError(f"Unsupported anonymous audit action: {action}")
        async with self.session_factory() as session:
            await write_anonymous_sensitive_audit(
                session, session_id=session_id, trace_id=trace_id
            )