import enum
from datetime import date, datetime
from decimal import Decimal

from sqlalchemy import Date, DateTime, Enum, ForeignKey, Numeric, String, Text, func
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.db.base import Base


class ServiceType(str, enum.Enum):
    maintenance = "maintenance"
    repair = "repair"
    fluid = "fluid"
    tire = "tire"
    inspection = "inspection"
    insurance = "insurance"
    tax = "tax"
    other = "other"


class FuelEntry(Base):
    __tablename__ = "fuel_entries"

    id: Mapped[int] = mapped_column(primary_key=True)
    car_id: Mapped[int] = mapped_column(ForeignKey("cars.id", ondelete="CASCADE"), index=True)
    entry_date: Mapped[date] = mapped_column(Date, index=True)
    odometer: Mapped[int]
    liters: Mapped[Decimal] = mapped_column(Numeric(8, 3))
    price_per_liter: Mapped[Decimal] = mapped_column(Numeric(10, 2))
    total_cost: Mapped[Decimal] = mapped_column(Numeric(12, 2))
    station: Mapped[str | None] = mapped_column(String(160))
    fuel_brand: Mapped[str | None] = mapped_column(String(80))
    is_full_tank: Mapped[bool] = mapped_column(default=True)
    notes: Mapped[str | None] = mapped_column(Text)
    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())

    car = relationship("Car", back_populates="fuel_entries")


class ServiceEntry(Base):
    __tablename__ = "service_entries"

    id: Mapped[int] = mapped_column(primary_key=True)
    car_id: Mapped[int] = mapped_column(ForeignKey("cars.id", ondelete="CASCADE"), index=True)
    entry_date: Mapped[date] = mapped_column(Date, index=True)
    odometer: Mapped[int | None]
    service_type: Mapped[ServiceType] = mapped_column(Enum(ServiceType), index=True)
    title: Mapped[str] = mapped_column(String(180))
    category: Mapped[str | None] = mapped_column(String(80))
    vendor: Mapped[str | None] = mapped_column(String(160))
    total_cost: Mapped[Decimal] = mapped_column(Numeric(12, 2))
    next_due_date: Mapped[date | None] = mapped_column(Date)
    next_due_odometer: Mapped[int | None]
    notes: Mapped[str | None] = mapped_column(Text)
    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())

    car = relationship("Car", back_populates="service_entries")