import requests

from app.infrastructure.service.backend.config import AI_PROVIDER, DEEPSEEK_API_BASE, DEEPSEEK_API_KEY, DEEPSEEK_MODEL, DIFY_API_BASE, DIFY_API_KEY, DIFY_USER, OPENAI_API_BASE, OPENAI_API_KEY, OPENAI_MODEL
from app.infrastructure.service.logging.log_service import log_event, new_trace_id


def do_openai_like(url, headers, payload):
    try:
        resp = requests.post(url, headers=headers, json=payload, timeout=60)
        data = resp.json() if resp.text else {}
        if resp.status_code >= 400:
            return f"抱歉，AI 服务请求失败（{resp.status_code}）"
        content = (((data or {}).get("choices") or [{}])[0].get("message") or {}).get("content", "")
        return content.strip() if content else "抱歉，AI 暂时没有合理的回复。"
    except Exception:
        return "抱歉，AI 服务暂时不可用，请稍后再试。"


def do_dify(url, headers, payload):
    try:
        resp = requests.post(url, headers=headers, json=payload, timeout=60)
        text = resp.text or ""
        if "data:" in text:
            answer = ""
            for line in text.splitlines():
                line = line.strip()
                if not line.startswith("data:"):
                    continue
                chunk = line[5:].strip()
                if not chunk or chunk == "[DONE]":
                    continue
                try:
                    j = requests.models.complexjson.loads(chunk)
                except Exception:
                    continue
                if "answer" in j:
                    answer += j["answer"]
            if answer.strip():
                return answer.strip()
        data = resp.json() if resp.text else {}
        if resp.status_code >= 400:
            return f"抱歉，Dify 服务请求失败（{resp.status_code}）"
        return (data.get("answer") or "抱歉，Dify 暂时没有合理的回复。").strip()
    except Exception:
        return "抱歉，Dify 服务暂时不可用，请稍后再试。"


def call_ai(prompt, user_id=""):
    trace_id = new_trace_id("ai")
    provider = AI_PROVIDER
    log_event("INFO", "ai", "ai.request", trace_id, "request", "ok", "发起AI请求", extra={"provider": provider, "user_id": user_id or "", "prompt_len": len(prompt or "")})
    if provider == "mock":
        result = "【自动回复】你刚才说了：" + (prompt or "")[:100]
        log_event("INFO", "ai", "ai.response", trace_id, "response", "ok", "AI回复完成", extra={"provider": provider, "reply_len": len(result)})
        return result
    if provider == "openai":
        result = do_openai_like(
            OPENAI_API_BASE.rstrip("/") + "/chat/completions",
            {"Content-Type": "application/json", "Authorization": f"Bearer {OPENAI_API_KEY}"},
            {"model": OPENAI_MODEL, "messages": [{"role": "system", "content": "你是一个专业的微信私域运营助手，用简洁自然的中文回复用户。"}, {"role": "user", "content": prompt}], "temperature": 0.7, "user": user_id or None},
        )
    elif provider == "deepseek":
        result = do_openai_like(
            DEEPSEEK_API_BASE.rstrip("/") + "/chat/completions",
            {"Content-Type": "application/json", "Authorization": f"Bearer {DEEPSEEK_API_KEY}"},
            {"model": DEEPSEEK_MODEL, "messages": [{"role": "system", "content": "你是一个简洁高效的微信助手。回复要求：一句话，不超过50字。"}, {"role": "user", "content": prompt}], "temperature": 0.7, "max_tokens": 100, "user": user_id or None},
        )
    elif provider == "dify":
        result = do_dify(
            DIFY_API_BASE.rstrip("/") + "/chat-messages",
            {"Content-Type": "application/json", "Authorization": f"Bearer {DIFY_API_KEY}"},
            {"inputs": {}, "query": prompt, "response_mode": "streaming", "user": user_id or DIFY_USER, "conversation_id": ""},
        )
    else:
        result = "AI_PROVIDER 未配置正确，请检查环境变量。"

    if result.startswith("抱歉") or "未配置正确" in result:
        log_event("WARNING", "ai", "ai.response", trace_id, "response", "failed", "AI回复异常或降级", reason="provider_error", extra={"provider": provider, "reply": result[:120]})
    else:
        log_event("INFO", "ai", "ai.response", trace_id, "response", "ok", "AI回复完成", extra={"provider": provider, "reply_len": len(result)})
    return result