Files
CoM-ghost_protocol/phantom/modules/vpn.py
T

40 lines
1.4 KiB
Python

"""WireGuard VPN server deployment module."""
CYAN = "\033[38;5;51m"
WHITE = "\033[38;5;255m"
GREY = "\033[38;5;244m"
RESET = "\033[0m"
def gather_config(config):
"""Gather WireGuard VPN configuration."""
print(f"\n{CYAN} ┌─ WireGuard VPN Configuration ──────────────────────┐{RESET}")
config["vpn_port"] = input(
f" {CYAN}{RESET} Listen port [{WHITE}51820{RESET}]: "
).strip() or "51820"
config["vpn_client_count"] = input(
f" {CYAN}{RESET} Number of client configs [{WHITE}3{RESET}]: "
).strip() or "3"
try:
config["vpn_client_count"] = int(config["vpn_client_count"])
except ValueError:
config["vpn_client_count"] = 3
config["vpn_dns"] = input(
f" {CYAN}{RESET} Client DNS server [{WHITE}1.1.1.1{RESET}]: "
).strip() or "1.1.1.1"
config["vpn_allowed_ips"] = input(
f" {CYAN}{RESET} Allowed IPs [{WHITE}0.0.0.0/0, ::/0{RESET}]: "
).strip() or "0.0.0.0/0, ::/0"
config["vpn_subnet"] = input(
f" {CYAN}{RESET} VPN subnet [{WHITE}10.66.66.0/24{RESET}]: "
).strip() or "10.66.66.0/24"
print(f" {CYAN}└─────────────────────────────────────────────────────┘{RESET}")
return config