"""add_uploads_and_achievements

Revision ID: 0003_uploads_achievements
Revises: 0002_groups_oauth
Create Date: 2026-02-03 21:00:00

"""
from alembic import op
import sqlalchemy as sa

revision = "0003_uploads_achievements"
down_revision = "0002_groups_oauth"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # Create uploads table
    op.create_table(
        "uploads",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("filename", sa.String(length=255), nullable=False),
        sa.Column("content_type", sa.String(length=100), nullable=False),
        sa.Column("file_path", sa.String(length=512), nullable=False),
        sa.Column("size_bytes", sa.Integer(), nullable=False),
        sa.Column("uploaded_by", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
    )
    op.create_index(op.f("ix_uploads_uploaded_by"), "uploads", ["uploaded_by"])

    # Add upload_id to modules table
    op.add_column("modules", sa.Column("upload_id", sa.Integer(), sa.ForeignKey("uploads.id", ondelete="SET NULL"), nullable=True))
    op.create_index(op.f("ix_modules_upload_id"), "modules", ["upload_id"])

    # Create achievements table
    op.create_table(
        "achievements",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("code", sa.String(length=100), nullable=False, unique=True),
        sa.Column("name", sa.String(length=255), nullable=False),
        sa.Column("description", sa.Text(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
    )
    op.create_index(op.f("ix_achievements_code"), "achievements", ["code"], unique=True)

    # Create user_achievements table
    op.create_table(
        "user_achievements",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
        sa.Column("achievement_id", sa.Integer(), sa.ForeignKey("achievements.id", ondelete="CASCADE"), nullable=False),
        sa.Column("earned_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
        sa.UniqueConstraint("user_id", "achievement_id", name="uq_user_achievement"),
    )
    op.create_index(op.f("ix_user_achievements_user_id"), "user_achievements", ["user_id"])
    op.create_index(op.f("ix_user_achievements_achievement_id"), "user_achievements", ["achievement_id"])

    # Create login_events table for tracking logins
    op.create_table(
        "login_events",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
        sa.Column("logged_in_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
    )
    op.create_index(op.f("ix_login_events_user_id"), "login_events", ["user_id"])
    op.create_index(op.f("ix_login_events_logged_in_at"), "login_events", ["logged_in_at"])


def downgrade() -> None:
    op.drop_table("login_events")
    op.drop_table("user_achievements")
    op.drop_table("achievements")
    op.drop_index(op.f("ix_modules_upload_id"), "modules")
    op.drop_column("modules", "upload_id")
    op.drop_table("uploads")