from __future__ import annotations
from datetime import datetime
from typing import Optional
from sqlalchemy import DateTime, Enum, ForeignKey, String, UniqueConstraint, func, Text, JSON, Boolean
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.sql.sqltypes import Enum as EnumType  # Явный импорт Enum для типизации
from app.db.session import Base
from app.models.post import PostType
from enum import Enum as PyEnum

class TemplateVisibility(str, PyEnum):
    private = "private"
    org = "org"
    public = "public"

class Template(Base):
    __tablename__ = "templates"
    __table_args__ = (
        UniqueConstraint("owner_id", "name", name="uq_template_owner_name"),
    )

    id: Mapped[int] = mapped_column(primary_key=True)
    owner_id: Mapped[int] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"), index=True)
    name: Mapped[str] = mapped_column(String(64))
    title: Mapped[Optional[str]] = mapped_column(String(128))
    type: Mapped[PostType] = mapped_column(EnumType(PostType), default=PostType.text)
    content: Mapped[str] = mapped_column(Text)
    keyboard_tpl: Mapped[Optional[list[dict]]] = mapped_column(JSON, nullable=True)
    parse_mode: Mapped[Optional[str]] = mapped_column(String(16), default="HTML")
    visibility: Mapped[TemplateVisibility] = mapped_column(EnumType(TemplateVisibility), default=TemplateVisibility.private)
    is_archived: Mapped[bool] = mapped_column(Boolean, default=False)
    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
    updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())