"""add goal-articulation columns to discovery_conversation and discovery_profile

Revision ID: 003
Revises: 002
Create Date: 2026-06-15

Phase 1 goal articulation: two new narrative prompts capture the person's own
near- and long-term goals (on the conversation), and two AI-articulated goal
summaries are stored alongside the Ikigai/enneagram profile.
"""
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:
    op.add_column(
        "discovery_conversation",
        sa.Column("prompt_goals_short", sa.Text(), nullable=True),
    )
    op.add_column(
        "discovery_conversation",
        sa.Column("prompt_goals_long", sa.Text(), nullable=True),
    )
    op.add_column(
        "discovery_profile",
        sa.Column("short_term_goals", sa.Text(), nullable=True),
    )
    op.add_column(
        "discovery_profile",
        sa.Column("long_term_goals", sa.Text(), nullable=True),
    )


def downgrade() -> None:
    op.drop_column("discovery_profile", "long_term_goals")
    op.drop_column("discovery_profile", "short_term_goals")
    op.drop_column("discovery_conversation", "prompt_goals_long")
    op.drop_column("discovery_conversation", "prompt_goals_short")