"""Add favorites tables (favorite_groups, favorite_containers)

Revision ID: 0017_add_favorites_tables
Revises: 0016
Create Date: 2025-12-22
"""

from __future__ import annotations

from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
from sqlalchemy import inspect


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


def upgrade() -> None:
    bind = op.get_bind()
    insp = inspect(bind)

    def _table_exists(name: str) -> bool:
        try:
            return insp.has_table(name)
        except Exception:
            return name in insp.get_table_names()

    def _index_exists(table: str, index: str) -> bool:
        try:
            return any(i.get('name') == index for i in insp.get_indexes(table))
        except Exception:
            return False

    if not _table_exists('favorite_groups'):
        op.create_table(
            'favorite_groups',
            sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
            sa.Column('user_id', sa.Integer(), nullable=True),
            sa.Column('name', sa.String(length=100), nullable=False),
            sa.Column('sort_order', sa.Integer(), nullable=False, server_default=sa.text('0')),
            sa.Column('color', sa.String(length=20), nullable=True),
            sa.Column('created_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
            sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
            sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
            sa.PrimaryKeyConstraint('id'),
        )

    if _table_exists('favorite_groups'):
        if not _index_exists('favorite_groups', 'ix_favorite_groups_user_id'):
            op.create_index('ix_favorite_groups_user_id', 'favorite_groups', ['user_id'])
        if not _index_exists('favorite_groups', 'ix_favorite_groups_sort_order'):
            op.create_index('ix_favorite_groups_sort_order', 'favorite_groups', ['user_id', 'sort_order'])
        if not _index_exists('favorite_groups', 'uq_favorite_groups_user_name'):
            op.create_index(
                'uq_favorite_groups_user_name',
                'favorite_groups',
                ['user_id', 'name'],
                unique=True,
            )

    if not _table_exists('favorite_containers'):
        op.create_table(
            'favorite_containers',
            sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
            sa.Column('user_id', sa.Integer(), nullable=True),
            sa.Column('docker_container_id', sa.Integer(), nullable=False),
            sa.Column('group_id', sa.Integer(), nullable=True),
            sa.Column('created_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
            sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
            sa.ForeignKeyConstraint(['docker_container_id'], ['docker_containers.id'], ondelete='CASCADE'),
            sa.ForeignKeyConstraint(['group_id'], ['favorite_groups.id'], ondelete='SET NULL'),
            sa.PrimaryKeyConstraint('id'),
        )

    if _table_exists('favorite_containers'):
        if not _index_exists('favorite_containers', 'ix_favorite_containers_user_id'):
            op.create_index('ix_favorite_containers_user_id', 'favorite_containers', ['user_id'])
        if not _index_exists('favorite_containers', 'ix_favorite_containers_docker_container_id'):
            op.create_index('ix_favorite_containers_docker_container_id', 'favorite_containers', ['docker_container_id'])
        if not _index_exists('favorite_containers', 'ix_favorite_containers_group_id'):
            op.create_index('ix_favorite_containers_group_id', 'favorite_containers', ['group_id'])
        if not _index_exists('favorite_containers', 'uq_favorite_containers_user_docker_container'):
            op.create_index(
                'uq_favorite_containers_user_docker_container',
                'favorite_containers',
                ['user_id', 'docker_container_id'],
                unique=True,
            )


def downgrade() -> None:
    op.drop_index('uq_favorite_containers_user_docker_container', table_name='favorite_containers')
    op.drop_index('ix_favorite_containers_group_id', table_name='favorite_containers')
    op.drop_index('ix_favorite_containers_docker_container_id', table_name='favorite_containers')
    op.drop_index('ix_favorite_containers_user_id', table_name='favorite_containers')
    op.drop_table('favorite_containers')

    op.drop_index('uq_favorite_groups_user_name', table_name='favorite_groups')
    op.drop_index('ix_favorite_groups_sort_order', table_name='favorite_groups')
    op.drop_index('ix_favorite_groups_user_id', table_name='favorite_groups')
    op.drop_table('favorite_groups')