"""任务进度 — poll or block on an operator.

    tripocli task status <operator_id>
    tripocli task wait <operator_id> -p <project_id> -o out.glb
"""

import os

from . import _common

TIMEOUT_DEFAULT = 900


def _status(ctx):
    items = ctx.client.progress([ctx.args.operator_id])
    if not items:
        raise ValueError(f"未找到任务 {ctx.args.operator_id}")
    task = items[0]
    ctx.out.data(
        task,
        human=lambda: (
            ctx.out.line(f"状态   : {task.get('status')}"),
            ctx.out.line(f"进度   : {task.get('progress')}%"),
            ctx.out.line(f"剩余   : {task.get('left_time')}s"),
        ),
    )
    return task


def _wait(ctx):
    args = ctx.args
    task = _common.wait_and_report(ctx, args.operator_id, timeout=args.timeout)

    output_path = None
    model_url = None
    if args.project:
        model_url, _detail = _common.fetch_model_url(ctx, args.project)
        if args.output and model_url:
            ctx.client.download(model_url, args.output)
            output_path = os.path.abspath(args.output)

    payload = {"task": task, "model_url": model_url, "output_file": output_path}
    ctx.out.data(
        payload,
        human=lambda: (
            ctx.out.line(f"✅ 任务状态: {task.get('status')}"),
            ctx.out.line(f"模型地址: {model_url}") if model_url else ctx.out.line(""),
            ctx.out.line(f"已保存: {output_path}") if output_path else ctx.out.line(""),
        ),
    )
    return payload


def register(subparsers):
    parser = subparsers.add_parser("task", help="任务进度查询与等待")
    sub = parser.add_subparsers(dest="task_action", required=True)

    status = sub.add_parser("status", help="查询任务状态")
    status.add_argument("operator_id", help="operator_id / task_id")
    status.set_defaults(func=_status)

    wait = sub.add_parser("wait", help="阻塞等待任务完成（可顺带下载）")
    wait.add_argument("operator_id", help="operator_id / task_id")
    wait.add_argument("-p", "--project", help="项目 ID，用于取最终模型地址")
    wait.add_argument("-o", "--output", help="下载保存路径")
    wait.add_argument("--timeout", type=int, default=TIMEOUT_DEFAULT, help=f"超时秒数（默认 {TIMEOUT_DEFAULT}）")
    wait.set_defaults(func=_wait)