# ============================================================
# 订单管理领域服务（moduleId: domain-orders, 可重生 ✅）
# 职责：销售订单只读投影 + P2 订单写入动作的纯应用逻辑
# 规则：新增/编辑/取消/完成订单均写主干世界，必须由调用方走 Harness 确认卡
# ============================================================
from __future__ import annotations                    # 前向类型引用

from datetime import datetime                         # 时间戳
from typing import Any                                # 类型标注

from server.timeutil import fmt_dt, today0            # 时间格式化

World = dict[str, Any]                                # 世界状态类型别名

# 当前数据模型的订单状态。后续订单池审批会扩展为更完整状态机。
ORDER_STATUSES = {"SUBMITTED", "CONFIRMED", "CANCELLED", "COMPLETED"}
CUSTOMER_LEVELS = {"VIP", "A", "B", "C"}


def _now() -> str:
    """当前时间字符串（与世界状态 createdAt/updatedAt 口径一致）。"""
    return fmt_dt(datetime.now())


def _product(world: World, product_id: int) -> dict[str, Any]:
    """取成品物料；不存在则抛出 ValueError，阻止脏订单进入世界状态。"""
    product = next((m for m in world["materials"]
                    if m["id"] == product_id and m["type"] == "FINISHED_PRODUCT"), None)
    if product is None:
        raise ValueError(f"产品不存在或不是成品：{product_id}")
    return product


def _next_item_id(world: World) -> int:
    """销售订单明细发号：扫描现有 items，避免改 WorldStore 计数器契约。"""
    max_id = 0
    for so in world.get("salesOrders", []):
        for item in so.get("items", []):
            if isinstance(item.get("id"), int):
                max_id = max(max_id, item["id"])
    return max_id + 1


def list_products(world: World) -> list[dict[str, Any]]:
    """成品下拉选项（P0 只读）。"""
    return [{
        "id": m["id"],
        "code": m["code"],
        "name": m["name"],
        "unit": m["unit"],
    } for m in world["materials"] if m["type"] == "FINISHED_PRODUCT"]


def list_orders(world: World) -> list[dict[str, Any]]:
    """销售订单列表投影（P0 只读；管理页数据源）。"""
    rows: list[dict[str, Any]] = []
    for so in world["salesOrders"]:
        items = so.get("items", [])
        first = items[0] if items else {}
        rows.append({
            "id": so["id"],
            "orderNo": so["orderNo"],
            "customerName": so["customerName"],
            "customerLevel": so["customerLevel"],
            "orderDate": so["orderDate"],
            "deliveryDate": so["deliveryDate"],
            "priority": so["priority"],
            "status": so["status"],
            "source": so.get("source", "MANUAL"),
            "isRush": so.get("isRush", False),
            "rushStrategy": so.get("rushStrategy"),
            "specialRequirements": so.get("specialRequirements", ""),
            "totalAmount": so.get("totalAmount", 0),
            "itemCount": len(items),
            "productId": first.get("productId"),
            "productCode": first.get("productCode", ""),
            "productName": first.get("productName", ""),
            "quantity": first.get("quantity", 0),
            "unit": first.get("unit", "件"),
            "updatedAt": so.get("updatedAt") or so.get("createdAt"),
        })
    rows.sort(key=lambda x: (x["status"] in ("CANCELLED", "COMPLETED"), x["deliveryDate"], x["priority"]))
    return rows


def find_order(world: World, *, order_id: int | None = None, order_no: str | None = None) -> dict[str, Any] | None:
    """按 ID 或订单号查订单（P0 只读）。"""
    if order_id is not None:
        return next((so for so in world["salesOrders"] if so["id"] == order_id), None)
    if order_no:
        return next((so for so in world["salesOrders"] if so["orderNo"].lower() == order_no.lower()), None)
    return None


def normalize_order_payload(world: World, payload: dict[str, Any]) -> dict[str, Any]:
    """订单写入载荷归一化与校验（P0 纯校验）。"""
    out = dict(payload)
    if out.get("id") in ("", None):
        out.pop("id", None)
    elif not isinstance(out["id"], int):
        out["id"] = int(out["id"])
    out["customerName"] = str(out.get("customerName") or "").strip()
    if not out["customerName"]:
        raise ValueError("客户名称不能为空")
    out["customerLevel"] = str(out.get("customerLevel") or "A").upper()
    if out["customerLevel"] not in CUSTOMER_LEVELS:
        raise ValueError("客户等级必须是 VIP/A/B/C")
    out["deliveryDate"] = str(out.get("deliveryDate") or "").strip()
    if not out["deliveryDate"]:
        raise ValueError("交期不能为空")
    out["priority"] = max(1, min(9, int(out.get("priority") or 5)))
    out["productId"] = int(out.get("productId") or 0)
    product = _product(world, out["productId"])
    out["productName"] = product["name"]
    out["productCode"] = product["code"]
    out["unit"] = product["unit"]
    out["quantity"] = max(1, int(float(out.get("quantity") or 1)))
    status = str(out.get("status") or "CONFIRMED").upper()
    if status not in ORDER_STATUSES:
        raise ValueError("订单状态必须是 SUBMITTED/CONFIRMED/CANCELLED/COMPLETED")
    out["status"] = status
    out["isRush"] = bool(out.get("isRush", False))
    out["rushStrategy"] = "STRATEGY_SHIFT" if out["isRush"] else None
    out["specialRequirements"] = str(out.get("specialRequirements") or "").strip()
    return out


def confirmation_for_order_action(world: World, action: str, payload: dict[str, Any]) -> tuple[str, list[str]]:
    """生成订单 P2 确认卡标题与影响摘要。"""
    if action == "order.upsert":
        p = normalize_order_payload(world, payload)
        verb = "编辑" if p.get("id") else "新增"
        title = f"{verb}销售订单"
        lines = [
            f"客户：{p['customerName']}（{p['customerLevel']}）",
            f"产品：{p['productName']} × {p['quantity']} {p['unit']}，交期 {p['deliveryDate']}",
            "批准后写入主干订单池，并影响后续新排产版本（P2）",
        ]
        return title, lines
    order = find_order(world, order_id=int(payload.get("id") or 0), order_no=payload.get("orderNo"))
    if order is None:
        raise ValueError("目标订单不存在")
    if action == "order.cancel":
        return f"取消订单 {order['orderNo']}", [
            f"客户：{order['customerName']}，当前状态 {order['status']}",
            "批准后该订单不再参与后续排产，历史版本不回写",
            "执行前自动建档，可回滚",
        ]
    if action == "order.complete":
        return f"完成订单 {order['orderNo']}", [
            f"客户：{order['customerName']}，当前状态 {order['status']}",
            "批准后订单及明细置为 COMPLETED，不再参与后续排产",
            "执行前自动建档，可回滚",
        ]
    raise ValueError(f"不支持的订单动作：{action}")


def apply_order_action(world: World, next_id, action: str, payload: dict[str, Any]) -> dict[str, Any]:
    """应用订单写入动作到内存世界（P2 动作的业务侧实现；调用方负责门禁/快照/审计/落盘）。"""
    if action == "order.upsert":
        p = normalize_order_payload(world, payload)
        if p.get("id"):
            order = find_order(world, order_id=p["id"])
            if order is None:
                raise ValueError("目标订单不存在")
            before_status = order["status"]
            order.update({
                "customerName": p["customerName"],
                "customerLevel": p["customerLevel"],
                "deliveryDate": p["deliveryDate"],
                "priority": p["priority"],
                "status": p["status"],
                "isRush": p["isRush"],
                "rushStrategy": p["rushStrategy"],
                "specialRequirements": p["specialRequirements"],
                "updatedAt": _now(),
            })
            if not order.get("items"):
                order["items"] = []
            item = order["items"][0] if order["items"] else None
            if item is None:
                item = {"id": _next_item_id(world), "orderId": order["id"], "lineNo": 1}
                order["items"].append(item)
            item.update({
                "productId": p["productId"],
                "productName": p["productName"],
                "productCode": p["productCode"],
                "quantity": p["quantity"],
                "unit": p["unit"],
                "bomVersion": item.get("bomVersion", "V1.0"),
                "routingVersion": item.get("routingVersion", "V1.0"),
                "status": "COMPLETED" if p["status"] == "COMPLETED"
                          else "CANCELLED" if p["status"] == "CANCELLED"
                          else "PENDING",
                "note": item.get("note", ""),
            })
            return {"order": order, "created": False, "beforeStatus": before_status}
        order_id = next_id("salesOrder")
        order_date = today0().strftime("%Y-%m-%d")
        product = _product(world, p["productId"])
        order = {
            "id": order_id,
            "orderNo": "SO" + today0().strftime("%Y%m%d") + f"{order_id:03d}",
            "customerId": f"CUST{order_id:03d}",
            "customerName": p["customerName"],
            "customerLevel": p["customerLevel"],
            "orderDate": order_date,
            "deliveryDate": p["deliveryDate"],
            "priority": p["priority"],
            "manualPriority": None,
            "status": p["status"],
            "source": "MANUAL",
            "specialRequirements": p["specialRequirements"],
            "totalAmount": p["quantity"] * 100,
            "isRush": p["isRush"],
            "rushStrategy": p["rushStrategy"],
            "changes": [],
            "createdBy": "planner",
            "createdAt": _now(),
            "updatedAt": _now(),
            "items": [{
                "id": _next_item_id(world),
                "orderId": order_id,
                "lineNo": 1,
                "productId": product["id"],
                "productName": product["name"],
                "productCode": product["code"],
                "quantity": p["quantity"],
                "unit": product["unit"],
                "bomVersion": "V1.0",
                "routingVersion": "V1.0",
                "status": "PENDING",
                "note": "",
            }],
        }
        world["salesOrders"].append(order)
        return {"order": order, "created": True, "beforeStatus": None}

    order = find_order(world, order_id=int(payload.get("id") or 0), order_no=payload.get("orderNo"))
    if order is None:
        raise ValueError("目标订单不存在")
    before_status = order["status"]
    if action == "order.cancel":
        order["status"] = "CANCELLED"
        for item in order.get("items", []):
            item["status"] = "CANCELLED"
    elif action == "order.complete":
        order["status"] = "COMPLETED"
        for item in order.get("items", []):
            item["status"] = "COMPLETED"
    else:
        raise ValueError(f"不支持的订单动作：{action}")
    order["updatedAt"] = _now()
    order.setdefault("changes", []).append({
        "at": _now(),
        "action": action,
        "from": before_status,
        "to": order["status"],
    })
    return {"order": order, "created": False, "beforeStatus": before_status}