from datetime import datetime, timezone
from typing import Any, Optional

from sqlalchemy import BigInteger, DateTime, ForeignKey, JSON, String, Text
from sqlalchemy.orm import Mapped, mapped_column

from app.db import Base


class GeneratedContract(Base):
    __tablename__ = "generated_contracts"

    id: Mapped[str] = mapped_column(String(64), primary_key=True)
    tenant_id: Mapped[str] = mapped_column(String(64), nullable=False, index=True)
    template_id: Mapped[str] = mapped_column(String(64), nullable=False, index=True)
    template_name: Mapped[str] = mapped_column(String(100), nullable=False)
    contract_type: Mapped[Optional[str]] = mapped_column(String(50), nullable=True)
    document_type_code: Mapped[Optional[str]] = mapped_column(String(64), nullable=True, index=True)
    generated_by: Mapped[str] = mapped_column(String(64), nullable=False, index=True)
    generated_by_name: Mapped[Optional[str]] = mapped_column(String(100), nullable=True)
    export_format: Mapped[str] = mapped_column(String(16), nullable=False)
    file_id: Mapped[Optional[str]] = mapped_column(String(64), ForeignKey("files.id"), nullable=True, index=True)
    file_name: Mapped[str] = mapped_column(String(255), nullable=False)
    mime_type: Mapped[str] = mapped_column(String(100), nullable=False)
    file_size: Mapped[int] = mapped_column(BigInteger, nullable=False)
    generated_content: Mapped[str] = mapped_column(Text, nullable=False)
    variable_values_json: Mapped[Optional[dict[str, str]]] = mapped_column(JSON, nullable=True)
    elements_snapshot_json: Mapped[Optional[dict[str, Any]]] = mapped_column(JSON, nullable=True)
    element_schema_id: Mapped[Optional[str]] = mapped_column(String(64), nullable=True, index=True)
    element_schema_version_label: Mapped[Optional[str]] = mapped_column(String(32), nullable=True)
    export_options_json: Mapped[Optional[dict[str, Any]]] = mapped_column(JSON, nullable=True)
    file_content_base64: Mapped[str] = mapped_column(Text, nullable=False)
    status: Mapped[str] = mapped_column(String(32), default="generated")
    lifecycle_status: Mapped[str] = mapped_column(String(32), default="generated")
    counterparty_name: Mapped[Optional[str]] = mapped_column(String(120), nullable=True)
    owner_name: Mapped[Optional[str]] = mapped_column(String(100), nullable=True)
    effective_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
    expires_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
    lifecycle_note: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
    lifecycle_updated_by: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
    lifecycle_updated_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))