haproxy-wi/app/modules/config/add.py

import os
import json

import modules.db.sql as sql
import modules.server.ssh as ssh_mod
import modules.common.common as common
import modules.config.config as config_mod
import modules.server.server as server_mod
import modules.roxywi.common as roxywi_common
import modules.roxy_wi_tools as roxy_wi_tools

time_zone = sql.get_setting('time_zone')
get_date = roxy_wi_tools.GetDate(time_zone)
get_config = roxy_wi_tools.GetConfigVar()


def get_userlists(config):
	return_config = ''
	with open(config, 'r') as f:
		for line in f:
			if line.startswith('userlist'):
				line = line.strip()
				return_config += line + ','

	return return_config


def show_userlist(server_ip: str) -> None:
	configs_dir = get_config.get_config_var('configs', 'haproxy_save_configs_dir')
	format_file = 'cfg'

	try:
		sections = get_userlists(configs_dir + roxywi_common.get_files(configs_dir, format_file)[0])
	except Exception as e:
		roxywi_common.logging('Roxy-WI server', str(e), roxywi=1)
		try:
			cfg = f'{configs_dir}{server_ip}-{get_date.return_date("config")}.{format_file}'
		except Exception as e:
			roxywi_common.logging('Roxy-WI server', f' Cannot generate a cfg path {e}', roxywi=1)
		try:
			error = config_mod.get_config(server_ip, cfg)
		except Exception as e:
			roxywi_common.logging('Roxy-WI server', f' Cannot download a config {e}', roxywi=1)
		try:
			sections = get_userlists(cfg)
		except Exception as e:
			roxywi_common.logging('Roxy-WI server', f' Cannot get Userlists from the config file {e}', roxywi=1)
			sections = 'error: Cannot get Userlists'

	print(sections)


def get_bwlist(color: str, group: str, list_name: str) -> None:
	lib_path = get_config.get_config_var('main', 'lib_path')
	list_path = f"{lib_path}/lists/{group}/{color}/{list_name}"

	try:
		with open(list_path, 'r') as f:
			print(f.read())
	except IOError as e:
		print(f"error: Cannot read {color} list: {e}")


def get_bwlists_for_autocomplete(color: str, group: str) -> None:
	lib_path = get_config.get_config_var('main', 'lib_path')
	list_path = f"{lib_path}/lists/{group}/{color}"
	lists = roxywi_common.get_files(list_path, "lst")

	for line in lists:
		print(line)


def create_bwlist(server_ip: str, list_name: str, color: str, group: str) -> None:
	lib_path = get_config.get_config_var('main', 'lib_path')
	list_name = f"{list_name.split('.')[0]}.lst"
	list_path = f"{lib_path}/lists/{group}/{color}/{list_name}"
	try:
		open(list_path, 'a').close()
		print('success: ')
		try:
			roxywi_common.logging(server_ip, f'A new list {color} {list_name} has been created', roxywi=1, login=1)
		except Exception:
			pass
	except IOError as e:
		print(f'error: Cannot create a new {color} list. {e}, ')


def save_bwlist(list_name: str, list_con: str, color: str, group: str, server_ip: str, action: str) -> None:
	lib_path = get_config.get_config_var('main', 'lib_path')
	list_path = f"{lib_path}/lists/{group}/{color}/{list_name}"
	try:
		with open(list_path, "w") as file:
			file.write(list_con)
	except IOError as e:
		print(f'error: Cannot save {color} list. {e}')

	path = sql.get_setting('haproxy_dir') + "/" + color
	servers = []

	if server_ip != 'all':
		servers.append(server_ip)

		masters = sql.is_master(server_ip)
		for master in masters:
			if master[0] is not None:
				servers.append(master[0])
	else:
		server = roxywi_common.get_dick_permit()
		for s in server:
			servers.append(s[2])

	for serv in servers:
		server_mod.ssh_command(serv, [f"sudo mkdir {path}"])
		server_mod.ssh_command(serv, [f"sudo chown $(whoami) {path}"])
		error = config_mod.upload(serv, f'{path}/{list_name}', list_path, dir='fullpath')

		if error:
			print(f'error: Upload fail: {error} , ')
		else:
			print(f'success: Edited {color} list was uploaded to {serv} , ')
			try:
				roxywi_common.logging(serv, f'Has been edited the {color} list {list_name}', roxywi=1, login=1)
			except Exception:
				pass

			server_id = sql.select_server_id_by_ip(server_ip=serv)
			haproxy_enterprise = sql.select_service_setting(server_id, 'haproxy', 'haproxy_enterprise')
			if haproxy_enterprise == '1':
				haproxy_service_name = "hapee-2.0-lb"
			else:
				haproxy_service_name = "haproxy"

			if action == 'restart':
				server_mod.ssh_command(serv, [f"sudo systemctl restart {haproxy_service_name}"])
			elif action == 'reload':
				server_mod.ssh_command(serv, [f"sudo systemctl reload {haproxy_service_name}"])


def delete_bwlist(list_name: str, color: str, group: str, server_ip: str) -> None:
	servers = []
	lib_path = get_config.get_config_var('main', 'lib_path')
	list_path = f"{lib_path}/lists/{group}/{color}/{list_name}"
	path = f"{sql.get_setting('haproxy_dir')}/{color}"

	try:
		os.remove(list_path)
	except IOError as e:
		print(f'error: Cannot delete {color} list from Roxy-WI server. {e} , ')

	if server_ip != 'all':
		servers.append(server_ip)

		masters = sql.is_master(server_ip)
		for master in masters:
			if master[0] is not None:
				servers.append(master[0])
	else:
		server = roxywi_common.get_dick_permit()
		for s in server:
			servers.append(s[2])

	for serv in servers:
		error = server_mod.ssh_command(serv, [f"sudo rm {path}/{list_name}"], return_err=1)

		if error:
			print(f'error: Deleting fail: {error} , ')
		else:
			print(f'success: the {color} list has been deleted on {serv} , ')
			try:
				roxywi_common.logging(serv, f'has been deleted the {color} list {list_name}', roxywi=1, login=1)
			except Exception:
				pass


def edit_map(map_name: str, group: str) -> None:
	lib_path = get_config.get_config_var('main', 'lib_path')
	list_path = f"{lib_path}/maps/{group}/{map_name}"

	try:
		with open(list_path, 'r') as f:
			print(f.read())
	except IOError as e:
		print(f"error: Cannot read {map_name} list: {e}")


def create_map(server_ip: str, map_name: str, group: str) -> None:
	lib_path = get_config.get_config_var('main', 'lib_path')
	map_name = f"{map_name.split('.')[0]}.map"
	map_path = f'{lib_path}/maps/{group}/'
	full_path = f'{map_path}/{map_name}'

	try:
		server_mod.subprocess_execute(f'mkdir -p {map_path}')
	except Exception as e:
		assert Exception(f'error: cannot create a local folder for maps: {e}')
	try:
		open(full_path, 'a').close()
		print('success: ')
		try:
			roxywi_common.logging(server_ip, f'A new map {map_name} has been created', roxywi=1, login=1)
		except Exception:
			pass
	except IOError as e:
		assert Exception(f'error: Cannot create a new {map_name} map. {e}, ')


def save_map(map_name: str, list_con: str, group: str, server_ip: str, action: str) -> None:
	lib_path = get_config.get_config_var('main', 'lib_path')
	map_path = f"{lib_path}/maps/{group}/{map_name}"
	try:
		with open(map_path, "w") as file:
			file.write(list_con)
	except IOError as e:
		print(f'error: Cannot save {map_name} list. {e}')

	path = sql.get_setting('haproxy_dir') + "/maps"
	servers = []

	if server_ip != 'all':
		servers.append(server_ip)

		masters = sql.is_master(server_ip)
		for master in masters:
			if master[0] is not None:
				servers.append(master[0])
	else:
		server = roxywi_common.get_dick_permit()
		for s in server:
			servers.append(s[2])

	for serv in servers:
		server_mod.ssh_command(serv, [f"sudo mkdir {path}"])
		server_mod.ssh_command(serv, [f"sudo chown $(whoami) {path}"])
		error = config_mod.upload(serv, f'{path}/{map_name}', map_path, dir='fullpath')

		if error:
			print(f'error: Upload fail: {error} , ')
		else:
			print(f'success: Edited {map_name} map was uploaded to {serv} , ')
			try:
				roxywi_common.logging(serv, f'Has been edited the map {map_name}', roxywi=1, login=1)
			except Exception:
				pass

			server_id = sql.select_server_id_by_ip(server_ip=serv)
			haproxy_enterprise = sql.select_service_setting(server_id, 'haproxy', 'haproxy_enterprise')
			if haproxy_enterprise == '1':
				haproxy_service_name = "hapee-2.0-lb"
			else:
				haproxy_service_name = "haproxy"

			if action == 'restart':
				server_mod.ssh_command(serv, [f"sudo systemctl restart {haproxy_service_name}"])
			elif action == 'reload':
				server_mod.ssh_command(serv, [f"sudo systemctl reload {haproxy_service_name}"])


def delete_map(map_name: str, group: str, server_ip: str) -> None:
	servers = []
	lib_path = get_config.get_config_var('main', 'lib_path')
	list_path = f"{lib_path}/maps/{group}/{map_name}"
	path = f"{sql.get_setting('haproxy_dir')}/maps"

	try:
		os.remove(list_path)
	except IOError as e:
		print(f'error: Cannot delete {map_name} map from Roxy-WI server. {e} , ')

	if server_ip != 'all':
		servers.append(server_ip)

		masters = sql.is_master(server_ip)
		for master in masters:
			if master[0] is not None:
				servers.append(master[0])
	else:
		server = roxywi_common.get_dick_permit()
		for s in server:
			servers.append(s[2])

	for serv in servers:
		error = server_mod.ssh_command(serv, [f"sudo rm {path}/{map_name}"], return_err=1)

		if error:
			print(f'error: Deleting fail: {error} , ')
		else:
			print(f'success: the {map_name} map has been deleted on {serv} , ')
			try:
				roxywi_common.logging(serv, f'has been deleted the {map_name} map', roxywi=1, login=1)
			except Exception:
				pass


def create_saved_option(option: str, group: int) -> None:
	if sql.insert_new_option(option, group):
		from jinja2 import Environment, FileSystemLoader
		env = Environment(loader=FileSystemLoader('templates/ajax'), autoescape=True)
		template = env.get_template('/new_option.html')
		template = template.render(options=sql.select_options(option=option))
		print(template)


def get_saved_option(group: str, term: str) -> None:
	options = sql.select_options(group=group, term=term)

	a = {}
	v = 0
	for i in options:
		a[v] = i.options
		v = v + 1

	print(json.dumps(a))


def create_saved_server(server: str, group: str, desc: str) -> None:
	if sql.insert_new_savedserver(server, desc, group):
		from jinja2 import Environment, FileSystemLoader
		env = Environment(loader=FileSystemLoader('templates/ajax'), autoescape=True)
		template = env.get_template('/new_saved_servers.html')

		template = template.render(server=sql.select_saved_servers(server=server))
		print(template)


def get_saved_servers(group: str, term: str) -> None:
	servers = sql.select_saved_servers(group=group, term=term)

	a = {}
	v = 0
	for i in servers:
		a[v] = {}
		a[v]['value'] = {}
		a[v]['desc'] = {}
		a[v]['value'] = i.server
		a[v]['desc'] = i.description
		v = v + 1

	print(json.dumps(a))


def get_le_cert(server_ip: str, lets_domain: str, lets_email: str) -> None:
	proxy = sql.get_setting('proxy')
	ssl_path = common.return_nice_path(sql.get_setting('cert_path'), is_service=0)
	haproxy_dir = sql.get_setting('haproxy_dir')
	script = "letsencrypt.sh"
	proxy_serv = ''
	ssh_settings = ssh_mod.return_ssh_keys_path(server_ip)

	os.system(f"cp scripts/{script} .")

	if proxy is not None and proxy != '' and proxy != 'None':
		proxy_serv = proxy

	commands = [
		f"chmod +x {script} &&  ./{script} PROXY={proxy_serv} haproxy_dir={haproxy_dir} DOMAIN={lets_domain} "
		f"EMAIL={lets_email} SSH_PORT={ssh_settings['port']} SSL_PATH={ssl_path} HOST={server_ip} USER={ssh_settings['user']} "
		f"PASS='{ssh_settings['password']}' KEY={ssh_settings['key']}"
	]

	output, error = server_mod.subprocess_execute(commands[0])

	if error:
		roxywi_common.logging('Roxy-WI server', error, roxywi=1)
		print(error)
	else:
		for line in output:
			if any(s in line for s in ("msg", "FAILED")):
				try:
					line = line.split(':')[1]
					line = line.split('"')[1]
					print(line + "<br>")
					break
				except Exception:
					print(output)
					break
		else:
			print('success: Certificate has been created')

	os.remove(script)


def get_ssl_cert(server_ip: str, cert_id: int) -> None:
	cert_path = sql.get_setting('cert_path')
	commands = [f"openssl x509 -in {cert_path}/{cert_id} -text"]

	try:
		server_mod.ssh_command(server_ip, commands, ip="1")
	except Exception as e:
		print(f'error: Cannot connect to the server {e.args[0]}')


def get_ssl_raw_cert(server_ip: str, cert_id: int) -> None:
	cert_path = sql.get_setting('cert_path')
	commands = [f"cat {cert_path}/{cert_id}"]

	try:
		server_mod.ssh_command(server_ip, commands, ip="1")
	except Exception as e:
		print(f'error: Cannot connect to the server {e.args[0]}')


def get_ssl_certs(server_ip: str) -> None:
	cert_path = sql.get_setting('cert_path')
	commands = [f"sudo ls -1t {cert_path} |grep -E 'pem|crt|key'"]
	try:
		server_mod.ssh_command(server_ip, commands, ip="1")
	except Exception as e:
		print(f'error: Cannot connect to the server: {e.args[0]}')


def del_ssl_cert(server_ip: str, cert_id: str) -> None:
	cert_path = sql.get_setting('cert_path')
	commands = [f"sudo rm -f {cert_path}/{cert_id}"]

	try:
		server_mod.ssh_command(server_ip, commands, ip="1")
	except Exception as e:
		print(f'error: Cannot delete the certificate {e.args[0]}')


def upload_ssl_cert(server_ip: str, ssl_name: str, ssl_cont: str) -> None:
	cert_path = sql.get_setting('cert_path')
	name = ''

	if ssl_name is None:
		print('error: Please enter a desired name')
	else:
		name = f"{ssl_name}.pem"

	try:
		with open(name, "w") as ssl_cert:
			ssl_cert.write(ssl_cont)
	except IOError as e:
		print(f'error: Cannot save the SSL key file: {e.args[0]}')
		return

	masters = sql.is_master(server_ip)
	for master in masters:
		if master[0] is not None:
			error = config_mod.upload(master[0], cert_path, name)
			if not error:
				print(f'success: the SSL file has been uploaded to {master[0]} into: {cert_path}/{name} <br/>')
	try:
		error = config_mod.upload(server_ip, cert_path, name)
		if not error:
			print(f'success: the SSL file has been uploaded to {server_ip} into: {cert_path}/{name}')
	except Exception as e:
		roxywi_common.logging('Roxy-WI server', e.args[0], roxywi=1)
	# try:
	# 	os.rename(name, cert_local_dir)
	# except OSError as e:
	# 	roxywi_common.logging('Roxy-WI server', e.args[0], roxywi=1)

	roxywi_common.logging(server_ip, f"add.py#ssl uploaded a new SSL cert {name}", roxywi=1, login=1)