from sqlalchemy import ARRAY, ForeignKey, Integer, Text
from sqlalchemy.orm import Mapped, mapped_column

from fabledassistant.models import Base
from fabledassistant.models.base import CreatedAtMixin


class NoteVersion(Base, CreatedAtMixin):
    __tablename__ = "note_versions"

    id: Mapped[int] = mapped_column(primary_key=True)
    note_id: Mapped[int] = mapped_column(Integer, ForeignKey("notes.id", ondelete="CASCADE"))
    user_id: Mapped[int] = mapped_column(Integer, ForeignKey("users.id", ondelete="CASCADE"))
    body: Mapped[str] = mapped_column(Text)
    title: Mapped[str] = mapped_column(Text, default="")
    tags: Mapped[list[str]] = mapped_column(ARRAY(Text), default=list)

    def to_dict(self, include_body: bool = True) -> dict:
        d: dict = {
            "id": self.id,
            "note_id": self.note_id,
            "user_id": self.user_id,
            "title": self.title,
            "tags": self.tags or [],
            "created_at": self.created_at.isoformat(),
        }
        if include_body:
            d["body"] = self.body
        return d