mirror of
https://github.com/eduardogsilva/wireguard_webadmin.git
synced 2025-04-19 08:55:12 +00:00
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
|
from django.http import JsonResponse
|
||
|
|
from django.views.decorators.http import require_http_methods
|
||
|
|
from django.contrib.auth.decorators import login_required
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
@login_required
|
||
|
|
@require_http_methods(["GET"])
|
||
|
|
def wireguard_status(request):
|
||
|
|
commands = {
|
||
|
|
'latest-handshakes': "wg show all latest-handshakes | expand | tr -s ' '",
|
||
|
|
'allowed-ips': "wg show all allowed-ips | expand | tr -s ' '",
|
||
|
|
'transfer': "wg show all transfer | expand | tr -s ' '",
|
||
|
|
'endpoints': "wg show all endpoints | expand | tr -s ' '",
|
||
|
|
}
|
||
|
|
|
||
|
|
output = {}
|
||
|
|
|
||
|
|
for key, command in commands.items():
|
||
|
|
process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
|
||
|
|
stdout, stderr = process.communicate()
|
||
|
|
|
||
|
|
if process.returncode != 0:
|
||
|
|
return JsonResponse({'error': stderr}, status=400)
|
||
|
|
|
||
|
|
current_interface = None
|
||
|
|
for line in stdout.strip().split('\n'):
|
||
|
|
parts = line.split()
|
||
|
|
if len(parts) >= 3:
|
||
|
|
interface, peer, value = parts[0], parts[1], " ".join(parts[2:])
|
||
|
|
current_interface = interface
|
||
|
|
else:
|
||
|
|
continue
|
||
|
|
|
||
|
|
if interface not in output:
|
||
|
|
output[interface] = {}
|
||
|
|
|
||
|
|
if peer not in output[interface]:
|
||
|
|
output[interface][peer] = {
|
||
|
|
'allowed-ips': [],
|
||
|
|
'latest-handshakes': '',
|
||
|
|
'transfer': {'tx': 0, 'rx': 0},
|
||
|
|
'endpoints': '',
|
||
|
|
}
|
||
|
|
|
||
|
|
if key == 'allowed-ips':
|
||
|
|
output[interface][peer]['allowed-ips'].append(value)
|
||
|
|
elif key == 'transfer':
|
||
|
|
tx, rx = value.split()[-2:]
|
||
|
|
output[interface][peer]['transfer'] = {'tx': int(tx), 'rx': int(rx)}
|
||
|
|
elif key == 'endpoints':
|
||
|
|
output[interface][peer]['endpoints'] = value
|
||
|
|
else:
|
||
|
|
output[interface][peer][key] = value
|
||
|
|
|
||
|
|
return JsonResponse(output)
|