"""Alerts resolve after N OK checks; rule changes reset state with a note

Revision ID: 0028_alert_resolve_and_reset
Revises: 0027_user_auth_source
Create Date: 2026-10-06
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa

revision: str = "0028_alert_resolve_and_reset"
down_revision: Union[str, None] = "0027_user_auth_source"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.add_column("alert_rules", sa.Column("consecutive_ok_required", sa.Integer,
                                           nullable=False, server_default="1"))
    op.add_column("alert_states", sa.Column("consecutive_ok_count", sa.Integer,
                                            nullable=False, server_default="0"))
    op.alter_column("alert_events", "metric_value", existing_type=sa.Float, nullable=True)
    op.add_column("alert_events", sa.Column("note", sa.Text, nullable=True))


def downgrade() -> None:
    op.drop_column("alert_events", "note")
    op.execute("UPDATE alert_events SET metric_value = 0 WHERE metric_value IS NULL")
    op.alter_column("alert_events", "metric_value", existing_type=sa.Float, nullable=False)
    op.drop_column("alert_states", "consecutive_ok_count")
    op.drop_column("alert_rules", "consecutive_ok_required")