"""Add General Affairs: can_be_scheduled, jadwal_shift, absensi.

Revision ID: x0b1c2d3e4f5
Revises: w9a0b1c2d3e4
Create Date: 2026-04-10
"""

from alembic import op
import sqlalchemy as sa

revision = "x0b1c2d3e4f5"
down_revision = "w9a0b1c2d3e4"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # 1. Add can_be_scheduled to master_role
    op.add_column(
        "master_role",
        sa.Column("can_be_scheduled", sa.Boolean(), nullable=False, server_default="false"),
    )

    # 2. Create jadwal_shift
    op.create_table(
        "jadwal_shift",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("spbu_id", sa.Integer(), sa.ForeignKey("master_spbu.id", ondelete="CASCADE"), nullable=False, index=True),
        sa.Column("user_id", sa.Integer(), sa.ForeignKey("master_user.id", ondelete="CASCADE"), nullable=False),
        sa.Column("shift_id", sa.Integer(), sa.ForeignKey("master_spbu_shift.id", ondelete="CASCADE"), nullable=False),
        sa.Column("tanggal", sa.Date(), nullable=False),
        sa.Column("created_by_id", sa.Integer(), sa.ForeignKey("master_user.id", ondelete="SET NULL"), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
        sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
        sa.UniqueConstraint("spbu_id", "user_id", "shift_id", "tanggal", name="uq_jadwal_shift"),
    )

    # 3. Create absensi
    op.create_table(
        "absensi",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("spbu_id", sa.Integer(), sa.ForeignKey("master_spbu.id", ondelete="CASCADE"), nullable=False, index=True),
        sa.Column("shift_id", sa.Integer(), sa.ForeignKey("master_spbu_shift.id", ondelete="CASCADE"), nullable=False),
        sa.Column("tanggal", sa.Date(), nullable=False),
        sa.Column("foto_url", sa.String(500), nullable=True),
        sa.Column("uploaded_by_id", sa.Integer(), sa.ForeignKey("master_user.id", ondelete="SET NULL"), nullable=True),
        sa.Column("uploaded_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("status", sa.String(20), nullable=False, server_default="pending"),
        sa.Column("reviewed_by_id", sa.Integer(), sa.ForeignKey("master_user.id", ondelete="SET NULL"), nullable=True),
        sa.Column("reviewed_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
        sa.UniqueConstraint("spbu_id", "shift_id", "tanggal", name="uq_absensi"),
    )


def downgrade() -> None:
    op.drop_table("absensi")
    op.drop_table("jadwal_shift")
    op.drop_column("master_role", "can_be_scheduled")