"""add magento integration tables

Revision ID: 0010_add_magento_tables
Revises: 0009_add_image_short_url
Create Date: 2026-02-27 00:00:00.000000

"""

from alembic import op
import sqlalchemy as sa


revision = "0010_add_magento_tables"
down_revision = "0009_add_image_short_url"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "magento_connections",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("tenant_id", sa.String(64), nullable=True),
        sa.Column("store_base_url", sa.Text(), nullable=False),
        sa.Column("magento_api_base_url", sa.Text(), nullable=True),
        sa.Column("rest_base_path", sa.String(64), nullable=True, server_default="V1"),
        sa.Column("store_code", sa.String(64), nullable=True),
        sa.Column("environment", sa.String(32), nullable=False),
        sa.Column("consumer_key", sa.String(255), nullable=False),
        sa.Column("consumer_secret", sa.Text(), nullable=False),
        sa.Column("access_token", sa.Text(), nullable=False),
        sa.Column("access_token_secret", sa.Text(), nullable=False),
        sa.Column("scopes", sa.dialects.postgresql.JSONB(), nullable=True),
        sa.Column("status", sa.String(32), nullable=False, server_default="active"),
        sa.Column("magento_version", sa.Text(), nullable=True),
        sa.Column("is_msi_enabled", sa.Boolean(), nullable=True),
        sa.Column("default_website_id", sa.Integer(), nullable=True),
        sa.Column("default_store_id", sa.Integer(), nullable=True),
        sa.Column("protected_fields", sa.dialects.postgresql.JSONB(), nullable=True),
        sa.Column("signature_method", sa.String(32), nullable=True, server_default="HMAC-SHA1"),
        sa.Column("last_verified_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("last_error", sa.Text(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), onupdate=sa.func.now(), nullable=False),
    )
    op.create_unique_constraint(
        "uq_magento_connections_tenant_store_env",
        "magento_connections",
        ["tenant_id", "store_base_url", "environment"],
    )

    op.create_table(
        "magento_sync_jobs",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("connection_id", sa.Integer(), sa.ForeignKey("magento_connections.id"), nullable=False),
        sa.Column("status", sa.String(32), nullable=False, server_default="started"),
        sa.Column("dry_run", sa.Boolean(), nullable=False, server_default=sa.false()),
        sa.Column("requested_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("finished_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("total_count", sa.Integer(), nullable=True),
        sa.Column("success_count", sa.Integer(), nullable=True, server_default="0"),
        sa.Column("error_count", sa.Integer(), nullable=True, server_default="0"),
        sa.Column("notes", sa.Text(), nullable=True),
    )
    op.create_index("ix_magento_sync_jobs_connection_id", "magento_sync_jobs", ["connection_id"])

    op.create_table(
        "magento_sync_items",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("job_id", sa.Integer(), sa.ForeignKey("magento_sync_jobs.id"), nullable=False),
        sa.Column("sku", sa.String(128), nullable=False),
        sa.Column("action", sa.String(16), nullable=True),
        sa.Column("payload", sa.dialects.postgresql.JSONB(), nullable=True),
        sa.Column("status", sa.String(32), nullable=False, server_default="pending"),
        sa.Column("error", sa.Text(), nullable=True),
        sa.Column("magento_response", sa.dialects.postgresql.JSONB(), nullable=True),
        sa.Column("attempts", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("duration_ms", sa.Integer(), nullable=True),
        sa.Column("http_status", sa.Integer(), nullable=True),
        sa.Column("idempotency_key", sa.Text(), nullable=True),
        sa.Column("remote_image_urls", sa.dialects.postgresql.JSONB(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
    )
    op.create_index("ix_magento_sync_items_job_id", "magento_sync_items", ["job_id"])
    op.create_index("ix_magento_sync_items_sku_job", "magento_sync_items", ["sku", "job_id"])

    op.create_table(
        "magento_media_map",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("connection_id", sa.Integer(), sa.ForeignKey("magento_connections.id"), nullable=False),
        sa.Column("sku", sa.String(128), nullable=False),
        sa.Column("remote_url", sa.Text(), nullable=False),
        sa.Column("content_sha256", sa.String(64), nullable=False),
        sa.Column("magento_entry_id", sa.Integer(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), onupdate=sa.func.now(), nullable=False),
    )
    op.create_unique_constraint(
        "uq_magento_media_map_conn_sku_url",
        "magento_media_map",
        ["connection_id", "sku", "remote_url"],
    )
    op.create_unique_constraint(
        "uq_magento_media_map_conn_sku_sha256",
        "magento_media_map",
        ["connection_id", "sku", "content_sha256"],
    )
    op.create_index("ix_magento_media_map_connection_id", "magento_media_map", ["connection_id"])

    op.create_table(
        "magento_product_override",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("sku", sa.String(128), nullable=False),
        sa.Column("connection_id", sa.Integer(), sa.ForeignKey("magento_connections.id"), nullable=False),
        sa.Column("field", sa.String(128), nullable=False),
        sa.Column("value", sa.Text(), nullable=False),
        sa.Column("magento_last_seen_updated_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("last_pushed_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("sync_lock_reason", sa.Text(), nullable=True),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
    )
    op.create_unique_constraint(
        "uq_magento_product_override_conn_sku_field",
        "magento_product_override",
        ["connection_id", "sku", "field"],
    )
    op.create_index("ix_magento_product_override_connection_id", "magento_product_override", ["connection_id"])


def downgrade() -> None:
    op.drop_table("magento_product_override")
    op.drop_table("magento_media_map")
    op.drop_table("magento_sync_items")
    op.drop_table("magento_sync_jobs")
    op.drop_table("magento_connections")