# vendored from jiangchang-platform-kit; do not edit here, edit the kit.
"""RPA 失败存证路径与截图。"""

from __future__ import annotations

import os
from datetime import datetime


def _artifacts_enabled() -> bool:
    v = (os.environ.get("OPENCLAW_ARTIFACTS_ON_FAILURE") or "1").strip().lower()
    return v not in ("0", "false", "no", "off")


def artifact_path(data_dir: str, batch_id: str, tag: str, ext: str = "png") -> str:
    """返回 {data_dir}/rpa-artifacts/{batch_id}/{tag}_{timestamp}.{ext}"""
    ts = datetime.now().strftime("%Y%m%d_%H%M%S")
    directory = os.path.join(data_dir, "rpa-artifacts", batch_id)
    os.makedirs(directory, exist_ok=True)
    return os.path.join(directory, f"{tag}_{ts}.{ext}")


async def capture_failure(page, data_dir: str, batch_id: str, tag: str) -> str:
    """受 OPENCLAW_ARTIFACTS_ON_FAILURE 控制，截图并返回路径；禁用时返回空字符串。"""
    if not _artifacts_enabled():
        return ""
    path = artifact_path(data_dir, batch_id, tag, ext="png")
    await page.screenshot(path=path, full_page=True)
    return path