from __future__ import annotations

import os

import pytest
from playwright.sync_api import Page

from e2e.pages.wanzhanggui_order_page import WanyoujiFulfillmentPage, WanzhangguiOrderPage
from e2e.utils.account_store import account_store


@pytest.mark.s_flow_014
@pytest.mark.e2e_ledger
@pytest.mark.e2e_chain
def test_s_flow_014_accept_upstream_order_from_hall(page: Page, settings, bug_reporter) -> None:
    wanzhanggui_phone = os.getenv("E2E_ORDER_WANZHANGGUI_PHONE", settings.wanzhanggui_tenant_phone)
    wanzhanggui_password = os.getenv("E2E_ORDER_WANZHANGGUI_PASSWORD", settings.wanzhanggui_tenant_password)

    store = account_store()
    order = resolve_s_flow_014_order(wanzhanggui_phone)
    if not order:
        pytest.skip("台账中未找到可供 S-FLOW-014 使用的丸友集待接单订单；请先执行 S-FLOW-007 或指定探测订单")

    source_order_no = order["source_order_no"]
    main_order_no = order.get("main_order_no", "")
    order_title = order.get("order_title", source_order_no)
    strategy_type = order.get("strategy_type") or os.getenv("E2E_ORDER_STRATEGY_TYPE", "单平台指定")
    expected_scope = "私单" if strategy_type == "单平台指定" else "公池单/私单"
    wanyouji_phone = resolve_wanyouji_phone(order)
    wanyouji_account = store.find(phone=wanyouji_phone)
    assert wanyouji_account is not None, f"台账中未找到 S-FLOW-014 丸友集账号：{wanyouji_phone}"

    fulfillment_page = WanyoujiFulfillmentPage(page, settings.tenant_home_url, bug_reporter)
    fulfillment_page.login_main_account(wanyouji_account.phone, wanyouji_account.password)
    bug_reporter.record_if_errors(account=wanyouji_account.phone, location="S-FLOW-014 丸友集账号登录")

    accepted_scope = fulfillment_page.accept_external_order_from_hall(
        source_order_no,
        order_title,
        expected_scope,
    )
    order_list_text = fulfillment_page.search_workshop_order_list(
        fulfillment_page.open_workshop_order_list(),
        source_order_no,
        order_title,
    )
    assert source_order_no in order_list_text or order_title in order_list_text, (
        f"S-FLOW-014 确认接单后丸友集订单列表未找到订单；source_order_no={source_order_no}; "
        f"text={order_list_text[:1500]}"
    )
    mark_s_flow_014_accepted(
        store=store,
        source_order_no=source_order_no,
        wanyouji_phone=wanyouji_account.phone,
        accepted_scope=accepted_scope,
        order_list_text=order_list_text,
    )
    bug_reporter.record_if_errors(
        account=wanyouji_account.phone,
        location=f"S-FLOW-014 丸友集确认接单并进入订单列表：{main_order_no or source_order_no}",
    )

    order_page = WanzhangguiOrderPage(page, settings.tenant_home_url, bug_reporter)
    order_page.login_main_account(wanzhanggui_phone, wanzhanggui_password)
    bug_reporter.record_if_errors(account=wanzhanggui_phone, location="S-FLOW-014 丸掌柜账号登录")

    detail_status_text = order_page.assert_order_detail_statuses(
        main_order_no=main_order_no,
        source_order_no=source_order_no,
        order_title=order_title,
        expected_main_status=["待接单", "代练中", "服务中"],
        expected_sub_status=["待接单-已推单", "代练中", "已接单"],
        expected_sub_order_list_status=["已接单", "代练中", "已发布"],
        expected_aftersale_absent=True,
    )
    bug_reporter.record_if_errors(
        account=wanzhanggui_phone,
        location=f"S-FLOW-014 丸掌柜订单详情接单后状态校验：{main_order_no or source_order_no}",
    )

    mark_s_flow_014_accepted(
        store=store,
        source_order_no=source_order_no,
        wanyouji_phone=wanyouji_account.phone,
        accepted_scope=accepted_scope,
        order_list_text=order_list_text,
        wanzhanggui_after_wanyouji_accept_detail_status_text=detail_status_text[:1500],
    )


def mark_s_flow_014_accepted(
    *,
    store,
    source_order_no: str,
    wanyouji_phone: str,
    accepted_scope: str,
    order_list_text: str,
    wanzhanggui_after_wanyouji_accept_detail_status_text: str = "",
) -> None:
    updates = {
        "status": "wanyouji_accepted",
        "current_flow_state": "wanyouji_accepted_without_player_dispatch",
        "wanyouji_accepted_by": wanyouji_phone,
        "wanyouji_accepted_scope": accepted_scope,
        "wanyouji_order_list_status_text": order_list_text[:1500],
        "wanzhanggui_main_status": "代练中",
        "wanzhanggui_sub_order_status": "代练中",
        "wanzhanggui_sub_order_list_status": "服务中",
        "wanyouji_parent_status": "已接单",
        "wanyouji_visibility": "workshop_order_list_visible",
        "available_for_cases": ["S-FLOW-015", "S-FLOW-018", "S-FLOW-021", "S-FLOW-058"],
    }
    if wanzhanggui_after_wanyouji_accept_detail_status_text:
        updates["wanzhanggui_after_wanyouji_accept_detail_status_text"] = (
            wanzhanggui_after_wanyouji_accept_detail_status_text
        )
    store.mark_order_consumed_by_flow_case(
        source_order_no=source_order_no,
        case_id="S-FLOW-014",
        **updates,
    )


def resolve_s_flow_014_order(wanzhanggui_phone: str) -> dict | None:
    explicit_main_order_no = os.getenv("E2E_S_FLOW_MAIN_ORDER_NO", "").strip()
    explicit_source_order_no = os.getenv("E2E_S_FLOW_SOURCE_ORDER_NO", "").strip()
    payload = account_store().load_payload()
    if explicit_main_order_no or explicit_source_order_no:
        for item in payload.get("dispatched_orders", []):
            if wanzhanggui_phone and item.get("wanzhanggui_phone") != wanzhanggui_phone:
                continue
            if explicit_main_order_no and item.get("main_order_no") != explicit_main_order_no:
                continue
            if explicit_source_order_no and item.get("source_order_no") != explicit_source_order_no:
                continue
            return item
        return None

    candidate = account_store().find_order_for_flow_case(
        case_id="S-FLOW-014",
        wanzhanggui_phone=wanzhanggui_phone,
        current_flow_state="wanyouji_pending_accept_rejectable",
    )
    if candidate:
        return candidate
    candidates = [
        item
        for item in payload.get("dispatched_orders", [])
        if (not wanzhanggui_phone or item.get("wanzhanggui_phone") == wanzhanggui_phone)
        and item.get("status") == "dispatched"
        and item.get("current_flow_state") == "wanyouji_pending_accept_rejectable"
        and "S-FLOW-014" in item.get("available_for_cases", [])
        and "S-FLOW-014" not in item.get("consumed_by_cases", [])
    ]
    return sorted(candidates, key=lambda item: item.get("updated_at", item.get("created_at", "")), reverse=True)[0] if candidates else None


def resolve_wanyouji_phone(order: dict) -> str:
    explicit = os.getenv("E2E_ORDER_WANYOUJI_PHONE", "").strip()
    if explicit:
        return explicit
    phones = order.get("wanyouji_phones") or []
    assert phones, f"S-FLOW-014 订单台账缺少 wanyouji_phones；order={order}"
    return phones[0]
