"""entangle: add missing fields + indexes to musehub_intel_entangle

Revision ID: 0010
Revises: 0009
"""

from __future__ import annotations

from typing import Union

import sqlalchemy as sa
from alembic import op

revision: str = "0010"
down_revision: Union[str, None] = "0009"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column(
        "musehub_intel_entangle",
        sa.Column("commits_both_active", sa.Integer(), nullable=True),
    )
    op.add_column(
        "musehub_intel_entangle",
        sa.Column("file_a", sa.String(512), nullable=True),
    )
    op.add_column(
        "musehub_intel_entangle",
        sa.Column("file_b", sa.String(512), nullable=True),
    )
    op.add_column(
        "musehub_intel_entangle",
        sa.Column("same_file", sa.Boolean(), nullable=True),
    )
    op.add_column(
        "musehub_intel_entangle",
        sa.Column("a_in_test", sa.Boolean(), nullable=True),
    )
    op.add_column(
        "musehub_intel_entangle",
        sa.Column("b_in_test", sa.Boolean(), nullable=True),
    )
    op.create_index(
        "ix_intel_entangle_repo_file_a", "musehub_intel_entangle", ["repo_id", "file_a"]
    )
    op.create_index(
        "ix_intel_entangle_repo_rate", "musehub_intel_entangle", ["repo_id", "co_change_rate"]
    )
    op.create_index(
        "ix_intel_entangle_repo_symbol_a", "musehub_intel_entangle", ["repo_id", "symbol_a"]
    )
    op.create_index(
        "ix_intel_entangle_repo_symbol_b", "musehub_intel_entangle", ["repo_id", "symbol_b"]
    )


def downgrade() -> None:
    op.drop_index("ix_intel_entangle_repo_symbol_b", table_name="musehub_intel_entangle")
    op.drop_index("ix_intel_entangle_repo_symbol_a", table_name="musehub_intel_entangle")
    op.drop_index("ix_intel_entangle_repo_rate", table_name="musehub_intel_entangle")
    op.drop_index("ix_intel_entangle_repo_file_a", table_name="musehub_intel_entangle")
    op.drop_column("musehub_intel_entangle", "b_in_test")
    op.drop_column("musehub_intel_entangle", "a_in_test")
    op.drop_column("musehub_intel_entangle", "same_file")
    op.drop_column("musehub_intel_entangle", "file_b")
    op.drop_column("musehub_intel_entangle", "file_a")
    op.drop_column("musehub_intel_entangle", "commits_both_active")