159 lines
4.4 KiB
Python
159 lines
4.4 KiB
Python
import getopt
|
|
import sys
|
|
import uuid
|
|
import pwd
|
|
import jinja2
|
|
import random
|
|
import os
|
|
import string
|
|
|
|
def randomString(stringLength=16):
|
|
letters = string.ascii_lowercase
|
|
return ''.join(random.choice(letters) for i in range(stringLength))
|
|
|
|
def read_conf(f):
|
|
ret = {}
|
|
f = open(f, "r")
|
|
lines = f.readlines()
|
|
for line in lines:
|
|
line = line.strip()
|
|
eline = line.split(' ')
|
|
if len(eline) >= 2:
|
|
ret[eline[0]] = eline[1]
|
|
return ret
|
|
|
|
def usage():
|
|
print("Usage: python setup.py [options]\n\n\
|
|
options:\n\
|
|
-h : show usage.\n\
|
|
-d domain : your domain - mydomain.tld.\n\
|
|
[-u uuid] : the uuid of the user. Optional.\n\
|
|
[-p path] : the path of the websocket. Optional.\n\
|
|
[-s subdomain] : your subdomain. Optional.\n\
|
|
[-e email] : your email. Optional.\n\
|
|
[-c conf] : load config from file. Optional.\n\n")
|
|
|
|
def main():
|
|
email = None
|
|
subdomain = None
|
|
domain = None
|
|
conf_file = None
|
|
uid = os.getuid()
|
|
gid = os.getgid()
|
|
v_uuid = None
|
|
v_path = None
|
|
|
|
try:
|
|
opts , _ = getopt.getopt(sys.argv[1:], "hd:s:e:c:u:p:")
|
|
except getopt.GetoptError as err:
|
|
print(str(err))
|
|
usage()
|
|
sys.exit(1)
|
|
|
|
for o, a in opts:
|
|
if o == "-h":
|
|
usage()
|
|
sys.exit(0)
|
|
elif o == "-d":
|
|
if domain != None:
|
|
print("Can specify maximum ONE domain.")
|
|
sys.exit(1)
|
|
else:
|
|
domain = a
|
|
elif o == "-s":
|
|
if subdomain != None:
|
|
print("Can specify maximum ONE subdomain.")
|
|
sys.exit(1)
|
|
else:
|
|
subdomain = a
|
|
elif o == "-e":
|
|
if email != None:
|
|
print("Can specify maximum ONE email.")
|
|
else:
|
|
email = a
|
|
elif o == "-c":
|
|
conf_file = a
|
|
elif o == "-u":
|
|
v_uuid = a
|
|
elif o == "-s":
|
|
v_path = a
|
|
|
|
# overwrite stuff with conf file
|
|
conf = read_conf(conf_file)
|
|
|
|
if "d" in conf:
|
|
domain = conf["d"]
|
|
if "s" in conf:
|
|
subdomain = conf["s"]
|
|
if "e" in conf:
|
|
email = conf["e"]
|
|
if "u" in conf:
|
|
v_uuid = conf["u"]
|
|
if "p" in conf:
|
|
v_path = conf["p"]
|
|
|
|
# generate settings if not specified
|
|
if v_uuid == None:
|
|
v_uuid = uuid.uuid4()
|
|
|
|
if v_path == None:
|
|
v_path = randomString()
|
|
|
|
if domain == None:
|
|
print("Must specify a domain.")
|
|
sys.exit(1)
|
|
|
|
server_name = None
|
|
if subdomain == None:
|
|
server_name = domain
|
|
else:
|
|
server_name = subdomain + "." + domain
|
|
|
|
# process docker-compose
|
|
with open("docker-compose.yml", "r") as file:
|
|
template = jinja2.Template(file.read())
|
|
|
|
output = template.render(uid = uid, gid = gid, domain = domain, \
|
|
subdomain = (subdomain if subdomain != None else ""), \
|
|
only_sub = ("true" if subdomain != None else "false"), \
|
|
email = ("dummy@dummy.com" if email == None else email))
|
|
|
|
with open("docker-compose.yml", "w") as file:
|
|
file.write(output)
|
|
|
|
# process v2ray/config
|
|
with open("v2ray/config.json", "r") as file:
|
|
template = jinja2.Template(file.read())
|
|
|
|
output = template.render(uuid = v_uuid, path = v_path)
|
|
|
|
with open("v2ray/config.json", "w") as file:
|
|
file.write(output)
|
|
|
|
# process nginx/nginx/site-confs/default
|
|
with open("nginx/nginx/site-confs/default", "r") as file:
|
|
template = jinja2.Template(file.read())
|
|
|
|
output = template.render(server_name = server_name, path = v_path)
|
|
|
|
with open("nginx/nginx/site-confs/default", "w") as file:
|
|
file.write(output)
|
|
|
|
# process client.json
|
|
with open("client.json", "r") as file:
|
|
template = jinja2.Template(file.read())
|
|
|
|
output = template.render(uuid = v_uuid, path = v_path, server_name = server_name)
|
|
|
|
with open("client.json", "w") as file:
|
|
file.write(output)
|
|
|
|
print("Processed all files. The detailed client config is written to client.conf.\n" + \
|
|
" Summary:\n" + \
|
|
" Server Address: " + server_name + "\n" \
|
|
" Path: " + v_path + "\n" \
|
|
" UUID: " + str(v_uuid) + "\n" \
|
|
"Please run docker-compose up -d to start the service.")
|
|
|
|
main()
|