"""绑定 — auto rigging with UniRig.

    tripocli rig -p <project_id> --type biped --spec mixamo --wait

A riggable check runs first (``pre_rig_check``); the model must be a humanoid in
T/A pose for good results.  Verified body of ``rigging_model``:
``{project_id, model_version, rig_type, spec}``.

骨骼预设（界面 5 个选项 → API 的 spec 值，取自 studio 前端数据表，并用实测请求核对）：
``Mixamo → mixamo``、``ActorCore → actorcore``、``UE5 Mannequin → unreal``、
``Unity Humanoid Compatible → unity``、``VRM 1.0 Humanoid → vrm``。
"""

from . import _common

NAME = "rig"
HELP = "自动绑定骨骼（人形/其他 + 5 种骨骼预设）"

RIG_TYPES = ("biped", "others")

# 界面预设名 → API spec 值
SPEC_ALIASES = {
    "mixamo": "mixamo",
    "actorcore": "actorcore",
    "actor_core": "actorcore",
    "unreal": "unreal",
    "ue5": "unreal",
    "ue5-mannequin": "unreal",
    "ue5_mannequin": "unreal",
    "unity": "unity",
    "unity-humanoid": "unity",
    "vrm": "vrm",
    "vrm1": "vrm",
}
DEFAULT_SPEC = "mixamo"


def resolve_spec(name):
    if not name:
        return DEFAULT_SPEC
    return SPEC_ALIASES.get(name.strip().lower(), name.strip())


def add_arguments(parser):
    _common.add_op_options(parser, "绑定完成后保存到该路径（可选）")
    parser.add_argument("--type", default="biped", choices=RIG_TYPES, help="绑定类型：biped=人形，others=其他")
    parser.add_argument(
        "--spec", default=DEFAULT_SPEC,
        help="骨骼预设: mixamo(默认) / actorcore / unreal(UE5 Mannequin) / unity / vrm",
    )
    parser.add_argument("--check-only", action="store_true", help="只做绑定前检查，不执行绑定")


def run(ctx):
    args = ctx.args
    model_version = _common.project_version(ctx.client, args.project, args.model)
    spec = resolve_spec(args.spec)

    check_body = {"project_id": args.project}
    if model_version:
        check_body["model_version"] = model_version
    check = ctx.client.get_data("POST", "/v2/studio/operation/pre_rig_check", body=check_body) or {}

    if args.check_only:
        ctx.out.data(
            check,
            human=lambda: (
                ctx.out.line(f"可绑定: {check.get('riggable')}"),
                ctx.out.line(f"绑定类型: {check.get('rig_type')}"),
            ),
        )
        return check

    ctx.out.line(f"🔎 绑定前检查: riggable={check.get('riggable')} type={check.get('rig_type')}")

    body = {"project_id": args.project, "rig_type": args.type, "spec": spec}
    if model_version:
        body["model_version"] = model_version
    data = ctx.client.get_data("POST", "/v2/studio/operation/rigging_model", body=body)
    return _common.run_operator(ctx, data, output=args.output, wait=args.wait, label="绑定")