"""Upload batches and their items (US05-02).

Revision ID: 0008_upload_batches
Revises: 0007_rename_plans
Create Date: 2026-08-16

One row per approved album folder handed to immich-go, plus the per-asset
pre-upload hashes that later stories verify the result against.
"""

import sqlalchemy as sa
from alembic import op

revision = "0008_upload_batches"
down_revision = "0007_rename_plans"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "upload_batches",
        sa.Column("id", sa.String(), primary_key=True),
        sa.Column("album", sa.String(), nullable=False),
        sa.Column("folder", sa.String(), nullable=False),
        sa.Column("album_name", sa.String(), nullable=False),
        # planned | running | cancelling | cancelled | succeeded | failed
        # | unknown_requires_verification
        sa.Column("state", sa.String(), nullable=False, server_default="planned"),
        # Preflight token this batch was approved against; re-checked before every
        # attempt, so changed bytes or decisions cannot be uploaded silently.
        sa.Column("preflight_token", sa.String(), nullable=False),
        sa.Column("allow_partial", sa.Boolean(), nullable=False, server_default="0"),
        sa.Column("command", sa.String(), nullable=True),  # JSON array, redacted
        sa.Column("uploader_version", sa.String(), nullable=True),
        sa.Column("asset_count", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("attempt_count", sa.Integer(), nullable=False, server_default="0"),
        # Bumped on every claim and used as the fencing token.
        sa.Column("version", sa.Integer(), nullable=False, server_default="1"),
        sa.Column("worker_id", sa.String(), nullable=True),
        sa.Column("report_path", sa.String(), nullable=True),
        sa.Column("report_bytes", sa.Integer(), nullable=True),
        sa.Column("report_truncated", sa.Boolean(), nullable=False, server_default="0"),
        sa.Column("exit_code", sa.Integer(), nullable=True),
        sa.Column("error_code", sa.String(), nullable=True),
        sa.Column("error_message", sa.String(), nullable=True),
        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("finished_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column(
            "created_at",
            sa.DateTime(timezone=True),
            nullable=False,
            server_default=sa.text("CURRENT_TIMESTAMP"),
        ),
        sa.Column(
            "updated_at",
            sa.DateTime(timezone=True),
            nullable=False,
            server_default=sa.text("CURRENT_TIMESTAMP"),
        ),
    )
    op.create_index("ix_upload_batches_album", "upload_batches", ["album"])
    op.create_index("ix_upload_batches_state", "upload_batches", ["state"])

    op.create_table(
        "upload_items",
        sa.Column(
            "batch_id",
            sa.String(),
            sa.ForeignKey("upload_batches.id", ondelete="CASCADE"),
            primary_key=True,
        ),
        sa.Column("asset_id", sa.String(), primary_key=True),
        sa.Column("path", sa.String(), nullable=False),
        # The bytes as they were when the batch was created; SHA-1 is what Immich
        # uses to recognise a file it already holds.
        sa.Column("sha256", sa.String(), nullable=True),
        sa.Column("sha1", sa.String(), nullable=True),
        sa.Column("state", sa.String(), nullable=False, server_default="pending"),
        sa.Column(
            "updated_at",
            sa.DateTime(timezone=True),
            nullable=False,
            server_default=sa.text("CURRENT_TIMESTAMP"),
        ),
    )


def downgrade() -> None:
    op.drop_table("upload_items")
    op.drop_table("upload_batches")