"""add reusable tools and assistant bindings

Revision ID: 20260710_0002
Revises: 20260709_0001
Create Date: 2026-07-10 00:00:00.000000
"""

from collections.abc import Sequence

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql


revision: str = "20260710_0002"
down_revision: str | Sequence[str] | None = "20260709_0001"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    op.create_table(
        "tools",
        sa.Column("id", sa.String(length=40), nullable=False),
        sa.Column("name", sa.String(length=128), nullable=False),
        sa.Column("function_name", sa.String(length=64), nullable=False),
        sa.Column("type", sa.String(length=24), nullable=False),
        sa.Column("description", sa.String(length=2048), server_default="", nullable=False),
        sa.Column(
            "definition",
            postgresql.JSONB(astext_type=sa.Text()),
            server_default=sa.text("'{}'::jsonb"),
            nullable=False,
        ),
        sa.Column(
            "secrets",
            postgresql.JSONB(astext_type=sa.Text()),
            server_default=sa.text("'{}'::jsonb"),
            nullable=False,
        ),
        sa.Column("status", sa.String(length=16), server_default="active", nullable=False),
        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.PrimaryKeyConstraint("id"),
    )
    op.create_index("ix_tools_function_name", "tools", ["function_name"], unique=True)
    op.create_index("ix_tools_type", "tools", ["type"])
    op.create_index("ix_tools_status", "tools", ["status"])

    op.create_table(
        "assistant_tool_bindings",
        sa.Column("assistant_id", sa.String(length=40), nullable=False),
        sa.Column("tool_id", sa.String(length=40), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
        sa.ForeignKeyConstraint(["assistant_id"], ["assistants.id"], ondelete="CASCADE"),
        sa.ForeignKeyConstraint(["tool_id"], ["tools.id"], ondelete="RESTRICT"),
        sa.PrimaryKeyConstraint("assistant_id", "tool_id"),
    )
    op.create_index(
        "ix_assistant_tool_bindings_tool_id",
        "assistant_tool_bindings",
        ["tool_id"],
    )


def downgrade() -> None:
    op.drop_index("ix_assistant_tool_bindings_tool_id", table_name="assistant_tool_bindings")
    op.drop_table("assistant_tool_bindings")
    op.drop_index("ix_tools_status", table_name="tools")
    op.drop_index("ix_tools_type", table_name="tools")
    op.drop_index("ix_tools_function_name", table_name="tools")
    op.drop_table("tools")