"""Sync DB with Models

Revision ID: 002
Revises: 001
Create Date: 2025-10-05 11:28:41.772294

"""

from enum import Enum
from typing import Sequence

import sqlalchemy as sa
from alembic import op

# revision identifiers, used by Alembic.
revision: str = '002'
down_revision: str | None = '001'
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


class EventCallbackStatus(Enum):
    ACTIVE = 'ACTIVE'
    DISABLED = 'DISABLED'
    COMPLETED = 'COMPLETED'
    ERROR = 'ERROR'


def upgrade() -> None:
    """Upgrade schema."""
    bind = op.get_bind()
    event_callback_status = sa.Enum(
        EventCallbackStatus,
        name='eventcallbackstatus',
    )
    event_callback_status.create(bind, checkfirst=True)

    op.add_column(
        'event_callback',
        sa.Column(
            'status',
            event_callback_status,
            nullable=False,
            server_default='ACTIVE',
        ),
    )
    op.add_column(
        'event_callback',
        sa.Column(
            'updated_at', sa.DateTime, nullable=False, server_default=sa.func.now()
        ),
    )
    op.drop_index('ix_event_callback_result_event_id')
    op.drop_column('event_callback_result', 'event_id')
    op.add_column(
        'event_callback_result', sa.Column('event_id', sa.String, nullable=True)
    )
    op.create_index(
        op.f('ix_event_callback_result_event_id'),
        'event_callback_result',
        ['event_id'],
        unique=False,
    )


def downgrade() -> None:
    """Downgrade schema."""
    bind = op.get_bind()
    event_callback_status = sa.Enum(
        EventCallbackStatus,
        name='eventcallbackstatus',
    )

    op.drop_column('event_callback', 'status')
    op.drop_column('event_callback', 'updated_at')
    op.drop_index('ix_event_callback_result_event_id')
    op.drop_column('event_callback_result', 'event_id')
    op.add_column(
        'event_callback_result', sa.Column('event_id', sa.UUID, nullable=True)
    )
    op.create_index(
        op.f('ix_event_callback_result_event_id'),
        'event_callback_result',
        ['event_id'],
        unique=False,
    )

    event_callback_status.drop(bind, checkfirst=True)