"""Drop CI tables — CI is a pro-tier feature removed before launch.

Tables dropped:
  musehub_ci_test_mappings
  musehub_ci_steps
  musehub_ci_jobs
  musehub_ci_runs
  musehub_ci_secrets

Column dropped:
  musehub_repos.ci_gate_config

Revision ID: 0024
Revises: 0023
"""
from __future__ import annotations

import sqlalchemy as sa
from alembic import op

revision = "0024"
down_revision = "0023"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.drop_column("musehub_repos", "ci_gate_config")
    op.drop_table("musehub_ci_test_mappings")
    op.drop_table("musehub_ci_steps")
    op.drop_table("musehub_ci_jobs")
    op.drop_table("musehub_ci_runs")
    op.drop_table("musehub_ci_secrets")


def downgrade() -> None:
    # Restore ci_gate_config on repos
    op.add_column(
        "musehub_repos",
        sa.Column("ci_gate_config", sa.JSON(), nullable=False, server_default="{}"),
    )

    # Recreate musehub_ci_runs (includes gate_results added by 0012).
    # proposal FK omitted — the original FK referenced musehub_pull_requests which was
    # renamed to musehub_proposals in 0011; omitting avoids name ambiguity.
    op.create_table(
        "musehub_ci_runs",
        sa.Column("run_id", sa.String(36), primary_key=True),
        sa.Column("repo_id", sa.String(36), sa.ForeignKey("musehub_repos.repo_id", ondelete="CASCADE"), nullable=False),
        sa.Column("commit_id", sa.String(255), nullable=False),
        sa.Column("branch", sa.String(255), nullable=False),
        sa.Column("pr_id", sa.String(36), nullable=True),
        sa.Column("trigger", sa.String(20), nullable=False, server_default="push"),
        sa.Column("workflow_name", sa.String(255), nullable=False, server_default="CI"),
        sa.Column("workflow_object_id", sa.String(255), nullable=True),
        sa.Column("status", sa.String(20), nullable=False, server_default="pending"),
        sa.Column("changed_symbols", sa.JSON, nullable=False, server_default="[]"),
        sa.Column("triggered_by", sa.String(255), nullable=False, server_default="system"),
        sa.Column("gate_results", sa.JSON(), nullable=False, server_default="{}"),
        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("completed_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
    )
    op.create_index("ix_musehub_ci_runs_repo_id", "musehub_ci_runs", ["repo_id"])
    op.create_index("ix_musehub_ci_runs_repo_status", "musehub_ci_runs", ["repo_id", "status"])
    op.create_index("ix_musehub_ci_runs_commit_id", "musehub_ci_runs", ["commit_id"])
    op.create_index("ix_musehub_ci_runs_pr_id", "musehub_ci_runs", ["pr_id"])
    op.create_index("ix_musehub_ci_runs_status", "musehub_ci_runs", ["status"])
    op.create_index("ix_musehub_ci_runs_created_at", "musehub_ci_runs", ["created_at"])

    op.create_table(
        "musehub_ci_jobs",
        sa.Column("job_id", sa.String(36), primary_key=True),
        sa.Column("run_id", sa.String(36), sa.ForeignKey("musehub_ci_runs.run_id", ondelete="CASCADE"), nullable=False),
        sa.Column("key", sa.String(255), nullable=False),
        sa.Column("name", sa.String(500), nullable=False),
        sa.Column("container_image", sa.String(500), nullable=False, server_default="python:3.14-slim"),
        sa.Column("needs", sa.JSON, nullable=False, server_default="[]"),
        sa.Column("condition", sa.Text, nullable=False, server_default=""),
        sa.Column("env_vars", sa.JSON, nullable=False, server_default="{}"),
        sa.Column("status", sa.String(20), nullable=False, server_default="pending"),
        sa.Column("exit_code", sa.Integer, nullable=True),
        sa.Column("runner_id", sa.String(255), nullable=True),
        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("completed_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
    )
    op.create_index("ix_musehub_ci_jobs_run_id", "musehub_ci_jobs", ["run_id"])
    op.create_index("ix_musehub_ci_jobs_status", "musehub_ci_jobs", ["status"])

    op.create_table(
        "musehub_ci_steps",
        sa.Column("step_id", sa.String(36), primary_key=True),
        sa.Column("job_id", sa.String(36), sa.ForeignKey("musehub_ci_jobs.job_id", ondelete="CASCADE"), nullable=False),
        sa.Column("order", sa.Integer, nullable=False, server_default="0"),
        sa.Column("name", sa.String(500), nullable=False, server_default=""),
        sa.Column("run_cmd", sa.Text, nullable=False),
        sa.Column("status", sa.String(20), nullable=False, server_default="pending"),
        sa.Column("log_output", sa.Text, nullable=False, server_default=""),
        sa.Column("exit_code", sa.Integer, nullable=True),
        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("completed_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
    )
    op.create_index("ix_musehub_ci_steps_job_id", "musehub_ci_steps", ["job_id"])
    op.create_index("ix_musehub_ci_steps_status", "musehub_ci_steps", ["status"])

    op.create_table(
        "musehub_ci_secrets",
        sa.Column("secret_id", sa.String(36), primary_key=True),
        sa.Column("repo_id", sa.String(36), sa.ForeignKey("musehub_repos.repo_id", ondelete="CASCADE"), nullable=False),
        sa.Column("name", sa.String(255), nullable=False),
        sa.Column("encrypted_value", sa.Text, nullable=False),
        sa.Column("description", sa.String(500), nullable=False, server_default=""),
        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.UniqueConstraint("repo_id", "name", name="uq_musehub_ci_secrets_repo_name"),
    )
    op.create_index("ix_musehub_ci_secrets_repo_id", "musehub_ci_secrets", ["repo_id"])

    op.create_table(
        "musehub_ci_test_mappings",
        sa.Column("mapping_id", sa.String(36), primary_key=True),
        sa.Column("run_id", sa.String(36), sa.ForeignKey("musehub_ci_runs.run_id", ondelete="CASCADE"), nullable=False, unique=True),
        sa.Column("repo_id", sa.String(36), sa.ForeignKey("musehub_repos.repo_id", ondelete="CASCADE"), nullable=False),
        sa.Column("commit_id", sa.String(255), nullable=False),
        sa.Column("changed_symbols", sa.JSON(), nullable=False, server_default="[]"),
        sa.Column("selected_tests", sa.JSON(), nullable=False, server_default="[]"),
        sa.Column("covered_addresses", sa.JSON(), nullable=False, server_default="[]"),
        sa.Column("uncovered_addresses", sa.JSON(), nullable=False, server_default="[]"),
        sa.Column("coverage_fraction", sa.Float(), nullable=False, server_default="0.0"),
        sa.Column("selected_count", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("total_estimated", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("NOW()")),
    )
    op.create_index("ix_musehub_ci_test_mappings_run_id", "musehub_ci_test_mappings", ["run_id"])
    op.create_index("ix_musehub_ci_test_mappings_repo_id", "musehub_ci_test_mappings", ["repo_id"])