"""Rebuild penerimaan — drop old single-table structure, add header/item/foto tables.

Revision ID: i4d5e6f7a8b9
Revises: h3c4d5e6f7a8
Create Date: 2026-04-03
"""

from alembic import op
import sqlalchemy as sa

revision = 'i4d5e6f7a8b9'
down_revision = 'h3c4d5e6f7a8'
branch_labels = None
depends_on = None


def upgrade() -> None:
    # Drop old penerimaan table (CASCADE removes any FKs referencing it)
    op.drop_index('ix_penerimaan_penebusan_id', table_name='penerimaan')
    op.drop_index('ix_penerimaan_spbu_tanggal', table_name='penerimaan')
    op.drop_table('penerimaan')

    # --- penerimaan (header) ---
    op.create_table(
        'penerimaan',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('spbu_id', sa.Integer(), nullable=False),
        sa.Column('penebusan_id', sa.Integer(), nullable=False),

        # Truck / delivery info (from Surat Pengantar Pertamina)
        sa.Column('tanggal', sa.Date(), nullable=False),
        sa.Column('tgl_jam_keluar_terminal', sa.DateTime(timezone=True), nullable=True),
        sa.Column('jam_tiba', sa.Time(), nullable=True),
        sa.Column('jam_berangkat', sa.Time(), nullable=True),
        sa.Column('no_polisi', sa.String(20), nullable=True),
        sa.Column('shipment_no', sa.String(50), nullable=True),
        sa.Column('nama_pengemudi', sa.Text(), nullable=True),
        sa.Column('no_lo', sa.String(100), nullable=True),

        # Density & Temperature
        sa.Column('density_obs', sa.Numeric(8, 4), nullable=True),
        sa.Column('temp_obs', sa.Numeric(6, 2), nullable=True),
        sa.Column('density_ons', sa.Numeric(8, 4), nullable=True),
        sa.Column('temp_ons', sa.Numeric(6, 2), nullable=True),

        sa.Column('catatan', sa.Text(), nullable=True),
        sa.Column('created_by_id', sa.Integer(), nullable=True),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),

        sa.ForeignKeyConstraint(['spbu_id'], ['master_spbu.id'], ondelete='CASCADE'),
        sa.ForeignKeyConstraint(['penebusan_id'], ['penebusan.id'], ondelete='RESTRICT'),
        sa.ForeignKeyConstraint(['created_by_id'], ['master_user.id'], ondelete='SET NULL'),
        sa.PrimaryKeyConstraint('id'),
    )
    op.create_index('ix_penerimaan_spbu_tanggal', 'penerimaan', ['spbu_id', 'tanggal'])
    op.create_index('ix_penerimaan_penebusan_id', 'penerimaan', ['penebusan_id'])

    # --- penerimaan_item ---
    op.create_table(
        'penerimaan_item',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('penerimaan_id', sa.Integer(), nullable=False),
        sa.Column('penebusan_item_id', sa.Integer(), nullable=True),
        sa.Column('produk_id', sa.Integer(), nullable=False),
        sa.Column('tangki_id', sa.Integer(), nullable=False),

        sa.Column('dipstick_sebelum_mm', sa.Numeric(10, 1), nullable=False),
        sa.Column('volume_sebelum', sa.Numeric(15, 3), nullable=False),
        sa.Column('dipstick_sesudah_mm', sa.Numeric(10, 1), nullable=False),
        sa.Column('volume_sesudah', sa.Numeric(15, 3), nullable=False),
        sa.Column('volume_diterima', sa.Numeric(15, 3), nullable=False),

        sa.ForeignKeyConstraint(['penerimaan_id'], ['penerimaan.id'], ondelete='CASCADE'),
        sa.ForeignKeyConstraint(['penebusan_item_id'], ['penebusan_item.id'], ondelete='SET NULL'),
        sa.ForeignKeyConstraint(['produk_id'], ['master_produk.id'], ondelete='RESTRICT'),
        sa.ForeignKeyConstraint(['tangki_id'], ['master_spbu_tangki.id'], ondelete='RESTRICT'),
        sa.PrimaryKeyConstraint('id'),
    )
    op.create_index('ix_penerimaan_item_penerimaan_id', 'penerimaan_item', ['penerimaan_id'])

    # --- penerimaan_foto ---
    op.create_table(
        'penerimaan_foto',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('penerimaan_id', sa.Integer(), nullable=False),
        sa.Column('penerimaan_item_id', sa.Integer(), nullable=True),
        sa.Column('tipe', sa.String(30), nullable=False),
        sa.Column('url', sa.String(500), nullable=False),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),

        sa.ForeignKeyConstraint(['penerimaan_id'], ['penerimaan.id'], ondelete='CASCADE'),
        sa.ForeignKeyConstraint(['penerimaan_item_id'], ['penerimaan_item.id'], ondelete='CASCADE'),
        sa.PrimaryKeyConstraint('id'),
    )
    op.create_index('ix_penerimaan_foto_penerimaan_id', 'penerimaan_foto', ['penerimaan_id'])


def downgrade() -> None:
    op.drop_index('ix_penerimaan_foto_penerimaan_id', table_name='penerimaan_foto')
    op.drop_table('penerimaan_foto')

    op.drop_index('ix_penerimaan_item_penerimaan_id', table_name='penerimaan_item')
    op.drop_table('penerimaan_item')

    op.drop_index('ix_penerimaan_penebusan_id', table_name='penerimaan')
    op.drop_index('ix_penerimaan_spbu_tanggal', table_name='penerimaan')
    op.drop_table('penerimaan')

    # Recreate old penerimaan table
    op.create_table(
        'penerimaan',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('spbu_id', sa.Integer(), nullable=False),
        sa.Column('penebusan_id', sa.Integer(), nullable=False),
        sa.Column('tangki_id', sa.Integer(), nullable=False),
        sa.Column('tanggal', sa.Date(), nullable=False),
        sa.Column('jam', sa.Time(), nullable=True),
        sa.Column('no_segel', sa.String(100), nullable=True),
        sa.Column('dipstick_sebelum_mm', sa.Numeric(10, 1), nullable=False),
        sa.Column('volume_sebelum', sa.Numeric(15, 3), nullable=False),
        sa.Column('dipstick_sesudah_mm', sa.Numeric(10, 1), nullable=False),
        sa.Column('volume_sesudah', sa.Numeric(15, 3), nullable=False),
        sa.Column('volume_diterima', sa.Numeric(15, 3), nullable=False),
        sa.Column('density', sa.Numeric(8, 4), nullable=True),
        sa.Column('surat_jalan_url', sa.String(500), nullable=True),
        sa.Column('catatan', sa.Text(), nullable=True),
        sa.Column('created_by_id', sa.Integer(), nullable=True),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.ForeignKeyConstraint(['spbu_id'], ['master_spbu.id'], ondelete='CASCADE'),
        sa.ForeignKeyConstraint(['penebusan_id'], ['penebusan.id'], ondelete='RESTRICT'),
        sa.ForeignKeyConstraint(['tangki_id'], ['master_spbu_tangki.id'], ondelete='RESTRICT'),
        sa.ForeignKeyConstraint(['created_by_id'], ['master_user.id'], ondelete='SET NULL'),
        sa.PrimaryKeyConstraint('id'),
    )
    op.create_index('ix_penerimaan_spbu_tanggal', 'penerimaan', ['spbu_id', 'tanggal'])
    op.create_index('ix_penerimaan_penebusan_id', 'penerimaan', ['penebusan_id'])