"""Add named taste profiles and speaker mappings

Revision ID: 003
Revises: 002
Create Date: 2026-02-22

"""
from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

revision: str = "003"
down_revision: Union[str, None] = "002"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # 1. Create profiles table
    op.create_table(
        "profiles",
        sa.Column("id", sa.BigInteger, primary_key=True),
        sa.Column("name", sa.Text, unique=True, nullable=False),
        sa.Column("display_name", sa.Text),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
    )

    # 2. Seed "default" profile
    op.execute("INSERT INTO profiles (id, name, display_name) VALUES (1, 'default', 'Default')")

    # 3. Create speaker_profile_mappings table
    op.create_table(
        "speaker_profile_mappings",
        sa.Column("id", sa.BigInteger, primary_key=True),
        sa.Column("speaker_name", sa.Text, unique=True, nullable=False),
        sa.Column("profile_id", sa.BigInteger, sa.ForeignKey("profiles.id"), nullable=False),
    )

    # 4. Add profile_id to listen_events (nullable — NULL means "default")
    op.add_column("listen_events", sa.Column("profile_id", sa.BigInteger, sa.ForeignKey("profiles.id")))

    # 5. Add profile_id to taste_profiles (nullable, unique)
    op.add_column("taste_profiles", sa.Column("profile_id", sa.BigInteger, sa.ForeignKey("profiles.id"), unique=True))

    # 6. Link existing "default" taste profile row to the default profile
    op.execute(
        "UPDATE taste_profiles SET profile_id = 1 WHERE name = 'default'"
    )


def downgrade() -> None:
    op.drop_column("taste_profiles", "profile_id")
    op.drop_column("listen_events", "profile_id")
    op.drop_table("speaker_profile_mappings")
    op.drop_table("profiles")