import enum
from datetime import datetime
from typing import Optional

from sqlalchemy import DECIMAL, Boolean, Column, DateTime
from sqlalchemy import Enum
from sqlalchemy import Enum as SAEnum
from sqlalchemy import ForeignKey, Integer, String, Text, UniqueConstraint
from sqlalchemy.orm import relationship

from src.apps.match.enums import (
    RequestType,
    Separator,
    Source,
    SourceEnum,
    Status,
    WineDbStatus,
)
from src.core.models.base import Base  # Assuming you're using a declarative base setup


class Retailer(Base):
    __tablename__ = "retailers"

    id = Column(Integer, primary_key=True, index=True)
    literal = Column(String, nullable=True)
    code = Column(String, unique=True, nullable=False)
    name = Column(String, nullable=True)
    email = Column(String, nullable=True)
    web_url = Column(Text, nullable=True)
    address = Column(Text, nullable=True)
    address_extended = Column(Text, nullable=True)
    postal_code = Column(String, nullable=True)
    city = Column(String, nullable=True)
    state = Column(String, nullable=True)
    phone = Column(String, nullable=True)
    ship_to = Column(String, nullable=True)

    country_id = Column(Integer, ForeignKey("countries.id"), nullable=True)
    active = Column(Boolean, default=True)

    date_created = Column(DateTime, default=datetime.utcnow)
    last_updated = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
    deleted_at = Column(DateTime, nullable=True)

    # Relationships
    country = relationship("Country")
    retailer_wines = relationship("RetailerWine", back_populates="retailer")


class RetailerWine(Base):
    __tablename__ = "retailer_wines"
    __table_args__ = (
        UniqueConstraint("retailer_id", "wine_id", "bottle_size_id", "sku", name="uix_retailer_wine_size_sku"),
    )

    id = Column(Integer, primary_key=True, index=True)
    literal = Column(String, nullable=True)
    name = Column(String, index=True, nullable=True)
    quantity = Column(Integer, nullable=True)
    tax_status = Column(String, nullable=True)
    description = Column(Text, nullable=True)
    sku = Column(String, nullable=True)
    url = Column(Text, nullable=True)
    currency = Column(String, nullable=True)
    price = Column(DECIMAL, nullable=True)
    usd_price = Column(DECIMAL, nullable=True)

    bottle_size_id = Column(Integer, ForeignKey("bottle_sizes.id"))
    retailer_id = Column(Integer, ForeignKey("retailers.id"))
    wine_id = Column(Integer, ForeignKey("wines.id"), nullable=True)

    available = Column(Boolean, default=True)
    last_available = Column(DateTime, nullable=True)

    date_created = Column(DateTime, default=datetime.utcnow)
    last_updated = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
    deleted_at = Column(DateTime, nullable=True)

    # Relationships
    bottle_size = relationship("BottleSize", back_populates="retailer_wines")
    retailer = relationship("Retailer", back_populates="retailer_wines")
    wine = relationship("Wine", back_populates="retailer_wines")