"""Initial migration

Revision ID: 93fb07842385
Revises:
Create Date: 2025-05-27 21:36:18.532068

"""

from typing import Sequence, Union

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

# revision identifiers, used by Alembic.
revision: str = "93fb07842385"
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:
    """Upgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    # Create user table
    op.create_table('user',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('email', sa.String(length=320), nullable=False),
                    sa.Column('hashed_password', sa.String(length=1024), nullable=False),
                    sa.Column('is_active', sa.Boolean(), nullable=False),
                    sa.Column('is_superuser', sa.Boolean(), nullable=False),
                    sa.Column('is_verified', sa.Boolean(), nullable=False),
                    sa.PrimaryKeyConstraint('id')
                    )
    op.create_index(op.f('ix_user_email'), 'user', ['email'], unique=True)

    # Create oauth account table
    op.create_table('oauth_account',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('oauth_name', sa.String(length=100), nullable=False),
                    sa.Column('access_token', sa.String(length=1024), nullable=False),
                    sa.Column('expires_at', sa.Integer(), nullable=True),
                    sa.Column('refresh_token', sa.String(length=1024), nullable=True),
                    sa.Column('account_id', sa.String(length=320), nullable=False),
                    sa.Column('account_email', sa.String(length=320), nullable=False),
                    sa.Column('user_id', sa.UUID(), nullable=False),
                    sa.ForeignKeyConstraint(['user_id'], ['user.id'], ondelete='cascade'),
                    sa.PrimaryKeyConstraint('id')
                    )
    op.create_index(op.f('ix_oauth_account_account_id'), 'oauth_account', ['account_id'], unique=False)
    op.create_index(op.f('ix_oauth_account_oauth_name'), 'oauth_account', ['oauth_name'], unique=False)

    # Create torrent table
    op.create_table('torrent',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('status', sa.Enum('finished', 'downloading', 'error', 'unknown', name='torrentstatus'), nullable=False),
                    sa.Column('title', sa.String(), nullable=False),
                    sa.Column('quality', sa.Enum('uhd', 'fullhd', 'hd', 'sd', 'unknown', name='quality'), nullable=False),
                    sa.Column('imported', sa.Boolean(), nullable=False),
                    sa.Column('hash', sa.String(), nullable=False),
                    sa.PrimaryKeyConstraint('id')
                    )

    # Create indexer query result table
    op.create_table('indexer_query_result',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('title', sa.String(), nullable=False),
                    sa.Column('download_url', sa.String(), nullable=False),
                    sa.Column('seeders', sa.Integer(), nullable=False),
                    sa.Column('flags', postgresql.ARRAY(sa.String()), nullable=True),
                    sa.Column('quality', sa.Enum('uhd', 'fullhd', 'hd', 'sd', 'unknown', name='quality'), nullable=False),
                    sa.Column('season', postgresql.ARRAY(sa.Integer()), nullable=True),
                    sa.Column('size', sa.BigInteger(), nullable=True),
                    sa.PrimaryKeyConstraint('id')
                    )

    # Create notification table
    op.create_table('notification',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('message', sa.String(), nullable=False),
                    sa.Column('read', sa.Boolean(), nullable=False),
                    sa.Column('timestamp', sa.DateTime(), nullable=False),
                    sa.PrimaryKeyConstraint('id')
                    )

    # Create show table
    op.create_table('show',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('external_id', sa.Integer(), nullable=False),
                    sa.Column('metadata_provider', sa.String(), nullable=False),
                    sa.Column('name', sa.String(), nullable=False),
                    sa.Column('overview', sa.String(), nullable=False),
                    sa.Column('year', sa.Integer(), nullable=True),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('external_id', 'metadata_provider')
                    )

    # Create movie table
    op.create_table('movie',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('external_id', sa.Integer(), nullable=False),
                    sa.Column('metadata_provider', sa.String(), nullable=False),
                    sa.Column('name', sa.String(), nullable=False),
                    sa.Column('overview', sa.String(), nullable=False),
                    sa.Column('year', sa.Integer(), nullable=True),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('external_id', 'metadata_provider')
                    )

    # Create season table
    op.create_table('season',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('show_id', sa.UUID(), nullable=False),
                    sa.Column('number', sa.Integer(), nullable=False),
                    sa.Column('external_id', sa.Integer(), nullable=False),
                    sa.Column('name', sa.String(), nullable=False),
                    sa.Column('overview', sa.String(), nullable=False),
                    sa.ForeignKeyConstraint(['show_id'], ['show.id'], ondelete='CASCADE'),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('show_id', 'number')
                    )

    # Create movie file table
    op.create_table('movie_file',
                    sa.Column('movie_id', sa.UUID(), nullable=False),
                    sa.Column('file_path_suffix', sa.String(), nullable=False),
                    sa.Column('quality', sa.Enum('uhd', 'fullhd', 'hd', 'sd', 'unknown', name='quality'), nullable=False),
                    sa.Column('torrent_id', sa.UUID(), nullable=True),
                    sa.ForeignKeyConstraint(['movie_id'], ['movie.id'], ondelete='CASCADE'),
                    sa.ForeignKeyConstraint(['torrent_id'], ['torrent.id'], ondelete='SET NULL'),
                    sa.PrimaryKeyConstraint('movie_id', 'file_path_suffix')
                    )

    # Create movie request table
    op.create_table('movie_request',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('movie_id', sa.UUID(), nullable=False),
                    sa.Column('wanted_quality', sa.Enum('uhd', 'fullhd', 'hd', 'sd', 'unknown', name='quality'), nullable=False),
                    sa.Column('min_quality', sa.Enum('uhd', 'fullhd', 'hd', 'sd', 'unknown', name='quality'), nullable=False),
                    sa.Column('authorized', sa.Boolean(), nullable=False),
                    sa.Column('requested_by_id', sa.UUID(), nullable=True),
                    sa.Column('authorized_by_id', sa.UUID(), nullable=True),
                    sa.ForeignKeyConstraint(['authorized_by_id'], ['user.id'], ondelete='SET NULL'),
                    sa.ForeignKeyConstraint(['movie_id'], ['movie.id'], ondelete='CASCADE'),
                    sa.ForeignKeyConstraint(['requested_by_id'], ['user.id'], ondelete='SET NULL'),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('movie_id', 'wanted_quality')
                    )

    # Create episode table
    op.create_table('episode',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('season_id', sa.UUID(), nullable=False),
                    sa.Column('number', sa.Integer(), nullable=False),
                    sa.Column('external_id', sa.Integer(), nullable=False),
                    sa.Column('title', sa.String(), nullable=False),
                    sa.ForeignKeyConstraint(['season_id'], ['season.id'], ondelete='CASCADE'),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('season_id', 'number')
                    )

    # Create season file table
    op.create_table('season_file',
                    sa.Column('season_id', sa.UUID(), nullable=False),
                    sa.Column('torrent_id', sa.UUID(), nullable=True),
                    sa.Column('file_path_suffix', sa.String(), nullable=False),
                    sa.Column('quality', sa.Enum('uhd', 'fullhd', 'hd', 'sd', 'unknown', name='quality'), nullable=False),
                    sa.ForeignKeyConstraint(['season_id'], ['season.id'], ondelete='CASCADE'),
                    sa.ForeignKeyConstraint(['torrent_id'], ['torrent.id'], ondelete='SET NULL'),
                    sa.PrimaryKeyConstraint('season_id', 'file_path_suffix')
                    )

    # Create season request table
    op.create_table('season_request',
                    sa.Column('id', sa.UUID(), nullable=False),
                    sa.Column('season_id', sa.UUID(), nullable=False),
                    sa.Column('wanted_quality', sa.Enum('uhd', 'fullhd', 'hd', 'sd', 'unknown', name='quality'), nullable=False),
                    sa.Column('min_quality', sa.Enum('uhd', 'fullhd', 'hd', 'sd', 'unknown', name='quality'), nullable=False),
                    sa.Column('requested_by_id', sa.UUID(), nullable=True),
                    sa.Column('authorized', sa.Boolean(), nullable=False),
                    sa.Column('authorized_by_id', sa.UUID(), nullable=True),
                    sa.ForeignKeyConstraint(['authorized_by_id'], ['user.id'], ondelete='SET NULL'),
                    sa.ForeignKeyConstraint(['requested_by_id'], ['user.id'], ondelete='SET NULL'),
                    sa.ForeignKeyConstraint(['season_id'], ['season.id'], ondelete='CASCADE'),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('season_id', 'wanted_quality')
                    )

    # ### end Alembic commands ###

def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_table('season_request')
    op.drop_table('season_file')
    op.drop_table('episode')
    op.drop_table('movie_request')
    op.drop_table('movie_file')
    op.drop_table('season')
    op.drop_table('movie')
    op.drop_table('show')
    op.drop_table('notification')
    op.drop_table('indexer_query_result')
    op.drop_table('torrent')
    op.drop_index(op.f('ix_oauth_account_oauth_name'), table_name='oauth_account')
    op.drop_index(op.f('ix_oauth_account_account_id'), table_name='oauth_account')
    op.drop_table('oauth_account')
    op.drop_index(op.f('ix_user_email'), table_name='user')
    op.drop_table('user')
    # ### end Alembic commands ###