"""Add dataset reference and provenance metadata columns."""

from alembic import op
import sqlalchemy as sa


revision = "202606120001"
down_revision = "20260611212435"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column("datasets", sa.Column("dataset_role", sa.Text(), nullable=False, server_default="source"))
    op.add_column("datasets", sa.Column("source_name", sa.Text(), nullable=True))
    op.add_column("datasets", sa.Column("reference_layer_name", sa.Text(), nullable=True))
    op.add_column("datasets", sa.Column("source_metadata", sa.JSON(), nullable=True))
    op.add_column("datasets", sa.Column("provenance_metadata", sa.JSON(), nullable=True))
    op.add_column("datasets", sa.Column("imported_at", sa.DateTime(timezone=True), server_default=sa.text("NOW()"), nullable=False))


def downgrade() -> None:
    op.drop_column("datasets", "imported_at")
    op.drop_column("datasets", "provenance_metadata")
    op.drop_column("datasets", "source_metadata")
    op.drop_column("datasets", "reference_layer_name")
    op.drop_column("datasets", "source_name")
    op.drop_column("datasets", "dataset_role")