"""Domain-owned AI capabilities; caller owns authorization and transactions."""

from fastapi import HTTPException
from pydantic import Field

from ..schemas import Contract, JobInput
from .capabilities import Capability, EmptyArgs


class JobArgs(Contract):
    job_id: str = Field(min_length=1, max_length=100)


async def list_jobs(ctx, args):
    return {"items": (await ctx.business.list_jobs())[:20]}


async def preview_create(ctx, args):
    return {"operation": args.model_dump(mode="json")}


async def preview_job(ctx, args):
    return {"job": await ctx.business.get_job_status(args.job_id)}


async def check_job(ctx, args, preview):
    current = await ctx.business.get_job_status(args.job_id)
    if current["updated_at"] != preview["job"]["updated_at"] or current["status"] != preview["job"]["status"]:
        raise HTTPException(409, "任务状态已变化，请重新确认操作")


async def cancel_job(ctx, args, preview):
    await check_job(ctx, args, preview)
    return await ctx.business.cancel_job(args.job_id)


async def retry_job(ctx, args, preview):
    await check_job(ctx, args, preview)
    return await ctx.business.retry_job(args.job_id)


async def wake_sync(runner, result):
    """Called only after the job and audit commit."""
    runner.wake.set()


async def cancel_sync(runner, result):
    """The durable cancel decision precedes interruption of the in-process task."""
    await runner.cancel(result["job_id"])


INSTRUCTIONS = "任务创建后返回任务信息并结束本轮，不要循环等待任务完成。不推测未执行操作已经成功。"


CAPABILITIES = (
    Capability(
        name="list_jobs",
        schema=EmptyArgs,
        description="查询最近的同步任务，不要循环轮询等待。",
        label="查询任务",
        renderer="jobs",
        effect="query",
        handler=list_jobs,
    ),
    Capability(
        name="get_job_status",
        schema=JobArgs,
        description="查询指定任务的状态、目标和错误，不要循环等待任务完成。",
        label="查看任务状态",
        renderer="jobs",
        effect="query",
        handler=lambda ctx, args: ctx.business.get_job_status(args.job_id),
    ),
    Capability(
        name="create_sync_job",
        schema=JobInput,
        description="提出同步或本地自相关任务，等待确认。full_sync 仅同步已提交；待提交必须用 daily_sync 并指定 submission、date_from/date_to（UTC），待提交按创建日、已提交按提交日逐天同步。alpha_refresh/pnl_refresh/self_correlation 使用固定 alpha_ids；自相关缺失 PnL 时自动补取，不触发平台检查。创建后立即返回任务 ID。",
        label="创建同步任务",
        renderer="jobs",
        effect="confirm",
        preview=preview_create,
        execute=lambda ctx, args, preview: ctx.business.create_sync_job(args),
        after_commit=wake_sync,
        refresh=("jobs",),
    ),
    Capability(
        name="cancel_job",
        schema=JobArgs,
        description="提出取消指定同步任务，等待用户确认。",
        label="取消任务",
        renderer="jobs",
        effect="confirm",
        preview=preview_job,
        execute=cancel_job,
        after_commit=cancel_sync,
        refresh=("jobs",),
    ),
    Capability(
        name="retry_job",
        schema=JobArgs,
        description="提出重试指定失败或暂停的同步任务，等待用户确认。",
        label="重试任务",
        renderer="jobs",
        effect="confirm",
        preview=preview_job,
        execute=retry_job,
        after_commit=wake_sync,
        refresh=("jobs",),
    ),
)