"""initial schema

Revision ID: 202605110001
Revises:
Create Date: 2026-05-11
"""

from collections.abc import Sequence

import sqlalchemy as sa
from alembic import op

revision: str = "202605110001"
down_revision: str | None = None
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


service_type = sa.Enum(
    "maintenance",
    "repair",
    "fluid",
    "tire",
    "inspection",
    "insurance",
    "tax",
    "other",
    name="servicetype",
    create_type=False,
)


def upgrade() -> None:
    op.create_table(
        "users",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("telegram_id", sa.BigInteger(), nullable=False),
        sa.Column("username", sa.String(length=128), nullable=True),
        sa.Column("first_name", sa.String(length=128), nullable=True),
        sa.Column("last_name", sa.String(length=128), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index(op.f("ix_users_telegram_id"), "users", ["telegram_id"], unique=True)

    op.create_table(
        "cars",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("owner_id", sa.Integer(), nullable=False),
        sa.Column("name", sa.String(length=160), nullable=False),
        sa.Column("make", sa.String(length=80), nullable=True),
        sa.Column("model", sa.String(length=80), nullable=True),
        sa.Column("year", sa.Integer(), nullable=True),
        sa.Column("plate_number", sa.String(length=32), nullable=True),
        sa.Column("vin", sa.String(length=32), nullable=True),
        sa.Column("fuel_type", sa.String(length=32), nullable=True),
        sa.Column("purchase_date", sa.Date(), nullable=True),
        sa.Column("purchase_price", sa.Numeric(12, 2), nullable=True),
        sa.Column("current_odometer", sa.Integer(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.ForeignKeyConstraint(["owner_id"], ["users.id"], ondelete="CASCADE"),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index(op.f("ix_cars_owner_id"), "cars", ["owner_id"], unique=False)

    op.create_table(
        "fuel_entries",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("car_id", sa.Integer(), nullable=False),
        sa.Column("entry_date", sa.Date(), nullable=False),
        sa.Column("odometer", sa.Integer(), nullable=False),
        sa.Column("liters", sa.Numeric(8, 3), nullable=False),
        sa.Column("price_per_liter", sa.Numeric(10, 2), nullable=False),
        sa.Column("total_cost", sa.Numeric(12, 2), nullable=False),
        sa.Column("station", sa.String(length=160), nullable=True),
        sa.Column("fuel_brand", sa.String(length=80), nullable=True),
        sa.Column("is_full_tank", sa.Boolean(), nullable=False),
        sa.Column("notes", sa.Text(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.ForeignKeyConstraint(["car_id"], ["cars.id"], ondelete="CASCADE"),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index(op.f("ix_fuel_entries_car_id"), "fuel_entries", ["car_id"], unique=False)
    op.create_index(op.f("ix_fuel_entries_entry_date"), "fuel_entries", ["entry_date"], unique=False)

    op.create_table(
        "service_entries",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("car_id", sa.Integer(), nullable=False),
        sa.Column("entry_date", sa.Date(), nullable=False),
        sa.Column("odometer", sa.Integer(), nullable=True),
        sa.Column("service_type", service_type, nullable=False),
        sa.Column("title", sa.String(length=180), nullable=False),
        sa.Column("category", sa.String(length=80), nullable=True),
        sa.Column("vendor", sa.String(length=160), nullable=True),
        sa.Column("total_cost", sa.Numeric(12, 2), nullable=False),
        sa.Column("next_due_date", sa.Date(), nullable=True),
        sa.Column("next_due_odometer", sa.Integer(), nullable=True),
        sa.Column("notes", sa.Text(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.ForeignKeyConstraint(["car_id"], ["cars.id"], ondelete="CASCADE"),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index(op.f("ix_service_entries_car_id"), "service_entries", ["car_id"], unique=False)
    op.create_index(op.f("ix_service_entries_entry_date"), "service_entries", ["entry_date"], unique=False)
    op.create_index(op.f("ix_service_entries_service_type"), "service_entries", ["service_type"], unique=False)


def downgrade() -> None:
    op.drop_index(op.f("ix_service_entries_service_type"), table_name="service_entries")
    op.drop_index(op.f("ix_service_entries_entry_date"), table_name="service_entries")
    op.drop_index(op.f("ix_service_entries_car_id"), table_name="service_entries")
    op.drop_table("service_entries")
    op.drop_index(op.f("ix_fuel_entries_entry_date"), table_name="fuel_entries")
    op.drop_index(op.f("ix_fuel_entries_car_id"), table_name="fuel_entries")
    op.drop_table("fuel_entries")
    op.drop_index(op.f("ix_cars_owner_id"), table_name="cars")
    op.drop_table("cars")
    op.drop_index(op.f("ix_users_telegram_id"), table_name="users")
    op.drop_table("users")
    service_type.drop(op.get_bind(), checkfirst=True)