"""invited by and check tokens

Revision ID: aa17ed273170
Revises: 173d66d27d78
Create Date: 2024-11-24 03:21:32.324284

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = 'aa17ed273170'
down_revision: Union[str, None] = '173d66d27d78'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.create_table(
        "tokens",
        sa.Column("id", sa.INTEGER(), primary_key=True, autoincrement=True),
        sa.Column("username", sa.VARCHAR(64), index=True),
        sa.Column("hash", sa.VARCHAR(128)),
        sa.Column("name", sa.VARCHAR(128)),
        sa.Column("created_at", sa.TIMESTAMP(), server_default=sa.func.now()),
    )
    op.create_foreign_key("token_username_fk", "tokens", "users",
                          ["username"], ["username"], ondelete="CASCADE")

    op.add_column("users", sa.Column("invited_by", sa.VARCHAR(64), default=None))



def downgrade() -> None:
    op.drop_column("users", "invited_by")
    op.drop_table("tokens")