"""Project GitHub import source metadata.

Revision ID: 060
Revises: 059
Create Date: 2026-05-01
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import UUID

revision = "060"
down_revision = "059"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column(
        "projects",
        sa.Column("source_kind", sa.String(40), nullable=False, server_default="generated"),
    )
    op.add_column(
        "projects",
        sa.Column("source_workspace_user_github_installation_id", UUID(as_uuid=True), nullable=True),
    )
    op.add_column("projects", sa.Column("source_repo_full_name", sa.String(255), nullable=True))
    op.add_column("projects", sa.Column("source_branch", sa.String(100), nullable=True))
    op.add_column("projects", sa.Column("source_root_dir", sa.String(255), nullable=True))
    op.add_column(
        "projects",
        sa.Column("source_import_env", sa.Boolean(), nullable=False, server_default=sa.true()),
    )
    op.add_column("projects", sa.Column("source_imported_commit_sha", sa.String(40), nullable=True))
    op.create_index("ix_projects_source_kind", "projects", ["source_kind"])
    op.create_index(
        "ix_projects_source_user_installation",
        "projects",
        ["source_workspace_user_github_installation_id"],
    )
    op.create_table(
        "project_import_sessions",
        sa.Column("project_id", UUID(as_uuid=True), primary_key=True),
        sa.Column(
            "workspace_id",
            UUID(as_uuid=True),
            sa.ForeignKey("workspaces.id", ondelete="CASCADE"),
            nullable=False,
        ),
        sa.Column("created_by", UUID(as_uuid=True), sa.ForeignKey("users.id"), nullable=False),
        sa.Column("provider", sa.String(40), nullable=False, server_default="github"),
        sa.Column("status", sa.String(30), nullable=False, server_default="started"),
        sa.Column("github_installation_id", sa.BigInteger(), nullable=True),
        sa.Column("workspace_user_github_installation_id", UUID(as_uuid=True), nullable=True),
        sa.Column("source_repo_full_name", sa.String(255), nullable=True),
        sa.Column("source_branch", sa.String(100), nullable=True),
        sa.Column("source_root_dir", sa.String(255), nullable=True),
        sa.Column(
            "created_at",
            sa.DateTime(timezone=True),
            server_default=sa.text("now()"),
            nullable=False,
        ),
        sa.Column(
            "updated_at",
            sa.DateTime(timezone=True),
            server_default=sa.text("now()"),
            nullable=False,
        ),
    )
    op.create_index("ix_project_import_sessions_workspace", "project_import_sessions", ["workspace_id"])
    op.create_index("ix_project_import_sessions_status", "project_import_sessions", ["status"])


def downgrade() -> None:
    op.drop_index("ix_project_import_sessions_status", table_name="project_import_sessions")
    op.drop_index("ix_project_import_sessions_workspace", table_name="project_import_sessions")
    op.drop_table("project_import_sessions")
    op.drop_index("ix_projects_source_user_installation", table_name="projects")
    op.drop_index("ix_projects_source_kind", table_name="projects")
    op.drop_column("projects", "source_imported_commit_sha")
    op.drop_column("projects", "source_import_env")
    op.drop_column("projects", "source_root_dir")
    op.drop_column("projects", "source_branch")
    op.drop_column("projects", "source_repo_full_name")
    op.drop_column("projects", "source_workspace_user_github_installation_id")
    op.drop_column("projects", "source_kind")