"""add mqtt pipeline tables

Revision ID: a1b2c3d4e5f6
Revises: 0b117fab4208
Create Date: 2026-03-17 12:30:00.000000

"""

from alembic import op
import sqlalchemy as sa
import sqlmodel.sql.sqltypes


# revision identifiers, used by Alembic.
revision = "a1b2c3d4e5f6"
down_revision = "0b117fab4208"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "terminal_message_raw",
        sa.Column("id", sa.Uuid(), nullable=False),
        sa.Column(
            "device_id", sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False
        ),
        sa.Column("topic", sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False),
        sa.Column("received_at", sa.DateTime(timezone=True), nullable=False),
        sa.Column("payload_bytes", sa.LargeBinary(), nullable=False),
        sa.Column("payload_size", sa.Integer(), nullable=False),
        sa.Column(
            "payload_sha256", sqlmodel.sql.sqltypes.AutoString(length=64), nullable=False
        ),
        sa.Column("qos", sa.Integer(), nullable=False),
        sa.Column("retain", sa.Boolean(), nullable=False),
        sa.Column(
            "parse_status", sqlmodel.sql.sqltypes.AutoString(length=32), nullable=False
        ),
        sa.Column("parse_attempts", sa.Integer(), nullable=False),
        sa.Column("next_retry_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column(
            "parse_error", sqlmodel.sql.sqltypes.AutoString(length=1024), nullable=True
        ),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index(
        op.f("ix_terminal_message_raw_device_id"),
        "terminal_message_raw",
        ["device_id"],
        unique=False,
    )
    op.create_index(
        op.f("ix_terminal_message_raw_next_retry_at"),
        "terminal_message_raw",
        ["next_retry_at"],
        unique=False,
    )
    op.create_index(
        op.f("ix_terminal_message_raw_parse_status"),
        "terminal_message_raw",
        ["parse_status"],
        unique=False,
    )
    op.create_index(
        op.f("ix_terminal_message_raw_payload_sha256"),
        "terminal_message_raw",
        ["payload_sha256"],
        unique=False,
    )
    op.create_index(
        op.f("ix_terminal_message_raw_received_at"),
        "terminal_message_raw",
        ["received_at"],
        unique=False,
    )

    op.create_table(
        "terminal_observation_event",
        sa.Column("id", sa.Uuid(), nullable=False),
        sa.Column("raw_id", sa.Uuid(), nullable=False),
        sa.Column(
            "device_id", sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False
        ),
        sa.Column("event_time", sa.DateTime(timezone=True), nullable=False),
        sa.Column(
            "event_type", sqlmodel.sql.sqltypes.AutoString(length=64), nullable=False
        ),
        sa.Column("payload_json", sa.JSON(), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
        sa.ForeignKeyConstraint(["raw_id"], ["terminal_message_raw.id"]),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index(
        op.f("ix_terminal_observation_event_device_id"),
        "terminal_observation_event",
        ["device_id"],
        unique=False,
    )
    op.create_index(
        op.f("ix_terminal_observation_event_event_time"),
        "terminal_observation_event",
        ["event_time"],
        unique=False,
    )
    op.create_index(
        op.f("ix_terminal_observation_event_event_type"),
        "terminal_observation_event",
        ["event_type"],
        unique=False,
    )
    op.create_index(
        op.f("ix_terminal_observation_event_raw_id"),
        "terminal_observation_event",
        ["raw_id"],
        unique=False,
    )

    op.create_table(
        "terminal_device_state",
        sa.Column(
            "device_id", sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False
        ),
        sa.Column("last_event_time", sa.DateTime(timezone=True), nullable=True),
        sa.Column("fix_status", sqlmodel.sql.sqltypes.AutoString(length=64), nullable=True),
        sa.Column("lat", sa.Float(), nullable=True),
        sa.Column("lon", sa.Float(), nullable=True),
        sa.Column("alt_m", sa.Float(), nullable=True),
        sa.Column("satellites_used", sa.Integer(), nullable=True),
        sa.Column("satellite_signal_summary", sa.JSON(), nullable=True),
        sa.Column("rtcm_type_summary", sa.JSON(), nullable=True),
        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
        sa.PrimaryKeyConstraint("device_id"),
    )

    op.create_table(
        "terminal_raw_archive_manifest",
        sa.Column("id", sa.Uuid(), nullable=False),
        sa.Column(
            "device_id", sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False
        ),
        sa.Column("range_start", sa.DateTime(timezone=True), nullable=False),
        sa.Column("range_end", sa.DateTime(timezone=True), nullable=False),
        sa.Column(
            "file_path", sqlmodel.sql.sqltypes.AutoString(length=1024), nullable=False
        ),
        sa.Column("file_size", sa.Integer(), nullable=False),
        sa.Column("sha256", sqlmodel.sql.sqltypes.AutoString(length=64), nullable=False),
        sa.Column("record_count", sa.Integer(), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
        sa.PrimaryKeyConstraint("id"),
        sa.UniqueConstraint("file_path"),
    )
    op.create_index(
        op.f("ix_terminal_raw_archive_manifest_device_id"),
        "terminal_raw_archive_manifest",
        ["device_id"],
        unique=False,
    )
    op.create_index(
        op.f("ix_terminal_raw_archive_manifest_range_end"),
        "terminal_raw_archive_manifest",
        ["range_end"],
        unique=False,
    )
    op.create_index(
        op.f("ix_terminal_raw_archive_manifest_range_start"),
        "terminal_raw_archive_manifest",
        ["range_start"],
        unique=False,
    )


def downgrade() -> None:
    op.drop_index(
        op.f("ix_terminal_raw_archive_manifest_range_start"),
        table_name="terminal_raw_archive_manifest",
    )
    op.drop_index(
        op.f("ix_terminal_raw_archive_manifest_range_end"),
        table_name="terminal_raw_archive_manifest",
    )
    op.drop_index(
        op.f("ix_terminal_raw_archive_manifest_device_id"),
        table_name="terminal_raw_archive_manifest",
    )
    op.drop_table("terminal_raw_archive_manifest")
    op.drop_table("terminal_device_state")
    op.drop_index(
        op.f("ix_terminal_observation_event_raw_id"),
        table_name="terminal_observation_event",
    )
    op.drop_index(
        op.f("ix_terminal_observation_event_event_type"),
        table_name="terminal_observation_event",
    )
    op.drop_index(
        op.f("ix_terminal_observation_event_event_time"),
        table_name="terminal_observation_event",
    )
    op.drop_index(
        op.f("ix_terminal_observation_event_device_id"),
        table_name="terminal_observation_event",
    )
    op.drop_table("terminal_observation_event")
    op.drop_index(
        op.f("ix_terminal_message_raw_received_at"), table_name="terminal_message_raw"
    )
    op.drop_index(
        op.f("ix_terminal_message_raw_payload_sha256"),
        table_name="terminal_message_raw",
    )
    op.drop_index(
        op.f("ix_terminal_message_raw_parse_status"), table_name="terminal_message_raw"
    )
    op.drop_index(
        op.f("ix_terminal_message_raw_next_retry_at"), table_name="terminal_message_raw"
    )
    op.drop_index(
        op.f("ix_terminal_message_raw_device_id"), table_name="terminal_message_raw"
    )
    op.drop_table("terminal_message_raw")