"""Add AWG S3/S4, MTU and vpn_installed

Revision ID: 0004_real_vpn
Revises: 0003_ssh_status
Create Date: 2026-07-25
"""

from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

revision: str = "0004_real_vpn"
down_revision: Union[str, None] = "0003_ssh_status"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.add_column("vpn_servers", sa.Column("s3", sa.Integer(), nullable=True))
    op.add_column("vpn_servers", sa.Column("s4", sa.Integer(), nullable=True))
    op.add_column(
        "vpn_servers",
        sa.Column("mtu", sa.Integer(), nullable=False, server_default="1420"),
    )
    op.add_column(
        "vpn_servers",
        sa.Column("vpn_installed", sa.Boolean(), nullable=False, server_default=sa.text("false")),
    )


def downgrade() -> None:
    op.drop_column("vpn_servers", "vpn_installed")
    op.drop_column("vpn_servers", "mtu")
    op.drop_column("vpn_servers", "s4")
    op.drop_column("vpn_servers", "s3")