"""area (topo do eixo abc), direcionador de gastos e despesa (lancamento)

Revision ID: fad2e725683e
Revises: ee8588821ee1
Create Date: 2026-09-10 14:00:00.000000

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


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


def upgrade() -> None:
    # --- area: novo topo do eixo Macroprocesso -> Processo -> Atividade ---
    op.create_table(
        'area',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('codigo', sa.String(length=20), nullable=False),
        sa.Column('nome', sa.String(length=200), nullable=False),
        sa.PrimaryKeyConstraint('id'),
        sa.UniqueConstraint('codigo'),
    )

    # --- macroprocesso ganha area_id ---
    op.add_column('macroprocesso', sa.Column('area_id', sa.Integer(), nullable=True))
    op.create_foreign_key('macroprocesso_area_id_fkey', 'macroprocesso', 'area', ['area_id'], ['id'])

    # backfill: macroprocesso pre-existente (dados de exemplo locais) ganha uma Area "a revisar"
    op.execute(
        """
        INSERT INTO area (codigo, nome)
        SELECT 'AREA_BACKFILL', 'Migrado automaticamente (revisar)'
        WHERE EXISTS (SELECT 1 FROM macroprocesso)
          AND NOT EXISTS (SELECT 1 FROM area WHERE codigo = 'AREA_BACKFILL')
        """
    )
    op.execute(
        """
        UPDATE macroprocesso SET area_id = (SELECT id FROM area WHERE codigo = 'AREA_BACKFILL')
        WHERE area_id IS NULL
        """
    )
    op.alter_column('macroprocesso', 'area_id', nullable=False)

    # --- direcionador: Atividade + criterio de rateio para Despesa ---
    op.create_table(
        'direcionador',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('codigo', sa.String(length=20), nullable=False),
        sa.Column('nome', sa.String(length=200), nullable=False),
        sa.Column('atividade_id', sa.Integer(), nullable=False),
        sa.Column('base', sa.String(length=20), nullable=False),
        sa.CheckConstraint("base in ('tempo', 'quantidade_produzida')", name='ck_direcionador_base'),
        sa.ForeignKeyConstraint(['atividade_id'], ['atividade.id'], ),
        sa.PrimaryKeyConstraint('id'),
        sa.UniqueConstraint('codigo'),
    )

    # --- despesa: lancamento (valor por Conta + Centro de Custo, rateado por Direcionador) ---
    op.create_table(
        'despesa',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('codigo', sa.String(length=20), nullable=False),
        sa.Column('plano_de_contas_id', sa.Integer(), nullable=False),
        sa.Column('centro_de_custo_id', sa.Integer(), nullable=False),
        sa.Column('direcionador_id', sa.Integer(), nullable=False),
        sa.Column('competencia', sa.Date(), nullable=False),
        sa.Column('valor', sa.Numeric(precision=14, scale=6), nullable=False),
        sa.Column('descricao', sa.String(length=500), nullable=True),
        sa.ForeignKeyConstraint(['plano_de_contas_id'], ['plano_de_contas.id'], ),
        sa.ForeignKeyConstraint(['centro_de_custo_id'], ['centro_de_custo.id'], ),
        sa.ForeignKeyConstraint(['direcionador_id'], ['direcionador.id'], ),
        sa.PrimaryKeyConstraint('id'),
        sa.UniqueConstraint('codigo'),
    )


def downgrade() -> None:
    op.drop_table('despesa')
    op.drop_table('direcionador')

    op.drop_constraint('macroprocesso_area_id_fkey', 'macroprocesso', type_='foreignkey')
    op.drop_column('macroprocesso', 'area_id')

    op.drop_table('area')