"""Add type, role, phone, day, time to website_enquiries

Revision ID: 0006
Revises: 0005
Create Date: 2026-06-24 00:00:00.000000
"""
from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

revision: str = "0006"
down_revision: Union[str, None] = "0005"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.add_column(
        "website_enquiries",
        sa.Column("type", sa.String(20), nullable=False, server_default="contact"),
    )
    op.add_column(
        "website_enquiries",
        sa.Column("role", sa.String(100), nullable=True),
    )
    op.add_column(
        "website_enquiries",
        sa.Column("phone", sa.String(50), nullable=True),
    )
    op.add_column(
        "website_enquiries",
        sa.Column("day", sa.String(100), nullable=True),
    )
    op.add_column(
        "website_enquiries",
        sa.Column("time", sa.String(100), nullable=True),
    )
    op.create_index("ix_website_enquiries_type", "website_enquiries", ["type"])


def downgrade() -> None:
    op.drop_index("ix_website_enquiries_type", table_name="website_enquiries")
    op.drop_column("website_enquiries", "time")
    op.drop_column("website_enquiries", "day")
    op.drop_column("website_enquiries", "phone")
    op.drop_column("website_enquiries", "role")
    op.drop_column("website_enquiries", "type")