"""initial_schema

Revision ID: 86cca6d7adcb
Revises:
Create Date: 2026-05-27 16:31:52.870967

"""
from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql

revision: str = '86cca6d7adcb'
down_revision: Union[str, None] = None
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.create_table(
        'users',
        sa.Column('id', sa.BigInteger(), nullable=False),
        sa.Column('username', sa.String(255), nullable=True),
        sa.Column('first_name', sa.String(255), nullable=False),
        sa.Column('last_name', sa.String(255), nullable=True),
        sa.Column('created_at', sa.DateTime(), nullable=False),
        sa.PrimaryKeyConstraint('id'),
    )

    op.create_table(
        'lotteries',
        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('creator_id', sa.BigInteger(), nullable=False),
        sa.Column('title', sa.String(255), nullable=False),
        sa.Column('description', sa.Text(), nullable=True),
        sa.Column('status', sa.String(20), nullable=False),
        sa.Column('target_chat_id', sa.BigInteger(), nullable=True),
        sa.Column('target_chat_title', sa.String(255), nullable=True),
        sa.Column('require_membership', sa.Boolean(), nullable=False),
        sa.Column('required_chat_ids', sa.JSON(), nullable=True),
        sa.Column('required_chat_titles', sa.JSON(), nullable=True),
        sa.Column('required_chat_types', sa.JSON(), nullable=True),
        sa.Column('required_chat_usernames', sa.JSON(), nullable=True),
        sa.Column('announcement_targets', sa.JSON(), nullable=True),
        sa.Column('max_participants', sa.Integer(), nullable=True),
        sa.Column('announcement_message_id', sa.BigInteger(), nullable=True),
        sa.Column('contact_info', sa.Text(), nullable=True),
        sa.Column('draw_at', sa.DateTime(), nullable=True),
        sa.Column('auto_draw_count', sa.Integer(), nullable=True),
        sa.Column('allow_multiple_wins', sa.Boolean(), server_default=sa.text('false'), nullable=False),
        sa.Column('invite_link_chat_ids', sa.JSON(), server_default=sa.text("'[]'"), nullable=False),
        sa.Column('creator_timezone', sa.String(64), nullable=True),
        sa.Column('created_at', sa.DateTime(), nullable=False),
        sa.Column('drawn_at', sa.DateTime(), nullable=True),
        sa.Column('draw_trigger', sa.String(20), nullable=True),
        sa.Column('required_chat_passphrases', sa.JSON(), nullable=True),
        sa.ForeignKeyConstraint(['creator_id'], ['users.id']),
        sa.PrimaryKeyConstraint('id'),
    )

    op.create_table(
        'prizes',
        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('lottery_id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('name', sa.String(255), nullable=False),
        sa.Column('description', sa.Text(), nullable=True),
        sa.Column('quantity', sa.Integer(), nullable=False),
        sa.Column('sort_order', sa.Integer(), nullable=False),
        sa.ForeignKeyConstraint(['lottery_id'], ['lotteries.id'], ondelete='CASCADE'),
        sa.PrimaryKeyConstraint('id'),
    )

    op.create_table(
        'participants',
        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('lottery_id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('user_id', sa.BigInteger(), nullable=False),
        sa.Column('joined_at', sa.DateTime(), nullable=False),
        sa.ForeignKeyConstraint(['lottery_id'], ['lotteries.id'], ondelete='CASCADE'),
        sa.ForeignKeyConstraint(['user_id'], ['users.id']),
        sa.PrimaryKeyConstraint('id'),
        sa.UniqueConstraint('lottery_id', 'user_id'),
    )

    op.create_table(
        'winners',
        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('lottery_id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('user_id', sa.BigInteger(), nullable=False),
        sa.Column('prize_id', postgresql.UUID(as_uuid=True), nullable=True),
        sa.Column('prize_name', sa.String(255), nullable=False),
        sa.Column('drawn_at', sa.DateTime(), nullable=False),
        sa.ForeignKeyConstraint(['lottery_id'], ['lotteries.id'], ondelete='CASCADE'),
        sa.ForeignKeyConstraint(['prize_id'], ['prizes.id'], ondelete='SET NULL'),
        sa.ForeignKeyConstraint(['user_id'], ['users.id']),
        sa.PrimaryKeyConstraint('id'),
    )

    op.create_table(
        'managed_chats',
        sa.Column('id', sa.BigInteger(), nullable=False),
        sa.Column('title', sa.String(255), nullable=False),
        sa.Column('username', sa.String(255), nullable=True),
        sa.Column('chat_type', sa.String(50), nullable=False),
        sa.Column('invite_link', sa.String(512), nullable=True),
        sa.Column('last_message_id', sa.BigInteger(), nullable=True),
        sa.Column('is_active', sa.Boolean(), nullable=False),
        sa.Column('added_at', sa.DateTime(), nullable=False),
        sa.PrimaryKeyConstraint('id'),
    )

    op.create_table(
        'passphrase_verifications',
        sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
        sa.Column('lottery_id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('chat_id', sa.BigInteger(), nullable=False),
        sa.Column('user_id', sa.BigInteger(), nullable=False),
        sa.Column('verified_at', sa.DateTime(), nullable=False),
        sa.ForeignKeyConstraint(['lottery_id'], ['lotteries.id'], ondelete='CASCADE'),
        sa.ForeignKeyConstraint(['user_id'], ['users.id']),
        sa.PrimaryKeyConstraint('id'),
        sa.UniqueConstraint('lottery_id', 'chat_id', 'user_id'),
    )


def downgrade() -> None:
    op.drop_table('passphrase_verifications')
    op.drop_table('managed_chats')
    op.drop_table('winners')
    op.drop_table('participants')
    op.drop_table('prizes')
    op.drop_table('lotteries')
    op.drop_table('users')