"""add missing shorewall rule columns

Revision ID: 0009
Revises: 0008
Create Date: 2026-03-01
"""
from alembic import op
import sqlalchemy as sa

revision = "0009"
down_revision = "0008"
branch_labels = None
depends_on = None

_NEW_COLS = [
    ("origdest",    sa.String(128)),
    ("rate_limit",  sa.String(64)),
    ("user_group",  sa.String(64)),
    ("mark",        sa.String(32)),
    ("connlimit",   sa.String(32)),
    ("time",        sa.String(128)),
    ("headers",     sa.String(128)),
    ("switch_name", sa.String(32)),
    ("helper",      sa.String(32)),
]


def upgrade() -> None:
    for col_name, col_type in _NEW_COLS:
        op.add_column("rules", sa.Column(col_name, col_type, server_default="''", nullable=False))


def downgrade() -> None:
    for col_name, _ in reversed(_NEW_COLS):
        op.drop_column("rules", col_name)