byceps/byceps

"""
byceps.services.board.dbmodels.posting
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

:Copyright: 2014-2024 Jochen Kupperschmidt
:License: Revised BSD (see `LICENSE` file for details)
"""

from datetime import datetime
from uuid import UUID

from sqlalchemy.orm import Mapped, mapped_column, relationship

from byceps.database import db
from byceps.services.board.models import PostingID, TopicID
from byceps.services.user.dbmodels.user import DbUser
from byceps.services.user.models.user import UserID
from byceps.util.instances import ReprBuilder

from .topic import DbTopic


class DbPosting(db.Model):
    """A posting."""

    __tablename__ = 'board_postings'

    id: Mapped[PostingID] = mapped_column(db.Uuid, primary_key=True)
    topic_id: Mapped[TopicID] = mapped_column(
        db.Uuid, db.ForeignKey('board_topics.id'), index=True
    )
    topic: Mapped[DbTopic] = relationship(DbTopic, backref='postings')
    created_at: Mapped[datetime] = mapped_column(default=datetime.utcnow)
    creator_id: Mapped[UserID] = mapped_column(
        db.Uuid, db.ForeignKey('users.id')
    )
    body: Mapped[str] = mapped_column(db.UnicodeText)
    last_edited_at: Mapped[datetime | None]
    last_edited_by_id: Mapped[UserID | None] = mapped_column(
        db.Uuid, db.ForeignKey('users.id')
    )
    last_edited_by: Mapped[DbUser | None] = relationship(
        DbUser, foreign_keys=[last_edited_by_id]
    )
    edit_count: Mapped[int] = mapped_column(default=0)
    hidden: Mapped[bool] = mapped_column(default=False)
    hidden_at: Mapped[datetime | None]
    hidden_by_id: Mapped[UserID | None] = mapped_column(
        db.Uuid, db.ForeignKey('users.id')
    )
    hidden_by: Mapped[DbUser | None] = relationship(
        DbUser, foreign_keys=[hidden_by_id]
    )

    def __init__(
        self,
        posting_id: PostingID,
        topic_id: TopicID,
        creator_id: UserID,
        body: str,
    ) -> None:
        self.id = posting_id
        self.topic_id = topic_id
        self.creator_id = creator_id
        self.body = body

    def is_initial_topic_posting(self) -> bool:
        return self.id == self.topic.initial_posting.id

    def __eq__(self, other) -> bool:
        return self.id == other.id

    def __repr__(self) -> str:
        builder = (
            ReprBuilder(self)
            .add_with_lookup('id')
            .add('topic', self.topic.title)
        )

        if self.hidden_by:
            builder.add_custom(f'hidden by {self.hidden_by.screen_name}')

        return builder.build()


class DbInitialTopicPostingAssociation(db.Model):
    __tablename__ = 'board_initial_topic_postings'

    topic_id: Mapped[TopicID] = mapped_column(
        db.Uuid, db.ForeignKey('board_topics.id'), primary_key=True
    )
    topic: Mapped[DbTopic] = relationship(
        DbTopic,
        backref=db.backref('initial_topic_posting_association', uselist=False),
    )
    posting_id: Mapped[PostingID] = mapped_column(
        db.Uuid, db.ForeignKey('board_postings.id'), unique=True
    )
    posting: Mapped[DbPosting] = relationship(DbPosting)

    def __init__(self, topic_id: TopicID, posting_id: PostingID) -> None:
        self.topic_id = topic_id
        self.posting_id = posting_id


class DbPostingReaction(db.Model):
    """A user reaction to a posting."""

    __tablename__ = 'board_posting_reactions'
    __table_args__ = (db.UniqueConstraint('user_id', 'posting_id', 'kind'),)

    id: Mapped[UUID] = mapped_column(db.Uuid, primary_key=True)
    created_at: Mapped[datetime]
    posting_id: Mapped[PostingID] = mapped_column(
        db.Uuid, db.ForeignKey('board_postings.id')
    )
    posting: Mapped[DbPosting] = relationship(
        DbPosting, backref=db.backref('reactions')
    )
    user_id: Mapped[UserID] = mapped_column(db.Uuid, db.ForeignKey('users.id'))
    user: Mapped[DbUser] = relationship(DbUser)
    kind: Mapped[str] = mapped_column(db.UnicodeText)

    def __init__(
        self,
        reaction_id: UUID,
        created_at: datetime,
        posting_id: PostingID,
        user_id: UserID,
        kind: str,
    ) -> None:
        self.id = reaction_id
        self.created_at = created_at
        self.posting_id = posting_id
        self.user_id = user_id
        self.kind = kind