99 lines
2.6 KiB
Python
99 lines
2.6 KiB
Python
# import wg0.conf to the database
|
|
import sys
|
|
import requests
|
|
import json
|
|
import configparser
|
|
import re
|
|
|
|
# usage:
|
|
# python import-wg-conf.py <wg0.conf> http://<ip>:<port> USER PASSWORD
|
|
|
|
# parse the command line
|
|
if len(sys.argv) != 5:
|
|
print("Usage: python import-wg-conf.py <wg0.conf> http://<ip>:<port> USER PASSWORD")
|
|
sys.exit(1)
|
|
|
|
# read the config file
|
|
with open(sys.argv[1], 'r') as f:
|
|
config_str = f.read()
|
|
|
|
# we need to use a trick:
|
|
# - toml can't parse the IP address
|
|
# - configparser will complain about multiple Peer sections
|
|
# we will replace [Peer] with [Peer1], [Peer2], etc.
|
|
|
|
# first, count the number of peers
|
|
peers = re.findall(r'\[Peer\]', config_str)
|
|
num_peers = len(peers)
|
|
# then replace [Peer] with [Peer1], [Peer2], etc.
|
|
for i in range(num_peers):
|
|
config_str = config_str.replace('[Peer]', '[Peer' + str(i+1) + ']', 1)
|
|
|
|
# remove the comment from '# friendly_name = ' line
|
|
config_str = config_str.replace('# friendly_name = ', 'friendly_name = ')
|
|
|
|
# parse the config file
|
|
config = configparser.ConfigParser()
|
|
config.read_string(config_str)
|
|
|
|
# iterate the sections
|
|
for section in config.sections():
|
|
# print header
|
|
print('[' + section + ']')
|
|
# iterate the options
|
|
for option in config.options(section):
|
|
print(option + ' = ' + config[section][option])
|
|
print('')
|
|
|
|
# login to the API
|
|
|
|
login_data = {
|
|
'username': sys.argv[3],
|
|
'password': sys.argv[4],
|
|
}
|
|
|
|
session = requests.Session()
|
|
|
|
response = session.post(sys.argv[2] + '/login', json=login_data)
|
|
|
|
|
|
# add the peers to wireguard-ui
|
|
|
|
# iterate the sections
|
|
for section in config.sections():
|
|
#if starts with "Peer"
|
|
if not section.startswith('Peer'):
|
|
continue
|
|
|
|
newclient_data = {
|
|
'name': 'clientname',
|
|
'email': '',
|
|
'allocated_ips': [
|
|
],
|
|
'allowed_ips': [
|
|
'10.123.0.0/24',
|
|
'172.16.0.0/12'
|
|
],
|
|
'extra_allowed_ips': [],
|
|
'use_server_dns': True,
|
|
'enabled': True,
|
|
'public_key': '',
|
|
'preshared_key': '',
|
|
}
|
|
|
|
for option in config.options(section):
|
|
if option == 'friendly_name':
|
|
newclient_data['name'] = config[section][option]
|
|
elif option == 'allowedips':
|
|
# this will become AllowedIPs in the configuration file
|
|
newclient_data['allocated_ips'] = config[section][option].split(',')
|
|
elif option == 'publickey':
|
|
newclient_data['public_key'] = config[section][option]
|
|
elif option == 'presharedkey':
|
|
newclient_data['preshared_key'] = config[section][option]
|
|
|
|
response = session.post('http://localhost:5000/new-client', json=newclient_data)
|
|
print(response.text)
|
|
#break
|
|
|