from .chain import Chain
from .base import BaseConfig, BaseMongoModel

from enum import Enum
from datetime import datetime
from pydantic import BaseModel, Field
from typing import Literal, Union, Annotated, Optional, List


class ProgressActionStatusEnum(str, Enum):
    PENDING = "pending"
    DONE = "done"
    FAILED = "failed"


class BaseProgressAction(BaseModel):
    action_type: Literal["wait", "comment"]

    status: Annotated[
        ProgressActionStatusEnum, Field(default=ProgressActionStatusEnum.PENDING)
    ]

    started_at: Annotated[Optional[datetime], Field(default=None)]
    finished_at: Annotated[Optional[datetime], Field(default=None)]
    error_text: Annotated[Optional[str], Field(default=None)]


class WaitProgressAction(BaseProgressAction):
    action_type: Literal["wait"]
    wait_for: int

    class Config(BaseConfig):
        pass


class CommentProgressAction(BaseProgressAction):
    action_type: Literal["comment"]
    text: Annotated[Optional[str], Field(default=None)]
    file_urls: Annotated[Optional[List[str]], Field(default=[])]

    class Config(BaseConfig):
        pass


Action = Annotated[
    Union[WaitProgressAction, CommentProgressAction],
    Field(description="action_type"),
]


class ProgressChain(BaseMongoModel):
    task_id: int
    recipients: Annotated[Optional[List[int]], Field(default=[])]
    name: Annotated[Optional[str], Field(default=None)]
    actions: Annotated[Optional[List[Action]], Field(default=[])]
    created_at: Annotated[datetime, Field(default_factory=datetime.utcnow)]

    @classmethod
    def create_from_chain(
        cls, chain: Chain, task_id: int, recipients: Optional[List[int]] = []
    ):
        return ProgressChain(
            task_id=task_id,
            recipients=recipients,
            name=chain.name,
            actions=map(Chain.model_dump, chain.actions),
        )

    @property
    def is_finished(self):
        return all(
            action.status is ProgressActionStatusEnum.DONE for action in self.actions
        )

    @property
    def has_failed(self):
        return any(
            action.status is ProgressActionStatusEnum.FAILED for action in self.actions
        )