from datetime import datetime

from pydantic import BaseModel, ConfigDict, Field

from models.service import ServiceEndpointDocument


class ProxyDocument(BaseModel):
    model_config = ConfigDict(extra="forbid")

    proxy_port: int = Field(..., description="Port exposed by the proxy")
    name: str = Field(..., description="Human readable proxy name")
    code: str = Field(..., description="Proxy source or identifier")
    service: ServiceEndpointDocument = Field(..., description="Associated upstream service")
    created_at: datetime | None = Field(default=None, description="Creation timestamp in UTC")
    updated_at: datetime | None = Field(default=None, description="Last update timestamp in UTC")


class ProxyCreateInput(BaseModel):
    model_config = ConfigDict(extra="forbid")

    proxy_port: int = Field(..., description="Port exposed by the proxy")
    name: str = Field(..., description="Human readable proxy name")
    code: str = Field(..., description="Proxy source or identifier")
    service: ServiceEndpointDocument = Field(..., description="Associated upstream service")


class ProxyUpdateInput(BaseModel):
    model_config = ConfigDict(extra="forbid")

    name: str | None = Field(default=None, description="Human readable proxy name")
    code: str | None = Field(default=None, description="Proxy source or identifier")
    service: ServiceEndpointDocument | None = Field(
        default=None,
        description="Associated upstream service",
    )