"""Apply bounded, backed-up gateway corrections to both requested directories."""
import hashlib
from pathlib import Path
import shutil

ROOTS=[Path('C:/kefu/wechat_rpa'),Path('C:/wechat_rpa')]
HELPER='''async def _dify_message_files(client, outlet, messages, image_b64, timeout):
    """Translate current-turn attachments without leaking image bytes into query."""
    urls = []
    for message in reversed(messages or []):
        if not isinstance(message, dict) or message.get("role") != "user":
            continue
        content = message.get("content")
        if isinstance(content, list):
            for part in content:
                if isinstance(part, dict) and part.get("type") == "image_url":
                    image = part.get("image_url") or {}
                    url = image.get("url", "") if isinstance(image, dict) else image
                    if isinstance(url, str) and url:
                        urls.append(url)
        break
    if image_b64:
        urls.append("data:image/png;base64," + image_b64)
    files = []
    seen = set()
    for url in urls:
        if url in seen:
            continue
        seen.add(url)
        if url.startswith("data:"):
            header, separator, data = url.partition(",")
            mime = header[5:].split(";")[0].lower()
            if not separator or ";base64" not in header or mime not in {"image/png", "image/jpeg", "image/webp", "image/gif"}:
                raise GatewayError("Dify 图片数据格式不受支持", retriable=False)
            files.append(await _dify_upload(client, outlet, data, timeout, mime=mime))
        elif url.startswith(("https://", "http://")):
            files.append({"type": "image", "transfer_method": "remote_url", "url": url})
        else:
            raise GatewayError("Dify 图片地址格式不受支持", retriable=False)
    return files or None


'''

for root in ROOTS:
    backup=root/'backups/recognition-audit-20260916/gateway'
    backup.mkdir(parents=True,exist_ok=True)
    path=root/'model_gateway.py'
    source=path.read_text(encoding='utf-8')
    assert 'async def _dify_message_files' not in source, 'Already applied'
    shutil.copy2(path,backup/path.name)
    old='''                if image_b64 and kind == "dify":
                    dify_files = [await _dify_upload(client, outlet, image_b64, timeout)]'''
    new='''                if kind == "dify":
                    dify_files = await _dify_message_files(client, outlet, messages, image_b64, timeout)'''
    assert old in source
    source=source.replace(old,new,1)
    marker='\n\nasync def _dify_upload('
    assert marker in source
    source=source.replace(marker,'''\n    except (GatewayError, httpx.HTTPError, ValueError) as exc:
        # Attachment preparation is part of this outlet, not a failure of every
        # concurrent model. Keep healthy candidates available to the caller.
        outlet.breaker.record(False)
        return {"provider": outlet.name, "text": "", "latency_ms": int((time.monotonic() - started) * 1000),
                "error": str(exc)[:200] or type(exc).__name__}


'''+HELPER+'async def _dify_upload(',1)
    source=source.replace('client: httpx.AsyncClient, outlet: Outlet, image_b64: str, timeout: float\n) -> dict:',
        'client: httpx.AsyncClient, outlet: Outlet, image_b64: str, timeout: float, *, mime: str = "image/png"\n) -> dict:',1)
    source=source.replace('data={"user": "wechat-rpa-vision"},','data={"user": "wechat-rpa"},',1)
    source=source.replace('files={"file": ("chat.png", base64.b64decode(image_b64), "image/png")},',
        'files={"file": ("chat." + {"image/png":"png", "image/jpeg":"jpg", "image/webp":"webp", "image/gif":"gif"}[mime], base64.b64decode(image_b64, validate=True), mime)},',1)
    source=source.replace('root = model_protocol.dify_api_root(outlet.config.get("base_url") or "")',
        'root = model_protocol.dify_api_root(outlet.config.get("base_url") or "", outlet.config.get("endpoint_mode") or "auto")',1)
    old='''                query = str(body.get("customer_text") or "")
                if not query:
                    for turn in reversed(messages):
                        if turn.get("role") == "user" and isinstance(turn.get("content"), str):
                            query = turn["content"]
                            break'''
    new='''                query = ""
                for turn in reversed(messages):
                    if isinstance(turn, dict) and turn.get("role") == "user":
                        query = model_protocol.message_text(turn.get("content"))
                        break
                if not query:
                    query = str(body.get("customer_text") or "")'''
    assert old in source
    source=source.replace(old,new,1)
    compile(source,str(path),'exec')
    path.write_text(source,encoding='utf-8')
    test=root/'test_model_gateway.py'
    shutil.copy2(test,backup/test.name)
    text=test.read_text(encoding='utf-8').replace('messages=[{"role": "system", "content": "忽略"},','messages=[{"role": "system", "content": "已审核知识参考：仅按适用条件回答"},',1)
    text=text.replace('self.assertEqual(payload["query"], "一天吃几次")','self.assertIn("一天吃几次", payload["query"])\n        self.assertIn("已审核知识参考", payload["query"])',1)
    test.write_text(text,encoding='utf-8')
    for name in ('test_rag_context_audit.py','test_gateway_media_audit.py'):
        shutil.copy2(Path(__file__).parent/name,root/name)
    print(str(root),hashlib.sha256(path.read_bytes()).hexdigest())