"""rename plans basic->premium, pro->max; reset ai_credits_used

Revision ID: fa32b9c8d1e0
Revises: c0d1e2f3g4h5
Create Date: 2026-06-11
"""
from alembic import op

revision = 'fa32b9c8d1e0'
down_revision = 'c0d1e2f3g4h5'
branch_labels = None
depends_on = None


def upgrade():
    # Rename plan values in organizations table
    op.execute("UPDATE public.organizations SET plan = 'premium' WHERE plan = 'basic'")
    op.execute("UPDATE public.organizations SET plan = 'max' WHERE plan = 'pro'")
    # Reset ai_credits_used to 0 for all orgs (counter semantics changed from tokens to credits)
    op.execute("UPDATE public.organizations SET ai_credits_used = 0")
    # Update plans table if it has rows
    op.execute("UPDATE public.plans SET name = 'premium' WHERE name = 'basic'")
    op.execute("UPDATE public.plans SET name = 'max' WHERE name = 'pro'")


def downgrade():
    op.execute("UPDATE public.organizations SET plan = 'basic' WHERE plan = 'premium'")
    op.execute("UPDATE public.organizations SET plan = 'pro' WHERE plan = 'max'")
    op.execute("UPDATE public.plans SET name = 'basic' WHERE name = 'premium'")
    op.execute("UPDATE public.plans SET name = 'pro' WHERE name = 'max'")