from __future__ import annotations

from typing import ClassVar

from django.db import models
from django.utils.translation import gettext_lazy as _


class SolutionTaskStatus(models.TextChoices):
    PENDING = "pending", _("待执行")
    RESEARCHING = "researching", _("检索中")
    GENERATING = "generating", _("生成中")
    COMPLETED = "completed", _("已完成")
    PARTIAL = "partial", _("部分完成")
    FAILED = "failed", _("失败")


class SolutionTask(models.Model):
    case_summary = models.TextField(verbose_name=_("案情简述"))
    keyword = models.CharField(max_length=255, blank=True, verbose_name=_("检索关键词"))
    credential = models.ForeignKey(
        "organization.AccountCredential",
        on_delete=models.PROTECT,
        related_name="solution_tasks",
        verbose_name=_("站点账号"),
    )
    research_task = models.ForeignKey(
        "legal_research.LegalResearchTask",
        on_delete=models.SET_NULL,
        null=True,
        blank=True,
        related_name="solution_tasks",
        verbose_name=_("案例检索任务"),
    )
    status = models.CharField(
        max_length=16,
        choices=SolutionTaskStatus.choices,
        default=SolutionTaskStatus.PENDING,
        verbose_name=_("状态"),
    )
    progress = models.PositiveIntegerField(default=0, verbose_name=_("进度"))
    message = models.CharField(max_length=255, blank=True, verbose_name=_("状态说明"))
    error = models.TextField(blank=True, verbose_name=_("错误信息"))
    llm_model = models.CharField(max_length=128, blank=True, verbose_name=_("LLM模型"))
    html_content = models.TextField(blank=True, verbose_name=_("HTML方案"))
    pdf_file = models.FileField(
        upload_to="legal_solution/pdf/",
        max_length=255,
        null=True,
        blank=True,
        verbose_name=_("PDF文件"),
    )
    created_by = models.ForeignKey(
        "organization.Lawyer",
        on_delete=models.SET_NULL,
        null=True,
        blank=True,
        related_name="solution_tasks",
        verbose_name=_("创建人"),
    )
    q_task_id = models.CharField(max_length=64, blank=True, verbose_name=_("队列任务ID"))
    started_at = models.DateTimeField(null=True, blank=True, verbose_name=_("开始时间"))
    finished_at = models.DateTimeField(null=True, blank=True, verbose_name=_("完成时间"))
    created_at = models.DateTimeField(auto_now_add=True, verbose_name=_("创建时间"))
    updated_at = models.DateTimeField(auto_now=True, verbose_name=_("更新时间"))

    class Meta:
        verbose_name = _("法律服务方案")
        verbose_name_plural = _("法律服务方案")
        ordering: ClassVar = ["-created_at"]

    def __str__(self) -> str:
        return f"{self.id} | {self.case_summary[:30]} | {self.status}"