"""add user id

Revision ID: d9faf8fb8642
Revises: 4ee21f978e6c
Create Date: 2026-04-27 17:24:17.892586

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision: str = 'd9faf8fb8642'
down_revision: Union[str, Sequence[str], None] = '4ee21f978e6c'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    users = op.create_table(
        "users",
        sa.Column("id", sa.Integer, primary_key=True, autoincrement=True),
        sa.Column("email", sa.String, nullable=False),
        sa.Column("name", sa.String),
        sa.Column("oidc_user_id", sa.Integer)
    )

    op.bulk_insert(users, [  { "email": "default" } ])

    with op.batch_alter_table("counters") as batch_op:
        batch_op.add_column(sa.Column("user_id", sa.Integer, insert_default=1))
        batch_op.create_foreign_key("fk_counters_user_id",
                              referent_table="users",
                              remote_cols=['id'],
                              local_cols=['user_id'])

    with op.batch_alter_table("entries") as batch_op:
        batch_op.add_column(sa.Column("user_id", sa.Integer, insert_default=1))
        batch_op.create_foreign_key("fk_entries_user_id",
                              referent_table="users",
                              remote_cols=['id'],
                              local_cols=['user_id'])


def downgrade() -> None:
    with op.batch_alter_table("counters") as batch_op:
        batch_op.drop_constraint("fk_counters_user_id", type_="foreignkey")
        batch_op.drop_column('user_id')

    with op.batch_alter_table("entries") as batch_op:
        batch_op.drop_constraint('fk_entries_user_id', type_='foreignkey')
        batch_op.drop_column('user_id')

    op.drop_table("users")