ERP/bot/app/handlers/supplier.py

"""Add-supplier flow: name + one of the 8 fixed Persian categories (required)."""
import logging

from app import drafts as draft_store
from app import observability as obs
from app.api.client import ApiError
from app.drafts import (
    KIND_SUPPLIER, merge_supplier_patch, new_supplier_payload, supplier_missing,
)
from app.keyboards import pick_buttons, simple_review_keyboard
from app.nlu import normalize_text
from app.texts import ERR_GENERIC, esc, supplier_committed_report, supplier_review_text

logger = logging.getLogger("bot.supplier")


async def handle_entities(ctx, chat_id, user, patch: dict, source: str | None,
                          trace_id: str | None = None) -> None:
    draft = await draft_store.get_active(chat_id, KIND_SUPPLIER)
    if draft is not None:
        payload = draft.payload
        merge_supplier_patch(payload, patch)
    else:
        payload = new_supplier_payload()
        merge_supplier_patch(payload, patch)
        draft = await draft_store.create(chat_id, KIND_SUPPLIER, payload, source=source)
    await obs.link_draft(trace_id, draft.id, draft.kind)
    await resolve_payload(ctx, payload, trace_id)
    await draft_store.save_payload(draft, missing={"fields": supplier_missing(payload)})
    await render(ctx, chat_id, draft, fresh=draft.review_message_id is None, trace_id=trace_id)


async def resolve_payload(ctx, payload: dict, trace_id: str | None = None) -> None:
    options = payload.setdefault("options", {})
    if not payload.get("category_id") and payload.get("category_name"):
        norm_cat = normalize_text(payload["category_name"])
        try:
            cats = await ctx.resolver.categories()
        except Exception:
            cats = []
        exact = [c for c in cats if normalize_text(c.name) == norm_cat or (c.extra or {}).get("slug") == norm_cat]
        if len(exact) == 1:
            payload["category_id"] = exact[0].id
            payload["category_name"] = exact[0].name
            options.pop("cat", None)
        else:
            partial = [c for c in cats if norm_cat in normalize_text(c.name) and len(norm_cat) >= 3]
            if len(partial) == 1:
                payload["category_id"] = partial[0].id
                payload["category_name"] = partial[0].name
                options.pop("cat", None)
            elif partial:
                options["cat"] = [{"id": c.id, "name": c.name} for c in partial]
            else:
                options.pop("cat", None)


async def apply_pick(ctx, chat_id, user, draft, field: str, choice: dict) -> None:
    if field == "cat":
        draft.payload["category_id"] = choice["id"]
        draft.payload["category_name"] = choice["name"]
        draft.payload.setdefault("options", {}).pop("cat", None)
    await draft_store.save_payload(draft, missing={"fields": supplier_missing(draft.payload)})
    await render(ctx, chat_id, draft, fresh=False)


def _keyboard(payload: dict, draft_id: str) -> dict:
    rows: list[list[dict]] = []
    options = (payload.get("options") or {}).get("cat") or []
    if options:
        rows += pick_buttons(draft_id, "cat", [o["name"] for o in options[:6]])
    return simple_review_keyboard(draft_id, KIND_SUPPLIER, rows)


async def render(ctx, chat_id, draft, fresh: bool = True, trace_id: str | None = None) -> None:
    payload = draft.payload
    missing = supplier_missing(payload)
    text = supplier_review_text(payload, missing)
    keyboard = _keyboard(payload, draft.id)
    if fresh:
        message = await ctx.send(chat_id, text, reply_markup=keyboard)
        draft.review_message_id = (message or {}).get("message_id")
        await draft_store.save_payload(draft, missing={"fields": missing},
                                       review_message_id=draft.review_message_id)
    else:
        await ctx.edit(chat_id, draft.review_message_id, text, reply_markup=keyboard)
        await draft_store.save_payload(draft, missing={"fields": missing})
    await obs.event(trace_id or await obs.latest_trace_for_draft(draft.id), "draft.render",
                    payload={"fresh": fresh, "missing": [m.get("key") for m in missing if not m.get("ok")]})


async def commit(ctx, chat_id, user, draft, trace_id: str | None = None) -> None:
    payload = draft.payload
    trace_id = trace_id or await obs.latest_trace_for_draft(draft.id)
    await resolve_payload(ctx, payload, trace_id)
    missing = {m["key"]: m["ok"] for m in supplier_missing(payload)}
    problems = [label for key, label, ok in [
        ("name", "نام تأمین‌کننده", missing.get("name")),
        ("category", "دسته‌بندی", missing.get("category")),
    ] if not ok]
    if problems:
        await ctx.send(chat_id, "ثبت ممکن نیست؛ اول این‌ها را مشخص کن:\n" +
                       "\n".join(f"❌ {esc(p)}" for p in problems))
        return
    body = {
        "name": payload["name"],
        "category_id": payload["category_id"],
        "phone": payload.get("phone"),
        "location": payload.get("location"),
        "contact_person": payload.get("contact_person"),
    }
    body = {k: v for k, v in body.items() if v}
    try:
        created = await ctx.api.create_supplier(body, actor_id=user.actor_id)
    except ApiError as exc:
        await ctx.send(chat_id, ERR_GENERIC.format(detail=esc(exc.message)))
        return
    await draft_store.set_status(draft.id, "committed")
    await obs.event(trace_id, "draft.commit", payload={"kind": "supplier",
                                                       "supplier_id": (created or {}).get("id")})
    await obs.set_trace_status(trace_id, "committed")
    ctx.resolver.invalidate()
    await ctx.send(chat_id, supplier_committed_report(created))