"""add_password_reset_table

Revision ID: d103d442dc24
Revises: 0f96348cdfd2
Create Date: 2025-08-22 08:27:58.468119

"""
from __future__ import annotations

from collections.abc import Sequence

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import mysql
import sqlmodel

# revision identifiers, used by Alembic.
revision: str = "d103d442dc24"
down_revision: str | Sequence[str] | None = "0f96348cdfd2"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    """Upgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table("password_resets",
    sa.Column("id", sa.Integer(), nullable=False),
    sa.Column("user_id", sa.BigInteger(), nullable=False),
    sa.Column("email", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
    sa.Column("reset_code", sqlmodel.sql.sqltypes.AutoString(length=8), nullable=False),
    sa.Column("created_at", sa.DateTime(), nullable=False),
    sa.Column("expires_at", sa.DateTime(), nullable=False),
    sa.Column("is_used", sa.Boolean(), nullable=False),
    sa.Column("used_at", sa.DateTime(), nullable=True),
    sa.Column("ip_address", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
    sa.Column("user_agent", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
    sa.ForeignKeyConstraint(["user_id"], ["lazer_users.id"], ),
    sa.PrimaryKeyConstraint("id")
    )
    op.create_index(op.f("ix_password_resets_email"), "password_resets", ["email"], unique=False)
    op.create_index(op.f("ix_password_resets_user_id"), "password_resets", ["user_id"], unique=False)
    op.drop_index(op.f("ix_two_factor_auth_id"), table_name="two_factor_auth")
    op.drop_index(op.f("ix_two_factor_auth_user_id"), table_name="two_factor_auth")
    op.drop_table("two_factor_auth")
    op.drop_index(op.f("ix_user_ip_history_id"), table_name="user_ip_history")
    op.drop_index(op.f("ix_user_ip_history_ip_address"), table_name="user_ip_history")
    op.drop_index(op.f("ix_user_ip_history_user_id"), table_name="user_ip_history")
    op.drop_table("user_ip_history")
    op.drop_index(op.f("ix_session_verification_user_id"), table_name="session_verification")
    op.drop_table("session_verification")
    op.alter_column("beatmapsets", "nsfw",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    op.alter_column("beatmapsets", "spotlight",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    op.alter_column("beatmapsets", "video",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    op.alter_column("beatmapsets", "can_be_hyped",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    op.alter_column("beatmapsets", "discussion_locked",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    op.alter_column("beatmapsets", "storyboard",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    op.alter_column("beatmapsets", "download_disabled",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    op.drop_index(op.f("uq_user_achievement"), table_name="lazer_user_achievements")
    op.alter_column("scores", "has_replay",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    op.alter_column("scores", "passed",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    op.alter_column("scores", "preserve",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=True)
    # ### end Alembic commands ###


def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.alter_column("scores", "preserve",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.alter_column("scores", "passed",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.alter_column("scores", "has_replay",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.create_index(op.f("uq_user_achievement"), "lazer_user_achievements", ["user_id", "achievement_id"], unique=True)
    op.alter_column("beatmapsets", "download_disabled",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.alter_column("beatmapsets", "storyboard",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.alter_column("beatmapsets", "discussion_locked",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.alter_column("beatmapsets", "can_be_hyped",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.alter_column("beatmapsets", "video",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.alter_column("beatmapsets", "spotlight",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.alter_column("beatmapsets", "nsfw",
               existing_type=mysql.TINYINT(display_width=1),
               nullable=False)
    op.create_table("session_verification",
    sa.Column("id", mysql.BIGINT(), autoincrement=True, nullable=False),
    sa.Column("user_id", mysql.BIGINT(), autoincrement=False, nullable=False),
    sa.Column("session_id", mysql.VARCHAR(length=255), nullable=False),
    sa.Column("ip_address", mysql.VARCHAR(length=45), nullable=False),
    sa.Column("is_verified", mysql.TINYINT(display_width=1), autoincrement=False, nullable=False),
    sa.Column("verified_at", mysql.DATETIME(), nullable=True),
    sa.Column("expires_at", mysql.DATETIME(), nullable=False),
    sa.ForeignKeyConstraint(["user_id"], ["lazer_users.id"], name=op.f("session_verification_ibfk_1")),
    sa.PrimaryKeyConstraint("id"),
    mysql_collate="utf8mb4_0900_ai_ci",
    mysql_default_charset="utf8mb4",
    mysql_engine="InnoDB"
    )
    op.create_index(op.f("ix_session_verification_user_id"), "session_verification", ["user_id"], unique=False)
    op.create_table("user_ip_history",
    sa.Column("id", mysql.BIGINT(), autoincrement=True, nullable=False),
    sa.Column("user_id", mysql.BIGINT(), autoincrement=False, nullable=False),
    sa.Column("ip_address", mysql.VARCHAR(length=45), nullable=False),
    sa.Column("first_seen", mysql.DATETIME(), nullable=False),
    sa.Column("last_seen", mysql.DATETIME(), nullable=False),
    sa.Column("usage_count", mysql.INTEGER(), autoincrement=False, nullable=False),
    sa.Column("is_trusted", mysql.TINYINT(display_width=1), autoincrement=False, nullable=False),
    sa.ForeignKeyConstraint(["user_id"], ["lazer_users.id"], name=op.f("user_ip_history_ibfk_1")),
    sa.PrimaryKeyConstraint("id"),
    mysql_collate="utf8mb4_0900_ai_ci",
    mysql_default_charset="utf8mb4",
    mysql_engine="InnoDB"
    )
    op.create_index(op.f("ix_user_ip_history_user_id"), "user_ip_history", ["user_id"], unique=False)
    op.create_index(op.f("ix_user_ip_history_ip_address"), "user_ip_history", ["ip_address"], unique=False)
    op.create_index(op.f("ix_user_ip_history_id"), "user_ip_history", ["id"], unique=False)
    op.create_table("two_factor_auth",
    sa.Column("id", mysql.BIGINT(), autoincrement=True, nullable=False),
    sa.Column("user_id", mysql.BIGINT(), autoincrement=False, nullable=False),
    sa.Column("verification_code", mysql.VARCHAR(length=8), nullable=False),
    sa.Column("expires_at", mysql.DATETIME(), nullable=False),
    sa.Column("is_used", mysql.TINYINT(display_width=1), autoincrement=False, nullable=False),
    sa.Column("ip_address", mysql.VARCHAR(length=45), nullable=False),
    sa.Column("trigger_reason", mysql.VARCHAR(length=50), nullable=False),
    sa.Column("verified_at", mysql.DATETIME(), nullable=True),
    sa.ForeignKeyConstraint(["user_id"], ["lazer_users.id"], name=op.f("two_factor_auth_ibfk_1")),
    sa.PrimaryKeyConstraint("id"),
    mysql_collate="utf8mb4_0900_ai_ci",
    mysql_default_charset="utf8mb4",
    mysql_engine="InnoDB"
    )
    op.create_index(op.f("ix_two_factor_auth_user_id"), "two_factor_auth", ["user_id"], unique=False)
    op.create_index(op.f("ix_two_factor_auth_id"), "two_factor_auth", ["id"], unique=False)
    op.drop_index(op.f("ix_password_resets_user_id"), table_name="password_resets")
    op.drop_index(op.f("ix_password_resets_email"), table_name="password_resets")
    op.drop_table("password_resets")
    # ### end Alembic commands ###