"""Add missing columns to schedules table for full scheduler support

Revision ID: 0002_add_schedule_columns
Revises: 0001_initial
Create Date: 2025-12-05
"""

from __future__ import annotations

from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision = "0002_add_schedule_columns"
down_revision = "0001_initial"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # Ajouter les colonnes manquantes à la table schedules
    op.add_column("schedules", sa.Column("description", sa.Text(), nullable=True))
    op.add_column("schedules", sa.Column("target_type", sa.String(), nullable=True, server_default="group"))
    op.add_column("schedules", sa.Column("extra_vars", sa.JSON(), nullable=True))
    op.add_column("schedules", sa.Column("timezone", sa.String(), nullable=True, server_default="America/Montreal"))
    op.add_column("schedules", sa.Column("start_at", sa.DateTime(timezone=True), nullable=True))
    op.add_column("schedules", sa.Column("end_at", sa.DateTime(timezone=True), nullable=True))
    op.add_column("schedules", sa.Column("last_status", sa.String(), nullable=True, server_default="never"))
    op.add_column("schedules", sa.Column("retry_on_failure", sa.Integer(), nullable=True, server_default="0"))
    op.add_column("schedules", sa.Column("timeout", sa.Integer(), nullable=True, server_default="3600"))
    op.add_column("schedules", sa.Column("run_count", sa.Integer(), nullable=True, server_default="0"))
    op.add_column("schedules", sa.Column("success_count", sa.Integer(), nullable=True, server_default="0"))
    op.add_column("schedules", sa.Column("failure_count", sa.Integer(), nullable=True, server_default="0"))

    # Ajouter hosts_impacted à schedule_runs
    op.add_column("schedule_runs", sa.Column("hosts_impacted", sa.Integer(), nullable=True, server_default="0"))


def downgrade() -> None:
    op.drop_column("schedule_runs", "hosts_impacted")
    op.drop_column("schedules", "failure_count")
    op.drop_column("schedules", "success_count")
    op.drop_column("schedules", "run_count")
    op.drop_column("schedules", "timeout")
    op.drop_column("schedules", "retry_on_failure")
    op.drop_column("schedules", "last_status")
    op.drop_column("schedules", "end_at")
    op.drop_column("schedules", "start_at")
    op.drop_column("schedules", "timezone")
    op.drop_column("schedules", "extra_vars")
    op.drop_column("schedules", "target_type")
    op.drop_column("schedules", "description")