from sqlalchemy import BigInteger, ForeignKey, String
from sqlalchemy.orm import Mapped, mapped_column, relationship

from ..db import Base, db
from .stop import _Stop


class UserLastStopSearchStopAssociations(Base):
    id = mapped_column(BigInteger, primary_key=True)
    user_mxid = mapped_column(
        String, ForeignKey("user_last_stop_search_results.user_mxid")
    )
    stop_id = mapped_column(BigInteger, ForeignKey("_stops.id"))

    __tablename__ = "user_last_stop_search_stop_associations"


class UserLastStopSearchResults(Base):
    db = db

    user_mxid = mapped_column(String, primary_key=True)
    request_content = mapped_column(String, nullable=False)
    stops: Mapped[_Stop] = relationship(
        _Stop, secondary="user_last_stop_search_stop_associations"
    )

    __tablename__ = "user_last_stop_search_results"