"""add document borrowing tables

Revision ID: bbbb_add_document_borrowing
Revises: a1b2c3d4e5f6
Create Date: 2026-04-21 00:00:00.000000
"""
from alembic import op
import sqlalchemy as sa

revision = 'bbbb_add_document_borrowing'
down_revision = 'a1b2c3d4e5f6'
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        'documents',
        sa.Column('id', sa.Integer(), primary_key=True),
        sa.Column('name', sa.String(255), nullable=False),
        sa.Column('doc_type', sa.String(20), nullable=False),
        sa.Column('location', sa.String(20), nullable=False, server_default='home'),
        sa.Column('status', sa.String(20), nullable=False, server_default='available'),
        sa.Column('custodian_id', sa.Integer(), sa.ForeignKey('users.id', ondelete='SET NULL'), nullable=True),
        sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
        sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now()),
    )
    op.create_index('ix_documents_status', 'documents', ['status'])
    op.create_index('ix_documents_custodian_id', 'documents', ['custodian_id'])

    op.create_table(
        'borrow_requests',
        sa.Column('id', sa.Integer(), primary_key=True),
        sa.Column('requester_id', sa.Integer(), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
        sa.Column('purpose', sa.Text(), nullable=False),
        sa.Column('status', sa.String(20), nullable=False, server_default='pending'),
        sa.Column('approved_by', sa.Integer(), sa.ForeignKey('users.id', ondelete='SET NULL'), nullable=True),
        sa.Column('approved_at', sa.DateTime(), nullable=True),
        sa.Column('rejection_reason', sa.Text(), nullable=True),
        sa.Column('completed_at', sa.DateTime(), nullable=True),
        sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
    )
    op.create_index('ix_borrow_requests_requester_id', 'borrow_requests', ['requester_id'])
    op.create_index('ix_borrow_requests_status', 'borrow_requests', ['status'])

    op.create_table(
        'borrow_request_items',
        sa.Column('id', sa.Integer(), primary_key=True),
        sa.Column('borrow_request_id', sa.Integer(), sa.ForeignKey('borrow_requests.id', ondelete='CASCADE'), nullable=False),
        sa.Column('document_id', sa.Integer(), sa.ForeignKey('documents.id', ondelete='RESTRICT'), nullable=False),
        sa.Column('status', sa.String(20), nullable=False, server_default='pending'),
        sa.Column('retrieved_at', sa.DateTime(), nullable=True),
        sa.Column('handed_over_at', sa.DateTime(), nullable=True),
        sa.Column('return_requested_at', sa.DateTime(), nullable=True),
        sa.Column('returned_at', sa.DateTime(), nullable=True),
        sa.Column('notes', sa.Text(), nullable=True),
    )
    op.create_index('ix_borrow_request_items_request_id', 'borrow_request_items', ['borrow_request_id'])
    op.create_index('ix_borrow_request_items_document_id', 'borrow_request_items', ['document_id'])


def downgrade() -> None:
    op.drop_index('ix_borrow_request_items_document_id', 'borrow_request_items')
    op.drop_index('ix_borrow_request_items_request_id', 'borrow_request_items')
    op.drop_table('borrow_request_items')

    op.drop_index('ix_borrow_requests_status', 'borrow_requests')
    op.drop_index('ix_borrow_requests_requester_id', 'borrow_requests')
    op.drop_table('borrow_requests')

    op.drop_index('ix_documents_custodian_id', 'documents')
    op.drop_index('ix_documents_status', 'documents')
    op.drop_table('documents')