"""groups_and_oauth

Revision ID: 0002_groups_oauth
Revises: 0001_initial
Create Date: 2026-02-03 20:00:00

"""
from alembic import op
import sqlalchemy as sa

revision = "0002_groups_oauth"
down_revision = "0001_initial"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column("users", sa.Column("is_active", sa.Boolean(), nullable=False, server_default=sa.true()))
    op.add_column("users", sa.Column("oauth_provider", sa.String(length=50), nullable=True))
    op.add_column("users", sa.Column("oauth_subject", sa.String(length=255), nullable=True))
    op.add_column("users", sa.Column("picture_url", sa.String(length=512), nullable=True))

    op.create_table(
        "groups",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("name", sa.String(length=255), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
    )
    op.create_index(op.f("ix_groups_name"), "groups", ["name"], unique=True)

    op.create_table(
        "group_memberships",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("group_id", sa.Integer(), sa.ForeignKey("groups.id", ondelete="CASCADE"), nullable=False),
        sa.Column("user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
        sa.UniqueConstraint("group_id", "user_id", name="uq_group_membership"),
    )
    op.create_index(op.f("ix_group_memberships_group_id"), "group_memberships", ["group_id"])
    op.create_index(op.f("ix_group_memberships_user_id"), "group_memberships", ["user_id"])

    op.create_table(
        "course_groups",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("course_id", sa.Integer(), sa.ForeignKey("courses.id", ondelete="CASCADE"), nullable=False),
        sa.Column("group_id", sa.Integer(), sa.ForeignKey("groups.id", ondelete="CASCADE"), nullable=False),
        sa.UniqueConstraint("course_id", "group_id", name="uq_course_group"),
    )
    op.create_index(op.f("ix_course_groups_course_id"), "course_groups", ["course_id"])
    op.create_index(op.f("ix_course_groups_group_id"), "course_groups", ["group_id"])


def downgrade() -> None:
    op.drop_table("course_groups")
    op.drop_table("group_memberships")
    op.drop_table("groups")
    op.drop_column("users", "picture_url")
    op.drop_column("users", "oauth_subject")
    op.drop_column("users", "oauth_provider")
    op.drop_column("users", "is_active")