from datetime import datetime, date
from typing import Optional
from sqlalchemy import String, Text, Float, Date, DateTime, ForeignKey, Enum, func
from sqlalchemy.orm import Mapped, mapped_column, relationship
import enum

from app.database import Base


class EntryType(str, enum.Enum):
    add = "add"
    remove = "remove"


class Entry(Base):
    __tablename__ = "entries"

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    user_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False, index=True)
    entry_type: Mapped[EntryType] = mapped_column(Enum(EntryType), nullable=False)
    date: Mapped[date] = mapped_column(Date, nullable=False)
    bourbon_name: Mapped[Optional[str]] = mapped_column(String(255))
    proof: Mapped[Optional[float]] = mapped_column(Float)
    amount_shots: Mapped[float] = mapped_column(Float, nullable=False, default=1.0)
    notes: Mapped[Optional[str]] = mapped_column(Text)
    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())

    user: Mapped["User"] = relationship("User", back_populates="entries")