"""Initial receipts schema

Revision ID: 001_initial
Revises:
Create Date: 2024-12-11

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
import sqlmodel

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


def upgrade() -> None:
    # Create receipts table
    op.create_table(
        'receipts',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('receipt_type', sa.Enum('BON_FISCAL', 'CHITANTA', name='receipttype'), nullable=False),
        sa.Column('direction', sa.Enum('CHELTUIALA', 'INCASARE', name='receiptdirection'), nullable=False),
        sa.Column('receipt_number', sa.String(length=50), nullable=True),
        sa.Column('receipt_series', sa.String(length=20), nullable=True),
        sa.Column('receipt_date', sa.Date(), nullable=False),
        sa.Column('amount', sa.Numeric(precision=15, scale=2), nullable=False),
        sa.Column('description', sa.String(length=500), nullable=True),
        sa.Column('expense_type_code', sa.String(length=20), nullable=True),
        sa.Column('company_id', sa.Integer(), nullable=False),
        sa.Column('partner_id', sa.Integer(), nullable=True),
        sa.Column('partner_name', sa.String(length=200), nullable=True),
        sa.Column('cash_register_id', sa.Integer(), nullable=True),
        sa.Column('cash_register_name', sa.String(length=100), nullable=True),
        sa.Column('cash_register_account', sa.String(length=20), nullable=True),
        sa.Column('status', sa.Enum('DRAFT', 'PENDING_REVIEW', 'APPROVED', 'REJECTED', 'SYNCED', name='receiptstatus'), nullable=False),
        sa.Column('created_by', sa.String(length=100), nullable=False),
        sa.Column('created_at', sa.DateTime(), nullable=False),
        sa.Column('updated_at', sa.DateTime(), nullable=False),
        sa.Column('submitted_at', sa.DateTime(), nullable=True),
        sa.Column('reviewed_by', sa.String(length=100), nullable=True),
        sa.Column('reviewed_at', sa.DateTime(), nullable=True),
        sa.Column('rejection_reason', sa.String(length=500), nullable=True),
        sa.Column('oracle_synced_at', sa.DateTime(), nullable=True),
        sa.Column('oracle_act_id', sa.Integer(), nullable=True),
        sa.Column('oracle_error', sa.String(length=500), nullable=True),
        sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_receipts_company_id'), 'receipts', ['company_id'], unique=False)
    op.create_index(op.f('ix_receipts_status'), 'receipts', ['status'], unique=False)
    op.create_index(op.f('ix_receipts_created_by'), 'receipts', ['created_by'], unique=False)
    op.create_index(op.f('ix_receipts_receipt_date'), 'receipts', ['receipt_date'], unique=False)

    # Create receipt_attachments table
    op.create_table(
        'receipt_attachments',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('receipt_id', sa.Integer(), nullable=False),
        sa.Column('filename', sa.String(length=255), nullable=False),
        sa.Column('stored_filename', sa.String(length=255), nullable=False),
        sa.Column('file_path', sa.String(length=500), nullable=False),
        sa.Column('file_size', sa.Integer(), nullable=False),
        sa.Column('mime_type', sa.String(length=100), nullable=False),
        sa.Column('uploaded_at', sa.DateTime(), nullable=False),
        sa.ForeignKeyConstraint(['receipt_id'], ['receipts.id'], ondelete='CASCADE'),
        sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_receipt_attachments_receipt_id'), 'receipt_attachments', ['receipt_id'], unique=False)

    # Create accounting_entries table
    op.create_table(
        'accounting_entries',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('receipt_id', sa.Integer(), nullable=False),
        sa.Column('entry_type', sa.Enum('DEBIT', 'CREDIT', name='entrytype'), nullable=False),
        sa.Column('account_code', sa.String(length=20), nullable=False),
        sa.Column('account_name', sa.String(length=200), nullable=True),
        sa.Column('amount', sa.Numeric(precision=15, scale=2), nullable=False),
        sa.Column('partner_id', sa.Integer(), nullable=True),
        sa.Column('cost_center_id', sa.Integer(), nullable=True),
        sa.Column('is_auto_generated', sa.Boolean(), nullable=False),
        sa.Column('modified_by', sa.String(length=100), nullable=True),
        sa.Column('modified_at', sa.DateTime(), nullable=True),
        sa.Column('sort_order', sa.Integer(), nullable=False),
        sa.ForeignKeyConstraint(['receipt_id'], ['receipts.id'], ondelete='CASCADE'),
        sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_accounting_entries_receipt_id'), 'accounting_entries', ['receipt_id'], unique=False)


def downgrade() -> None:
    op.drop_index(op.f('ix_accounting_entries_receipt_id'), table_name='accounting_entries')
    op.drop_table('accounting_entries')

    op.drop_index(op.f('ix_receipt_attachments_receipt_id'), table_name='receipt_attachments')
    op.drop_table('receipt_attachments')

    op.drop_index(op.f('ix_receipts_receipt_date'), table_name='receipts')
    op.drop_index(op.f('ix_receipts_created_by'), table_name='receipts')
    op.drop_index(op.f('ix_receipts_status'), table_name='receipts')
    op.drop_index(op.f('ix_receipts_company_id'), table_name='receipts')
    op.drop_table('receipts')

    # Drop enums (SQLite doesn't actually use these, but for consistency)
    op.execute("DROP TYPE IF EXISTS receipttype")
    op.execute("DROP TYPE IF EXISTS receiptdirection")
    op.execute("DROP TYPE IF EXISTS receiptstatus")
    op.execute("DROP TYPE IF EXISTS entrytype")