"""Add email confirmation columns"""

from __future__ import annotations

from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision = '20240927_0006'
down_revision = '20240927_0005'
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column(
        'users',
        sa.Column('is_email_confirmed', sa.Boolean(), server_default=sa.sql.expression.true(), nullable=False),
    )
    op.add_column('users', sa.Column('email_confirmation_token', sa.String(length=128), nullable=True))
    op.add_column('users', sa.Column('email_confirmed_at', sa.DateTime(timezone=True), nullable=True))
    op.execute("UPDATE users SET is_email_confirmed = 1")


def downgrade() -> None:
    op.drop_column('users', 'email_confirmed_at')
    op.drop_column('users', 'email_confirmation_token')
    op.drop_column('users', 'is_email_confirmed')