"""Add domain summary columns and agent provenance columns to proposals + review enhancements (issue #37 Phase 1b)

Revision ID: 0046
Revises: 0045
"""
from __future__ import annotations

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects.postgresql import ARRAY, JSONB

revision: str = "0046"
down_revision: str = "0045"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # MIDI domain summary columns
    op.add_column("musehub_proposals", sa.Column("midi_tracks_changed", sa.Integer, nullable=False, server_default="0"))
    op.add_column("musehub_proposals", sa.Column("midi_notes_delta", sa.Integer, nullable=False, server_default="0"))
    op.add_column("musehub_proposals", sa.Column("harmonic_tension_delta", sa.Float, nullable=True))

    # Payment domain summary columns
    op.add_column("musehub_proposals", sa.Column("payment_claim_count", sa.Integer, nullable=False, server_default="0"))
    op.add_column("musehub_proposals", sa.Column("payment_ledger_delta_nano", sa.BigInteger, nullable=False, server_default="0"))
    op.add_column("musehub_proposals", sa.Column("payment_avax_address", sa.String(255), nullable=True))

    # Agent provenance columns
    op.add_column("musehub_proposals", sa.Column("agent_model", sa.String(255), nullable=True))
    op.add_column("musehub_proposals", sa.Column("agent_spawned_by", sa.String(255), nullable=True))

    # Review dimensional fields
    op.add_column("musehub_proposal_reviews", sa.Column("reviewed_domains", ARRAY(sa.Text), nullable=False, server_default="{}"))
    op.add_column("musehub_proposal_reviews", sa.Column("domain_risk_acknowledged", JSONB, nullable=False, server_default="{}"))
    op.add_column("musehub_proposal_reviews", sa.Column("suggested_merge_strategy", sa.String(50), nullable=True))


def downgrade() -> None:
    op.drop_column("musehub_proposal_reviews", "suggested_merge_strategy")
    op.drop_column("musehub_proposal_reviews", "domain_risk_acknowledged")
    op.drop_column("musehub_proposal_reviews", "reviewed_domains")

    op.drop_column("musehub_proposals", "agent_spawned_by")
    op.drop_column("musehub_proposals", "agent_model")
    op.drop_column("musehub_proposals", "payment_avax_address")
    op.drop_column("musehub_proposals", "payment_ledger_delta_nano")
    op.drop_column("musehub_proposals", "payment_claim_count")
    op.drop_column("musehub_proposals", "harmonic_tension_delta")
    op.drop_column("musehub_proposals", "midi_notes_delta")
    op.drop_column("musehub_proposals", "midi_tracks_changed")