"""AI获客 - 客户咨询接口"""
from fastapi import APIRouter, HTTPException
from pydantic import BaseModel
import json, os, datetime, subprocess, urllib.request

router = APIRouter(prefix="/api/cma/lead", tags=["AI获客"])

# ai-cs 实盘分流脚本（跨 profile 调用）
# 必须以 ai-cs 身份 + 其 HERMES_HOME/HOME 运行：脚本内用 `hermes send` 复用该 profile 的企微凭据；
# cma-backend.service 的 Environment 为空，不显式传入会解析到非 ai-cs 的 profile（root 默认 profile）。
AI_CS_USER = "ai-cs"
AI_CS_HOME = "/var/lib/wecom-ai-cs/profile"
AI_CS_ROUTING_SCRIPT = AI_CS_HOME + "/scripts/routing-live.py"

class LeadRequest(BaseModel):
    name: str
    phone: str
    requirement: str = ""
    source: str = "sxbh.ltd"

def notify_wecom(name, phone, requirement, source):
    """推送到企微全员群"""
    msg = {
        "msgtype": "markdown",
        "markdown": {
            "content": f"## 🔔 新客户咨询\n**姓名**: {name}\n**电话**: {phone}\n**需求**: {requirement}\n**来源**: {source}\n**时间**: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M')}"
        }
    }
    try:
        data = json.dumps(msg).encode('utf-8')
        req = urllib.request.Request(
            "http://127.0.0.1:8800/send",
            data=data,
            headers={"Content-Type": "application/json"},
            method="POST"
        )
        urllib.request.urlopen(req, timeout=5)
    except:
        pass  # 通知失败不影响主流程

def notify_boss(name, phone, requirement):
    """额外推给任总（经 ai-cs 实盘分流脚本；失败不影响主流程）"""
    try:
        subprocess.run(
            ["runuser", "-u", AI_CS_USER, "--",
             "env", f"HOME={AI_CS_HOME}", f"HERMES_HOME={AI_CS_HOME}",
             f"USER={AI_CS_USER}", f"LOGNAME={AI_CS_USER}",
             "/usr/bin/python3", AI_CS_ROUTING_SCRIPT,
             requirement or "（未填需求）",
             f"{name} {phone}",
             "website:sxbh.ltd"],
            timeout=30, capture_output=True,
        )
    except Exception:
        pass

@router.post("")
def create_lead(data: LeadRequest):
    """接收客户咨询并保存"""
    record = data.model_dump()
    record["timestamp"] = datetime.datetime.now().isoformat()
    record["status"] = "new"
    
    log_dir = "/root/leads"
    os.makedirs(log_dir, exist_ok=True)
    with open(f"{log_dir}/leads.json", "a") as f:
        f.write(json.dumps(record, ensure_ascii=False) + "\n")
    
    # 异步推送企微通知
    notify_wecom(data.name, data.phone, data.requirement, data.source)
    # 加性：额外推给任总 DM（口径与 ai-cs 客服分流一致）
    notify_boss(data.name, data.phone, data.requirement)
    
    return {"success": True, "message": "咨询已提交，我们将在30分钟内联系您"}