"""remove auth columns from users

Revision ID: a1b2c3d4e5f6
Revises: 1855836abf6c
Create Date: 2026-03-13

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
import sqlmodel


# revision identifiers, used by Alembic.
revision: str = 'a1b2c3d4e5f6'
down_revision: Union[str, Sequence[str], None] = '1855836abf6c'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """Drop password_hash and is_admin columns, delete admin users."""
    # Delete admin users before dropping the column
    op.execute("DELETE FROM users WHERE is_admin = 1")

    # SQLite requires table recreation for column drops
    with op.batch_alter_table('users', schema=None) as batch_op:
        batch_op.drop_column('password_hash')
        batch_op.drop_column('is_admin')


def downgrade() -> None:
    """Re-add password_hash and is_admin columns."""
    with op.batch_alter_table('users', schema=None) as batch_op:
        batch_op.add_column(sa.Column('password_hash', sa.String(), nullable=False, server_default=''))
        batch_op.add_column(sa.Column('is_admin', sa.Boolean(), nullable=False, server_default='0'))