"""Add item image and unique barcode per company.

Revision ID: 013_item_image_barcode
Revises: 012_brands
Create Date: 2026-08-05

"""

from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

revision: str = "013_item_image_barcode"
down_revision: Union[str, None] = "012_brands"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    conn = op.get_bind()
    inspector = sa.inspect(conn)
    columns = {col["name"] for col in inspector.get_columns("items")}
    indexes = {idx["name"] for idx in inspector.get_indexes("items")}
    uniques = {uq.get("name") for uq in inspector.get_unique_constraints("items")}

    if "image" not in columns:
        op.add_column("items", sa.Column("image", sa.String(length=500), nullable=True))

    if "barcode" not in columns:
        op.add_column("items", sa.Column("barcode", sa.String(length=100), nullable=True))

    # Empty string is not NULL — normalize so unique (company_id, barcode) allows blanks
    op.execute(sa.text("UPDATE items SET barcode = NULL WHERE barcode IS NOT NULL AND TRIM(barcode) = ''"))

    if "ix_items_barcode" not in indexes:
        op.create_index("ix_items_barcode", "items", ["barcode"], unique=False)

    if "uq_items_company_barcode" not in uniques:
        op.create_unique_constraint(
            "uq_items_company_barcode",
            "items",
            ["company_id", "barcode"],
        )


def downgrade() -> None:
    conn = op.get_bind()
    inspector = sa.inspect(conn)
    columns = {col["name"] for col in inspector.get_columns("items")}
    indexes = {idx["name"] for idx in inspector.get_indexes("items")}
    uniques = {uq.get("name") for uq in inspector.get_unique_constraints("items")}

    if "uq_items_company_barcode" in uniques:
        op.drop_constraint("uq_items_company_barcode", "items", type_="unique")
    if "ix_items_barcode" in indexes:
        op.drop_index("ix_items_barcode", table_name="items")
    if "image" in columns:
        op.drop_column("items", "image")