"""add status and updated_at to callback

Revision ID: 080
Revises: 079
Create Date: 2025-11-05 00:00:00.000000

"""

from enum import Enum
from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

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


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


def upgrade() -> None:
    """Upgrade schema."""
    status = sa.Enum(EventCallbackStatus, name='eventcallbackstatus')
    status.create(op.get_bind(), checkfirst=True)
    op.add_column(
        'event_callback',
        sa.Column('status', 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."""
    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,
    )
    op.execute('DROP TYPE eventcallbackstatus')