"""add missing shorewall snat columns

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

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

_NEW_COLS = [
    ("proto",       sa.String(16)),
    ("port",        sa.String(64)),
    ("ipsec",       sa.String(128)),
    ("mark",        sa.String(32)),
    ("user_group",  sa.String(64)),
    ("switch_name", sa.String(32)),
    ("origdest",    sa.String(128)),
    ("probability", sa.String(16)),
]


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


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