forked from uncloud/uncloud
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
from django.db import transaction
|
|
|
|
from .models import *
|
|
from .selectors import *
|
|
from .tasks import *
|
|
|
|
@transaction.atomic
|
|
def create_wireguard_vpn(owner, public_key, network_mask):
|
|
|
|
pool = get_suitable_pools(network_mask)[0]
|
|
count = pool.wireguardvpn_set.count()
|
|
|
|
# Try re-using previously used networks first
|
|
try:
|
|
free_lease = WireGuardVPNFreeLeases.objects.get(vpnpool=pool)
|
|
|
|
vpn = WireGuardVPN.objects.create(owner=owner,
|
|
vpnpool=pool,
|
|
pool_index=free_lease.pool_index,
|
|
wireguard_public_key=public_key)
|
|
|
|
free_lease.delete()
|
|
|
|
except WireGuardVPNFreeLeases.DoesNotExist:
|
|
# First object
|
|
if count == 0:
|
|
vpn = WireGuardVPN.objects.create(owner=owner,
|
|
vpnpool=pool,
|
|
pool_index=0,
|
|
wireguard_public_key=public_key)
|
|
|
|
else: # Select last network and try +1 it
|
|
last_net = WireGuardVPN.objects.filter(vpnpool=pool).order_by('pool_index').last()
|
|
|
|
next_index = last_net.pool_index + 1
|
|
|
|
if next_index <= pool.max_pool_index:
|
|
vpn = WireGuardVPN.objects.create(owner=owner,
|
|
vpnpool=pool,
|
|
pool_index=next_index,
|
|
wireguard_public_key=public_key)
|
|
|
|
|
|
config = pool.wireguard_config
|
|
server = pool.vpn_server_hostname
|
|
wg_name = pool.wg_name
|
|
|
|
configure_wireguard_server_on_host.apply_async((wg_name, config),
|
|
queue=server)
|
|
|
|
return vpn
|