1"""add shadowtls protocol
 2
 3Revision ID: a015ed8ce227
 4Revises: be0032100c07
 5Create Date: 2024-11-29 21:41:06.896405
 6
 7"""
 8
 9from alembic import op
10
11# revision identifiers, used by Alembic.
12revision = "a015ed8ce227"
13down_revision = "be0032100c07"
14branch_labels = None
15depends_on = None
16
17
18def upgrade() -> None:
19    bind = op.get_bind()
20    dialect = bind.dialect.name
21
22    if dialect == "postgresql":
23        op.execute("ALTER TYPE proxytypes ADD VALUE 'ShadowTLS'")
24
25    elif dialect in ("mariadb", "mysql"):
26        op.execute(
27            "ALTER TABLE inbounds MODIFY protocol ENUM('VMess', 'VLESS', 'Trojan', 'Shadowsocks', 'Shadowsocks2022', 'Hysteria2', 'WireGuard', 'TUIC', 'ShadowTLS')"
28        )
29
30
31def downgrade() -> None:
32    bind = op.get_bind()
33    dialect = bind.dialect.name
34
35    if dialect in ("mysql", "mariadb"):
36        op.execute(
37            "ALTER TABLE inbounds MODIFY protocol ENUM('VMess', 'VLESS', 'Trojan', 'Shadowsocks', 'Shadowsocks2022', 'Hysteria2', 'WireGuard', 'TUIC')"
38        )