from datetime import datetime
from typing import Literal

from beanie import Document
from bson import ObjectId
from pydantic import Field, field_validator
from pydantic import BaseModel
from pymongo import IndexModel


class Deployment(Document):
    deployment_id: str

    deployment_stage: str
    deployment_status: Literal["started", "failed", "succeeded", "aborted"]

    deployment_target_env: Literal["alpha", "prod"]
    deployment_ttl_hours: int = 2

    deployment_project_id: str
    deployment_project_name: str
    deployment_product_id: str
    deployment_product_name: str
    deployment_git_url: str
    deployment_git_sha256: str
    deployment_reason: str

    deployed_by: str
    created_at: datetime = datetime.now()
    updated_at: datetime = datetime.now()

    class Settings:
        name = "deployment"
        indexes = [
            IndexModel([("deployment_product_id", 1), ("created_at", 1)]),
            IndexModel([("deployment_id", 1), ("deployment_status", 1)]),
            IndexModel([("deployment_id", 1), ("deployment_stage", 1)], unique=True)
        ]

class InitDeploymentRequest(BaseModel):
    product_id: str
    sha256: str
    target_env: str
    user_id: str
    reason: str = "not provided"
    ttl_hours: int = 3

class CheckDeploymentStatusRequest(BaseModel):
    product_id: str
    target_env: str
    user_id: str