"""Home Assistant: Steward holds a sign-in, in place of a token and a webhook

Revision ID: homeassistant_002_sign_in
Revises: homeassistant_001_connection
Create Date: 2026-10-31
"""
from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

revision: str = "homeassistant_002_sign_in"
down_revision: Union[str, None] = "homeassistant_001_connection"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # A saved token can't be refreshed like a sign-in, so that connection signs in again.
    op.execute("DELETE FROM homeassistant_connection")
    op.drop_column("homeassistant_connection", "webhook_id")
    op.drop_column("homeassistant_connection", "automation")
    op.drop_column("homeassistant_updates", "triggered_before")
    op.alter_column("homeassistant_connection", "token", new_column_name="refresh_token")
    op.add_column("homeassistant_connection", sa.Column("client_id", sa.String(500), nullable=False))
    op.add_column("homeassistant_connection",
                  sa.Column("can_install", sa.Boolean, nullable=False, server_default=sa.false()))
    op.add_column("homeassistant_connection",
                  sa.Column("signed_in_by", sa.String(100), nullable=False, server_default=""))
    op.add_column("homeassistant_connection", sa.Column("signed_in_at", sa.DateTime(timezone=True), nullable=True))


def downgrade() -> None:
    op.execute("DELETE FROM homeassistant_connection")
    op.drop_column("homeassistant_connection", "signed_in_at")
    op.drop_column("homeassistant_connection", "signed_in_by")
    op.drop_column("homeassistant_connection", "can_install")
    op.drop_column("homeassistant_connection", "client_id")
    op.alter_column("homeassistant_connection", "refresh_token", new_column_name="token")
    op.add_column("homeassistant_updates", sa.Column("triggered_before", sa.String(64), nullable=True))
    op.add_column("homeassistant_connection", sa.Column("automation", sa.String(20), nullable=True))
    op.add_column("homeassistant_connection", sa.Column("webhook_id", sa.Text, nullable=False))