"""add poi tables

Revision ID: c5d6e7f8a9b0
Revises: b4c7d8e9f0a1
Create Date: 2026-02-08 12:00:00.000000

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
import sqlmodel


# revision identifiers, used by Alembic.
revision: str = 'c5d6e7f8a9b0'
down_revision: Union[str, None] = 'b4c7d8e9f0a1'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """Create pointofinterest and poidistance tables."""
    op.create_table('pointofinterest',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('user_id', sa.Integer(), nullable=False),
        sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
        sa.Column('address', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
        sa.Column('latitude', sa.Float(), nullable=False),
        sa.Column('longitude', sa.Float(), nullable=False),
        sa.Column('created_at', sa.DateTime(), nullable=True,
                  server_default=sa.func.now()),
        sa.ForeignKeyConstraint(['user_id'], ['user.id']),
        sa.PrimaryKeyConstraint('id'),
    )
    op.create_index(op.f('ix_pointofinterest_user_id'),
                    'pointofinterest', ['user_id'], unique=False)

    op.create_table('poidistance',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('listing_id', sa.Integer(), nullable=False),
        sa.Column('listing_type', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
        sa.Column('poi_id', sa.Integer(), nullable=False),
        sa.Column('travel_mode', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
        sa.Column('duration_seconds', sa.Integer(), nullable=False),
        sa.Column('distance_meters', sa.Integer(), nullable=False),
        sa.Column('computed_at', sa.DateTime(), nullable=True,
                  server_default=sa.func.now()),
        sa.ForeignKeyConstraint(['poi_id'], ['pointofinterest.id']),
        sa.PrimaryKeyConstraint('id'),
        sa.UniqueConstraint('listing_id', 'listing_type', 'poi_id', 'travel_mode',
                            name='uq_poi_distance_listing_poi_mode'),
    )
    op.create_index(op.f('ix_poidistance_listing_id'),
                    'poidistance', ['listing_id'], unique=False)
    op.create_index(op.f('ix_poidistance_listing_type'),
                    'poidistance', ['listing_type'], unique=False)
    op.create_index(op.f('ix_poidistance_poi_id'),
                    'poidistance', ['poi_id'], unique=False)


def downgrade() -> None:
    """Drop poidistance and pointofinterest tables."""
    op.drop_index(op.f('ix_poidistance_poi_id'), table_name='poidistance')
    op.drop_index(op.f('ix_poidistance_listing_type'), table_name='poidistance')
    op.drop_index(op.f('ix_poidistance_listing_id'), table_name='poidistance')
    op.drop_table('poidistance')

    op.drop_index(op.f('ix_pointofinterest_user_id'), table_name='pointofinterest')
    op.drop_table('pointofinterest')