# -*- coding: utf-8 -*-
"""
upsert_snapshot.py —— 结算快照编排（统一调用各取数脚本）
================================================================================

功能：一键抓取 UpsertUserAll 需要的全部基线数据，整理成「节点字典」。

★ 本模块**只做编排**，取数逻辑都在各自的 api 脚本里（userdata.py / user_map.py …）。

--------------------------------------------------------------------------------
为什么需要快照
--------------------------------------------------------------------------------
服务器对 userData 是**整表覆盖**语义，对差分节点是**只回传变化项**语义。
要把本局的变化正确表达出来，必须先有一份「登录时的服务器状态」作基线，
再把本局的影响叠加到基线上 —— 这就是快照的作用。

真机对照（source/2026-10-05.log）：

    | 节点                | Get 侧（基线）    | Upsert 侧（本局后）  |
    |---------------------|-------------------|----------------------|
    | userData.playCount  | 1257              | 1258        (+1)     |
    | userMapList         | 104 条            | 仅 1 条（distance 变）|
    | userMusicDetailList | 739 组            | 仅 1 条（playCount+1）|
    | userItemList        | 1070 条           | []（无变化）         |

--------------------------------------------------------------------------------
调用示例
--------------------------------------------------------------------------------
    from api import fetch_upsert_snapshot

    snapshot = fetch_upsert_snapshot(user_id, token=token)
    # 之后直接传给 UpsertUserAll(snapshot=snapshot, ...)
"""

from __future__ import annotations

import json
from typing import Any, Dict, List, Optional, Sequence

from config import CLIENT_ID, PLACE_ID, PLAY_SHOP_ITEM_IDS, REGION_ID, TIMEOUT_MS

from .log_setup import log_debug, log_info, log_warning
from .maimai_net import MaimaiNetClient

# 各取数脚本（一个接口一个脚本）
from .userdata import UserData
from .user_extend import UserExtend
from .user_option import UserOption
from .user_character import UserCharacter
from .user_ghost import UserGhost
from .user_map import UserMapAll
from .user_login_bonus import UserLoginBonusAll
from .user_rating import UserRating
from .user_item import UserItemAll
from .user_music import UserMusicDetailAll
from .user_course import UserCourseAll
from .user_charge import UserCharge
from .user_favorite import UserFavoriteAll
from .user_activity import UserActivity
from .user_mission import UserMissionData
from .user_intimate import UserIntimate
from .user_shop_stock import UserShopStock
from .user_favorite_item import UserFavoriteItemAll
from .user_kaleidx_scope import UserKaleidxScope


#: 默认要抓的节点（顺序即抓取顺序）
#: ★ 只列 UpsertUserAll 真正需要的；纯展示类接口（排行/活动/大会…）不在此列。
DEFAULT_SNAPSHOT_NODES: tuple = (
    "userData",
    "userExtend",
    "userOption",
    "userCharacterList",
    "userMapList",
    "userLoginBonusList",
    "userRatingList",
    "userItemList",
    "userMusicDetailList",
    "userCourseList",
    "userChargeList",
    "userFavoriteList",
    "userActivityList",
    "userMissionDataList",
    "userIntimateList",
    "userShopItemStockList",
    "userFavoritemusicList",
    "userKaleidxScopeList",
    "userGhost",
)


def fetch_upsert_snapshot(
    user_id: int,
    *,
    token: str = "",
    client_id: str = CLIENT_ID,
    place_id: int = PLACE_ID,
    region_id: int = REGION_ID,
    shop_item_ids: Optional[Sequence[int]] = None,
    nodes: Optional[Sequence[str]] = None,
    timeout_ms: Optional[int] = None,
    client: Optional[MaimaiNetClient] = None,
    strict: bool = False,
) -> Dict[str, Any]:
    """
    抓取结算所需的全部基线数据。

    参数：
        user_id:       用户 ID（必填）
        token:         会话令牌（可选，多数取数接口不需要）
        client_id:     机台短值
        place_id:      店铺 ID
        region_id:     区域 ID
        shop_item_ids: 商店商品清单（可选，默认 config.PLAY_SHOP_ITEM_IDS）
        nodes:         要抓的节点（可选，默认 DEFAULT_SNAPSHOT_NODES）
        timeout_ms:    超时毫秒
        client:        可复用的 MaimaiNetClient
        strict:        True = 任一接口失败就抛异常；
                       False（默认）= 记录失败并跳过，保证链路不崩

    返回：
        {
          "userData":       <GetUserDataApi 原始响应>,
          "userExtend":     <GetUserExtendApi 原始响应>,
          "userMapList":    {"userMapList": [...]},       # 已摊平/翻页
          "userMusicDetailList": {"userMusicList": [...]},# 已摊平成单条
          ...
          "_failed": [ {"node":.., "error":..}, ... ]
        }
        这个结构可以直接传给 UpsertUserAll(snapshot=...)，也可 json 落盘。
    """
    wanted = list(nodes) if nodes else list(DEFAULT_SNAPSHOT_NODES)
    log_info("抓取结算快照：user_id=" + str(user_id) + " 节点数=" + str(len(wanted)))

    raw: Dict[str, Any] = {}
    failed: List[Dict[str, str]] = []

    def _call(node: str, fn, *args, **kwargs):
        """逐节点容错：单个接口失败不影响整体。"""
        try:
            raw[node] = fn(*args, **kwargs)
            return True
        except Exception as exc:
            failed.append({"node": node, "error": type(exc).__name__ + ": " + str(exc)})
            log_warning("节点 " + node + " 抓取失败：" + type(exc).__name__ + ": " + str(exc))
            if strict:
                raise
            return False

    common = dict(client_id=client_id, timeout_ms=timeout_ms, client=client)

    # ---- 逐个抓取（顺序与 DEFAULT_SNAPSHOT_NODES 一致）----
    if "userData" in wanted:
        _call("userData", UserData, user_id, **common)

    if "userExtend" in wanted:
        _call("userExtend", UserExtend, user_id, **common)

    if "userOption" in wanted:
        _call("userOption", UserOption, user_id, **common)

    if "userCharacterList" in wanted:
        _call("userCharacterList", UserCharacter, user_id, **common)

    if "userMapList" in wanted:
        _call("userMapList", UserMapAll, user_id, **common)

    if "userLoginBonusList" in wanted:
        _call("userLoginBonusList", UserLoginBonusAll, user_id, **common)

    if "userRatingList" in wanted:
        _call("userRatingList", UserRating, user_id, **common)

    if "userItemList" in wanted:
        _call("userItemList", UserItemAll, user_id, **common)

    if "userMusicDetailList" in wanted:
        _call("userMusicDetailList", UserMusicDetailAll, user_id, **common)

    if "userCourseList" in wanted:
        _call("userCourseList", UserCourseAll, user_id, **common)

    if "userChargeList" in wanted:
        _call("userChargeList", UserCharge, user_id, **common)

    if "userFavoriteList" in wanted:
        # ★ 该接口额外带一份**原始** itemKind 副本（_raw），
        #   供 UpsertUserAll 生成 isNewFavoriteList 位串用
        #   —— 真机 itemKind=10 回的是 null，正是它产生了位串里的 "1"。
        if _call("userFavoriteList", UserFavoriteAll, user_id, **common):
            fav_raw = (raw.get("userFavoriteList") or {}).get("_raw")
            if fav_raw:
                raw["_raw_favorite"] = fav_raw

    if "userActivityList" in wanted:
        _call("userActivityList", UserActivity, user_id, **common)

    if "userMissionDataList" in wanted:
        # ★ 该接口的响应里同时带着 userWeeklyData（周界），
        #   它是 Upsert 侧 userWeeklyData 节点的来源，会被自动提取。
        _call("userMissionDataList", UserMissionData, user_id, **common)

    if "userIntimateList" in wanted:
        _call("userIntimateList", UserIntimate, user_id, **common)

    if "userShopItemStockList" in wanted:
        ids = list(shop_item_ids) if shop_item_ids is not None else list(PLAY_SHOP_ITEM_IDS)
        if ids:
            _call("userShopItemStockList", UserShopStock, user_id,
                  shop_item_id_list=ids, **common)
        else:
            log_debug("未配置 PLAY_SHOP_ITEM_IDS，跳过商店库存节点")

    if "userFavoritemusicList" in wanted:
        _call("userFavoritemusicList", UserFavoriteItemAll, user_id, 1, **common)

    if "userKaleidxScopeList" in wanted:
        _call("userKaleidxScopeList", UserKaleidxScope, user_id, **common)

    if "userGhost" in wanted:
        _call("userGhost", UserGhost, user_id, **common)

    raw["_failed"] = failed
    log_info("结算快照抓取完成：成功 " + str(len(raw) - 1) + " 个节点，失败 "
             + str(len(failed)) + " 个")
    return raw


def save_snapshot(snapshot: Dict[str, Any], path: str) -> str:
    """把快照写到 JSON 文件（便于离线复用 / 人工核对）。"""
    with open(path, "w", encoding="utf-8") as fp:
        json.dump(snapshot, fp, ensure_ascii=False, indent=2, default=str)
    log_info("快照已保存：" + path)
    return path


def load_snapshot(path: str) -> Dict[str, Any]:
    """从 JSON 文件读回快照。"""
    with open(path, "r", encoding="utf-8") as fp:
        data = json.load(fp)
    log_info("快照已加载：" + path)
    return data


def summarize_snapshot(snapshot: Dict[str, Any]) -> Dict[str, int]:
    """
    统计快照各节点的条数，便于快速核对（返回 {节点名: 条数}）。

    条数含义：
        userData 这类单元素数组 -> 1
        userMusicDetailList 这类列表 -> 摊平后的条数
    """
    out: Dict[str, int] = {}
    for node, value in (snapshot or {}).items():
        if node.startswith("_"):
            continue
        if isinstance(value, list):
            out[node] = len(value)
        elif isinstance(value, dict):
            # 从响应里挑出列表
            found = False
            for key, sub in value.items():
                if isinstance(sub, list):
                    out[node] = len(sub)
                    found = True
                    break
            if not found:
                out[node] = 1
        else:
            out[node] = 0
    return out


# =============================================================================
# 单独运行（调试）
# =============================================================================

if __name__ == "__main__":
    import sys

    from .log_setup import log_error, setup_logging, teardown_logging

    def _main() -> int:
        setup_logging("upsert_snapshot_test")
        try:
            if len(sys.argv) < 2:
                print("用法: python -m api.upsert_snapshot <user_id> [--token TOKEN] [--save 路径]")
                return 1

            user_id = int(sys.argv[1])
            token = ""
            save_path = ""
            args = sys.argv[2:]
            i = 0
            while i < len(args):
                if args[i] == "--token" and i + 1 < len(args):
                    token = args[i + 1]
                    i += 2
                elif args[i] == "--save" and i + 1 < len(args):
                    save_path = args[i + 1]
                    i += 2
                else:
                    i += 1

            snap = fetch_upsert_snapshot(user_id, token=token)
            summary = summarize_snapshot(snap)
            print("")
            for node, count in summary.items():
                print("   " + node.ljust(28) + "= " + str(count) + " 条")
            if save_path:
                save_snapshot(snap, save_path)
            return 0
        except Exception as exc:
            log_error("抓取快照失败：" + str(exc))
            return 1
        finally:
            teardown_logging()

    raise SystemExit(_main())