from __future__ import annotations

from alembic import op
import sqlalchemy as sa
from datetime import datetime

# revision identifiers, used by Alembic.
revision = "20250915_000001"
down_revision = None
branch_labels = None
depends_on = None


def upgrade() -> None:
	op.create_table(
		"users",
		sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
		sa.Column("email", sa.String(length=255), nullable=True),
		sa.Column("mobile", sa.String(length=32), nullable=True),
		sa.Column("first_name", sa.String(length=100), nullable=True),
		sa.Column("last_name", sa.String(length=100), nullable=True),
		sa.Column("password_hash", sa.String(length=255), nullable=False),
		sa.Column("referral_code", sa.String(length=32), nullable=False),
		sa.Column("referred_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
		sa.Column("is_active", sa.Boolean(), nullable=False, server_default=sa.text("1")),
		sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.text("CURRENT_TIMESTAMP")),
		sa.Column("updated_at", sa.DateTime(), nullable=False, server_default=sa.text("CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP")),
	)
	op.create_index("ix_users_email", "users", ["email"], unique=True)
	op.create_index("ix_users_mobile", "users", ["mobile"], unique=True)
	op.create_index("ix_users_referral_code", "users", ["referral_code"], unique=True)

	op.create_table(
		"api_keys",
		sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
		sa.Column("user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
		sa.Column("key_hash", sa.String(length=128), nullable=False),
		sa.Column("key_type", sa.String(length=16), nullable=False),
		sa.Column("name", sa.String(length=100), nullable=True),
		sa.Column("scopes", sa.String(length=500), nullable=True),
		sa.Column("device_id", sa.String(length=100), nullable=True),
		sa.Column("user_agent", sa.String(length=255), nullable=True),
		sa.Column("ip", sa.String(length=64), nullable=True),
		sa.Column("expires_at", sa.DateTime(), nullable=True),
		sa.Column("last_used_at", sa.DateTime(), nullable=True),
		sa.Column("revoked_at", sa.DateTime(), nullable=True),
		sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.text("CURRENT_TIMESTAMP")),
	)
	op.create_index("ix_api_keys_key_hash", "api_keys", ["key_hash"], unique=True)
	op.create_index("ix_api_keys_user_id", "api_keys", ["user_id"], unique=False)

	op.create_table(
		"captchas",
		sa.Column("id", sa.String(length=40), primary_key=True),
		sa.Column("code_hash", sa.String(length=128), nullable=False),
		sa.Column("expires_at", sa.DateTime(), nullable=False),
		sa.Column("attempts", sa.Integer(), nullable=False, server_default=sa.text("0")),
		sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.text("CURRENT_TIMESTAMP")),
	)

	op.create_table(
		"password_resets",
		sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
		sa.Column("user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
		sa.Column("token_hash", sa.String(length=128), nullable=False),
		sa.Column("expires_at", sa.DateTime(), nullable=False),
		sa.Column("used_at", sa.DateTime(), nullable=True),
		sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.text("CURRENT_TIMESTAMP")),
	)
	op.create_index("ix_password_resets_token_hash", "password_resets", ["token_hash"], unique=True)
	op.create_index("ix_password_resets_user_id", "password_resets", ["user_id"], unique=False)


def downgrade() -> None:
	op.drop_index("ix_password_resets_user_id", table_name="password_resets")
	op.drop_index("ix_password_resets_token_hash", table_name="password_resets")
	op.drop_table("password_resets")

	op.drop_table("captchas")

	op.drop_index("ix_api_keys_user_id", table_name="api_keys")
	op.drop_index("ix_api_keys_key_hash", table_name="api_keys")
	op.drop_table("api_keys")

	op.drop_index("ix_users_mobile", table_name="users")
	op.drop_index("ix_users_email", table_name="users")
	op.drop_table("users")